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;
    }


}