From 1e15bb33cf38603c136380a6e6c669d65d54c57e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=98=BF=E6=B2=90?= <babamu@126.com>
Date: Mon, 1 Apr 2024 16:37:36 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E7=94=A8=E6=88=B7=E5=A4=B4?=
 =?UTF-8?q?=E5=83=8F=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/net/maku/system/controller/SysUserController.java   | 10 ++++++++++
 .../main/java/net/maku/system/service/SysUserService.java    |  3 +++
 .../net/maku/system/service/impl/SysUserServiceImpl.java     | 12 ++++++++++++
 3 files changed, 25 insertions(+)

diff --git a/maku-boot-system/src/main/java/net/maku/system/controller/SysUserController.java b/maku-boot-system/src/main/java/net/maku/system/controller/SysUserController.java
index bbde0ad..2a98a5c 100644
--- a/maku-boot-system/src/main/java/net/maku/system/controller/SysUserController.java
+++ b/maku-boot-system/src/main/java/net/maku/system/controller/SysUserController.java
@@ -18,6 +18,7 @@ import net.maku.system.service.SysPostService;
 import net.maku.system.service.SysUserPostService;
 import net.maku.system.service.SysUserRoleService;
 import net.maku.system.service.SysUserService;
+import net.maku.system.vo.SysUserAvatarVO;
 import net.maku.system.vo.SysUserBaseVO;
 import net.maku.system.vo.SysUserPasswordVO;
 import net.maku.system.vo.SysUserVO;
@@ -100,6 +101,15 @@ public class SysUserController {
         return Result.ok();
     }
 
+    @PutMapping("avatar")
+    @Operation(summary = "修改登录用户头像")
+    @OperateLog(type = OperateTypeEnum.UPDATE)
+    public Result<String> avatar(@RequestBody SysUserAvatarVO avatar) {
+        sysUserService.updateAvatar(avatar);
+
+        return Result.ok();
+    }
+
     @PutMapping("password")
     @Operation(summary = "修改密码")
     @OperateLog(type = OperateTypeEnum.UPDATE)
diff --git a/maku-boot-system/src/main/java/net/maku/system/service/SysUserService.java b/maku-boot-system/src/main/java/net/maku/system/service/SysUserService.java
index 2cd7882..85faa18 100644
--- a/maku-boot-system/src/main/java/net/maku/system/service/SysUserService.java
+++ b/maku-boot-system/src/main/java/net/maku/system/service/SysUserService.java
@@ -5,6 +5,7 @@ import net.maku.framework.mybatis.service.BaseService;
 import net.maku.system.entity.SysUserEntity;
 import net.maku.system.query.SysRoleUserQuery;
 import net.maku.system.query.SysUserQuery;
+import net.maku.system.vo.SysUserAvatarVO;
 import net.maku.system.vo.SysUserBaseVO;
 import net.maku.system.vo.SysUserVO;
 import org.springframework.web.multipart.MultipartFile;
@@ -27,6 +28,8 @@ public interface SysUserService extends BaseService<SysUserEntity> {
 
     void updateLoginInfo(SysUserBaseVO vo);
 
+    void updateAvatar(SysUserAvatarVO avatar);
+
     void delete(List<Long> idList);
 
     SysUserVO getByMobile(String mobile);
diff --git a/maku-boot-system/src/main/java/net/maku/system/service/impl/SysUserServiceImpl.java b/maku-boot-system/src/main/java/net/maku/system/service/impl/SysUserServiceImpl.java
index 7db23ec..9102c3f 100644
--- a/maku-boot-system/src/main/java/net/maku/system/service/impl/SysUserServiceImpl.java
+++ b/maku-boot-system/src/main/java/net/maku/system/service/impl/SysUserServiceImpl.java
@@ -22,6 +22,7 @@ import net.maku.system.enums.SuperAdminEnum;
 import net.maku.system.query.SysRoleUserQuery;
 import net.maku.system.query.SysUserQuery;
 import net.maku.system.service.*;
+import net.maku.system.vo.SysUserAvatarVO;
 import net.maku.system.vo.SysUserBaseVO;
 import net.maku.system.vo.SysUserExcelVO;
 import net.maku.system.vo.SysUserVO;
@@ -162,6 +163,17 @@ public class SysUserServiceImpl extends BaseServiceImpl<SysUserDao, SysUserEntit
     }
 
     @Override
+    public void updateAvatar(SysUserAvatarVO avatar) {
+        SysUserEntity entity = new SysUserEntity();
+        entity.setId(SecurityUser.getUserId());
+        entity.setAvatar(avatar.getAvatar());
+        updateById(entity);
+
+        // 删除用户缓存
+        tokenStoreCache.deleteUser(TokenUtils.getAccessToken());
+    }
+
+    @Override
     public void delete(List<Long> idList) {
         // 删除用户
         removeByIds(idList);