package quant.rich.emoney.entity.sqlite;

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import org.apache.commons.lang3.StringUtils;
import org.springframework.util.CollectionUtils;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import com.fasterxml.jackson.databind.JsonNode;

import lombok.Data;
import lombok.experimental.Accessors;
import quant.rich.emoney.mybatis.typehandler.CommaListTypeHandler;
import quant.rich.emoney.mybatis.typehandler.JsonStringTypeHandler;

/**
 * 目前能想到的计划任务有三种
 * <ul>
 * <li>指标的抓取
 * <li>个股策略(/strategy, id=9400)的抓取
 * <li>选股策略的抓取
 * </ul>
 * 未来可能存在更多的计划任务, 所以需要考虑如何设计才能更好地兼容后续添加的任务类型
 */
@Data
@Accessors(chain = true)
@TableName(value = "plan", autoResultMap = true)
public class Plan {

    /**
     * 键 ID
     */
    @TableId(type = IdType.AUTO)
    private String planId;

    /**
     * 计划任务表达式
     */
    private String cronExpr;

    /**
     * 计划名称
     */
    private String planName;

    /**
     * 指标代码
     */
    private String indexCode;
    
    /**
     * 需要抓取的指标周期
     */
    @TableField(typeHandler = CommaListTypeHandler.class)
    private List<String> periods;

    /**
     * 参数
     */
    @TableField(typeHandler = JsonStringTypeHandler.class)
    private JsonNode params;
    
    /**
     * 是否启用
     */
    private Boolean enabled;
    
    /**
     * 交易日检查，若为 true，则任务执行时判断当日是否为交易日，若是则不执行任务，
     * 反之无论是否为交易日都执行
     */
    private Boolean openDayCheck;
    
    /**
     * 设置抓取周期并去重
     * @param periods
     * @return
     */
    public Plan setPeriods(List<String> periods) {
        if (CollectionUtils.isEmpty(periods)) {
            this.periods = Collections.emptyList();
            return this;
        }
        Set<String> hashSet = new HashSet<>();
        periods.forEach(s -> {
            if (StringUtils.isNotBlank(s)) {
                hashSet.add(s.trim());
            }
        });
        this.periods = new ArrayList<>(hashSet);
        return this;
    }
    
    /**
     * 获取抓取周期。已去重
     * @return
     */
    public List<String> getPeriods() {
        setPeriods(periods);
        return periods;
    }

}