Redisを使ってExpressとWebSocketのセッションを共有する

constexpress=require('express')constWebSocket=require('ws')constRedis=require('ioredis')constSession=require('express-session')constRedisStore=require('connect-redis')(Session)// ---------------------------------------------------// sessionデータを保管するstoreを用意する// ---------------------------------------------------// Redisのhostとportは環境変数から読み込むconst{REDIS_HOST,REDIS_PORT}=process.env// Redis用インスタンスを生成する// セッション＆通知バッファリング用constredis=newRedis({host:REDIS_HOST||'redis',// ※Dockerで使用する前提でこんな風に設定してますport:REDIS_PORT||6379})// セッションストアを生成するconststore=newRedisStore({client:redis})// ---------------------------------------------------// ここからexpressの設定（httpセッション）// ---------------------------------------------------// expressを生成するconstapp=express()app.use(express.json())// Redisに紐づいたセッションプロバイダーを作るconstsession=Session({store,secret:'hogehoge',// お好きにsecretを指定してくださいresave:false,saveUninitialized:false,cookie:{secure:'auto'}})// expressにセッションプロバイダーを紐づけるapp.use(session)// port 3000 でlinten開始consthttpServer=app.listen(3000)// expressを初期化する段階で、Redisに紐づいたセッションプロバイダーと差し替えるだけなので、// すでに稼働しているプログラムの改変も最小限の作業で済む（かも）// あとは普通にexpressの処理を記述すればOK!.........// ログイン処理app.post('login',(req,res,next)=>{// ログインチェック......// ログイン完了時の処理// DBなどから取得したユーザ情報をセッションに書き込むイメージ user = ユーザ情報req.session.user=user// ログイン成功時にセッションID（req.sessionID）をクライアントに返す！  // これを後でWebSocketを通して返却してもらい紐づけに使用するべしres.json({login:true,sessId:req.sessionID})})// ---------------------------------------------------// ここからWebSocketの設定//// WebSocketサーバを生成するconstwsServer=newWebSocket.Server({server:httpServer.server})wsServer.on('connection',(ws)=>{ws.on('open',()=>{// ここでは、redisSessionId というパラメータにセッションIdを保管することにしますws.redisSessionId=''})ws.on('message',(message)=>{// データを復号するconstmsg=JSON.parse(message)// セッションidをwsに紐づける処理// この例では、クライアントから次のようなデータを受け取る想定// { cmd: 'connect_session', sessionID: 'xxxxxxxxxxxxx' }if(msg.cmd==='connect_session'){ws.redisSessionId=msg.sessionID// クライアントにセッションidを受け取ったことを報告しておくreturnws.send(JSON.stringify({result:'received_session_id'}))}......})......})// ---------------------------------------------------// 何らかのイベント処理（WebHookなど）//consthogeEvent=(params)=>{// wsServerには複数のsocketが接続されている// wsServer.clientsをループで処理して個々のWebSocketに対して処理を行うwsServer.clients.forEach(async(ws)=>{// Sessionを取得するconstsession=!ws.redisSessionId?awaitgetSession(ws.redisSessionId)// セッションデータ読込関数を呼ぶ:{}console.log(session.user)......})}// ---------------------------------------------------// セッションデータの読み込み関数//constgetSession=(sid)=>{returnnewPromise((resolve,reject)=>{// Redisに紐づいたセッションストアからセッションデータを取得するstore.get(sid,(err,session)=>{if(err){returnreject(err)}resolve(session?session:{})})})}