fix bug

database/new.go

@@ -25,6 +25,7 @@ var (
 // options: 数据库连接选项
 // 返回: GORM数据库实例
 func NewDatabase(driver string, dsn []string, options *types.SqlOptions) (db *gorm.DB, err error) {
+	MigrateTables = make([]any, 0)
 	driver = strings.ToLower(driver)
 	switch driver {
 	case "mysql":
@@ -133,3 +134,13 @@ func NewPostgres(dsn []string, options *types.SqlOptions) (gormDb *gorm.DB, err
 
 	return gormDb, nil
 }
+
+// AppendMigrate 调用此函数后，会在数据库初始化时自动迁移表结构
+//
+// - table: 需要自动迁移的表
+func AppendMigrate(table any) {
+	if MigrateTables == nil {
+		MigrateTables = make([]any, 0)
+	}
+	MigrateTables = append(MigrateTables, table)
+}

errcode/errcode.go

@@ -11,6 +11,7 @@ import (
 
 // HTTP请求头相关错误码，起始码:1000
 var (
+	AllErrors              = make(map[int]string)
 	ErrHeaderRequestId     = NewError(1001, "Header Request-Id Not Found")    // 请求ID头缺失
 	ErrHeaderAuthorization = NewError(1002, "Header Authorization Not Found") // 授权头缺失
 	ErrHeaderSecretKey     = NewError(1003, "Header Secret-Key Not Found")    // 密钥头缺失
@@ -81,6 +82,7 @@ var (
 // code: 错误码
 // msg: 错误消息
 func NewError(code int, msg string) error {
+	AllErrors[code] = msg
 	return status.New(codes.Code(code), msg).Err()
 }
 
@@ -88,6 +90,7 @@ func NewError(code int, msg string) error {
 // code: 错误码
 // msg: 错误消息
 func ErrFatal(code int, msg string) error {
+	AllErrors[code] = msg
 	return status.New(codes.Code(code), msg).Err()
 }
 
@@ -95,5 +98,6 @@ func ErrFatal(code int, msg string) error {
 // code: 错误码
 // msg: 错误消息，会自动转换为大写
 func ErrNotFound(code int, msg string) error {
+	AllErrors[code] = strings.ToUpper(msg)
 	return status.New(codes.Code(code), strings.ToUpper(msg)).Err()
 }