package com.mes.service.impl;

import com.mes.entity.PlcTestTask;
import com.mes.mapper.PlcTestTaskMapper;
import com.mes.service.PlcTestTaskService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;

import java.util.Date;
import java.util.List;

/**
 * PLC测试任务服务实现类
 * 
 * @author huang
 * @date 2025/11/04
 */
@Slf4j
@Service
public class PlcTestTaskServiceImpl implements PlcTestTaskService {
    
    @Autowired
    private PlcTestTaskMapper plcTestTaskMapper;
    
    /**
     * 创建新任务
     */
    @Override
    public PlcTestTask createTask(PlcTestTask task) {
        try {
            if (task == null) {
                throw new RuntimeException("创建任务失败：task对象不能为空");
            }
            
            if (!StringUtils.hasText(task.getProjectId())) {
                throw new RuntimeException("创建任务失败：projectId不能为空");
            }
            
            // 一些必上字段应一次设置
            if (!StringUtils.hasText(task.getModule())) {
                task.setModule("default");
            }
            
            if (!StringUtils.hasText(task.getOperationMode())) {
                task.setOperationMode("MANUAL");
            }
            if (!StringUtils.hasText(task.getStatus())) {
                task.setStatus("PENDING");
            }
            
            if (task.getStartTime() == null) {
                task.setStartTime(new Date());
            }
            
            if (task.getCreatedAt() == null) {
                task.setCreatedAt(new Date());
            }
            
            if (task.getUpdatedAt() == null) {
                task.setUpdatedAt(new Date());
            }
            
            log.info("即将插入任务：projectId={}, module={}, status={}", 
                task.getProjectId(), task.getModule(), task.getStatus());
            
            int insertCount = plcTestTaskMapper.insert(task);
            
            if (insertCount <= 0) {
                throw new RuntimeException("创建任务失败：数据库插入返回0行");
            }
            
            log.info("创建PLC测试任务成功，id: {}, projectId: {}", task.getId(), task.getProjectId());
            return task;
        } catch (Exception e) {
            log.error("创建PLC测试任务失败", e);
            throw new RuntimeException("创建PLC测试任务失败: " + e.getMessage(), e);
        }
    }
    
    /**
     * 根据id查询任务
     */
    @Override
    public PlcTestTask getTaskById(Long id) {
        if (id == null) {
            log.warn("查询任务失败：id不能为空");
            return null;
        }
        try {
            return plcTestTaskMapper.selectById(id);
        } catch (Exception e) {
            log.error("查询任务失败，id: {}", id, e);
            return null;
        }
    }
    
    /**
     * 查询项目的所有任务
     */
    @Override
    public List<PlcTestTask> getTasksByProjectId(String projectId) {
        try {
            if (!StringUtils.hasText(projectId)) {
                log.warn("查询任务失败：projectId为空");
                return null;
            }
            
            List<PlcTestTask> tasks = plcTestTaskMapper.selectRunningTasks(projectId);
            log.debug("查询项目[{}]的任务，共{}条", projectId, tasks != null ? tasks.size() : 0);
            return tasks;
        } catch (Exception e) {
            log.error("查询项目任务失败，projectId: {}", projectId, e);
            throw new RuntimeException("查询项目任务失败", e);
        }
    }
    
    /**
     * 查询项目的运行中任务
     */
    @Override
    public List<PlcTestTask> getRunningTasks(String projectId) {
        try {
            if (!StringUtils.hasText(projectId)) {
                log.warn("查询运行中任务失败：projectId为空");
                return null;
            }
            
            List<PlcTestTask> tasks = plcTestTaskMapper.selectRunningTasks(projectId);
            log.debug("查询项目[{}]的运行中任务，共{}条", projectId, tasks != null ? tasks.size() : 0);
            return tasks;
        } catch (Exception e) {
            log.error("查询运行中任务失败，projectId: {}", projectId, e);
            throw new RuntimeException("查询运行中任务失败", e);
        }
    }
    
    /**
     * 更新任务状态
     */
    @Override
    public boolean updateTaskStatus(Long id, String status) {
        if (id == null || !StringUtils.hasText(status)) {
            log.warn("更新任务状态失败：id或status不能为空");
            return false;
        }
        try {
            PlcTestTask task = new PlcTestTask();
            task.setId(id);
            task.setStatus(status);
            task.setUpdatedAt(new Date());
            
            int result = plcTestTaskMapper.updateById(task);
            log.info("更新任务状态成功，id: {}, status: {}", id, status);
            return result > 0;
        } catch (Exception e) {
            log.error("更新任务状态失败，id: {}, status: {}", id, status, e);
            return false;
        }
    }
    
    /**
     * 任务完成
     */
    @Override
    public boolean completeTask(Long id, String status, Integer duration, String result, String errorMessage) {
        if (id == null || !StringUtils.hasText(status)) {
            log.warn("完成任务失败：id或status不能为空");
            return false;
        }
        try {
            PlcTestTask task = new PlcTestTask();
            task.setId(id);
            task.setStatus(status);
            task.setDuration(duration);
            task.setResult(result);
            task.setErrorMessage(errorMessage);
            task.setEndTime(new Date());
            task.setUpdatedAt(new Date());
            
            int updateResult = plcTestTaskMapper.updateById(task);
            log.info("完成任务成功，id: {}, status: {}", id, status);
            return updateResult > 0;
        } catch (Exception e) {
            log.error("完成任务失败，id: {}, status: {}", id, status, e);
            return false;
        }
    }
    
    /**
     * 删除任务
     */
    @Override
    public boolean deleteTask(Long id) {
        if (id == null) {
            log.warn("删除任务失败：id不能为空");
            return false;
        }
        try {
            int result = plcTestTaskMapper.deleteById(id);
            log.info("删除任务成功，id: {}", id);
            return result > 0;
        } catch (Exception e) {
            log.error("删除任务失败，id: {}", id, e);
            return false;
        }
    }
    
    
}