1 Star 0 Fork 0

linxing / youye-core

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
api.go 3.52 KB
一键复制 编辑 原始数据 按行查看 历史
Mark 提交于 2024-03-18 11:31 . feat: app add plugin
package api
import (
"errors"
"fmt"
"net/http"
"gitee.com/linxing_3/youye-core/logger"
"gitee.com/linxing_3/youye-core/sdk"
"gitee.com/linxing_3/youye-core/sdk/pkg"
"gitee.com/linxing_3/youye-core/sdk/pkg/response"
respgin "gitee.com/linxing_3/youye-core/sdk/pkg/response/gin"
"gitee.com/linxing_3/youye-core/sdk/service"
"gitee.com/linxing_3/youye-core/storage"
"gitee.com/linxing_3/youye-core/tools/language"
vd "github.com/bytedance/go-tagexpr/v2/validator"
"github.com/gin-gonic/gin"
"github.com/gin-gonic/gin/binding"
"gorm.io/gorm"
)
var DefaultLanguage = "zh-CN"
type Api struct {
Context *gin.Context
Logger *logger.Helper
Orm *gorm.DB
Errors error
Cache storage.AdapterCache
respWriter response.IRespWriter
}
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.respWriter = respgin.NewGinRespWriter(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) {
db, err := pkg.GetOrm(e.Context)
if err != nil {
e.Logger.Error(http.StatusInternalServerError, err, "数据库连接获取失败")
return nil, err
}
return db, 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
}
db, err := pkg.GetOrm(e.Context)
if err != nil {
e.Logger.Error(http.StatusInternalServerError, err, "数据库连接获取失败")
e.AddError(err)
}
e.Orm = db
return e
}
func (e *Api) MakeService(c *service.Context) *Api {
c.Log = e.Logger
c.Orm = e.Orm
c.Cache = sdk.App.GetCacheAdapter()
e.Cache = c.Cache
return e
}
// Error 通常错误数据处理
func (e Api) Error(code int, err error, msg string) {
response.Error(e.respWriter, code, err, msg)
}
// OK 通常成功数据处理
func (e Api) OK(data interface{}, msg string) {
response.OK(e.respWriter, data, msg)
}
// PageOK 分页数据处理
func (e Api) PageOK(result interface{}, count int, pageIndex int, pageSize int, msg string) {
response.PageOK(e.respWriter, result, count, pageIndex, pageSize, msg)
}
// Custom 兼容函数
func (e Api) Custom(data gin.H) {
response.Custum(e.respWriter, 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]
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/linxing_3/youye-core.git
git@gitee.com:linxing_3/youye-core.git
linxing_3
youye-core
youye-core
v0.0.1-202404151115

搜索帮助

344bd9b3 5694891 D2dac590 5694891