From 4f171763ce5ee065a704e5d08fcbd5a3379dec85 Mon Sep 17 00:00:00 2001
From: 潘志宝 <979469083@qq.com>
Date: 星期三, 13 十一月 2024 14:22:31 +0800
Subject: [PATCH] 返回预测结果

---
 iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/DmModuleServiceImpl.java |   72 ++++++++++++++++++++++-------------
 1 files changed, 45 insertions(+), 27 deletions(-)

diff --git a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/DmModuleServiceImpl.java b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/DmModuleServiceImpl.java
index cefe08a..2ed8c90 100644
--- a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/DmModuleServiceImpl.java
+++ b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/service/impl/DmModuleServiceImpl.java
@@ -1,17 +1,16 @@
 package com.iailab.module.model.mcs.pre.service.impl;
 
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
-import com.baomidou.mybatisplus.core.metadata.IPage;
-import com.iailab.framework.common.page.PageData;
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.iailab.framework.common.pojo.PageResult;
-import com.iailab.framework.common.service.impl.BaseServiceImpl;
 import com.iailab.module.model.mcs.pre.dao.DmModuleDao;
 import com.iailab.module.model.mcs.pre.entity.DmModuleEntity;
 import com.iailab.module.model.mcs.pre.service.DmModuleService;
 import com.iailab.module.model.mcs.pre.vo.DmModulePageReqVO;
 import org.apache.commons.lang3.StringUtils;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
-import org.springframework.transaction.annotation.Transactional;
+import org.springframework.util.CollectionUtils;
 
 import java.util.*;
 
@@ -19,19 +18,36 @@
  * @author PanZhibao
  * @date 2021年04月23日 8:36
  */
-@Service("dmModuleService")
-public class DmModuleServiceImpl extends BaseServiceImpl<DmModuleDao, DmModuleEntity> implements DmModuleService {
+@Service
+public class DmModuleServiceImpl extends ServiceImpl<DmModuleDao, DmModuleEntity> implements DmModuleService {
+
+    @Autowired
+    private DmModuleDao dmModuleDao;
 
     @Override
     public PageResult<DmModuleEntity> queryPage(DmModulePageReqVO reqVO) {
-        return baseDao.selectPage(reqVO);
+        return dmModuleDao.selectPage(reqVO);
     }
 
     @Override
     public List<DmModuleEntity> list(Map<String, Object> params) {
         QueryWrapper<DmModuleEntity> wrapper = getWrapper(params);
-        wrapper.orderByDesc("CREATE_TIME");
-        return baseDao.selectList(wrapper);
+        wrapper.orderByDesc("create_time");
+        return dmModuleDao.selectList(wrapper);
+    }
+
+    @Override
+    public DmModuleEntity info(String id) {
+        return dmModuleDao.selectById(id);
+    }
+
+    @Override
+    public DmModuleEntity getModuleByItemId(String itemId) {
+        List<DmModuleEntity> list = dmModuleDao.getModuleByItemId(itemId);
+        if (CollectionUtils.isEmpty(list)) {
+            return null;
+        }
+        return list.get(0);
     }
 
     private QueryWrapper<DmModuleEntity> getWrapper(Map<String, Object> params) {
@@ -46,29 +62,22 @@
     }
 
     @Override
-    @Transactional(rollbackFor = Exception.class)
-    public void saveModule(DmModuleEntity module) {
-        module.setId(UUID.randomUUID().toString());
-        Calendar calendar = Calendar.getInstance();
-        calendar.set(Calendar.SECOND, 0);
-        calendar.set(Calendar.MILLISECOND, 0);
-        module.setPredicttime(calendar.getTime());
-        module.setCollecttime(calendar.getTime());
-        module.setUpdateTime(new Date());
-        baseDao.insert(module);
+    public void saveModule(DmModuleEntity entity) {
+        entity.setId(UUID.randomUUID().toString());
+        entity.setCreateTime(new Date());
+        entity.setUpdateTime(new Date());
+        dmModuleDao.insert(entity);
     }
 
     @Override
-    @Transactional(rollbackFor = Exception.class)
-    public void update(DmModuleEntity module) {
-        module.setUpdateTime(new Date());
-        baseDao.updateById(module);
+    public void update(DmModuleEntity entity) {
+        entity.setUpdateTime(new Date());
+        dmModuleDao.updateById(entity);
     }
 
     @Override
-    @Transactional(rollbackFor = Exception.class)
     public void deleteBatch(String[] moduleIds) {
-        baseDao.deleteBatchIds(Arrays.asList(moduleIds));
+        dmModuleDao.deleteBatchIds(Arrays.asList(moduleIds));
     }
 
     @Override
@@ -76,8 +85,17 @@
         Map<String, Object> params = new HashMap<>(1);
         params.put("moduletype", moduletype);
         QueryWrapper<DmModuleEntity> wrapper = getWrapper(params);
-        return baseDao.selectList(wrapper);
+        return dmModuleDao.selectList(wrapper);
+    }
 
+    @Override
+    public void updatePredictTime(String id, Date predictTime) {
+        DmModuleEntity entity = dmModuleDao.selectById(id);
+        if (entity == null) {
+            return;
+        }
+        entity.setPredicttime(predictTime);
+        dmModuleDao.updateById(entity);
     }
 
     @Override
@@ -88,7 +106,7 @@
         moduleWrapper.ne(StringUtils.isNotBlank(id), "id", id);
         moduleWrapper.and(wrapper -> wrapper.eq("modulename", modulename));
 
-        List<DmModuleEntity> list = baseDao.selectList(moduleWrapper);
+        List<DmModuleEntity> list = dmModuleDao.selectList(moduleWrapper);
         return list.size();
     }
 

--
Gitblit v1.9.3