From 47c21942bd2ba890ab554a15766c5faf31ef51e3 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Wed, 14 Aug 2024 05:31:45 +0800
Subject: [PATCH] 1
---
websocketSerivce/src/main/java/org/example/controller/UserController.java | 270 ++++++++++++++++++++++++++++++++++++++++++++++++++---
1 files changed, 253 insertions(+), 17 deletions(-)
diff --git a/websocketSerivce/src/main/java/org/example/controller/UserController.java b/websocketSerivce/src/main/java/org/example/controller/UserController.java
index e37b31b..0be2408 100644
--- a/websocketSerivce/src/main/java/org/example/controller/UserController.java
+++ b/websocketSerivce/src/main/java/org/example/controller/UserController.java
@@ -1,17 +1,33 @@
package org.example.controller;
-
+import cn.hutool.core.convert.Convert;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.core.type.TypeReference;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.google.gson.Gson;
+import com.google.gson.GsonBuilder;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang.StringUtils;
import org.example.common.ServerResponse;
import org.example.dao.ConfigCurrencyMapper;
-import org.example.pojo.ConfigCurrency;
-import org.example.pojo.User;
+import org.example.dao.LogMapper;
+import org.example.dao.MenuMapper;
+import org.example.pojo.*;
+import org.example.pojo.vo.DeleteConfigVo;
import org.example.pojo.vo.SaveConfigVo;
+import org.example.pojo.vo.UpdateUserVo;
import org.example.server.impl.UserServiceImpl;
-import org.example.util.MD5Util;
+import org.example.util.*;
import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RequestParam;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
+
+import javax.servlet.http.HttpServletRequest;
+import java.security.SecureRandom;
+import java.sql.Date;
+import java.time.LocalDate;
+import java.util.*;
+import java.util.stream.Collectors;
/**
* @program: demo
@@ -20,6 +36,8 @@
**/
@RestController
@RequestMapping("/user")
+@CrossOrigin(origins = "*")
+@Slf4j
public class UserController {
@Autowired
@@ -28,21 +46,130 @@
@Autowired
private UserServiceImpl userService;
+ @Autowired
+ private LogMapper logMapper;
+
+ @Autowired
+ private MenuMapper menuMapper;
+
+ @PostMapping("/login")
+ @ResponseBody
+ public ServerResponse saveUser(@RequestParam("account") String account
+ , @RequestParam("password") String password, HttpServletRequest request) throws JsonProcessingException {
+ try {
+ User user = userService.getOne(new LambdaQueryWrapper<User>().eq(User::getAccount, account));
+ if(null == user){
+ return ServerResponse.createByErrorMsg("用户不存在");
+ }
+
+ if (!MD5Util.verify(password, user.getPassword())) {
+ return ServerResponse.createByErrorMsg("密码错误");
+ }
+ List<Menu> menus = menuMapper.selectList(new LambdaQueryWrapper<Menu>());
+ if(user.getIsRoot() == 1){
+ String token = JwtUtil.getToken(user);
+ Map<String,Object> map = new HashMap<>();
+ map.put("token",token);
+ map.put("user",user);
+ map.put("menu",menus);
+ extracted(account, request);
+ RedisUtil.set(user.getId().toString(),token);
+ return ServerResponse.createBySuccess(map);
+ }
+
+ //判断是否锁定
+ if(user.getIsLock() == 1){
+ return ServerResponse.createByErrorMsg("账号已被锁定");
+ }
+
+ //判断是否到期
+ if(new java.util.Date().after(user.getEndTime())){
+ return ServerResponse.createByErrorMsg("账号已到期");
+ }
+
+ String token = JwtUtil.getToken(user);
+ Map<String,Object> map = new HashMap<>();
+ map.put("token",token);
+ List<Menu> menuList = menus.stream().filter(f -> f.getIsRoot() == 0).collect(Collectors.toList());
+ map.put("menu",menuList);
+ map.put("user",user);
+ RedisUtil.set(user.getId().toString(),token);
+
+ extracted(account, request);
+ return ServerResponse.createBySuccess(map);
+ }catch (Exception e){
+ e.printStackTrace();
+ log.error("登录异常："+e.getMessage());
+ }
+ return ServerResponse.createByErrorMsg("系统异常");
+ }
+
+ private void extracted(String account, HttpServletRequest request) {
+ String ip = IpAddressUtil.getIpAddress(request);
+ String address = IpAddressUtil.getIpPossessionByFile(ip);
+ Log log = new Log();
+ log.setIp(ip);
+ log.setAccount(account);
+ log.setLoginTime(new java.util.Date());
+ log.setAddress(address);
+ logMapper.insert(log);
+ }
+
@PostMapping("/saveConfig")
- public ServerResponse saveConfig(SaveConfigVo saveConfigVo) {
+ public ServerResponse saveConfig(@RequestBody SaveConfigVo saveConfigVo,HttpServletRequest request) {
+ String token = request.getHeader("token");
+ User user = JwtUtil.verify(token);
saveConfigVo.getCurrencyList().forEach(f->{
ConfigCurrency currency = new ConfigCurrency();
- currency.setUserId(saveConfigVo.getUserId());
+ currency.setUserId(user.getId());
currency.setCurrency(f.getCurrency());
currency.setBuy(f.getBuy());
currency.setSell(f.getSell());
currencyMapper.insert(currency);
});
+ List<ConfigCurrency> configCurrencies = currencyMapper.selectList(new LambdaQueryWrapper<ConfigCurrency>().eq(ConfigCurrency::getUserId, user.getId()));
+ String key = "config_";
+ Gson gson = new GsonBuilder().setPrettyPrinting().create();
+ String json = gson.toJson(configCurrencies);
+ RedisUtil.set(key+user.getId(),json);
return ServerResponse.createBySuccess();
}
+ @GetMapping("/deleteConfig")
+ public ServerResponse deleteConfig(@RequestParam("id") int id,HttpServletRequest request) {
+ String token = request.getHeader("token");
+ User user = JwtUtil.verify(token);
+ currencyMapper.deleteById(id);
+ List<ConfigCurrency> configCurrencies = currencyMapper.selectList(new LambdaQueryWrapper<ConfigCurrency>().eq(ConfigCurrency::getUserId, user.getId()));
+ String key = "config_";
+ Gson gson = new GsonBuilder().setPrettyPrinting().create();
+ String json = gson.toJson(configCurrencies);
+ RedisUtil.set(key+user.getId(),json);
+ return ServerResponse.createBySuccess();
+ }
+
+ @GetMapping("/selectConfig")
+ public ServerResponse selectConfig(@RequestParam(value = "currency", required = false) String currency,
+ @RequestParam("pageNum") int pageNum,
+ @RequestParam("pageSize") int pageSize,
+ HttpServletRequest request) {
+ String token = request.getHeader("token");
+ User user = JwtUtil.verify(token);
+ Page<ConfigCurrency> page = new Page<>(pageNum, pageSize);
+ Page<ConfigCurrency> configCurrencyPage = currencyMapper.selectPage(page, new LambdaQueryWrapper<ConfigCurrency>()
+ .eq(StringUtils.isNotEmpty(currency), ConfigCurrency::getCurrency, currency).eq(ConfigCurrency::getUserId,user.getId()));
+ return ServerResponse.createBySuccess(configCurrencyPage);
+
+ }
+
+
@PostMapping("/saveUser")
- public ServerResponse saveUser(User user) {
+ public ServerResponse saveUser(@RequestBody User user) {
+ long count = userService.count(new LambdaQueryWrapper<User>().eq(User::getAccount, user.getAccount()));
+ if(count > 0){
+ return ServerResponse.createByErrorMsg("账号已存在");
+ }
+ user.setAddTime(Date.valueOf(LocalDate.now()));
user.setPassword(MD5Util.encrypt(user.getPassword()));
userService.save(user);
return ServerResponse.createBySuccess();
@@ -51,21 +178,130 @@
@PostMapping("/deleteUser")
public ServerResponse deleteUser(@RequestParam("id") int id) {
User byId = userService.getById(id);
- if(null == byId){
+ if(null == byId || byId.getIsRoot() == 1){
return ServerResponse.createByErrorMsg("用户不存在");
}
userService.removeById(id);
return ServerResponse.createBySuccess();
}
- @PostMapping("/updateUser")
- public ServerResponse deleteUser(User user) {
- User byId = userService.getById(user.getId());
- if(null == byId){
+ @GetMapping("/selectById")
+ public ServerResponse selectById(@RequestParam("id") Integer id) {
+ User byId = userService.getById(id);
+ if(null == byId || byId.getIsRoot() == 1){
return ServerResponse.createByErrorMsg("用户不存在");
}
- user.setPassword(MD5Util.encrypt(user.getPassword()));
+ return ServerResponse.createBySuccess(byId);
+ }
+
+ @PostMapping("/updateUser")
+ public ServerResponse deleteUser(@RequestBody UpdateUserVo updateUserVo) {
+ User user = userService.getById(updateUserVo.getId());
+ if(null == user || user.getIsRoot() == 1){
+ return ServerResponse.createByErrorMsg("用户不存在");
+ }
+ user.setAccount(updateUserVo.getAccount());
+ user.setEndTime(updateUserVo.getEndTime());
+ user.setIsLock(updateUserVo.getIsLock());
userService.updateById(user);
return ServerResponse.createBySuccess();
}
+
+ @PostMapping("/updatePassword")
+ public ServerResponse updatePassword(@RequestParam("id") Integer id,@RequestParam("password") String password) {
+ User user = userService.getById(id);
+ if(null == user || user.getIsRoot() == 1){
+ return ServerResponse.createByErrorMsg("用户不存在");
+ }
+ user.setPassword(MD5Util.encrypt(password));
+ userService.updateById(user);
+ return ServerResponse.createBySuccess();
+ }
+
+ @GetMapping("/selectUser")
+ public ServerResponse selectUser(@RequestParam(value = "account", required = false) String account,
+ @RequestParam("pageNum") int pageNum,
+ @RequestParam("pageSize") int pageSize) {
+ Page<User> page = new Page<>(pageNum, pageSize);
+ Page<User> pageList = userService.page(page, new LambdaQueryWrapper<User>()
+ .like(StringUtils.isNotEmpty(account), User::getAccount, account)
+ .eq(User::getIsShow,0)
+ .eq(User::getIsRoot,0)
+ .orderByDesc(User::getAddTime));
+ return ServerResponse.createBySuccess(pageList);
+ }
+
+ @GetMapping("/selectLog")
+ public ServerResponse selectLog(@RequestParam(value = "account", required = false) String account,
+ @RequestParam("pageNum") int pageNum,
+ @RequestParam("pageSize") int pageSize) {
+ Page<Log> page = new Page<>(pageNum, pageSize);
+ Page<Log> pageList = logMapper.selectPage(page, new LambdaQueryWrapper<Log>()
+ .like(StringUtils.isNotEmpty(account), Log::getAccount, account)
+ .orderByDesc(Log::getLoginTime));
+ return ServerResponse.createBySuccess(pageList);
+ }
+
+ @GetMapping("/addMark")
+ public ServerResponse addMark(@RequestParam(value = "favorites") String favorites,
+ HttpServletRequest request) {
+ String token = request.getHeader("token");
+ User user = JwtUtil.verify(token);
+
+ if (user == null) {
+ // 处理用户验证失败的情况，例如返回未授权错误
+ return ServerResponse.createByErrorMsg("标记失败");
+ }
+
+ String mark = RedisUtil.get(user.getId() + "_mark");
+
+ if (StringUtils.isNotEmpty(mark)) {
+ // 使用 StringJoiner 或者 StringBuilder 来拼接字符串
+ mark += "," + favorites;
+ } else {
+ mark = favorites;
+ }
+
+ // 更新 Redis 中的标记
+ RedisUtil.set(user.getId() + "_mark", mark);
+
+ // 返回操作成功的响应
+ return ServerResponse.createBySuccess("标记成功");
+ }
+
+ @GetMapping("/deleteMark")
+ @ResponseBody
+ public ServerResponse deleteMark(@RequestParam(value = "favorites") String favorites,
+ HttpServletRequest request) {
+ String token = request.getHeader("token");
+ User user = JwtUtil.verify(token);
+
+ String mark = RedisUtil.get(user.getId() + "_mark");
+
+ if (StringUtils.isNotEmpty(mark)) {
+ // 使用逗号分隔符将标记分割成数组
+ String[] marks = mark.split(",");
+
+ // 创建一个新的标记列表，用于存储没有要删除的标记
+ List<String> updatedMarks = new ArrayList<>();
+
+ // 遍历现有的标记，将不等于要删除的标记加入新列表
+ for (String m : marks) {
+ if (!m.equals(favorites)) {
+ updatedMarks.add(m);
+ }
+ }
+
+ // 将新的标记列表转换为逗号分隔的字符串
+ mark = String.join(",", updatedMarks);
+
+ // 更新 Redis 中的标记
+ RedisUtil.set(user.getId() + "_mark", mark);
+
+
+ }
+ // 返回操作成功的响应
+ return ServerResponse.createBySuccess("取消标记成功");
+ }
+
}
--
Gitblit v1.9.3