新建

src/main/java/com/zmzm/financial/common/controller/AuthController.java

@@ -0,0 +1,67 @@
+package com.zmzm.financial.common.controller;
+
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.zmzm.financial.common.entity.User;
+import com.zmzm.financial.common.entity.UserToken;
+import com.zmzm.financial.common.service.IUserService;
+import com.zmzm.financial.common.service.IUserTokenService;
+import com.zmzm.financial.util.MD5Utils;
+import com.zmzm.financial.util.R;
+import com.zmzm.financial.util.ShiroUtils;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+import java.util.Date;
+
+@Slf4j
+@RestController("auth")
+@RequestMapping("auth")
+public class AuthController {
+
+    @Autowired
+    private IUserService userService;
+    @Autowired
+    private IUserTokenService userTokenService;
+
+    //登录
+    @RequestMapping("/login")
+    public R login(@RequestBody User user){
+        User userEntity = userService.getOne(new LambdaQueryWrapper<User>()
+                .eq(User::getAccount, user.getAccount()));
+        if(userEntity == null){
+            return R.error(500,"账号不存在");
+        }else {
+            if (MD5Utils.getSaltverifyMD5(user.getPassword(),userEntity.getPassword())){
+                UserToken userToken = userTokenService.createToken(userEntity.getId());
+                return R.ok("登录成功").put("userToken",userToken);
+            }else {
+                return R.error(500,"密码不正确，请重试");
+            }
+        }
+    }
+
+    //注册
+    @RequestMapping("/register")
+    public R register(@RequestBody User user){
+        long count = userService.count(new LambdaQueryWrapper<User>().eq(User::getAccount, user.getAccount()));
+        if(count>0){
+            return R.error(500,"账号已注册");
+        }
+        String saltMD5 = MD5Utils.getSaltMD5(user.getPassword());
+        user.setPassword(saltMD5);
+        userService.save(user);
+        return R.ok();
+    }
+
+    //退出
+    @RequestMapping("/logout")
+    public R logout(){
+        UserToken userToken = userTokenService.getById(ShiroUtils.getUserId());
+        userToken.setExpireTime(new Date());
+        userTokenService.updateById(userToken);
+        return R.ok();
+    }
+
+}