From ea4884b44a6b44c8c4a2948fc643bdea6656a210 Mon Sep 17 00:00:00 2001
From: lidongdong <1459917685@qq.com>
Date: 星期日, 09 十月 2022 16:16:16 +0800
Subject: [PATCH] 修改登录测试

---
 flower_city/src/main/java/com/dg/core/controller/UserController.java |   74 +++++++++++++++++++++++++++++++++----
 1 files changed, 66 insertions(+), 8 deletions(-)

diff --git a/flower_city/src/main/java/com/dg/core/controller/UserController.java b/flower_city/src/main/java/com/dg/core/controller/UserController.java
index 4dfe743..6c8441b 100644
--- a/flower_city/src/main/java/com/dg/core/controller/UserController.java
+++ b/flower_city/src/main/java/com/dg/core/controller/UserController.java
@@ -4,20 +4,19 @@
 import com.dg.core.HttpStatus;
 import com.dg.core.ResultData;
 import com.dg.core.annotation.Authorization;
+import com.dg.core.db.gen.entity.HuaChengSysUser;
 import com.dg.core.db.gen.entity.SysUser;
 import com.dg.core.manager.TokenManager;
+import com.dg.core.service.IHuaChengSysUserService;
 import com.dg.core.util.SmsUtil;
 import com.dg.core.util.TableDataInfo;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.data.redis.core.RedisTemplate;
-import org.springframework.data.repository.query.Param;
 import org.springframework.util.Assert;
 import org.springframework.web.bind.annotation.*;
 
-import javax.annotation.Resource;
 import javax.servlet.http.HttpServletRequest;
 import java.time.LocalDateTime;
 import java.util.HashMap;
@@ -37,20 +36,79 @@
     @Autowired
     TokenManager tokenManager;
 
+    /**
+     * 花城e+用户接口
+     */
+    @Autowired
+    IHuaChengSysUserService iHuaChengSysUserService;
+
     @ApiOperation("登录接口")
     @PostMapping(path = "/login")
     public ResultData login(@RequestParam String account, @RequestParam String password, HttpServletRequest request) {
         Assert.notNull(account, "username can not be empty");
         Assert.notNull(password, "password can not be empty");
 
-        SysUser user = IUserService.getUserByAccount(account);
-        if (user == null || !password.equals(user.getPassword())) {
+        SysUser user=null;
+        //管理员登录
+        if(StringUtils.equals("zigonggaoadmin",account) || StringUtils.equals("ceshirenyuan",account) )
+        {
+            user= IUserService.getUserByAccount(account);
+
+            if(user==null || !password.equals(user.getPassword()))
+            {
+                //提示用户名或密码错误
+                return ResultData.fail(HttpStatus.UNAUTHORIZED, "用户名或密码错误");
+            }
+            String token = tokenManager.getTokenByUserId(user.getUserId());
+            System.out.println("token "+token);
+            if (token == null) {
+                //生成一个token,保存用户登录状态
+                token = tokenManager.createToken(user.getUserId(), user.getUserId());
+            }
+
+            user.setLoginDate(LocalDateTime.now());
+            //更新登录时间
+            IUserService.updateConfig(user);
+
+            Map<String, Object> map = new HashMap<>();
+            map.put("token", token);
+            map.put("userId", user.getUserId());
+            map.put("userName", user.getUserName());
+
+            return ResultData.success(map);
+        }
+
+        HuaChengSysUser huaChengSysUser=iHuaChengSysUserService.selectData("",account,"");
+        if(huaChengSysUser==null || !password.equals(huaChengSysUser.getPassword()))
+        {
             //提示用户名或密码错误
             return ResultData.fail(HttpStatus.UNAUTHORIZED, "用户名或密码错误");
         }
 
+        //本地表查询用户
+        user = IUserService.selectData(huaChengSysUser.getUserId()+"");
+        if (user == null) {
+            //本地没有此用户  新用户
+            SysUser newSysUser=new SysUser();
+            //默认  1系统后台用户
+            newSysUser.setUserType("1");
+            newSysUser.setPhonenumber(huaChengSysUser.getPhone());
+            newSysUser.setUserId(huaChengSysUser.getUserId());
+            newSysUser.setUserName(huaChengSysUser.getNickName());
+            newSysUser.setSex(huaChengSysUser.getSex());
+            newSysUser.setAvatar(huaChengSysUser.getImageUrl());
+            newSysUser.setCreateTime(LocalDateTime.now());
+            newSysUser.setUpdateTime(LocalDateTime.now());
+            IUserService.insertConfig(newSysUser);
+        }
+
+        user = IUserService.selectData(huaChengSysUser.getUserId()+"");
+
+        if (user == null) {
+            return ResultData.error("用户不存在!请联系管理员");
+        }
+
         String token = tokenManager.getTokenByUserId(user.getUserId());
-//        String token = tokenManager.createToken(user.getUserId(), user.getRoleId());
         System.out.println("token "+token);
         if (token == null) {
             //生成一个token,保存用户登录状态
@@ -204,7 +262,7 @@
             return error("userId不能为空");
         }
 
-        SysUser sysUser=IUserService.getUserById(Long.valueOf(userId));
+        SysUser sysUser=IUserService.selectData(userId);
         //删除标志(0代表存在 2代表删除)
         sysUser.setDelFlag("2");
         return toAjax(IUserService.updateConfig(sysUser));
@@ -232,7 +290,7 @@
             return error("密码不能为空");
         }
 
-        SysUser sysUser=IUserService.getUserById(config.getUserId());
+        SysUser sysUser=IUserService.selectData(config.getUserId()+"");
         if(sysUser==null)
         {
             return error("用户不存在");

--
Gitblit v1.7.1