1

common/apis/api.go

@@ -0,0 +1,135 @@
+package apis
+
+import (
+	"errors"
+	"fmt"
+
+	"github.com/gin-gonic/gin"
+	"github.com/gin-gonic/gin/binding"
+	"github.com/go-admin-team/go-admin-core/logger"
+	"github.com/go-admin-team/go-admin-core/sdk/api"
+	"github.com/go-admin-team/go-admin-core/sdk/pkg"
+	"github.com/go-admin-team/go-admin-core/sdk/pkg/response"
+	"gorm.io/gorm"
+
+	"go-admin/common/service"
+)
+
+type Api struct {
+	Context *gin.Context
+	Logger  *logger.Helper
+	Orm     *gorm.DB
+	Errors  error
+}
+
+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 = api.GetRequestLogger(c)
+	return e
+}
+
+// GetLogger 获取上下文提供的日志
+func (e *Api) GetLogger() *logger.Helper {
+	return api.GetRequestLogger(e.Context)
+}
+
+func (e *Api) Bind(d interface{}, bindings ...binding.Binding) *Api {
+	var err error
+	if len(bindings) == 0 {
+		bindings = append(bindings, binding.JSON, nil)
+	}
+	for i := range bindings {
+		switch bindings[i] {
+		case binding.JSON:
+			err = e.Context.ShouldBindWith(d, binding.JSON)
+		case binding.XML:
+			err = e.Context.ShouldBindWith(d, binding.XML)
+		case binding.Form:
+			err = e.Context.ShouldBindWith(d, binding.Form)
+		case binding.Query:
+			err = e.Context.ShouldBindWith(d, binding.Query)
+		case binding.FormPost:
+			err = e.Context.ShouldBindWith(d, binding.FormPost)
+		case binding.FormMultipart:
+			err = e.Context.ShouldBindWith(d, binding.FormMultipart)
+		case binding.ProtoBuf:
+			err = e.Context.ShouldBindWith(d, binding.ProtoBuf)
+		case binding.MsgPack:
+			err = e.Context.ShouldBindWith(d, binding.MsgPack)
+		case binding.YAML:
+			err = e.Context.ShouldBindWith(d, binding.YAML)
+		case binding.Header:
+			err = e.Context.ShouldBindWith(d, binding.Header)
+		default:
+			err = e.Context.ShouldBindUri(d)
+		}
+		if err != nil {
+			e.AddError(err)
+		}
+	}
+	return e
+}
+
+// GetOrm 获取Orm DB
+func (e *Api) GetOrm() (*gorm.DB, error) {
+	db, err := pkg.GetOrm(e.Context)
+	if err != nil {
+		e.Error(500, 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.Logger.Error(500, err, "at MakeOrm logger is nil")
+		e.AddError(err)
+		return e
+	}
+	db, err := pkg.GetOrm(e.Context)
+	if err != nil {
+		e.Logger.Error(500, err, "数据库连接获取失败")
+		e.AddError(err)
+	}
+	e.Orm = db
+	return e
+}
+
+func (e *Api) MakeService(c *service.Service) *Api {
+	c.Log = e.Logger
+	c.Orm = e.Orm
+	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)
+}