代码提交

k94314517

2024-07-24 6a475442955b0ffb480facb92f6a23aa325034ec

 server/web/src/main/java/com/doumee/api/web/ApiController.java

@@ -1,5 +1,6 @@
package com.doumee.api.web;

import com.alibaba.fastjson.JSONObject;
import com.doumee.config.Jwt.JwtTokenUtil;
import com.doumee.core.constants.ResponseStatus;
import com.doumee.core.exception.BusinessException;
@@ -8,6 +9,7 @@
import com.doumee.service.business.MemberService;
import com.doumee.service.business.UsersService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.apache.poi.ss.formula.ptg.MemAreaPtg;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
@@ -54,6 +56,10 @@
        return obj != null ? (Long) obj : null;
    }

    protected String getToken() {
        Object obj = this.getRequest().getAttribute(JwtTokenUtil.HEADER_KEY);
        return obj != null ? (String) obj : null;
    }


    protected Long getUserId() {
@@ -64,25 +70,35 @@
        return obj != null ? (Long) obj : null;
    }
    protected Users getLoginUserInfo() {
      Long userId = getUserId();
      if(userId== null){
          throw new BusinessException(ResponseStatus.SERVER_ERROR.getCode(),"未登录");
      }
        Users user = usersService.findById(userId);
      return user;
        Long userId = getUserId();
        if(userId== null){
            throw new BusinessException(ResponseStatus.SERVER_ERROR.getCode(),"未登录");
        }
        Object obj = this.getRequest().getAttribute(JwtTokenUtil.UserInfo);
        String userInfo = obj != null ? (String) obj : null;
        Users user = new Users();
        if(StringUtils.isNotBlank(userInfo)){
            user = JSONObject.toJavaObject(JSONObject.parseObject(userInfo),Users.class);
        }
        return user;
    }
    protected Member getLoginMemberInfo() {
      Long userId = getMemberId();
      if(userId== null){
          return null;
      }
        Member user = memberService.findById(userId);
      return user;
        Object obj = this.getRequest().getAttribute(JwtTokenUtil.UserInfo);
        String userInfo = obj != null ? (String) obj : null;
        Member member = new Member();
        if(StringUtils.isNotBlank(userInfo)){
            member = JSONObject.toJavaObject(JSONObject.parseObject(userInfo),Member.class);
        }
      return member;
    }

    protected Long getUserType() {
    protected String getUserType() {
        Object obj = this.getRequest().getAttribute(JwtTokenUtil.UserType);
        return obj != null ? (Long) obj : null;
        return obj != null ? (String) obj : null;
    }

}