From 7804263c6061aef813f0db27cb3046f746572606 Mon Sep 17 00:00:00 2001
From: admin <weikou2014>
Date: 星期六, 06 三月 2021 16:07:26 +0800
Subject: [PATCH] 后台管理优化

---
 src/main/java/com/yeshi/buwan/service/imp/UserService.java |   12 ++++++++++--
 1 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/yeshi/buwan/service/imp/UserService.java b/src/main/java/com/yeshi/buwan/service/imp/UserService.java
index 75b446e..a6dc65d 100644
--- a/src/main/java/com/yeshi/buwan/service/imp/UserService.java
+++ b/src/main/java/com/yeshi/buwan/service/imp/UserService.java
@@ -12,6 +12,7 @@
 import com.yeshi.buwan.dto.user.QQUserInfo;
 import com.yeshi.buwan.exception.user.LoginUserException;
 import com.yeshi.buwan.exception.user.RegisterUserException;
+import com.yeshi.buwan.service.inter.LoginUserService;
 import org.hibernate.HibernateException;
 import org.hibernate.Session;
 import org.springframework.cache.annotation.Cacheable;
@@ -44,6 +45,9 @@
 
     @Resource
     private LoginUserExtraDao loginUserExtraDao;
+
+    @Resource
+    private LoginUserService loginUserService;
 
 
     // 鐢ㄦ埛鎿嶄綔
@@ -257,7 +261,7 @@
      *
      * @param dto
      */
-    public void register(LoginInfoDto dto) throws RegisterUserException {
+    public LoginUser register(LoginInfoDto dto) throws RegisterUserException {
         LoginUser loginUser = null;
         LoginUserExtra loginUserExtra = null;
         switch (dto.getLoginType()) {
@@ -295,6 +299,7 @@
                 loginUser.setLoginType(LoginUser.LOGIN_TYPE_QQ);
                 loginUser.setQqOpenId(dto.getQqUserInfo().getOpenId());
                 loginUser.setName(dto.getQqUserInfo().getName());
+                loginUser.setPortrait(dto.getQqUserInfo().getPortrait());
 
 
                 loginUserExtra = new LoginUserExtra();
@@ -316,6 +321,7 @@
                 loginUser.setWxUnionId(dto.getWeiXinUser().getUnionid());
                 loginUser.setName(dto.getWeiXinUser().getNickname());
                 loginUser.setWxOpenId(dto.getWeiXinUser().getOpenid());
+                loginUser.setPortrait(dto.getWeiXinUser().getHeadimgurl());
 
 
                 loginUserExtra = new LoginUserExtra();
@@ -371,7 +377,9 @@
 
         loginUserExtra.setId(loginUser.getId());
         loginUserExtra.setCreateTime(new Date());
-        loginUserExtraDao.save(loginUserExtra);
+        loginUserService.initExtra(loginUserExtra);
+
+        return loginUser;
     }
 
     public LoginUser getLoginUser(String id) {

--
Gitblit v1.8.0