From 021169555a8b2b9dd4d1c09ea547799d493aa925 Mon Sep 17 00:00:00 2001
From: richjohnrambo <richjohnrambo@gmail.com>
Date: Sun, 27 Sep 2026 00:40:16 +0800
Subject: [PATCH] fix: use group message shard directory

---
 server/auth_audit_test.go                | 13 +++++++++++++
 server/internal_admin_query_resources.go | 13 +++++++++----
 2 files changed, 22 insertions(+), 4 deletions(-)

diff --git a/server/auth_audit_test.go b/server/auth_audit_test.go
index eac2cb2..f6c9be0 100644
--- a/server/auth_audit_test.go
+++ b/server/auth_audit_test.go
@@ -89,6 +89,19 @@ func TestGroupMessageQuerySupportsOperationalFilters(t *testing.T) {
 	}
 }
 
+func TestLegacyGroupMessagesUseShardDirectory(t *testing.T) {
+	query := strings.ToLower(internalLegacyGroupMessagesSelect(types.TenantID(9)))
+	if strings.Contains(query, "im_group_message_topic_shards") {
+		t.Fatalf("legacy group message query must not use removed topic shard routes: %s", query)
+	}
+	if !strings.Contains(query, "im_group_message_shards") {
+		t.Fatalf("legacy group message query must use group message shard directory: %s", query)
+	}
+	if !strings.Contains(query, "legacy.created_at>=route.shard_start") || !strings.Contains(query, "legacy.created_at<route.shard_end") {
+		t.Fatalf("legacy group message query must exclude rows covered by shard windows: %s", query)
+	}
+}
+
 func TestLoginAuditQuerySpecsExposeUserIDAsString(t *testing.T) {
 	spec := internalTenantResourceSpecs["user-login-logs"]
 	if spec == nil {
diff --git a/server/internal_admin_query_resources.go b/server/internal_admin_query_resources.go
index 6efd79c..7662422 100644
--- a/server/internal_admin_query_resources.go
+++ b/server/internal_admin_query_resources.go
@@ -461,14 +461,19 @@ func internalGroupMessageSource(tenantID types.TenantID) (string, error) {
 		}
 		selects = append(selects, fmt.Sprintf(`SELECT message_id AS id,tenant_id,topic,seqid,from_user,msg_type,status,content,head,reply_seqid,mentioned_user_ids,created_at,updated_at FROM %s`, table))
 	}
+	selects = append(selects, internalLegacyGroupMessagesSelect(tenantID))
+	return "(" + strings.Join(selects, " UNION ALL ") + ")", nil
+}
+
+func internalLegacyGroupMessagesSelect(tenantID types.TenantID) string {
 	// legacy 行内联租户条件（tenantID 来自租户注册表数值 ID，非用户输入）
-	selects = append(selects, fmt.Sprintf(`SELECT id,tenant_id,topic COLLATE utf8mb4_0900_ai_ci AS topic,seqid,from_user,
+	return fmt.Sprintf(`SELECT id,tenant_id,topic COLLATE utf8mb4_0900_ai_ci AS topic,seqid,from_user,
 		msg_type COLLATE utf8mb4_0900_ai_ci AS msg_type,status COLLATE utf8mb4_0900_ai_ci AS status,
 		content,head,reply_seqid,mentioned_user_ids,created_at,updated_at
 		FROM im_group_messages AS legacy WHERE legacy.tenant_id=%d AND NOT EXISTS (
-			SELECT 1 FROM im_group_message_topic_shards AS route
-			WHERE route.topic=legacy.topic COLLATE utf8mb4_0900_ai_ci AND legacy.seqid BETWEEN route.min_seq AND route.max_seq)`, int64(tenantID)))
-	return "(" + strings.Join(selects, " UNION ALL ") + ")", nil
+			SELECT 1 FROM im_group_message_shards AS route
+			WHERE route.tenant_id=legacy.tenant_id AND route.status<>'archived'
+				AND legacy.created_at>=route.shard_start AND legacy.created_at<route.shard_end)`, int64(tenantID))
 }
 
 func validInternalGroupMessageTable(table string) bool {