flag 자체 구현

This commit is contained in:
2023-06-21 14:13:30 +09:00
parent 94b6d1499a
commit 0d8752e661
7 changed files with 197 additions and 27 deletions

View File

@ -12,6 +12,7 @@ import (
"sync"
common "repositories.action2quare.com/ayo/gocommon"
"repositories.action2quare.com/ayo/gocommon/flagx"
"repositories.action2quare.com/ayo/gocommon/logger"
"github.com/go-redis/redis/v8"
@ -20,6 +21,8 @@ import (
"go.mongodb.org/mongo-driver/bson/primitive"
)
var noSessionFlag = flagx.Bool("nosession", false, "nosession=[true|false]")
const (
connStateCachePrefix = "conn_state_"
connStateScript = `
@ -289,7 +292,7 @@ func (ws *WebsocketHandler) RegisterHandlers(ctx context.Context, serveMux *http
}
sh.url = common.MakeHttpHandlerPattern(prefix, region, "ws")
sh.redisSync = ws.RedisSync
if *common.NoSessionFlag {
if *noSessionFlag {
serveMux.HandleFunc(sh.url, sh.upgrade_nosession)
} else {
serveMux.HandleFunc(sh.url, sh.upgrade)