feat: split workbench core mysql access

internal/store/core_mysql.go

@@ -0,0 +1,116 @@
+package store
+
+import (
+	"context"
+	"strings"
+
+	"git.118im.com/im_team/im_workbench/internal/config"
+	"github.com/jmoiron/sqlx"
+)
+
+// CoreStore provides read-only access to shared IM data.
+type CoreStore struct {
+	db *sqlx.DB
+}
+
+func coreStoreOwnsDB(primaryDSN, coreDSN string) bool {
+	return strings.TrimSpace(primaryDSN) != strings.TrimSpace(coreDSN)
+}
+
+func openCoreDSN(cfg config.MySQLConfig) (*CoreStore, error) {
+	db, err := openMySQLDB(cfg)
+	if err != nil {
+		return nil, err
+	}
+	return &CoreStore{db: db}, nil
+}
+
+func (s *CoreStore) Close() error {
+	if s == nil || s.db == nil {
+		return nil
+	}
+	return s.db.Close()
+}
+
+func (s *CoreStore) fileUploadMetadata(ctx context.Context, tenantID int64, fileID string) fileUploadMetadata {
+	var meta fileUploadMetadata
+	decodedID, storageName := parseUploadFileID(fileID)
+	locationCandidates := uploadLocationCandidates(fileID, storageName)
+	if decodedID <= 0 && len(locationCandidates) == 0 {
+		return meta
+	}
+	if decodedID > 0 {
+		err := s.db.GetContext(ctx, &meta, `SELECT IFNULL(mimetype,'') AS mimetype,IFNULL(size,0) AS size,IFNULL(etag,'') AS etag,IFNULL(location,'') AS location,IFNULL(original_name,'') AS original_name FROM im_fileuploads WHERE tenant_id=? AND id=? LIMIT 1`, tenantID, decodedID)
+		if err == nil {
+			return meta
+		}
+	}
+	for _, candidate := range locationCandidates {
+		err := s.db.GetContext(ctx, &meta, `SELECT IFNULL(mimetype,'') AS mimetype,IFNULL(size,0) AS size,IFNULL(etag,'') AS etag,IFNULL(location,'') AS location,IFNULL(original_name,'') AS original_name
+			FROM im_fileuploads WHERE tenant_id=? AND (location=? OR location LIKE ?) ORDER BY createdat DESC LIMIT 1`, tenantID, candidate, "%/"+candidate)
+		if err == nil {
+			return meta
+		}
+	}
+	return meta
+}
+
+func (s *CoreStore) populateUserSummaries(ctx context.Context, tenantID int64, ids []int64, result map[int64]UserSummary) error {
+	query, args, err := sqlx.In(`SELECT user_id,nickname AS display_name,IFNULL(avatar_file_id,'') AS avatar_file_id
+		FROM im_user_profiles WHERE tenant_id=? AND user_id IN (?) ORDER BY user_id ASC`, tenantID, ids)
+	if err != nil {
+		return err
+	}
+	var profileRows []struct {
+		UserID       int64  `db:"user_id"`
+		DisplayName  string `db:"display_name"`
+		AvatarFileID string `db:"avatar_file_id"`
+	}
+	if err := s.db.SelectContext(ctx, &profileRows, s.db.Rebind(query), args...); err != nil && !isMissingTable(err) {
+		return err
+	}
+	for _, row := range profileRows {
+		summary := result[row.UserID]
+		if summary.UserID == 0 {
+			summary.UserID = row.UserID
+		}
+		if summary.DisplayName == "" {
+			summary.DisplayName = strings.TrimSpace(row.DisplayName)
+		}
+		if summary.AvatarFileID == "" {
+			summary.AvatarFileID = row.AvatarFileID
+		}
+		result[row.UserID] = summary
+	}
+
+	query, args, err = sqlx.In(`SELECT id AS user_id,
+			COALESCE(
+				NULLIF(JSON_UNQUOTE(JSON_EXTRACT(public,'$.display_name')),''),
+				NULLIF(JSON_UNQUOTE(JSON_EXTRACT(public,'$.fn')),''),
+				NULLIF(JSON_UNQUOTE(JSON_EXTRACT(public,'$.nickname')),''),
+				NULLIF(JSON_UNQUOTE(JSON_EXTRACT(public,'$.name')),''),
+				''
+			) AS display_name
+		FROM im_users WHERE tenant_id=? AND id IN (?)`, tenantID, ids)
+	if err != nil {
+		return err
+	}
+	var userRows []struct {
+		UserID      int64  `db:"user_id"`
+		DisplayName string `db:"display_name"`
+	}
+	if err := s.db.SelectContext(ctx, &userRows, s.db.Rebind(query), args...); err != nil && !isMissingTable(err) {
+		return err
+	}
+	for _, row := range userRows {
+		summary := result[row.UserID]
+		if summary.UserID == 0 {
+			summary.UserID = row.UserID
+		}
+		if summary.DisplayName == "" {
+			summary.DisplayName = strings.TrimSpace(row.DisplayName)
+		}
+		result[row.UserID] = summary
+	}
+	return nil
+}