/* Ċ Ġ* ĠĠ ĠCopyright Ġ( C ) Ġ 2 0 1 7 - 2 0 1 8 ĠZe ĠH ao ĠX iao Ċ Ġ* Ċ Ġ* ĠĠ ĠLicensed Ġunder Ġthe ĠApache ĠLicense , ĠVersion Ġ 2 . 0 Ġ( the Ġ" License "); Ċ Ġ* ĠĠ Ġyou Ġmay Ġnot Ġuse Ġthis Ġfile Ġexcept Ġin Ġcompliance Ġwith Ġthe ĠLicense . Ċ Ġ* ĠĠ ĠYou Ġmay Ġobtain Ġa Ġcopy Ġof Ġthe ĠLicense Ġat Ċ Ġ* Ċ Ġ* ĠĠĠĠĠĠ Ġhttp :// www . apache . org / licenses / LICENSE - 2 . 0 Ċ Ġ* Ċ Ġ* ĠĠ ĠUnless Ġrequired Ġby Ġapplicable Ġlaw Ġor Ġagreed Ġto Ġin Ġwriting , Ġsoftware Ċ Ġ* ĠĠ Ġdistributed Ġunder Ġthe ĠLicense Ġis Ġdistributed Ġon Ġan Ġ" AS ĠIS " ĠBASIS , Ċ Ġ* ĠĠ ĠWITHOUT ĠWARRANTIES ĠOR ĠCONDITIONS ĠOF ĠANY ĠKIND , Ġeither Ġexpress Ġor Ġimplied . Ċ Ġ* ĠĠ ĠSee Ġthe ĠLicense Ġfor Ġthe Ġspecific Ġlanguage Ġgoverning Ġpermissions Ġand Ċ Ġ* ĠĠ Ġlimitations Ġunder Ġthe ĠLicense . Ċ Ġ*/ Ċ package Ġcom . github . kv nx iao . spring . web flux . chat room . config Ċ Ċ import Ġcom . github . kv nx iao . spring . web flux . chat room . handler . websocket . lob by . Lobby Socket Handler Ċ import Ġcom . github . kv nx iao . spring . web flux . chat room . handler . websocket . room . Room Socket Handler Ċ import Ġcom . github . kv nx iao . spring . web flux . chat room . model . Session Ċ import Ġorg . springframework . beans . factory . annotation . Autowired Ċ import Ġorg . springframework . context . annotation . Bean Ċ import Ġorg . springframework . context . annotation . Configuration Ċ import Ġorg . springframework . web . reactive . Handler Mapping Ċ import Ġorg . springframework . web . reactive . Handler Result Ċ import Ġorg . springframework . web . reactive . handler . Simple Url Handler Mapping Ċ import Ġorg . springframework . web . reactive . socket . WebSocket Handler Ċ import Ġorg . springframework . web . reactive . socket . server . support . Handshake WebSocket Service Ċ import Ġorg . springframework . web . reactive . socket . server . support . WebSocket Handler Adapter Ċ import Ġorg . springframework . web . server . Server Web Exchange Ċ import Ġreactor . core . publisher . Mono Ċ Ċ @ Configuration Ċ class ĠWebSocket Configuration Ġ@ Autowired Ġconstructor ( ĊĠĠĠ Ġprivate Ġval Ġlob by Socket Handler : ĠL obby Socket Handler , ĊĠĠĠ Ġprivate Ġval Ġroom Socket Handler : ĠRoom Socket Handler Ċ ) Ġ{ ĊĊĠĠĠ Ġ@ Bean ĊĠĠĠ Ġfun Ġweb Socket Mapping (): ĠHandler Mapping Ġ{ ĊĠĠĠĠĠĠĠ Ġreturn ĠSimple Url Handler Mapping (). apply Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠ Ġorder Ġ= Ġ 1 ĊĠĠĠĠĠĠĠĠĠĠĠ Ġurl Map Ġ= Ġmap Of < String , ĠWebSocket Handler >( ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ"/ lob by / ws " Ġto Ġlob by Socket Handler , ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ"/ room /*/ ws " Ġto Ġroom Socket Handler ĊĠĠĠĠĠĠĠĠĠĠĠ Ġ) ĊĠĠĠĠĠĠĠ Ġ} ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġ@ Bean ĊĠĠĠ Ġfun Ġhandler Adapter (): ĠWebSocket Handler Adapter Ġ= Ġobject Ġ: ĠWebSocket Handler Adapter () Ġ{ ĊĠĠĠĠĠĠĠ Ġoverride Ġfun Ġhandle ( exchange : ĠServer Web Exchange , Ġhandler : ĠAny ): ĠMono < Handler Result > Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠ Ġval Ġhandshake WebSocket Service Ġ= Ġthis . web Socket Service Ġas ĠHand shake WebSocket Service ĊĠĠĠĠĠĠĠĠĠĠĠ Ġhandshake WebSocket Service . set Session Attribute Predicate Ġ{ Ġit Ġ== ĠSession . ROOM _ ID Ġ|| Ġit Ġ== ĠSession . USER Ġ} ĊĠĠĠĠĠĠĠĠĠĠĠ Ġreturn Ġsuper . handle ( exchange , Ġhandler ) ĊĠĠĠĠĠĠĠ Ġ} ĊĠĠĠ Ġ} Ċ } Ċ