ayo/tavern
6
0
Fork 2
Code Issues Pull Requests Projects Releases Wiki Activity

tavern ccu 메트릭 수집

Browse Source
This commit is contained in:
mountain
2023-11-24 16:30:47 +09:00
parent e169e72f66
commit 2e2f7a151e
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
core/tavern.go
View File

@ -14,6 +14,7 @@ import (
"repositories.action2quare.com/ayo/gocommon" "repositories.action2quare.com/ayo/gocommon"
"repositories.action2quare.com/ayo/gocommon/flagx" "repositories.action2quare.com/ayo/gocommon/flagx"
"repositories.action2quare.com/ayo/gocommon/logger" "repositories.action2quare.com/ayo/gocommon/logger"
"repositories.action2quare.com/ayo/gocommon/metric"
"repositories.action2quare.com/ayo/gocommon/session" "repositories.action2quare.com/ayo/gocommon/session"
"repositories.action2quare.com/ayo/gocommon/wshandler" "repositories.action2quare.com/ayo/gocommon/wshandler"
@ -79,6 +80,7 @@ func New(context context.Context, wsh *wshandler.WebsocketHandler) (*Tavern, err
} }
func (tv *Tavern) Cleanup() { func (tv *Tavern) Cleanup() {
metric.ConcurrentUser.Set(0)
tv.mongoClient.Close() tv.mongoClient.Close()
} }
@ -136,6 +138,7 @@ func (tv *Tavern) LeaveChannel(ctx wshandler.ApiCallContext) {
} }
func (tv *Tavern) ClientConnected(conn *websocket.Conn, callby *wshandler.Sender) { func (tv *Tavern) ClientConnected(conn *websocket.Conn, callby *wshandler.Sender) {
metric.ConcurrentUser.Add(1)
tv.redison.Del(tv.redison.Context(), callby.Accid.Hex()) tv.redison.Del(tv.redison.Context(), callby.Accid.Hex())
_, err := tv.redison.JSONSet(callby.Accid.Hex(), "$", bson.M{"_ts": time.Now().UTC().Unix()}) _, err := tv.redison.JSONSet(callby.Accid.Hex(), "$", bson.M{"_ts": time.Now().UTC().Unix()})
if err != nil { if err != nil {
@ -144,6 +147,7 @@ func (tv *Tavern) ClientConnected(conn *websocket.Conn, callby *wshandler.Sender
} }
func (tv *Tavern) ClientDisconnected(msg string, callby *wshandler.Sender) { func (tv *Tavern) ClientDisconnected(msg string, callby *wshandler.Sender) {
metric.ConcurrentUser.Add(-1)
tv.redison.Del(tv.redison.Context(), callby.Accid.Hex()).Result() tv.redison.Del(tv.redison.Context(), callby.Accid.Hex()).Result()
} }