|
|
package connection
|
|
|
|
|
|
// GroupStrategy 分组策略接口
|
|
|
// 用于定义如何将连接分配到分组
|
|
|
type GroupStrategy interface {
|
|
|
// GetGroupID 获取连接所属的分组ID
|
|
|
// 如果连接不属于任何分组，返回空字符串
|
|
|
GetGroupID(conn ConnectionInterface) string
|
|
|
|
|
|
// Match 检查连接是否匹配指定的分组ID
|
|
|
Match(conn ConnectionInterface, groupID string) bool
|
|
|
}
|
|
|
|
|
|
// StringGroupStrategy 字符串分组策略（现有实现）
|
|
|
// 直接使用字符串作为分组ID
|
|
|
type StringGroupStrategy struct{}
|
|
|
|
|
|
// NewStringGroupStrategy 创建字符串分组策略
|
|
|
func NewStringGroupStrategy() GroupStrategy {
|
|
|
return &StringGroupStrategy{}
|
|
|
}
|
|
|
|
|
|
// GetGroupID 字符串策略不支持自动获取分组ID
|
|
|
// 需要通过AddToGroup手动添加
|
|
|
func (s *StringGroupStrategy) GetGroupID(conn ConnectionInterface) string {
|
|
|
// 字符串策略不支持自动分组，返回空
|
|
|
return ""
|
|
|
}
|
|
|
|
|
|
// Match 字符串策略不支持匹配检查
|
|
|
// 分组关系由管理器维护
|
|
|
func (s *StringGroupStrategy) Match(conn ConnectionInterface, groupID string) bool {
|
|
|
// 字符串策略的匹配由管理器维护的groups map决定
|
|
|
// 这里返回false，由管理器自己处理
|
|
|
return false
|
|
|
}
|
|
|
|
|
|
// TagGroupStrategy 标签分组策略
|
|
|
// 基于连接的attributes中的标签进行分组
|
|
|
type TagGroupStrategy struct {
|
|
|
tagKey string // 标签键（如 "group"、"room"等）
|
|
|
}
|
|
|
|
|
|
// NewTagGroupStrategy 创建标签分组策略
|
|
|
// tagKey: 用于分组的标签键名
|
|
|
func NewTagGroupStrategy(tagKey string) GroupStrategy {
|
|
|
return &TagGroupStrategy{
|
|
|
tagKey: tagKey,
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// GetGroupID 从连接的attributes中获取分组ID
|
|
|
func (s *TagGroupStrategy) GetGroupID(conn ConnectionInterface) string {
|
|
|
// 检查连接是否有GetAttribute方法
|
|
|
if attrConn, ok := conn.(interface{ GetAttribute(key string) interface{} }); ok {
|
|
|
if tagValue := attrConn.GetAttribute(s.tagKey); tagValue != nil {
|
|
|
if groupID, ok := tagValue.(string); ok {
|
|
|
return groupID
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
return ""
|
|
|
}
|
|
|
|
|
|
// Match 检查连接的标签是否匹配分组ID
|
|
|
func (s *TagGroupStrategy) Match(conn ConnectionInterface, groupID string) bool {
|
|
|
return s.GetGroupID(conn) == groupID
|
|
|
}
|
|
|
|
|
|
// ConditionGroupStrategy 条件分组策略
|
|
|
// 基于连接属性的条件表达式进行分组
|
|
|
type ConditionGroupStrategy struct {
|
|
|
condition func(conn ConnectionInterface) string // 条件函数，返回分组ID
|
|
|
}
|
|
|
|
|
|
// NewConditionGroupStrategy 创建条件分组策略
|
|
|
// condition: 条件函数，根据连接属性返回分组ID
|
|
|
func NewConditionGroupStrategy(condition func(conn ConnectionInterface) string) GroupStrategy {
|
|
|
return &ConditionGroupStrategy{
|
|
|
condition: condition,
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// GetGroupID 通过条件函数获取分组ID
|
|
|
func (s *ConditionGroupStrategy) GetGroupID(conn ConnectionInterface) string {
|
|
|
if s.condition != nil {
|
|
|
return s.condition(conn)
|
|
|
}
|
|
|
return ""
|
|
|
}
|
|
|
|
|
|
// Match 通过条件函数检查是否匹配
|
|
|
func (s *ConditionGroupStrategy) Match(conn ConnectionInterface, groupID string) bool {
|
|
|
return s.GetGroupID(conn) == groupID
|
|
|
}
|
|
|
|