作者：殷天文

www.jianshu.com/p/b6f5ec98d790

通常系統(tǒng)都會(huì)限制同一個(gè)賬號(hào)的登錄人數(shù)，多人登錄要么限制后者登錄，要么踢出前者，Spring Security 提供了這樣的功能，本文講解一下在沒有使用Security的時(shí)候如何手動(dòng)實(shí)現(xiàn)這個(gè)功能

demo 技術(shù)選型

• SpringBoot

• JWT

• Filter

• Redis + Redisson

JWT（token）存儲(chǔ)在Redis中，類似 JSessionId-Session的關(guān)系，用戶登錄后每次請求在Header中攜帶jwt
如果你是使用session的話，也完全可以借鑒本文的思路，只是代碼上需要加些改動(dòng)

兩種實(shí)現(xiàn)思路

比較時(shí)間戳

維護(hù)一個(gè) username: jwtToken 這樣的一個(gè) key-value 在Reids中, Filter邏輯如下

?

圖片不清可點(diǎn)開放大

public?class?CompareKickOutFilter?extends?KickOutFilter?{@Autowiredprivate?UserService?userService;@Overridepublic?boolean?isAccessAllowed(HttpServletRequest?request,?HttpServletResponse?response)?{String?token?=?request.getHeader("Authorization");String?username?=?JWTUtil.getUsername(token);String?userKey?=?PREFIX?+?username;RBucket<String>?bucket?=?redissonClient.getBucket(userKey);String?redisToken?=?bucket.get();if?(token.equals(redisToken))?{return?true;}?else?if?(StringUtils.isBlank(redisToken))?{bucket.set(token);}?else?{Long?redisTokenUnixTime?=?JWTUtil.getClaim(redisToken,?"createTime").asLong();Long?tokenUnixTime?=?JWTUtil.getClaim(token,?"createTime").asLong();//?token?>?redisToken?則覆蓋if?(tokenUnixTime.compareTo(redisTokenUnixTime)?>?0)?{bucket.set(token);}?else?{//?注銷當(dāng)前tokenuserService.logout(token);sendJsonResponse(response,?4001,?"您的賬號(hào)已在其他設(shè)備登錄");return?false;}}return?true;} }

隊(duì)列踢出

public?class?QueueKickOutFilter?extends?KickOutFilter?{/***?踢出之前登錄的/之后登錄的用戶?默認(rèn)踢出之前登錄的用戶*/private?boolean?kickoutAfter?=?false;/***?同一個(gè)帳號(hào)最大會(huì)話數(shù)?默認(rèn)1*/private?int?maxSession?=?1;public?void?setKickoutAfter(boolean?kickoutAfter)?{this.kickoutAfter?=?kickoutAfter;}public?void?setMaxSession(int?maxSession)?{this.maxSession?=?maxSession;}@Overridepublic?boolean?isAccessAllowed(HttpServletRequest?request,?HttpServletResponse?response)?throws?Exception?{String?token?=?request.getHeader("Authorization");UserBO?currentSession?=?CurrentUser.get();Assert.notNull(currentSession,?"currentSession?cannot?null");String?username?=?currentSession.getUsername();String?userKey?=?PREFIX?+?"deque_"?+?username;String?lockKey?=?PREFIX_LOCK?+?username;RLock?lock?=?redissonClient.getLock(lockKey);lock.lock(2,?TimeUnit.SECONDS);try?{RDeque<String>?deque?=?redissonClient.getDeque(userKey);//?如果隊(duì)列里沒有此token，且用戶沒有被踢出；放入隊(duì)列if?(!deque.contains(token)?&&?currentSession.isKickout()?==?false)?{deque.push(token);}//?如果隊(duì)列里的sessionId數(shù)超出最大會(huì)話數(shù)，開始踢人while?(deque.size()?>?maxSession)?{String?kickoutSessionId;if?(kickoutAfter)?{?//?如果踢出后者kickoutSessionId?=?deque.removeFirst();}?else?{?//?否則踢出前者kickoutSessionId?=?deque.removeLast();}try?{RBucket<UserBO>?bucket?=?redissonClient.getBucket(kickoutSessionId);UserBO?kickoutSession?=?bucket.get();if?(kickoutSession?!=?null)?{//?設(shè)置會(huì)話的kickout屬性表示踢出了kickoutSession.setKickout(true);bucket.set(kickoutSession);}}?catch?(Exception?e)?{}}//?如果被踢出了，直接退出，重定向到踢出后的地址if?(currentSession.isKickout())?{//?會(huì)話被踢出了try?{//?注銷userService.logout(token);sendJsonResponse(response,?4001,?"您的賬號(hào)已在其他設(shè)備登錄");}?catch?(Exception?e)?{}return?false;}}?finally?{if?(lock.isHeldByCurrentThread())?{lock.unlock();LOGGER.info(Thread.currentThread().getName()?+?"?unlock");}?else?{LOGGER.info(Thread.currentThread().getName()?+?"?already?automatically?release?lock");}}return?true;}}

比較兩種方法

第一種方法邏輯簡單粗暴, 只維護(hù)一個(gè)key-value 不需要使用鎖，非要說缺點(diǎn)的話沒有第二種方法靈活。

第二種方法我很喜歡，代碼很優(yōu)雅靈活，但是邏輯相對麻煩一些，而且為了保證線程安全地操作隊(duì)列，要使用分布式鎖。目前我們項(xiàng)目中使用的是第一種方法

演示

下載地址:

https://gitee.com/yintianwen7/taven-springboot-learning/tree/master/login-control

運(yùn)行項(xiàng)目，訪問localhost:8887 demo中沒有存儲(chǔ)用戶信息，隨意輸入用戶名密碼，用戶名相同則被踢出

訪問 localhost:8887/index.html 彈出用戶信息, 代表當(dāng)前用戶有效

另一個(gè)瀏覽器登錄相同用戶名，回到第一個(gè)瀏覽器刷新頁面，提示被踢出

application.properties中選擇開啟哪種過濾器模式，默認(rèn)是比較時(shí)間戳踢出，開啟隊(duì)列踢出 queue-filter.enabled=true

總結(jié)

以上是生活随笔為你收集整理的厉害了，教你用 Spring Boot 控制并发登录人数的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。