package com.muyu.domain.model;

import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.extern.log4j.Log4j2;

import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * @author DongZeLiang
 * @version 1.0
 * @description 任务执行模型
 * @date 2023/12/4
 */
@Data
@Log4j2
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class TaskModel {

    /**
     * 任务状态 默认为false状态
     */
    private final AtomicBoolean unifiedStatus = new AtomicBoolean(Boolean.FALSE);

    /**
     * 任务是否可以执行
     * @return 是否有任务执行
     */
    public synchronized boolean isExecution(){
        // 为true表示任务在执行
        if (unifiedStatus.get()){
            // 就算状态为true若执行线程为null或者线程为不活跃也可以执行也可以执行任务
            return this.currentThread == null || !this.currentThread.isAlive();
        }
        return true;
    }

    /**
     * 当前线程
     */
    private Thread currentThread ;
    /**
     * 任务名称
     */
    private String taskName;
    /**
     * 任务执行总数
     */
    private Integer taskExecutionSum;

    /**
     * 任务执行时间
     */
    private long taskStartTime;

    /**
     * 提交当前线程任务
     * @param task 线程任务
     * @param taskName 任务名称
     * @param taskExecutionSum 任务总数
     */
    public synchronized void submit(String taskName,Integer taskExecutionSum, Thread task){
        if (!this.isExecution()){
            throw new RuntimeException("["+this.taskName+"]的任务正在进行中，请等待任务执行完成再次发布一键任务");
        }
        unifiedStatus.set(Boolean.TRUE);
        this.currentThread = task;
        this.currentThread.start();
        this.taskName = taskName;
        this.taskExecutionSum = taskExecutionSum;
        this.taskSuccessSum = new AtomicInteger();
        this.taskErrorSum = new AtomicInteger();
        this.taskStartTime = System.currentTimeMillis();
        log.info("[{}]任务执行开始", this.taskName);
    }

    /**
     * 执行结束
     */
    public void down(){
        log.info("[{}]任务执行结束，耗时：[{}]MS", this.taskName, System.currentTimeMillis() - taskStartTime);
        this.currentThread = null;
        this.taskName = null;
        this.taskExecutionSum = 0;
        this.taskSuccessSum.set(0);
        this.taskErrorSum.set(0);
        unifiedStatus.set(Boolean.FALSE);
    }

    /**
     * 任务成功执行总数
     */
    private AtomicInteger taskSuccessSum;

    /**
     * 累计成功
     */
    public void incrementSuccess(){
        this.taskSuccessSum.incrementAndGet();
    }
    /**
     * 获取成功数量
     */
    public int getSuccessSum(){
       return this.taskSuccessSum.incrementAndGet();
    }

    /**
     * 任务执行失败总数
     */
    private AtomicInteger taskErrorSum;

    /**
     * 累计失败
     */
    public void incrementError(){
        this.taskErrorSum.incrementAndGet();
    }
    /**
     * 获取失败数量
     */
    public int getErrorSum(){
        return this.taskErrorSum.incrementAndGet();
    }
}