go-svc-configure/internal/service/aliasmanagement/save_table_field_alias.go

package aliasmanagement

import (
	"context"
	"fmt"
	"regexp"

	"git.x2erp.com/qdy/go-base/ctx"
	"git.x2erp.com/qdy/go-base/logger"
	"git.x2erp.com/qdy/go-base/model/response"
	"git.x2erp.com/qdy/go-base/util"
	"git.x2erp.com/qdy/go-db/factory/database"
	"git.x2erp.com/qdy/go-svc-configure/internal/tables"
	"github.com/google/uuid"
	"github.com/jmoiron/sqlx"
)

// SaveTableFieldAlias 保存字段别名字典（创建或更新）
func SaveTableFieldAlias(req *TableFieldAliasRequest, ctx context.Context, dbFactory *database.DBFactory, reqCtx *ctx.RequestContext) *response.QueryResult[TableFieldAliasDetail] {
	logger.Debug("SaveTableFieldAlias-开始保存字段别名字典")

	// 参数验证
	if err := validateTableFieldAliasRequest(req); err != nil {
		logger.ErrorC(reqCtx, fmt.Sprintf("参数验证失败: %v", err))
		return util.CreateErrorResult[TableFieldAliasDetail](fmt.Sprintf("参数验证失败: %v", err), reqCtx)
	}

	// 获取数据库连接并开始事务
	db := dbFactory.GetDB()
	tx, err := db.BeginTxx(ctx, nil)
	if err != nil {
		logger.ErrorC(reqCtx, fmt.Sprintf("开始事务失败: %v", err))
		return util.CreateErrorResult[TableFieldAliasDetail](fmt.Sprintf("开始事务失败: %v", err), reqCtx)
	}

	defer func() {
		if p := recover(); p != nil {
			tx.Rollback()
			panic(p)
		}
	}()

	// 获取当前用户
	creator := reqCtx.UserID
	if creator == "" {
		creator = "system"
	}

	// 检查字段别名是否已存在（基于field_alias字段的唯一索引）
	aliasExists, err := checkTableFieldAliasExists(ctx, tx, req.FieldAlias)
	if err != nil {
		tx.Rollback()
		logger.ErrorC(reqCtx, fmt.Sprintf("检查字段别名存在性失败: %v", err))
		return util.CreateErrorResult[TableFieldAliasDetail](fmt.Sprintf("检查字段别名存在性失败: %v", err), reqCtx)
	}

	var tableFieldAlias tables.DicTableFieldAliasDB
	if aliasExists {
		// 更新字段别名（根据ID或field_alias）
		tableFieldAlias, err = updateTableFieldAlias(ctx, tx, req)
		if err != nil {
			tx.Rollback()
			logger.ErrorC(reqCtx, fmt.Sprintf("更新字段别名失败: %v", err))
			return util.CreateErrorResult[TableFieldAliasDetail](fmt.Sprintf("更新字段别名失败: %v", err), reqCtx)
		}
		logger.Debug(fmt.Sprintf("更新字段别名成功: %s", req.FieldAlias))
	} else {
		// 检查是否有软删除的记录
		softDeletedExists, err := checkSoftDeletedTableFieldAliasExists(ctx, tx, req.FieldAlias)
		if err != nil {
			tx.Rollback()
			logger.ErrorC(reqCtx, fmt.Sprintf("检查软删除字段别名存在性失败: %v", err))
			return util.CreateErrorResult[TableFieldAliasDetail](fmt.Sprintf("检查软删除字段别名存在性失败: %v", err), reqCtx)
		}

		if softDeletedExists {
			// 恢复软删除的字段别名
			tableFieldAlias, err = restoreTableFieldAlias(ctx, tx, req)
			if err != nil {
				tx.Rollback()
				logger.ErrorC(reqCtx, fmt.Sprintf("恢复软删除字段别名失败: %v", err))
				return util.CreateErrorResult[TableFieldAliasDetail](fmt.Sprintf("恢复软删除字段别名失败: %v", err), reqCtx)
			}
			logger.Debug(fmt.Sprintf("恢复软删除字段别名成功: %s", req.FieldAlias))
		} else {
			// 插入新的字段别名
			tableFieldAlias, err = insertTableFieldAlias(ctx, tx, req, creator)
			if err != nil {
				tx.Rollback()
				logger.ErrorC(reqCtx, fmt.Sprintf("插入字段别名失败: %v", err))
				return util.CreateErrorResult[TableFieldAliasDetail](fmt.Sprintf("插入字段别名失败: %v", err), reqCtx)
			}
			logger.Debug(fmt.Sprintf("插入字段别名成功: %s", req.FieldAlias))
		}
	}

	// 提交事务
	if err := tx.Commit(); err != nil {
		logger.ErrorC(reqCtx, fmt.Sprintf("提交事务失败: %v", err))
		return util.CreateErrorResult[TableFieldAliasDetail](fmt.Sprintf("提交事务失败: %v", err), reqCtx)
	}

	logger.Debug(fmt.Sprintf("成功保存字段别名字典: %s", req.FieldAlias))

	// 构建返回结果
	detail := TableFieldAliasDetail{
		TableFieldAlias: tableFieldAlias,
	}

	return util.CreateSuccessResultData[TableFieldAliasDetail](detail, reqCtx)
}

// validateTableFieldAliasRequest 验证字段别名字典请求
func validateTableFieldAliasRequest(req *TableFieldAliasRequest) error {
	if req.FieldID == "" {
		return fmt.Errorf("字段ID不能为空")
	}

	// 验证字段ID格式：仅允许字母、数字、下划线
	match, _ := regexp.MatchString("^[a-zA-Z0-9_.]+$", req.FieldID)
	if !match {
		return fmt.Errorf("字段ID只能包含字母、数字、下划线和点")
	}

	if req.TableID == "" {
		return fmt.Errorf("表ID不能为空")
	}

	match, _ = regexp.MatchString("^[a-zA-Z0-9_]+$", req.TableID)
	if !match {
		return fmt.Errorf("表ID只能包含字母、数字、下划线")
	}

	if req.FieldName == "" {
		return fmt.Errorf("字段名称不能为空")
	}

	match, _ = regexp.MatchString("^[a-zA-Z0-9_]+$", req.FieldName)
	if !match {
		return fmt.Errorf("字段名称只能包含字母、数字、下划线")
	}

	if req.FieldAlias == "" {
		return fmt.Errorf("字段别名不能为空")
	}

	match, _ = regexp.MatchString("^[a-zA-Z0-9_]+$", req.FieldAlias)
	if !match {
		return fmt.Errorf("字段别名只能包含字母、数字、下划线")
	}

	// 描述和查询条件可以为空，不需要验证

	return nil
}

// checkTableFieldAliasExists 检查字段别名是否存在（仅活跃记录，基于field_alias）
func checkTableFieldAliasExists(ctx context.Context, tx *sqlx.Tx, fieldAlias string) (bool, error) {
	var count int
	query := "SELECT COUNT(*) FROM dic_table_field_alias WHERE field_alias = ? AND deleted_at IS NULL"
	err := tx.GetContext(ctx, &count, query, fieldAlias)
	return count > 0, err
}

// checkSoftDeletedTableFieldAliasExists 检查字段别名是否被软删除（基于field_alias）
func checkSoftDeletedTableFieldAliasExists(ctx context.Context, tx *sqlx.Tx, fieldAlias string) (bool, error) {
	var count int
	query := "SELECT COUNT(*) FROM dic_table_field_alias WHERE field_alias = ? AND deleted_at IS NOT NULL"
	err := tx.GetContext(ctx, &count, query, fieldAlias)
	return count > 0, err
}

// restoreTableFieldAlias 恢复软删除的字段别名
func restoreTableFieldAlias(ctx context.Context, tx *sqlx.Tx, req *TableFieldAliasRequest) (tables.DicTableFieldAliasDB, error) {
	query := `
		UPDATE dic_table_field_alias 
		SET deleted_at = NULL, field_id = ?, table_id = ?, field_name = ?, description = ?, where = ?, updated_at = CURRENT_TIMESTAMP
		WHERE field_alias = ? AND deleted_at IS NOT NULL
	`
	_, err := tx.ExecContext(ctx, query,
		req.FieldID,
		req.TableID,
		req.FieldName,
		req.Description,
		req.WhereCondition,
		req.FieldAlias,
	)
	if err != nil {
		return tables.DicTableFieldAliasDB{}, err
	}

	// 查询恢复后的记录
	var tableFieldAlias tables.DicTableFieldAliasDB
	selectQuery := `
		SELECT id, field_id, table_id, field_name, field_alias, description, where, created_at, updated_at, deleted_at
		FROM dic_table_field_alias
		WHERE field_alias = ? AND deleted_at IS NULL
	`
	err = tx.GetContext(ctx, &tableFieldAlias, selectQuery, req.FieldAlias)
	return tableFieldAlias, err
}

// insertTableFieldAlias 插入字段别名
func insertTableFieldAlias(ctx context.Context, tx *sqlx.Tx, req *TableFieldAliasRequest, creator string) (tables.DicTableFieldAliasDB, error) {
	// 生成ID
	id := req.ID
	if id == "" {
		id = uuid.New().String()
	}

	query := `
		INSERT INTO dic_table_field_alias (id, field_id, table_id, field_name, field_alias, description, where, created_at, updated_at)
		VALUES (?, ?, ?, ?, ?, ?, ?, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)
	`
	logger.Debug(fmt.Sprintf("insertTableFieldAlias - 执行插入: query=%s, id=%s, field_id=%s, table_id=%s, field_alias=%s", query, id, req.FieldID, req.TableID, req.FieldAlias))

	_, err := tx.ExecContext(ctx, query,
		id,
		req.FieldID,
		req.TableID,
		req.FieldName,
		req.FieldAlias,
		req.Description,
		req.WhereCondition,
	)
	if err != nil {
		logger.Error(fmt.Sprintf("insertTableFieldAlias - 插入失败: %v", err))
		return tables.DicTableFieldAliasDB{}, err
	}

	// 查询刚插入的记录
	var tableFieldAlias tables.DicTableFieldAliasDB
	selectQuery := `
		SELECT id, field_id, table_id, field_name, field_alias, description, where, created_at, updated_at, deleted_at
		FROM dic_table_field_alias
		WHERE field_alias = ? AND deleted_at IS NULL
	`
	err = tx.GetContext(ctx, &tableFieldAlias, selectQuery, req.FieldAlias)
	return tableFieldAlias, err
}

// updateTableFieldAlias 更新字段别名
func updateTableFieldAlias(ctx context.Context, tx *sqlx.Tx, req *TableFieldAliasRequest) (tables.DicTableFieldAliasDB, error) {
	// 使用ID更新（如果提供ID），否则使用field_alias
	var query string
	var queryArgs []interface{}

	if req.ID != "" {
		query = `
			UPDATE dic_table_field_alias
			SET field_id = ?, table_id = ?, field_name = ?, description = ?, where = ?, updated_at = CURRENT_TIMESTAMP
			WHERE id = ? AND deleted_at IS NULL
		`
		queryArgs = []interface{}{req.FieldID, req.TableID, req.FieldName, req.Description, req.WhereCondition, req.ID}
	} else {
		query = `
			UPDATE dic_table_field_alias
			SET field_id = ?, table_id = ?, field_name = ?, description = ?, where = ?, updated_at = CURRENT_TIMESTAMP
			WHERE field_alias = ? AND deleted_at IS NULL
		`
		queryArgs = []interface{}{req.FieldID, req.TableID, req.FieldName, req.Description, req.WhereCondition, req.FieldAlias}
	}

	_, err := tx.ExecContext(ctx, query, queryArgs...)
	if err != nil {
		return tables.DicTableFieldAliasDB{}, err
	}

	// 查询更新后的记录
	var tableFieldAlias tables.DicTableFieldAliasDB
	selectQuery := `
		SELECT id, field_id, table_id, field_name, field_alias, description, where, created_at, updated_at, deleted_at
		FROM dic_table_field_alias
		WHERE field_alias = ? AND deleted_at IS NULL
	`
	err = tx.GetContext(ctx, &tableFieldAlias, selectQuery, req.FieldAlias)
	return tableFieldAlias, err
}