iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/sche/service/impl/StScheduleSchemeServiceImpl.java

@@ -1,16 +1,110 @@
package com.iailab.module.model.mcs.sche.service.impl;

import com.baomidou.dynamic.datasource.annotation.DSTransactional;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.iailab.framework.common.pojo.PageResult;
import com.iailab.framework.common.service.impl.BaseServiceImpl;
import com.iailab.framework.common.util.object.BeanUtils;
import com.iailab.framework.common.util.object.ConvertUtils;
import com.iailab.module.model.api.mcs.dto.AlarmConfigRespDTO;
import com.iailab.module.model.api.mcs.dto.StScheduleSchemeDTO;
import com.iailab.module.model.mcs.pre.entity.MmPredictAlarmConfigEntity;
import com.iailab.module.model.mcs.sche.dao.StScheduleSchemeDao;
import com.iailab.module.model.mcs.sche.entity.StScheduleSchemeEntity;
import com.iailab.module.model.mcs.sche.service.StScheduleSchemeService;
import com.iailab.module.model.mcs.sche.vo.StScheduleSchemePageReqVO;
import com.iailab.module.model.mcs.sche.vo.StScheduleSchemeSaveReqVO;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.util.*;

/**
 * @author PanZhibao
 * @date 2021年08月11日 12:06
 */
@Service("stScheduleSchemeService")
@Service
public class StScheduleSchemeServiceImpl extends BaseServiceImpl<StScheduleSchemeDao, StScheduleSchemeEntity>
        implements StScheduleSchemeService {

    @Override
    public PageResult<StScheduleSchemeEntity> page(StScheduleSchemePageReqVO reqVO) {
        return baseDao.selectPage(reqVO);
    }

    @Override
    public void create(StScheduleSchemeSaveReqVO createReqVO) {
        StScheduleSchemeEntity entity = BeanUtils.toBean(createReqVO, StScheduleSchemeEntity.class);
        entity.setId(UUID.randomUUID().toString());
        baseDao.insert(entity);
    }

    @Override
    public void update(StScheduleSchemeSaveReqVO updateReqVO) {
        StScheduleSchemeEntity entity = BeanUtils.toBean(updateReqVO, StScheduleSchemeEntity.class);
        baseDao.updateById(entity);
    }

    @Override
    public void delete(String id) {
        baseDao.deleteById(id);
    }

    @Override
    public StScheduleSchemeEntity getInfo(String id) {
        return baseDao.selectById(id);
    }

    @Override
    public StScheduleSchemeEntity getByCode(String code) {
        QueryWrapper<StScheduleSchemeEntity> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("code", code);
        return baseDao.selectOne(queryWrapper);
    }

    @Override
    public List<StScheduleSchemeDTO> list(Map<String, Object> params) {
        QueryWrapper<StScheduleSchemeEntity> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("trigger_method", params.get("triggerMethod"));
        queryWrapper.eq("trigger_condition", params.get("triggerCondition"));
        List<StScheduleSchemeEntity> list = baseDao.selectList(queryWrapper);
        return ConvertUtils.sourceToTarget(list, StScheduleSchemeDTO.class);
    }

    @Override
    public void updateTime(String id, Date scheduleTime, String statusCode) {
        StScheduleSchemeEntity entity = new StScheduleSchemeEntity();
        entity.setId(id);
        entity.setRunStatus(statusCode);
        entity.setScheduleTime(scheduleTime);
        baseDao.updateById(entity);
    }

    @Override
    @DSTransactional(rollbackFor = Exception.class)
    public void enableByIds(String[] ids) {
        if (CollectionUtils.isEmpty(Arrays.asList(ids))) {
            return;
        }
        Arrays.asList(ids).forEach(item -> {
            StScheduleSchemeEntity entity = new StScheduleSchemeEntity();
            entity.setId(item);
            entity.setStatus(0);
            baseDao.updateById(entity);
        });
    }

    @Override
    @DSTransactional(rollbackFor = Exception.class)
    public void disableByIds(String[] ids) {
        if (CollectionUtils.isEmpty(Arrays.asList(ids))) {
            return;
        }
        Arrays.asList(ids).forEach(item -> {
            StScheduleSchemeEntity entity = new StScheduleSchemeEntity();
            entity.setId(item);
            entity.setStatus(1);
            baseDao.updateById(entity);
        });
    }
}