ApiServiceImpl.java 2.47 KB
package com.huaheng.control.management.service.impl;

import java.util.ArrayList;
import java.util.Comparator;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import javax.annotation.Resource;

import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import com.huaheng.control.management.dto.Task;
import com.huaheng.control.management.dto.TaskPackage;
import com.huaheng.control.management.service.ApiService;
import com.huaheng.control.management.utils.HuahengRedisUtils;

import cn.hutool.core.date.DatePattern;
import cn.hutool.core.date.DateUtil;

@Service
public class ApiServiceImpl implements ApiService {

    @Resource
    private HuahengRedisUtils huahengRedisUtils;

    @Override
    public void saveTaskInfo(Task task) {
        long timestamp = System.currentTimeMillis();
        task.setTimestamp(timestamp);
        task.setTimestampFormat(DateUtil.format(new Date(timestamp), DatePattern.NORM_DATETIME_MS_PATTERN));
        huahengRedisUtils.saveTaskStatus(task.getTaskId(), task.getTaskStatus(), task);
    }

    @Override
    public Map<String, Map<String, Task>> queryTaskInfos() {
        return huahengRedisUtils.getTaskDetailsWithTaskKeys(huahengRedisUtils.getAllTaskKeys());
    }

    @Override
    public List<TaskPackage> queryTaskPackageInfos() {
        List<TaskPackage> taskPackageList = new ArrayList<TaskPackage>();
        Map<String, Map<String, Task>> taskDetails = huahengRedisUtils.getTaskDetailsWithTaskKeys(huahengRedisUtils.getAllTaskKeys());
        if (CollectionUtils.isEmpty(taskDetails)) {
            return taskPackageList;
        }
        for (Entry<String, Map<String, Task>> entry : taskDetails.entrySet()) {
            TaskPackage taskPackage = new TaskPackage();

            // 按时间戳顺序遍历,后面的覆盖前面的
            entry.getValue().values().stream().sorted(Comparator.comparingLong(Task::getTimestamp)).forEach(task -> {
                BeanUtils.copyProperties(task, taskPackage);
                taskPackage.setUpdateBy(task.getReporter());
                taskPackage.setUpdateTime(task.getTimestampFormat());
            });
            taskPackageList.add(taskPackage);
        }
        taskPackageList
            .sort(Comparator.comparing(TaskPackage::getTaskExecutionOrder, Comparator.nullsLast(String::compareTo)).thenComparing(TaskPackage::getTaskId));
        return taskPackageList;
    }
}