初始代码

yunzhupaas-oauth/yunzhupaas-oauth-api/pom.xml

@@ -0,0 +1,22 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <parent>
+        <artifactId>yunzhupaas-oauth</artifactId>
+        <groupId>com.yunzhupaas</groupId>
+        <version>5.2.0-RELEASE</version>
+    </parent>
+    <modelVersion>4.0.0</modelVersion>
+
+    <artifactId>yunzhupaas-oauth-api</artifactId>
+
+    <dependencies>
+        <dependency>
+            <groupId>com.yunzhupaas</groupId>
+            <artifactId>yunzhupaas-oauth-entity</artifactId>
+            <version>${project.version}</version>
+        </dependency>
+    </dependencies>
+
+</project>

yunzhupaas-oauth/yunzhupaas-oauth-api/src/main/java/com/yunzhupaas/service/AuthService.java

@@ -0,0 +1,16 @@
+package com.yunzhupaas.service;
+
+import com.yunzhupaas.base.ActionResult;
+import com.yunzhupaas.exception.LoginException;
+import com.yunzhupaas.model.LoginVO;
+
+import java.util.Map;
+
+public interface AuthService {
+    ActionResult<LoginVO> login(Map<String, String> parameters) throws LoginException;
+
+    ActionResult kickoutByToken(String... tokens);
+
+    ActionResult kickoutByUserId(String userId, String tenantId);
+}
+

yunzhupaas-oauth/yunzhupaas-oauth-api/src/main/java/com/yunzhupaas/util/AuthUtil.java

@@ -0,0 +1,95 @@
+package com.yunzhupaas.util;
+
+import com.google.common.collect.ImmutableMap;
+import com.yunzhupaas.base.ActionResult;
+import com.yunzhupaas.base.UserInfo;
+import com.yunzhupaas.exception.LoginException;
+import com.yunzhupaas.model.LoginVO;
+import com.yunzhupaas.service.AuthService;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.util.Map;
+import java.util.Optional;
+
+/**
+ * 内部登录、退出用户工具
+ */
+@Slf4j
+@Component
+public class AuthUtil {
+
+    public static AuthService authApi;
+
+    @Autowired
+    public void setAuthApi(AuthService authApi) {
+        AuthUtil.authApi = authApi;
+    }
+
+    /**
+     * 登录临时用户
+     * 此用户已经登录将返回现有用户Token
+     * 未登录将直接使用用户ID进行免密登录返回Token
+     *
+     * @param userId 用户ID
+     * @param tenantId 租户ID
+     * @return
+     */
+    public static String loginTempUser(String userId, String tenantId){
+        return loginTempUser(userId, tenantId, false);
+    }
+
+    /**
+     * 登录临时用户
+     * 此用户已经登录将返回现有用户Token
+     * 未登录将直接使用用户ID进行免密登录返回Token
+     *
+     * @param userId 用户ID
+     * @param tenantId 租户ID
+     * @param limited 是否是限制型临时用户(无法登录主系统前端)
+     * @return
+     */
+    public static String loginTempUser(String userId, String tenantId, boolean limited){
+        Map<String, String> loginInfo = ImmutableMap.of(
+                "grant_type", "tempuser",
+                "token", UserProvider.getInnerAuthToken(),
+                "userId", userId,
+                "tenantId", Optional.ofNullable(tenantId).orElse(StringUtil.NULLSTR),
+                "limited", String.valueOf(limited));
+        String errMsg;
+        ActionResult<LoginVO> result = authApi.login(loginInfo);
+        if(Constants.SUCCESS.equals(result.getCode())){
+            return result.getData().getToken();
+        }
+        log.error("登录临时用户失败: {}", result.getMsg());
+        throw new LoginException(result.getMsg());
+    }
+
+    /**
+     * 踢出用户, 用户将收到Websocket下线通知
+     * 执行流程：认证服务退出用户->用户踢出监听->消息服务发送Websocket推送退出消息
+     * @param tokens
+     */
+    public static void kickoutByToken(String... tokens){
+        authApi.kickoutByToken(tokens);
+    }
+
+    /**
+     * 踢出用户, 用户将收到Websocket下线通知
+     * 执行流程：认证服务退出用户->用户踢出监听->消息服务发送Websocket推送退出消息
+     * @param userId
+     */
+    public static void kickoutByUserId(String userId){
+        String tenantId = TenantHolder.getDatasourceId();
+        if(tenantId == null) {
+            UserInfo userInfo = UserProvider.getUser();
+            if (userInfo.getUserId() == null) {
+                throw new RuntimeException("请设置UserInfo");
+            }
+            tenantId = userInfo.getTenantId();
+        }
+        authApi.kickoutByUserId(userId, tenantId);
+    }
+}
+