LandingPageGenerationTask.java 3.0 KB
// landingpage/entity/LandingPageGenerationTask.java
package com.aigeo.landingpage.entity;

import com.aigeo.common.enums.TaskStatus;
import jakarta.persistence.*;
import java.time.LocalDateTime;



@Entity
@Table(name = "ai_landing_page_generation_tasks")
public class LandingPageGenerationTask {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Integer id;

    @Column(name = "project_id", nullable = false)
    private Integer projectId;

    @Column(name = "user_id", nullable = false)
    private Integer userId;

    @Enumerated(EnumType.STRING)
    private TaskStatus status;

    private Byte progress;

    @Column(name = "dify_api_config_id")
    private Integer difyApiConfigId;

    @Column(name = "prompt_template_id")
    private Integer promptTemplateId;

    @Column(name = "final_prompt_snapshot")
    private String finalPromptSnapshot;

    @Column(name = "error_message")
    private String errorMessage;

    @Column(name = "created_at")
    
    private LocalDateTime createdAt;

    @Column(name = "completed_at")
    
    private LocalDateTime completedAt;

    // Constructors
    public LandingPageGenerationTask() {}

    // Getters and Setters
    public Integer getId() {
        return id;
    }

    public void setId(Integer id) {
        this.id = id;
    }

    public Integer getProjectId() {
        return projectId;
    }

    public void setProjectId(Integer projectId) {
        this.projectId = projectId;
    }

    public Integer getUserId() {
        return userId;
    }

    public void setUserId(Integer userId) {
        this.userId = userId;
    }

    public TaskStatus getStatus() {
        return status;
    }

    public void setStatus(TaskStatus status) {
        this.status = status;
    }

    public Byte getProgress() {
        return progress;
    }

    public void setProgress(Byte progress) {
        this.progress = progress;
    }

    public Integer getDifyApiConfigId() {
        return difyApiConfigId;
    }

    public void setDifyApiConfigId(Integer difyApiConfigId) {
        this.difyApiConfigId = difyApiConfigId;
    }

    public Integer getPromptTemplateId() {
        return promptTemplateId;
    }

    public void setPromptTemplateId(Integer promptTemplateId) {
        this.promptTemplateId = promptTemplateId;
    }

    public String getFinalPromptSnapshot() {
        return finalPromptSnapshot;
    }

    public void setFinalPromptSnapshot(String finalPromptSnapshot) {
        this.finalPromptSnapshot = finalPromptSnapshot;
    }

    public String getErrorMessage() {
        return errorMessage;
    }

    public void setErrorMessage(String errorMessage) {
        this.errorMessage = errorMessage;
    }

    public LocalDateTime getCreatedAt() {
        return createdAt;
    }

    public void setCreatedAt(LocalDateTime createdAt) {
        this.createdAt = createdAt;
    }

    public LocalDateTime getCompletedAt() {
        return completedAt;
    }

    public void setCompletedAt(LocalDateTime completedAt) {
        this.completedAt = completedAt;
    }
}