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 }