package api import ( "errors" "fmt" "net/http" vd "github.com/bytedance/go-tagexpr/v2/validator" "github.com/gin-gonic/gin" "github.com/gin-gonic/gin/binding" "gorm.io/gorm" "gogs.baozhida.cn/zoie/OAuth-core/logger" "gogs.baozhida.cn/zoie/OAuth-core/pkg" "gogs.baozhida.cn/zoie/OAuth-core/pkg/response" "gogs.baozhida.cn/zoie/OAuth-core/sdk" "gogs.baozhida.cn/zoie/OAuth-core/service" "gogs.baozhida.cn/zoie/OAuth-core/storage" "gogs.baozhida.cn/zoie/OAuth-core/tools/language" ) var DefaultLanguage = "zh-CN" type Api struct { Context *gin.Context Logger *logger.Helper Orm *gorm.DB Errors error Cache storage.AdapterCache } func (e *Api) AddError(err error) { if e.Errors == nil { e.Errors = err } else if err != nil { e.Logger.Error(err) e.Errors = fmt.Errorf("%v; %w", e.Errors, err) } } // MakeContext 设置http上下文 func (e *Api) MakeContext(c *gin.Context) *Api { e.Context = c e.Logger = GetRequestLogger(c) return e } // GetLogger 获取上下文提供的日志 func (e Api) GetLogger() *logger.Helper { return GetRequestLogger(e.Context) } // Bind 参数校验 func (e *Api) Bind(d interface{}, bindings ...binding.Binding) *Api { var err error if len(bindings) == 0 { bindings = constructor.GetBindingForGin(d) } for i := range bindings { if bindings[i] == nil { err = e.Context.ShouldBindUri(d) } else { err = e.Context.ShouldBindWith(d, bindings[i]) } if err != nil && err.Error() == "EOF" { e.Logger.Warn("request body is not present anymore. ") err = nil continue } if err != nil { e.AddError(err) break } } //vd.SetErrorFactory(func(failPath, msg string) error { // return fmt.Errorf(`"validation failed: %s %s"`, failPath, msg) //}) if err1 := vd.Validate(d); err1 != nil { e.AddError(err1) } return e } // GetOrm 获取Orm DB func (e Api) GetOrm() (*gorm.DB, error) { ormDB, err := pkg.GetOrm(e.Context) if err != nil { e.Logger.Error(http.StatusInternalServerError, err, "数据库连接获取失败") return nil, err } return ormDB, nil } // MakeOrm 设置Orm DB func (e *Api) MakeOrm() *Api { var err error if e.Logger == nil { err = errors.New("at MakeOrm logger is nil") e.AddError(err) return e } ormDB, err := pkg.GetOrm(e.Context) if err != nil { e.Logger.Error(http.StatusInternalServerError, err, "数据库连接获取失败") e.AddError(err) } e.Orm = ormDB return e } func (e *Api) MakeService(c *service.Service) *Api { c.Log = e.Logger c.Orm = e.Orm c.Cache = sdk.Runtime.GetCacheAdapter() e.Cache = c.Cache return e } // Error 通常错误数据处理 func (e Api) Error(code int, err error, msg string) { response.Error(e.Context, code, err, msg) } // OK 通常成功数据处理 func (e Api) OK(data interface{}, msg string) { response.OK(e.Context, data, msg) } // PageOK 分页数据处理 func (e Api) PageOK(result interface{}, count int, pageIndex int, pageSize int, msg string) { response.PageOK(e.Context, result, count, pageIndex, pageSize, msg) } // Custom 兼容函数 func (e Api) Custom(data gin.H) { response.Custum(e.Context, data) } func (e Api) Translate(form, to interface{}) { pkg.Translate(form, to) } // getAcceptLanguage 获取当前语言 func (e *Api) getAcceptLanguage() string { languages := language.ParseAcceptLanguage(e.Context.GetHeader("Accept-Language"), nil) if len(languages) == 0 { return DefaultLanguage } return languages[0] }