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

ccu metric 추가

Browse Source
This commit is contained in:
mountain
2023-11-16 19:59:25 +09:00
parent be15f3f854
commit f96058057d
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
core/tavern.go
View File

@ -7,6 +7,7 @@ import (
"net"
"net/http"
"strings"
"sync/atomic"
"time"
"github.com/go-redis/redis/v8"
@ -14,6 +15,7 @@ import (
"repositories.action2quare.com/ayo/gocommon"
"repositories.action2quare.com/ayo/gocommon/flagx"
"repositories.action2quare.com/ayo/gocommon/logger"
"repositories.action2quare.com/ayo/gocommon/metric"
"repositories.action2quare.com/ayo/gocommon/session"
"repositories.action2quare.com/ayo/gocommon/wshandler"
@ -21,6 +23,8 @@ import (
)
var devflag = flagx.Bool("dev", false, "")
var totalCCUWriter = metric.NewMetric(metric.MetricGuage, "concurrent_user", "current connected user count")
var ccu = int64(0)
type TavernConfig struct {
session.SessionConfig `json:",inline"`
@ -79,6 +83,7 @@ func New(context context.Context, wsh *wshandler.WebsocketHandler) (*Tavern, err
}
func (tv *Tavern) Cleanup() {
totalCCUWriter(0)
tv.mongoClient.Close()
}
@ -136,6 +141,7 @@ func (tv *Tavern) LeaveChannel(ctx wshandler.ApiCallContext) {
}
func (tv *Tavern) ClientConnected(conn *websocket.Conn, callby *wshandler.Sender) {
totalCCUWriter(float64(atomic.AddInt64(&ccu, 1)))
tv.redison.Del(tv.redison.Context(), callby.Accid.Hex())
_, err := tv.redison.JSONSet(callby.Accid.Hex(), "$", bson.M{"_ts": time.Now().UTC().Unix()})
if err != nil {
@ -144,6 +150,8 @@ func (tv *Tavern) ClientConnected(conn *websocket.Conn, callby *wshandler.Sender
}
func (tv *Tavern) ClientDisconnected(msg string, callby *wshandler.Sender) {
totalCCUWriter(float64(atomic.AddInt64(&ccu, -1)))
tv.redison.Del(tv.redison.Context(), callby.Accid.Hex()).Result()
}