package com.yice.webadmin.app.model;

import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import com.yice.common.core.base.mapper.BaseModelMapper;
import com.yice.common.core.base.model.BaseModel;
import com.yice.common.core.util.MyCommonUtil;
import com.yice.webadmin.app.vo.PromptOptimizeBatchVo;
import lombok.Data;
import lombok.EqualsAndHashCode;
import org.mapstruct.Mapper;
import org.mapstruct.factory.Mappers;

import java.util.Date;

/**
 * PromptOptimizeBatch实体对象。
 *
 * @author linking
 * @date 2023-04-13
 */
@Data
@EqualsAndHashCode(callSuper = true)
@TableName(value = "lmp_prompt_optimize_batch")
public class PromptOptimizeBatch extends BaseModel {

    /**
     * 批量优化任务ID。
     */
    @TableId(value = "batch_task_id")
    private Long batchTaskId;

    /**
     * 任务名称。
     */
    private String taskName;

    /**
     * 任务描述。
     */
    private String taskDescribe;

    /**
     * 优化前数据集ID。
     */
    private Long sourceDatasetId;

    /**
     * 优化前数据集名称。
     */
    private String sourceDatasetName;

    /**
     * 优化前数据集版本。
     */
    private Long sourceDatasetVersion;

    /**
     * 优化后数据集ID。
     */
    private Long destDatasetId;

    /**
     * 优化后数据集名称。
     */
    private String destDatasetName;

    /**
     * 优化后数据集版本。
     */
    private Long destDatasetVersion;

    /**
     * 质量优化。
     */
    private Integer qualityQptimality;

    /**
     * 缩短提示词。
     */
    private Integer shortenPrompt;

    /**
     * 迭代轮次。
     */
    private Integer epoch;

    /**
     * 思维链条。
     */
    private Integer chain;

    /**
     * 任务状态。
     */
    private Integer taskStatus;

    /**
     * 优化成功数据量。
     */
    private Long successCount;

    /**
     * 优化失败数据量。
     */
    private Long failCount;

    /**
     * 完成时间。
     */
    private Date finishTime;

    /**
     * task_name LIKE搜索字符串。
     */
    @TableField(exist = false)
    private String searchString;

    public void setSearchString(String searchString) {
        this.searchString = MyCommonUtil.replaceSqlWildcard(searchString);
    }

    @Mapper
    public interface PromptOptimizeBatchModelMapper extends BaseModelMapper<PromptOptimizeBatchVo, PromptOptimizeBatch> {
    }

    public static final PromptOptimizeBatchModelMapper INSTANCE = Mappers.getMapper(PromptOptimizeBatchModelMapper.class);
}
