1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package com.iailab.module.model.matlab.service.impl;
 
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.iailab.framework.common.service.impl.BaseServiceImpl;
import com.iailab.framework.common.util.object.ConvertUtils;
import com.iailab.module.model.matlab.dao.MlProjectModelDao;
import com.iailab.module.model.matlab.dto.MlModelDTO;
import com.iailab.module.model.matlab.dto.MlProjectModelDTO;
import com.iailab.module.model.matlab.entity.MlProjectModelEntity;
import com.iailab.module.model.matlab.service.MlProjectModelService;
import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Service;
 
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.UUID;
 
/**
 * @description:
 * @author: dzd
 * @date: 2024/8/22 14:41
 **/
@Service
public class MlProjectModelServiceImpl extends BaseServiceImpl<MlProjectModelDao, MlProjectModelEntity> implements MlProjectModelService {
    @Override
    public List<MlProjectModelDTO> list(Map<String, Object> params) {
        List<MlProjectModelEntity> entityList = baseDao.selectList(getWrapper(params));
 
        return ConvertUtils.sourceToTarget(entityList, MlProjectModelDTO.class);
    }
 
    @Override
    public void deleteByMap(Map<String,Object> map) {
        baseDao.delete(getWrapper(map));
    }
 
    @Override
    public void create(List<MlModelDTO> models, String projectId) {
        List<MlProjectModelEntity> list = new ArrayList<>(models.size());
        for (MlModelDTO model : models) {
            MlProjectModelEntity projectModel = new MlProjectModelEntity();
            projectModel.setId(UUID.randomUUID().toString());
            projectModel.setProjectId(projectId);
            projectModel.setModelId(model.getId());
            list.add(projectModel);
        }
        baseDao.insert(list);
    }
 
    @Override
    public List<MlModelDTO> getProjectModel(Map<String, Object> params) {
        return baseDao.getProjectModel(params);
    }
 
    public QueryWrapper<MlProjectModelEntity> getWrapper(Map<String, Object> params){
        String id = (String)params.get("id");
        String projectId = (String)params.get("projectId");
        String modelId = (String)params.get("modelId");
 
        QueryWrapper<MlProjectModelEntity> wrapper = new QueryWrapper<>();
        wrapper.eq(StringUtils.isNotBlank(id), "id", id)
                .eq(StringUtils.isNotBlank(projectId), "project_id", projectId)
                .eq(StringUtils.isNotBlank(modelId),"model_id",modelId);
 
        return wrapper;
    }
 
 
}