66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
//go:build mysql
|
|
// +build mysql
|
|
|
|
package mysql
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
t "github.com/tinode/chat/server/store/types"
|
|
)
|
|
|
|
func tenantShardSQL(tenantID t.TenantID, query string) string {
|
|
query = strings.ReplaceAll(query, "im_topics", tenantTopicsTable(tenantID))
|
|
query = strings.ReplaceAll(query, "im_group_members", tenantGroupMembersTable(tenantID))
|
|
query = strings.ReplaceAll(query, "im_subscriptions", tenantSubscriptionsTable(tenantID))
|
|
return strings.ReplaceAll(query, "im_system_notifications", tenantSystemNotificationsTable(tenantID))
|
|
}
|
|
|
|
func tenantTablePrefix(tenantID t.TenantID) string {
|
|
return fmt.Sprintf("im_%03d_", tenantID)
|
|
}
|
|
|
|
func tenantFriendshipsTable(tenantID t.TenantID) string {
|
|
return tenantTablePrefix(tenantID) + "friendships"
|
|
}
|
|
|
|
func tenantMessagesTable(tenantID t.TenantID) string {
|
|
return tenantTablePrefix(tenantID) + "messages"
|
|
}
|
|
|
|
func tenantTopicsTable(tenantID t.TenantID) string {
|
|
return tenantTablePrefix(tenantID) + "topics"
|
|
}
|
|
|
|
func tenantGroupMembersTable(tenantID t.TenantID) string {
|
|
return tenantTablePrefix(tenantID) + "group_members"
|
|
}
|
|
|
|
func tenantSubscriptionsTable(tenantID t.TenantID) string {
|
|
return tenantTablePrefix(tenantID) + "subscriptions"
|
|
}
|
|
|
|
func tenantSystemNotificationsTable(tenantID t.TenantID) string {
|
|
return tenantTablePrefix(tenantID) + "system_notifications"
|
|
}
|
|
|
|
func tenantGroupMessagesTable(tenantID t.TenantID, month string) (string, bool) {
|
|
if !validShardMonth(month) {
|
|
return "", false
|
|
}
|
|
return fmt.Sprintf("%sgroup_messages_%s", tenantTablePrefix(tenantID), month), true
|
|
}
|
|
|
|
func validShardMonth(month string) bool {
|
|
if len(month) != 6 {
|
|
return false
|
|
}
|
|
for i := range month {
|
|
if month[i] < '0' || month[i] > '9' {
|
|
return false
|
|
}
|
|
}
|
|
return month[4:] >= "01" && month[4:] <= "12"
|
|
}
|