SEEC-DataAnalysis2.0/src/main/java/com/seecoder/dataanalysis/logic/parser/impl/LogParsingServiceImpl.java

package com.seecoder.dataanalysis.logic.parser.impl;

import com.alibaba.fastjson.JSONException;
import com.seecoder.dataanalysis.data.dao.devcloud.DevcloudInfoDao;
import com.seecoder.dataanalysis.data.dao.eval.EvalExamInfoDao;
import com.seecoder.dataanalysis.data.dao.eval.EvalExamRecordDao;
import com.seecoder.dataanalysis.data.entity.devcloud.*;
import com.seecoder.dataanalysis.data.entity.eval.EvalExamInfo;
import com.seecoder.dataanalysis.data.entity.eval.EvalExamRecord;
import com.seecoder.dataanalysis.logic.parser.LogParsingService;
import com.seecoder.dataanalysis.logic.parser.ParsingException;
import com.seecoder.dataanalysis.logic.parser.convertor.LogConvertor;
import com.seecoder.dataanalysis.util.JsonUtil;
import com.seecoder.dataanalysis.util.Messages;
import com.seecoder.dataanalysis.vo.*;
import com.seecoder.dataanalysis.vo.devcloud.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import javax.annotation.PostConstruct;
import java.util.*;

/**
 * @author miaomuzhi
 * @since 2022/1/30
 */
@Service
public class LogParsingServiceImpl implements LogParsingService {

    /**
     * 根据日志元数据操作类型(op_type)查找对应的类名
     */
    private Map<Integer, String> opTypesTable;

    private EvalExamRecordDao evalExamRecordDao;
    private EvalExamInfoDao evalExamInfoDao;
    private DevcloudInfoDao devcloudInfoDao;

    @Autowired
    public void setEvalExamRecordDao(EvalExamRecordDao evalExamRecordDao) {
        this.evalExamRecordDao = evalExamRecordDao;
    }
    @Autowired
    public void setEvalExamInfoDao(EvalExamInfoDao evalExamInfoDao) {
        this.evalExamInfoDao = evalExamInfoDao;
    }
    @Autowired
    public void setDevcloudInfoDao(DevcloudInfoDao devcloudInfoDao) {
        this.devcloudInfoDao = devcloudInfoDao;
    }


    @Override
    public SimpleLog parseLog(String log) throws ParsingException{
        try {
            return JsonUtil.toObject(log, SimpleLog.class);
        } catch (JSONException e) {
            e.printStackTrace();
            throw new ParsingException(e.toString(), 0, log);
        }
    }

    @Override
    @SuppressWarnings("unchecked")
    public <T> T parseLogData(SimpleLog simpleLog, boolean save) throws ParsingException{
        if (simpleLog == null) {
            return null;
        }

        try {
            String className = parseOpType(simpleLog.getOpType());
            Class clazz = Class.forName(className);
            Object data = JsonUtil.toObject(simpleLog.getData(), clazz);

            if (save) {
                saveToDb(data, className);
            }
            return (T) data;
        } catch (ClassNotFoundException|ClassCastException e){
            e.printStackTrace();
            throw new ParsingException(e.toString(), 4, simpleLog.getData());
        }  catch (JSONException e) {
            e.printStackTrace();
            throw new ParsingException(e.toString(), 1, simpleLog.getData());
        }
    }

    @Override
    public String parseOpType(int opType) throws ParsingException {
        if (!opTypesTable.containsKey(opType)) {
            throw new ParsingException("不支持的操作类型：" + opType, 2);
        }
        return opTypesTable.get(opType);
    }


    @PostConstruct
    protected void buildOpTypesTable(){
        opTypesTable = new HashMap<>();
        opTypesTable.put(0, EvalExamInfoDTO.class.getName());  //试卷数据
        opTypesTable.put(1, EvalExamInfoDTO.class.getName());  //诗句更新数据
        opTypesTable.put(2, EvalExamRecordDTO.class.getName());  //学生提交试卷

        opTypesTable.put(100, BugInfoDTO.class.getName());  //修改缺陷信息
        opTypesTable.put(101, CommitInfoDTO.class.getName());  //用户进行commit
        opTypesTable.put(102, TestInfoDTO.class.getName());  //用户创建并执行接口测试
        opTypesTable.put(103, PipelineCreationDTO.class.getName());  //用户创建流水线
        opTypesTable.put(104, PipelineDeletionDTO.class.getName());  //用户删除流水线
        opTypesTable.put(105, PipelineDeployDTO.class.getName());  //用户部署流水线
        opTypesTable.put(106, TreeNodeInfoDTO.class.getName());  //用户插入/更新节点
        opTypesTable.put(107, TreeNodeDeletionDTO.class.getName());  //用户删除节点
    }


    private void saveToDb(Object data, String className) throws ParsingException {
        if (className.equals(EvalExamInfoDTO.class.getName())) {
            EvalExamInfo info = LogConvertor.convertToEvalExamInfoEntity((EvalExamInfoDTO) data);
            evalExamInfoDao.save(info);

        } else if (className.equals(EvalExamRecordDTO.class.getName())) {
            EvalExamRecord record = LogConvertor.convertToEvalExamRecordEntity((EvalExamRecordDTO) data);
            evalExamRecordDao.save(record);

        } else if (className.equals(BugInfoDTO.class.getName())) {
            BugInfoDTO bugInfoDTO = (BugInfoDTO) data;
            if (bugInfoDTO.getProjectId() == null) {
                throw new ParsingException(Messages.PROJECT_ID_NOT_FOUND, 3, data.toString());
            }

            BugInfo info = LogConvertor.convertToBugInfoEntity(bugInfoDTO);
            Optional<DevcloudInfo> devcloudInfo = devcloudInfoDao.findById(bugInfoDTO.getProjectId());
            DevcloudInfo devcloudInfoEntity = devcloudInfo.orElseGet(() -> createDevCloudInfo(bugInfoDTO.getProjectId()));
            devcloudInfoEntity.getBugList().add(info);
            devcloudInfoDao.save(devcloudInfoEntity);

        } else if (className.equals(CommitInfoDTO.class.getName())) {
            CommitInfoDTO commitInfoDTO = (CommitInfoDTO) data;
            if (commitInfoDTO.getProjectId() == null) {
                throw new ParsingException(Messages.PROJECT_ID_NOT_FOUND, 3, data.toString());
            }

            CommitInfo info = LogConvertor.convertToCommitInfoEntity(commitInfoDTO);
            Optional<DevcloudInfo> devcloudInfo = devcloudInfoDao.findById(commitInfoDTO.getProjectId());
            DevcloudInfo devcloudInfoEntity = devcloudInfo.orElseGet(() -> createDevCloudInfo(commitInfoDTO.getProjectId()));
            devcloudInfoEntity.getCommitList().add(info);
            devcloudInfoDao.save(devcloudInfoEntity);

        } else if (className.equals(TestInfoDTO.class.getName())) {
            TestInfoDTO testInfoDTO = (TestInfoDTO) data;
            if (testInfoDTO.getProjectId() == null) {
                throw new ParsingException(Messages.PROJECT_ID_NOT_FOUND, 3, data.toString());
            }

            TestInfo info = LogConvertor.convertToTestInfoEntity(testInfoDTO);
            Optional<DevcloudInfo> devcloudInfo = devcloudInfoDao.findById(testInfoDTO.getProjectId());
            DevcloudInfo devcloudInfoEntity = devcloudInfo.orElseGet(() -> createDevCloudInfo(testInfoDTO.getProjectId()));
            devcloudInfoEntity.getTestList().add(info);
            devcloudInfoDao.save(devcloudInfoEntity);

        } else if (className.equals(PipelineCreationDTO.class.getName())) {
            PipelineCreationDTO creationDTO = (PipelineCreationDTO) data;
            if (creationDTO.getProjectId() == null) {
                throw new ParsingException(Messages.PROJECT_ID_NOT_FOUND, 3, data.toString());
            }

            Optional<DevcloudInfo> devcloudInfo = devcloudInfoDao.findById(creationDTO.getProjectId());
            DevcloudInfo devcloudInfoEntity = devcloudInfo.orElseGet(() -> createDevCloudInfo(creationDTO.getProjectId()));
            PipelineInfo pipelineInfo = new PipelineInfo();
            pipelineInfo.setPipelineId(creationDTO.getPipelineId());
            devcloudInfoEntity.getPipelineList().add(pipelineInfo);
            devcloudInfoDao.save(devcloudInfoEntity);

        } else if (className.equals(PipelineDeletionDTO.class.getName())) {
            PipelineDeletionDTO deletionDTO = (PipelineDeletionDTO) data;
            Optional<DevcloudInfo> devcloudInfo = devcloudInfoDao.findById(deletionDTO.getProjectId());
            //TODO 暂不处理删除部署

        } else if (className.equals(PipelineDeployDTO.class.getName())) {
            PipelineDeployDTO deployDTO = (PipelineDeployDTO) data;

            boolean foundPipeline = false;
            for (DevcloudInfo devcloudInfo: devcloudInfoDao.findAll()) {
                for (PipelineInfo info: devcloudInfo.getPipelineList()) {
                    if (info.getPipelineId().equals(deployDTO.getPipelineId())) {
                        info.setUserId(deployDTO.getUserId());  // 更新用户id
                        int successInc = (deployDTO.getSuccess() ? 1 : 0);
                        info.setSuccess(info.getSuccess() + successInc);
                        info.setTotal(info.getTotal() + 1);
                        foundPipeline = true;
                        break;
                    }
                }

                if (foundPipeline) {
                    devcloudInfoDao.save(devcloudInfo);
                    break;
                }
            }

        } else if (className.equals(TreeNodeInfoDTO.class.getName())) {
            TreeNodeInfoDTO nodeInfoDTO = (TreeNodeInfoDTO) data;
            if (nodeInfoDTO.getProjectId() == null) {
                throw new ParsingException(Messages.PROJECT_ID_NOT_FOUND, 3, data.toString());
            }

            TreeNodeInfo info = LogConvertor.convertToTreeNodeInfoEntity(nodeInfoDTO);
            Optional<DevcloudInfo> devcloudInfo = devcloudInfoDao.findById(nodeInfoDTO.getProjectId());
            DevcloudInfo devcloudInfoEntity = devcloudInfo.orElseGet(() -> createDevCloudInfo(nodeInfoDTO.getProjectId()));
            devcloudInfoEntity.getTreeNodeList().add(info);
            devcloudInfoDao.save(devcloudInfoEntity);

        }  else if (className.equals(TreeNodeDeletionDTO.class.getName())) {
            TreeNodeDeletionDTO deletionDTO = (TreeNodeDeletionDTO) data;

            for (DevcloudInfo devcloudInfo: devcloudInfoDao.findAll()) {
                boolean updated = false;
                List<TreeNodeInfo> deleteList = new ArrayList<>();
                for (TreeNodeInfo info: devcloudInfo.getTreeNodeList()) {
                    if (deletionDTO.getNodeIds().contains(info.getNodeId())) {
                        deleteList.add(info);
                        updated = true;
                    }
                }

                if (updated) {
                    devcloudInfo.getTreeNodeList().removeAll(deleteList);
                    devcloudInfoDao.save(devcloudInfo);
                }
            }
        }
    }

    private DevcloudInfo createDevCloudInfo(Integer projectId) {
        DevcloudInfo devcloudInfo = new DevcloudInfo(projectId, new ArrayList<>(), new ArrayList<>(),
                new ArrayList<>(), new ArrayList<>(), new ArrayList<>());
        this.devcloudInfoDao.save(devcloudInfo);
        return devcloudInfo;
    }
}