feat: prepare for adding sso

Author: easepan

pom.xml

@@ -275,8 +275,8 @@
 
     <developers>
         <developer>
-            <name>zhazhapan</name>
-            <email>tao@zhazhapan.com</email>
+            <name>easepan</name>
+            <email>easepan@qq.com</email>
             <url>https://blog.csdn.net/qq_26954773</url>
         </developer>
     </developers>

src/main/java/org/code4everything/boot/base/ReferenceUtils.java

@@ -0,0 +1,29 @@
+package org.code4everything.boot.base;
+
+import java.lang.ref.Reference;
+import java.util.Objects;
+
+/**
+ * 引用工具类
+ *
+ * @author pantao
+ * @since 2019/6/26
+ **/
+public class ReferenceUtils {
+
+    private ReferenceUtils() {}
+
+    /**
+     * 获取引用值
+     *
+     * @param reference 值的引用
+     * @param <T> 值类型
+     *
+     * @return 值
+     *
+     * @since 1.1.5
+     */
+    public static <T> T safeUnwrap(Reference<T> reference) {
+        return Objects.isNull(reference) ? null : reference.get();
+    }
+}

src/main/java/org/code4everything/boot/web/sso/LoginVO.java

@@ -0,0 +1,41 @@
+package org.code4everything.boot.web.sso;
+
+import org.code4everything.boot.base.bean.BaseBean;
+
+import java.io.Serializable;
+
+/**
+ * @author pantao
+ * @since 2019/6/26
+ **/
+public class LoginVO implements BaseBean, Serializable {
+
+    private static final long serialVersionUID = -3238074886033708538L;
+
+    private String token;
+
+    private String redirectUrl;
+
+    public LoginVO() {}
+
+    public LoginVO(String token, String redirectUrl) {
+        this.token = token;
+        this.redirectUrl = redirectUrl;
+    }
+
+    public String getToken() {
+        return token;
+    }
+
+    public void setToken(String token) {
+        this.token = token;
+    }
+
+    public String getRedirectUrl() {
+        return redirectUrl;
+    }
+
+    public void setRedirectUrl(String redirectUrl) {
+        this.redirectUrl = redirectUrl;
+    }
+}

src/main/java/org/code4everything/boot/web/sso/SSOAware.java

@@ -0,0 +1,50 @@
+package org.code4everything.boot.web.sso;
+
+import cn.hutool.core.util.IdUtil;
+import org.code4everything.boot.web.mvc.Response;
+
+import javax.annotation.Nullable;
+
+/**
+ * @author pantao
+ * @since 2019/6/26
+ **/
+public interface SSOAware {
+
+    /**
+     * 生成令牌
+     *
+     * @return
+     */
+    default String generateToken() {
+        return IdUtil.simpleUUID();
+    }
+
+    /**
+     * 检测用户是否已经登录
+     *
+     * @param sid 会话唯一标识符
+     * @param addr
+     * @param url 用户当前的页面路径
+     *
+     * @return 验证结果
+     *
+     * @since 1.1.5
+     */
+    Response<String> validateSession(String sid, String addr, @Nullable String url);
+
+    /**
+     * 用户登录成功后，注册子系统
+     *
+     * @param sid 会话唯一标识符
+     * @param addr 子系统地址
+     * @param user 用户
+     *
+     * @return 注册结果
+     *
+     * @since 1.1.5
+     */
+    Response<LoginVO> registerSubSystem(String sid, String addr, Object user);
+
+    Response<Object> getUser(String sid);
+}

src/main/java/org/code4everything/boot/web/sso/SSOUtils.java

@@ -0,0 +1,15 @@
+package org.code4everything.boot.web.sso;
+
+/**
+ * @author pantao
+ * @since 2019/6/26
+ **/
+public class SSOUtils {
+
+    private static final SSOAware aware = new SimpleSSO();
+
+    private SSOUtils() {}
+
+    public static String validateSession(SSOAware ssoAware, String sid) {
+    }
+}

src/main/java/org/code4everything/boot/web/sso/SimpleSSO.java

@@ -0,0 +1,94 @@
+package org.code4everything.boot.web.sso;
+
+import cn.hutool.core.util.StrUtil;
+import org.code4everything.boot.base.ReferenceUtils;
+import org.code4everything.boot.base.constant.IntegerConsts;
+import org.code4everything.boot.web.mvc.Response;
+
+import javax.annotation.Nullable;
+import java.lang.ref.SoftReference;
+import java.lang.ref.WeakReference;
+import java.util.HashSet;
+import java.util.Map;
+import java.util.Objects;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
+
+/**
+ * @author pantao
+ * @since 2019/6/26
+ **/
+class SimpleSSO implements SSOAware {
+
+    private String loginUrl;
+
+    private int sessionTimeout;
+
+    /**
+     * 缓存登录的用户
+     *
+     * @since 1.1.5
+     */
+    private Map<String, SoftReference<Object>> userCache = new ConcurrentHashMap<>(1024);
+
+    /**
+     * 缓存注册的子系统
+     *
+     * @since 1.1.5
+     */
+    private Map<String, SoftReference<Set<String>>> subSysCache = new ConcurrentHashMap<>(1024);
+
+    /**
+     * 缓存重定向路径
+     *
+     * @since 1.1.5
+     */
+    private Map<String, WeakReference<String>> urlCache = new ConcurrentHashMap<>(1024);
+
+    public SimpleSSO() {
+        this("/error", IntegerConsts.ONE_DAY_SECONDS);
+    }
+
+    public SimpleSSO(String loginUrl, int sessionTimeout) {
+        this.loginUrl = loginUrl;
+    }
+
+    public SimpleSSO loginUrl(String loginUrl) {
+        this.loginUrl = loginUrl;
+        return this;
+    }
+
+    @Override
+    public Response<String> validateSession(String sid, String addr, @Nullable String url) {
+        if (userCache.containsKey(sid)) {
+            addSubSys(sid, addr);
+            return new Response<>("用户已登录", generateToken());
+        }
+        if (StrUtil.isNotEmpty(url)) {
+            urlCache.put(sid, new WeakReference<>(url));
+        }
+        return new Response<String>().error("请求用户登录").setData(loginUrl);
+    }
+
+    @Override
+    public Response<LoginVO> registerSubSystem(String sid, String addr, Object user) {
+        userCache.put(sid, new SoftReference<>(user));
+        addSubSys(sid, addr);
+        String url = ReferenceUtils.safeUnwrap(urlCache.get(sid));
+        return new Response<>("登录成功", new LoginVO(generateToken(), url));
+    }
+
+    @Override
+    public Response<Object> getUser(String sid) {
+        return new Response<>(userCache.containsKey(sid) ? userCache.get(sid).get() : null);
+    }
+
+    private void addSubSys(String sid, String addr) {
+        Set<String> subs = ReferenceUtils.safeUnwrap(subSysCache.get(sid));
+        if (Objects.isNull(subs)) {
+            subs = new HashSet<>(8);
+            subSysCache.put(sid, new SoftReference<>(subs));
+        }
+        subs.add(addr);
+    }
+}

src/main/java/org/code4everything/boot/web/sso/package-info.java

@@ -0,0 +1,6 @@
+/**
+ * 单点登录的简单封装
+ *
+ * @since 1.1.5
+ */
+package org.code4everything.boot.web.sso;