From 7c13eb909a140c736a67555fabe60966f5d2c065 Mon Sep 17 00:00:00 2001
From: zhanghua <314079846@qq.com>
Date: 星期四, 23 三月 2023 17:51:36 +0800
Subject: [PATCH] 账号增加时间限制

---
 ycl-common/src/main/java/com/ycl/service/user/impl/UmsAdminServiceImpl.java |   80 +++++++++++++++++++++++++++++----------
 1 files changed, 59 insertions(+), 21 deletions(-)

diff --git a/ycl-common/src/main/java/com/ycl/service/user/impl/UmsAdminServiceImpl.java b/ycl-common/src/main/java/com/ycl/service/user/impl/UmsAdminServiceImpl.java
index d96a0ed..c8235e2 100644
--- a/ycl-common/src/main/java/com/ycl/service/user/impl/UmsAdminServiceImpl.java
+++ b/ycl-common/src/main/java/com/ycl/service/user/impl/UmsAdminServiceImpl.java
@@ -1,9 +1,13 @@
 package com.ycl.service.user.impl;
 
 import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.util.PageUtil;
 import cn.hutool.core.util.StrUtil;
+import com.alibaba.druid.sql.PagerUtils;
+import com.baomidou.mybatisplus.core.conditions.Wrapper;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.segments.MergeSegments;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.ycl.bo.AdminUserDetails;
@@ -45,9 +49,7 @@
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletRequest;
 import java.text.SimpleDateFormat;
-import java.util.ArrayList;
-import java.util.Date;
-import java.util.List;
+import java.util.*;
 import java.util.function.Consumer;
 import java.util.stream.Collectors;
 
@@ -85,8 +87,9 @@
 
     @Override
     public UmsAdmin getAdminByUsername(String username) {
-        UmsAdmin admin = getCacheService().getAdmin(username);
-        if (admin != null) return admin;
+        // UmsAdmin admin = getCacheService().getAdmin(username);
+        // if (admin != null) return admin;
+        UmsAdmin admin = null;
         QueryWrapper<UmsAdmin> wrapper = new QueryWrapper<>();
         wrapper.lambda().eq(UmsAdmin::getUsername, username);
         List<UmsAdmin> adminList = list(wrapper);
@@ -109,6 +112,9 @@
         //鏌ヨ鏄惁鏈夌浉鍚岀敤鎴峰悕鐨勭敤鎴�
         QueryWrapper<UmsAdmin> wrapper = new QueryWrapper<>();
         wrapper.lambda().eq(UmsAdmin::getUsername, umsAdmin.getUsername());
+        if (umsAdmin.getId() != null) {
+            wrapper.lambda().ne(UmsAdmin::getId, umsAdmin.getId());
+        }
         List<UmsAdmin> umsAdminList = list(wrapper);
         if (umsAdminList.size() > 0) {
             return null;
@@ -116,15 +122,35 @@
         //灏嗗瘑鐮佽繘琛屽姞瀵嗘搷浣�
         String encodePassword = passwordEncoder.encode(umsAdmin.getPassword());
 
-        if ( !umsAdmin.getPassword().matches("^(?=.*\\d)(?=.*[a-z])(?=.*[A-Z])[a-zA-Z0-9]{8,16}$")) {
+        if (umsAdmin.getId() == null && !umsAdmin.getPassword().matches("^(?=.*\\d)(?=.*[a-z])(?=.*[A-Z])[a-zA-Z0-9]{8,16}$")) {
             Asserts.fail("瀵嗙爜闀垮害8鍒�16浣嶄笖瀵嗙爜涓殑瀛楃蹇呴』鍖呭惈瀛楁瘝锛堝ぇ鍐欐垨鑰呭皬鍐欙級鍜屽繀椤诲寘鍚暟瀛楋紝涓嶈兘鍖呭惈绌烘牸");
         }
-        if(umsAdmin.getUsername().equals(umsAdmin.getPassword())){
-           Asserts.fail("瀵嗙爜涓嶈兘鍜岀櫥褰曞悕瀹屽叏涓�鑷�");
+        if (umsAdmin.getUsername().equals(umsAdmin.getPassword())) {
+            Asserts.fail("瀵嗙爜涓嶈兘鍜岀櫥褰曞悕瀹屽叏涓�鑷�");
         }
         umsAdmin.setPassword(encodePassword);
-        baseMapper.insert(umsAdmin);
+        if (umsAdminParam.getDays() != null) {
+            Date date = new Date();
+            Calendar calendar = new GregorianCalendar();
+            calendar.setTime(date);
+            // 鎶婃棩鏈熷線鍚庡鍔犱竴澶�,鏁存暟  寰�鍚庢帹,璐熸暟寰�鍓嶇Щ鍔�
+            calendar.add(Calendar.DATE, umsAdminParam.getDays());
+            // 杩欎釜鏃堕棿灏辨槸鏃ユ湡寰�鍚庢帹涓�澶╃殑缁撴灉
+            date = calendar.getTime();
+            umsAdmin.setExpirationDate(date);
+        } else {
+            umsAdmin.setExpirationDate(new Date("2200-01-01"));
+        }
+        if (umsAdmin.getId() != null) {
+            baseMapper.updateById(umsAdmin);
+        } else {
+            baseMapper.insert(umsAdmin);
+        }
 
+        LambdaQueryWrapper<UmsAdminRoleRelation> deleteWrapper = new QueryWrapper<UmsAdminRoleRelation>().lambda()
+                .eq(UmsAdminRoleRelation::getAdminId, umsAdmin.getId());
+
+        umsAdminRoleRelationService.remove(deleteWrapper);
         //娣诲姞瑙掕壊鎺堟潈
         List<Long> ids = umsAdminParam.getRoleIds();
         if (!ids.isEmpty()) {
@@ -138,6 +164,8 @@
             umsAdminRoleRelationService.saveBatch(roleIds);
         }
 
+        umsDepartManageMapper.deletedByDepartId(umsAdmin.getId());
+
         //娣诲姞閮ㄩ棬
         UmsDepartManage departManage = new UmsDepartManage();
         departManage.setUserId(umsAdmin.getId());
@@ -146,7 +174,7 @@
         departManage.setUpdateTime(new Date());
         umsDepartManageMapper.insert(departManage);
 
-        baseMapper.updateById(umsAdmin);
+        // baseMapper.updateById(umsAdmin);
         return umsAdmin;
     }
 
@@ -167,10 +195,10 @@
         //灏嗗瘑鐮佽繘琛屽姞瀵嗘搷浣�
         String encodePassword = passwordEncoder.encode(umsAdmin.getPassword());
 
-        if ( !umsAdmin.getPassword().matches("^(?=.*\\d)(?=.*[a-z])(?=.*[A-Z])[a-zA-Z0-9]{8,16}$")) {
+        if (!umsAdmin.getPassword().matches("^(?=.*\\d)(?=.*[a-z])(?=.*[A-Z])[a-zA-Z0-9]{8,16}$")) {
             Asserts.fail("瀵嗙爜闀垮害8鍒�16浣嶄笖瀵嗙爜涓殑瀛楃蹇呴』鍖呭惈瀛楁瘝锛堝ぇ鍐欐垨鑰呭皬鍐欙級鍜屽繀椤诲寘鍚暟瀛楋紝涓嶈兘鍖呭惈绌烘牸");
         }
-        if(umsAdmin.getUsername().equals(umsAdmin.getPassword())){
+        if (umsAdmin.getUsername().equals(umsAdmin.getPassword())) {
             Asserts.fail("瀵嗙爜涓嶈兘鍜岀櫥褰曞悕瀹屽叏涓�鑷�");
         }
         umsAdmin.setPassword(encodePassword);
@@ -236,6 +264,7 @@
             if (!userDetails.isEnabled()) {
                 Asserts.fail("甯愬彿宸茶绂佺敤");
             }
+
             UsernamePasswordAuthenticationToken authentication = new UsernamePasswordAuthenticationToken(userDetails, null, userDetails.getAuthorities());
             SecurityContextHolder.getContext().setAuthentication(authentication);
             //鏍规嵁鐢ㄦ埛id锛岀敤鎴峰鍚�
@@ -369,7 +398,7 @@
                 && !param.getNewPassword().matches("^(?=.*\\d)(?=.*[a-z])(?=.*[A-Z])[a-zA-Z0-9]{8,16}$")) {
             return -1;
         }
-        if(param.getUsername().equals(param.getNewPassword())){
+        if (param.getUsername().equals(param.getNewPassword())) {
             return -4;
         }
         QueryWrapper<UmsAdmin> wrapper = new QueryWrapper<>();
@@ -393,6 +422,11 @@
         //鑾峰彇鐢ㄦ埛淇℃伅
         UmsAdmin admin = getAdminByUsername(username);
         if (admin != null) {
+            if (admin.getExpirationDate() != null) {
+                if (!admin.getExpirationDate().after(new Date())) {
+                    Asserts.fail("璐﹀彿宸茶繃鏈熻鑱旂郴绠$悊鍛�");
+                }
+            }
             List<UmsMenu> resourceList = getResourceList(admin.getId());
             return new AdminUserDetails(admin, resourceList);
         }
@@ -432,9 +466,13 @@
         if (pageUserVO.getJobTitle() != null && !("".equals(pageUserVO.getJobTitle()))) {
             umsAdmin.setJobTitle(pageUserVO.getJobTitle());
         }
-        Page<UmsAdmin> page1 = baseMapper.selectCondPage(page, umsAdmin);
-
-        return page1;
+        PageUtil.setFirstPageNo(1);
+        int offset = PageUtil.getStart(current, pageSize);
+        List<UmsAdmin> list = baseMapper.selectCondList(umsAdmin, offset, pageSize);
+        Long total = baseMapper.selectCondTotal(umsAdmin);
+        page.setRecords(list);
+        page.setTotal(total);
+        return page;
     }
 
     @Override
@@ -456,7 +494,7 @@
                             roleStrBuffer.append(",");
                         }
                     });
-                    roleStrBuffer.deleteCharAt(roleStrBuffer.length()-1);
+                    roleStrBuffer.deleteCharAt(roleStrBuffer.length() - 1);
                     res1.setRoles(roleStrBuffer.toString());
                     StringBuffer departStrBuffer = new StringBuffer();
                     List<UmsDepart> departs = item.getDepart();
@@ -467,19 +505,19 @@
                             departStrBuffer.append(",");
                         }
                     });
-                    departStrBuffer.deleteCharAt(departStrBuffer.length()-1);
+                    departStrBuffer.deleteCharAt(departStrBuffer.length() - 1);
                     res1.setDepart(departStrBuffer.toString());
                     res1.setJobTitle(item.getJobTitle());
                     Date createTime = item.getCreateTime();
                     SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
-                    if (createTime != null){
+                    if (createTime != null) {
                         res1.setCreateTime(sdf.format(createTime));
                     }
 
                     Integer status = item.getStatus();
-                    if (status.equals(1)){
+                    if (status.equals(1)) {
                         res1.setStatus("鍚敤");
-                    }else {
+                    } else {
                         res1.setStatus("绂佺敤");
                     }
                     return res1;

--
Gitblit v1.8.0