智慧农业后台管理
guoshilong
2022-11-08 72b971c98c46531064d74b68fc8fe8864ed0b544
src/main/java/org/springblade/modules/auth/endpoint/BladeTokenEndPoint.java
@@ -35,6 +35,7 @@
import org.springblade.core.secure.BladeUser;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.core.tenant.annotation.NonDS;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.support.Kv;
import org.springblade.core.tool.utils.Func;
import org.springblade.core.tool.utils.StringUtil;
@@ -68,6 +69,13 @@
   private final BladeRedis bladeRedis;
   private final JwtProperties jwtProperties;
   @ApiLog("根据token获取登录用户")
   @GetMapping("/getuser/token")
   public R<Object> getUserLoginInfoByToken() {
      String token = JwtUtil.getToken(WebUtil.getRequest().getHeader(TokenConstant.HEADER));
      return R.data(bladeRedis.get(token));
   }
   @ApiLog("登录用户验证")
   @PostMapping("/oauth/token")
@@ -106,8 +114,8 @@
      }
      Kv authInfo1 = TokenUtil.createAuthInfo(userInfo);
      String access_token = authInfo1.get("access_token").toString();
      // 存入redis并设置过期时间为30分钟
      bladeRedis.setEx(access_token, access_token, Duration.ofMinutes(1));
      // 存入redis并设置过期时间为60分钟
      bladeRedis.setEx(access_token, authInfo1, Duration.ofMinutes(60));
      return TokenUtil.createAuthInfo(userInfo);
   }
@@ -116,10 +124,12 @@
   @ApiOperation(value = "退出登录")
   public Kv logout() {
      BladeUser user = AuthUtil.getUser();
      String token = JwtUtil.getToken(WebUtil.getRequest().getHeader(TokenConstant.HEADER));
      // 清空redis保存的token
      if (user != null && jwtProperties.getState()) {
         String token = JwtUtil.getToken(WebUtil.getRequest().getHeader(TokenConstant.HEADER));
         JwtUtil.removeAccessToken(user.getTenantId(), String.valueOf(user.getUserId()), token);
      }
      bladeRedis.del(token);
      return Kv.create().set("success", "true").set("msg", "success");
   }