feat: add kafka admin integration

server/store/group_message_runtime.go

@@ -69,6 +69,7 @@ type groupMessageRuntime interface {
 	RemoveCachedSetMember(tenantID types.TenantID, namespace, id, member string) error
 	CachedSetContains(tenantID types.TenantID, namespace, id, member string) (bool, error)
 	AllowRate(tenantID types.TenantID, scope string, limit int, window time.Duration) (bool, int, time.Duration, error)
+	ClaimShortState(namespace, id string, ttl time.Duration) (bool, error)
 }
 
 var groupMsgRuntime groupMessageRuntime
@@ -88,6 +89,14 @@ func MarkUserOnline(tenantID types.TenantID, uid types.Uid, sid, nodeID, device,
 	return groupMsgRuntime.MarkUserOnline(tenantID, uid, sid, nodeID, device, userAgent)
 }
 
+func ClaimShortState(namespace, id string, ttl time.Duration) (bool, bool, error) {
+	if groupMsgRuntime == nil || namespace == "" || id == "" {
+		return false, false, nil
+	}
+	ok, err := groupMsgRuntime.ClaimShortState(namespace, id, ttl)
+	return ok, true, err
+}
+
 func RefreshUserPresence(tenantID types.TenantID, uid types.Uid, sid, nodeID string) error {
 	if groupMsgRuntime == nil || !tenantID.IsValid() || uid.IsZero() || sid == "" {
 		return nil
@@ -169,6 +178,13 @@ func DeleteCachedObject(tenantID types.TenantID, namespace, id string) error {
 	return groupMsgRuntime.DeleteCachedObject(tenantID, namespace, id)
 }
 
+func InvalidateHotMessages(tenantID types.TenantID, topic string, ranges []types.Range) error {
+	if groupMsgRuntime == nil || !tenantID.IsValid() || topic == "" {
+		return nil
+	}
+	return groupMsgRuntime.InvalidateHotMessages(tenantID, topic, ranges)
+}
+
 func AddCachedSetMember(tenantID types.TenantID, namespace, id, member string) error {
 	if groupMsgRuntime == nil || !tenantID.IsValid() || namespace == "" || id == "" || member == "" {
 		return nil
@@ -730,6 +746,13 @@ func (r *redisGroupMessageRuntime) AllowRate(tenantID types.TenantID, scope stri
 	return count <= int64(limit), int(count), ttl, nil
 }
 
+func (r *redisGroupMessageRuntime) ClaimShortState(namespace, id string, ttl time.Duration) (bool, error) {
+	if ttl <= 0 {
+		ttl = time.Duration(r.cfg.ShortTTL) * time.Second
+	}
+	return r.client.SetNX(context.Background(), r.shortStateKey(namespace, id), "1", ttl).Result()
+}
+
 func (r *redisGroupMessageRuntime) hotSeqIDs(tenantID types.TenantID, topic string, opts *types.QueryOpt, limit int) ([]int64, bool, error) {
 	if opts != nil && len(opts.IdRanges) > 0 {
 		seqs := expandRanges(opts.IdRanges, limit)
@@ -851,6 +874,10 @@ func (r *redisGroupMessageRuntime) rateKey(tenantID types.TenantID, scope string
 	return r.key(tenantID, "rate", scope)
 }
 
+func (r *redisGroupMessageRuntime) shortStateKey(namespace, id string) string {
+	return r.key(types.ZeroTenantID, "short", namespace, id)
+}
+
 func (r *redisGroupMessageRuntime) cacheTTL(namespace string) time.Duration {
 	ttl := r.cfg.CacheTTL
 	if namespace == "verification" || namespace == "action_token" {