From cac8f41a47915de2a30a8784fe1fc35c364e86df Mon Sep 17 00:00:00 2001
From: jiaosong <jiaosong6760@dingtalk.com>
Date: 星期二, 10 十月 2023 18:27:08 +0800
Subject: [PATCH] #pgsql 运营监控

---
 server/services/src/main/java/com/doumee/service/business/impl/SitesServiceImpl.java |   38 ++++++++++++++++++++++++++++++++++++--
 1 files changed, 36 insertions(+), 2 deletions(-)

diff --git a/server/services/src/main/java/com/doumee/service/business/impl/SitesServiceImpl.java b/server/services/src/main/java/com/doumee/service/business/impl/SitesServiceImpl.java
index 95be646..24230ce 100644
--- a/server/services/src/main/java/com/doumee/service/business/impl/SitesServiceImpl.java
+++ b/server/services/src/main/java/com/doumee/service/business/impl/SitesServiceImpl.java
@@ -1,5 +1,9 @@
 package com.doumee.service.business.impl;
 
+import com.doumee.core.constants.Constants;
+import com.doumee.core.constants.ResponseStatus;
+import com.doumee.core.exception.BusinessException;
+import com.doumee.core.model.LoginUserInfo;
 import com.doumee.core.model.PageData;
 import com.doumee.core.model.PageWrap;
 import com.doumee.core.utils.Utils;
@@ -10,10 +14,12 @@
 import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import org.apache.shiro.SecurityUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.util.CollectionUtils;
 
+import java.util.Date;
 import java.util.List;
 
 /**
@@ -29,6 +35,27 @@
 
     @Override
     public String create(Sites sites) {
+        LoginUserInfo principal = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
+
+        QueryWrapper<Sites> wrapper = new QueryWrapper<>();
+        wrapper.lambda()
+                .eq(Sites::getCode,sites.getCode());
+        Integer count = sitesMapper.selectCount(wrapper);
+        if (count > 0){
+            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(),"绔欑偣缂栧彿宸插瓨鍦�");
+        }
+        Sites insert = new Sites();
+        insert.setCreateDate(new Date());
+        insert.setCreator(principal.getId());
+        insert.setEditDate(new Date());
+        insert.setEditor(principal.getId());
+        insert.setIsdeleted(Constants.ZERO);
+        insert.setInfo(sites.getInfo());
+        insert.setName(sites.getName());
+        insert.setCode(sites.getCode());
+        insert.setStatus(sites.getStatus());
+        insert.setLockNum(sites.getLockNum());
+        insert.setIp(sites.getIp());
         sitesMapper.insert(sites);
         return sites.getId();
     }
@@ -54,7 +81,14 @@
 
     @Override
     public void updateById(Sites sites) {
-        sitesMapper.updateById(sites);
+        LoginUserInfo principal = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
+        Sites update = new Sites();
+        update.setCreateDate(new Date());
+        update.setEditor(principal.getId());
+        update.setName(sites.getName());
+        update.setLockNum(sites.getLockNum());
+        update.setIp(sites.getIp());
+        sitesMapper.updateById(update);
     }
 
     @Override
@@ -83,7 +117,7 @@
         QueryWrapper<Sites> wrapper = new QueryWrapper<>(sites);
         return sitesMapper.selectList(wrapper);
     }
-  
+
     @Override
     public PageData<Sites> findPage(PageWrap<Sites> pageWrap) {
         IPage<Sites> page = new Page<>(pageWrap.getPage(), pageWrap.getCapacity());

--
Gitblit v1.9.3