package com.iailab.module.model.matlab.service.impl;

import com.iailab.framework.common.service.impl.BaseServiceImpl;
import com.iailab.framework.common.util.object.ConvertUtils;
import com.iailab.module.model.matlab.dao.MlModelMethodDao;
import com.iailab.module.model.matlab.dto.MlModelMethodDTO;
import com.iailab.module.model.matlab.entity.MlModelMethodEntity;
import com.iailab.module.model.matlab.entity.MlModelMethodSettingEntity;
import com.iailab.module.model.matlab.service.MlModelMethodService;
import com.iailab.module.model.matlab.service.MlModelMethodSettingService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.util.*;

/**
 * 
 *
 * @author Dzd 
 * @since 1.0.0 2025-02-08
 */
@Service
public class MlModelMethodServiceImpl extends BaseServiceImpl<MlModelMethodDao, MlModelMethodEntity> implements MlModelMethodService {

    @Autowired
    private MlModelMethodSettingService mlModelMethodSettingService;
    @Override
    public void insertList(List<MlModelMethodDTO> list, String modelId) {
        List<MlModelMethodEntity> methodEntities = new ArrayList<>();
        List<MlModelMethodSettingEntity> settingEntities = new ArrayList<>();
        for (int i = 0; i < list.size(); i++) {
            MlModelMethodEntity entity = ConvertUtils.sourceToTarget(list.get(i), MlModelMethodEntity.class);
            String methodId = UUID.randomUUID().toString();
            entity.setId(methodId);
            entity.setMlModelId(modelId);
            entity.setSort(i);
            methodEntities.add(entity);
            if (!CollectionUtils.isEmpty(list.get(i).getMethodSettings())) {
                List<MlModelMethodSettingEntity> settingEntityList = ConvertUtils.sourceToTarget(list.get(i).getMethodSettings(), MlModelMethodSettingEntity.class);
                for (int j = 0; j < settingEntityList.size(); j++) {
                    MlModelMethodSettingEntity mlModelMethodSettingEntity = settingEntityList.get(j);
                    mlModelMethodSettingEntity.setId(UUID.randomUUID().toString());
                    mlModelMethodSettingEntity.setMlModelMethodId(methodId);
                    mlModelMethodSettingEntity.setSort(j);
                    settingEntities.add(mlModelMethodSettingEntity);
                }
            }
        }
        baseDao.insert(methodEntities);
        mlModelMethodSettingService.insertList(settingEntities);
    }

    @Override
    public void deleteModelMethod(String modelId) {
        Map<String,Object> map = new HashMap<>();
        map.put("ml_model_id", modelId);
        baseDao.deleteByMap(map);
    }
}