AI-Backend/java_compile/src/main/java/cn/seecoder/ai/service/ConfigService.java

package cn.seecoder.ai.service;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import cn.seecoder.ai.analysis.FeatureProject;
import cn.seecoder.ai.model.vo.ConfigVO;
import cn.seecoder.ai.dao.*;
import cn.seecoder.ai.dao.entity.Config;
import cn.seecoder.ai.dao.entity.HeaderInfo;
import cn.seecoder.ai.dao.entity.Model;
import cn.seecoder.ai.dao.entity.ModelType;
import cn.seecoder.ai.model.bo.ConfigDetailBO;
import cn.seecoder.ai.model.bo.ConfigModelBO;
import cn.seecoder.ai.model.bo.ConfigBO;
import cn.seecoder.ai.model.bo._ConfigBO;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

/**
 * Config-related service Created by twenbo on 2016/11/16. Changed by lucas on 2016/11/18.
 */
@Slf4j
@Service
public class ConfigService {

    @Autowired
    private ConfigDAO configDao;
    @Autowired
    private FileInfoDAO fileInfoDao;
    @Autowired
    private ModelTypeDAO modelTypeDao;
    @Autowired
    private HeaderInfoDAO headerInfoDao;
    @Autowired
    private ModelDAO modelDao;
    @Autowired
    private FeatureProject featureProject;

    /**
     * @param configVO
     * @return
     */

    public String addConfig(ConfigVO configVO) {
        Config configInfo = new Config();
        configInfo.setFileInfoId(Integer.parseInt(configVO.getFileInfoId()));
        configInfo.setUserId(Integer.parseInt(configVO.getUserId()));
        configInfo.setModelTypeId(Integer.parseInt(configVO.getModelTypeId()));
        configInfo.setConfName(configVO.getConfName());
        configInfo.setFieldIds(JSON.toJSON(configVO.getFieldIds()).toString());
        int configId = configDao.save(configInfo).getId();
        return String.valueOf(configId);
    }

    /**
     * @param userId
     * @return
     */
    public List<ConfigBO> getAllConfigs(int userId) {
        List<ConfigBO> rsl = configDao.findByUserId(userId).stream().map((config) -> {
            String fileName = fileInfoDao.findById(config.getFileInfoId()).getFilename();
            String modelTypeName = modelTypeDao.findById(config.getModelTypeId()) == null ? "" :
                    modelTypeDao.findById(config.getModelTypeId()).getModelTypeName();
            return new ConfigBO(String.valueOf(config.getFileInfoId()),
                    fileName,
                    String.valueOf(config.getId()),
                    String.valueOf(config.getModelTypeId()),
                    modelTypeName,
                    config.getConfName(),
                    JSONObject.parseArray(config.getFieldIds(), String.class).stream().map(fieldId ->
                            headerInfoDao.findById(Integer.parseInt(fieldId))
                    ).collect(Collectors.toList()));
        }).collect(Collectors.toList());
        return rsl;
    }

    /**
     * @param userId
     * @return
     */
    public List<_ConfigBO> _getAllConfigsByUserId(int userId) {
        List<_ConfigBO> rsl = configDao.findByUserId(userId).stream().map((config) -> {
            String fileName = fileInfoDao.findById(config.getFileInfoId()).getFilename();
            _ConfigBO configPojo = new _ConfigBO(String.valueOf(config.getFileInfoId()),
                    fileName,
                    String.valueOf(config.getId()),
                    config.getConfName());
            return configPojo;
        }).collect(Collectors.toList());
        return rsl;
    }

    /**
     * @param configId
     * @return
     */
    public ConfigDetailBO getConfigInfoByConfigId(int configId) {
        List<ConfigModelBO> modelInfos = new ArrayList<>();
        List<Model> models = modelDao.findByConfigId(configId);
        for (Model model : models) {
            ModelType modelType = modelTypeDao.findById(model.getModelTypeId());
            int flag = -1;
            if (model.isTrained() && model.getResOfModel() != null) {
                flag = 1;
            } else if (model.isTrained() && model.getResOfModel() == null) {
                flag = 0;
            }
            modelInfos.add(new ConfigModelBO(
                    model.getModelName(), modelType == null ? "自动训练" : modelType.getModelDetailName(), flag
            ));
        }

        Config config = configDao.findById(configId);
        int modelTypeId=config.getModelTypeId();
        String modelTypeName = modelTypeDao.findById(config.getModelTypeId()) == null ? "" : modelTypeDao.findById(config.getModelTypeId()).getModelTypeName();
        String modelTypeDes = modelTypeDao.findById(config.getModelTypeId()) == null ? "" : modelTypeDao.findById(config.getModelTypeId()).getModelDes();
        String modelDetailDes = modelTypeDao.findById(config.getModelTypeId()) == null ? "" : modelTypeDao.findById(config.getModelTypeId()).getModelDetailDes();
        ConfigDetailBO configPojo = new ConfigDetailBO(
                String.valueOf(config.getId()),
                String.valueOf(modelTypeId),
                modelTypeName,
                modelTypeDes,
                modelDetailDes,
                JSONObject.parseArray(config.getFieldIds(), String.class).stream().map(fieldId ->
                        headerInfoDao.findById(Integer.parseInt(fieldId))
                ).collect(Collectors.toList()),
                modelInfos);
        return configPojo;
    }

    /**
     * @param configId
     * @param confName
     * @param trainFeatureList
     * @return
     */
    public boolean changeConfig(int configId, String confName, List<String> trainFeatureList) {
        Config configInfo = configDao.findById(configId);
        if(configInfo==null){
            return false;
        }
        configInfo.setConfName(confName);
        configInfo.setFieldIds(JSON.toJSON(trainFeatureList).toString());
        configDao.save(configInfo);
        return true;
    }

    public boolean deleteConfigById(int configId) {
        Config configInfo = configDao.findById(configId);
        if(configInfo==null){
            return false;
        }
        List<Model> modelList = modelDao.findByConfigId(configId);
        if (modelList != null) {
            for (Model model : modelList) {
                modelDao.delete(model);
            }
        }
        configDao.delete(configDao.findById(configId));
        return true;
    }

    public List<HeaderInfo> getSelectedFeature(int fileId, String fieldName) {
        log.info("FeatureSelector: ", fileId + ", " + fieldName);
        List<HeaderInfo> res = featureProject.featureSelector(fileId, fieldName);
        return res;
    }
}