wshandler와 분리 중

core/richconn.go

@@ -1,73 +1,97 @@
 package core
 
 import (
-	"context"
-
-	"repositories.action2quare.com/ayo/gocommon/logger"
-	"repositories.action2quare.com/ayo/gocommon/wshandler"
-
-	"github.com/go-redis/redis/v8"
-	"go.mongodb.org/mongo-driver/bson/primitive"
+	"fmt"
+	"strings"
+	"sync"
 )
 
-type richConnOuter struct {
-	wsh *wshandler.WebsocketHandler
-	rc  *wshandler.Richconn
+type connection struct {
+	locker  sync.Mutex
+	alias   string
+	tags    []string
+	onClose map[string]func()
 }
 
-func (sub richConnOuter) JoinTag(region string, tag primitive.ObjectID, tid primitive.ObjectID, hint string) error {
-	sub.wsh.JoinTag(region, tag, tid, sub.rc, hint)
+func (rc *connection) addTag(name, val string) {
+	rc.locker.Lock()
+	defer rc.locker.Unlock()
 
-	wsh := sub.wsh
-	sub.rc.RegistOnCloseFunc(tag.Hex(), func() {
-		wsh.LeaveTag(region, tag, tid)
-	})
-	return nil
-}
-
-func (sub richConnOuter) LeaveTag(region string, tag primitive.ObjectID, tid primitive.ObjectID, hint string) error {
-	sub.SetStateInTag(region, tag, tid, "", hint)
-	return sub.wsh.LeaveTag(region, tag, tid)
-}
-
-func (sub richConnOuter) SetStateInTag(region string, tag primitive.ObjectID, tid primitive.ObjectID, state string, hint string) error {
-	return sub.wsh.SetStateInTag(region, tag, tid, state, hint)
-}
-
-func (sub richConnOuter) TurnGroupOnline(key string, gid primitive.ObjectID, score float64) error {
-	gidhex := gid.Hex()
-	_, err := sub.wsh.RedisSync.ZAdd(context.Background(), key, &redis.Z{Score: score, Member: gidhex}).Result()
-	if err != nil {
-		logger.Error("TurnGroupOnline failed. redis.ZAdd return err :", err)
-		return err
+	prefix := name + "="
+	for i, tag := range rc.tags {
+		if strings.HasPrefix(tag, prefix) {
+			rc.tags[i] = prefix + val
+			return
+		}
 	}
-
-	sub.rc.RegistOnCloseFunc(key, func() {
-		sub.wsh.RedisSync.ZRem(context.Background(), key, gidhex)
-	})
-
-	return nil
+	rc.tags = append(rc.tags, prefix+val)
 }
 
-func (sub richConnOuter) TurnGroupOffline(key string, gid primitive.ObjectID) error {
-	f := sub.rc.UnregistOnCloseFunc(key)
-	if f != nil {
+func (rc *connection) getTag(name string) string {
+	rc.locker.Lock()
+	defer rc.locker.Unlock()
+
+	prefix := name + "="
+	for _, tag := range rc.tags {
+		if strings.HasPrefix(tag, prefix) {
+			return tag[len(prefix):]
+		}
+	}
+	return ""
+}
+
+func (rc *connection) removeTag(name string, val string) {
+	rc.locker.Lock()
+	defer rc.locker.Unlock()
+
+	whole := fmt.Sprintf("%s=%s", name, val)
+	for i, tag := range rc.tags {
+		if tag == whole {
+			if i == 0 && len(rc.tags) == 1 {
+				rc.tags = nil
+			} else {
+				lastidx := len(rc.tags) - 1
+				if i < lastidx {
+					rc.tags[i] = rc.tags[lastidx]
+				}
+				rc.tags = rc.tags[:lastidx]
+			}
+			return
+		}
+	}
+}
+
+func (rc *connection) registOnCloseFunc(name string, f func()) {
+	rc.locker.Lock()
+	defer rc.locker.Unlock()
+
+	if rc.onClose == nil {
 		f()
-	} else {
-		sub.wsh.RedisSync.ZRem(context.Background(), key, gid.Hex())
+		return
 	}
-	return nil
+	rc.onClose[name] = f
 }
 
-func (sub richConnOuter) SendMessage(doc []byte) error {
-	return sub.rc.WriteBytes(doc)
+func (rc *connection) hasOnCloseFunc(name string) bool {
+	rc.locker.Lock()
+	defer rc.locker.Unlock()
+
+	if rc.onClose == nil {
+		return false
+	}
+
+	_, ok := rc.onClose[name]
+	return ok
 }
 
-func (sub richConnOuter) SendMessageToTag(region string, tag primitive.ObjectID, msg []byte) error {
-	sub.wsh.BroadcastRaw(region, tag, msg)
-	return nil
-}
+func (rc *connection) unregistOnCloseFunc(name string) (out func()) {
+	rc.locker.Lock()
+	defer rc.locker.Unlock()
 
-func (sub richConnOuter) CloseOnPurpose() error {
-	return sub.rc.Close()
+	if rc.onClose == nil {
+		return
+	}
+	out = rc.onClose[name]
+	delete(rc.onClose, name)
+	return
 }