From 3ce8d58220cfe2b7330c7d59243a7d5ad578f824 Mon Sep 17 00:00:00 2001
From: TianZhou <3570544406@qq.com>
Date: Tue, 23 Jul 2024 10:45:53 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E6=88=90=E4=BA=86=E7=94=A8=E6=88=B7?=
 =?UTF-8?q?=E4=B8=AD=E5=BF=83?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../net/maku/maku/controller/TUserController.java  | 27 ++++++++++++++++++----
 .../src/main/java/net/maku/maku/vo/TUserVO.java    |  4 ++++
 2 files changed, 27 insertions(+), 4 deletions(-)

diff --git a/maku-server/src/main/java/net/maku/maku/controller/TUserController.java b/maku-server/src/main/java/net/maku/maku/controller/TUserController.java
index 9466208..ba27ed9 100644
--- a/maku-server/src/main/java/net/maku/maku/controller/TUserController.java
+++ b/maku-server/src/main/java/net/maku/maku/controller/TUserController.java
@@ -81,21 +81,40 @@ public class TUserController {
         if(ObjectUtils.isNull(entity)){
             throw new ServerException(ErrorCode.REFRESH_TOKEN_INVALID);
         }
-
-        return Result.ok(entity);
+        TUserEntity p = tUserService.getById(entity.getId());
+        redisCache.set(token, p);
+        return Result.ok(p);
     }
 
     @GetMapping("/recharge")
     @Operation(summary = "充值")
-    public Result<String> recharge(HttpServletRequest request,@RequestParam(value = "balance",required =false) int balance){
+    public Result<String> recharge(HttpServletRequest request,@RequestParam(value = "balance") int balance){
         String token = request.getHeader("token");
-        TUserEntity entity = (TUserEntity) redisCache.get(token);;
+        TUserEntity entity = (TUserEntity) redisCache.get(token);
+        if(ObjectUtils.isNull(entity)){
+            throw new ServerException(ErrorCode.REFRESH_TOKEN_INVALID);
+        }
         int money=entity.getBalance();
         entity.setBalance(money+balance);
         tUserService.updateById(entity);
         return Result.ok();
     }
 
+    @PutMapping("/userInfo")
+    @Operation(summary = "更新资料")
+    public Result<String> userInfo(HttpServletRequest request,@RequestBody TUserVO vo){
+        String token = request.getHeader("token");
+        TUserEntity tokenUser = (TUserEntity) redisCache.get(token);
+        TUserEntity entity = tUserService.getById(tokenUser.getId());
+        if(ObjectUtils.isNull(entity)){
+            throw new ServerException(ErrorCode.REFRESH_TOKEN_INVALID);
+        }
+        entity.setEmail(vo.getEmail());
+        entity.setPhone(vo.getPassword());
+        tUserService.updateById(entity);
+        return Result.ok("success");
+    }
+
 
     @PostMapping("/logout")
     @Operation(summary = "退出登录")
diff --git a/maku-server/src/main/java/net/maku/maku/vo/TUserVO.java b/maku-server/src/main/java/net/maku/maku/vo/TUserVO.java
index 97912fa..6c98b92 100644
--- a/maku-server/src/main/java/net/maku/maku/vo/TUserVO.java
+++ b/maku-server/src/main/java/net/maku/maku/vo/TUserVO.java
@@ -31,6 +31,10 @@ public class TUserVO implements Serializable {
 
 	private Integer status;
 
+	private String email;
+
+	private String phonne;
+
 	@JsonFormat(pattern = DateUtils.DATE_TIME_PATTERN)
 	private Date createTime;