baize/binding/bind_item.go

package binding

import (
	"git.sxidc.com/go-framework/baize/api"
	"git.sxidc.com/go-framework/baize/binding/request"
	"git.sxidc.com/go-framework/baize/domain"
	"git.sxidc.com/go-tools/utils/reflectutils"
	"git.sxidc.com/go-tools/utils/strutils"
	"git.sxidc.com/service-supports/fserr"
	"net/http"
	"reflect"
)

func Bind[O any](binder *Binder, item *BindItem[O], middlewares ...api.Handler) {
	item.bind(binder, middlewares...)
}

// BindItem 通用BindItem
type BindItem[O any] struct {
	Method string
	*SimpleBindItem[O]
}

func (item *BindItem[O]) bind(binder *Binder, middlewares ...api.Handler) {
	if strutils.IsStringEmpty(item.Path) {
		panic("需要指定路径")
	}

	if strutils.IsStringEmpty(item.Method) {
		panic("需要指定方法")
	}

	if item.ResponseFunc == nil {
		panic("需要指定响应函数")
	}

	if item.ServiceFunc == nil {
		panic("需要指定应用服务函数")
	}

	// 给单个路由增加中间件
	handlers := []api.Handler{
		func(c *api.Context) {
			itemDTO := item.DTO

			// 有请求数据
			if itemDTO != nil {
				// 将请求数据解析到dto中
				if item.DTOBindFunc != nil {
					err := item.DTOBindFunc(c, itemDTO)
					if err != nil {
						var outputZero O
						item.ResponseFunc(c, http.StatusBadRequest, outputZero, err)
						return
					}
				} else {
					switch item.Method {
					case http.MethodPost:
						fallthrough
					case http.MethodPut:
						err := request.JsonBody(c, itemDTO)
						if err != nil {
							var outputZero O
							item.ResponseFunc(c, http.StatusBadRequest, outputZero, err)
							return
						}
					case http.MethodGet:
						err := request.QueryParams(c, itemDTO)
						if err != nil {
							var outputZero O
							item.ResponseFunc(c, http.StatusBadRequest, outputZero, err)
							return
						}
					case http.MethodDelete:
						err := request.PathParams(c, itemDTO)
						if err != nil {
							var outputZero O
							item.ResponseFunc(c, http.StatusBadRequest, outputZero, err)
							return
						}
					}
				}
			}

			// 进行领域对象转化
			var domainObjects []domain.Object
			if item.FormDomainObjectsFunc != nil {
				innerDomainObjects, err := item.FormDomainObjectsFunc(c, itemDTO)
				if err != nil {
					var outputZero O
					item.ResponseFunc(c, http.StatusOK, outputZero, err)
					return
				}

				domainObjects = innerDomainObjects
			} else {
				if item.Objects != nil && len(item.Objects) != 0 {
					for _, object := range item.Objects {
						if object == nil {
							continue
						}

						objectType := reflect.TypeOf(object)
						if !reflectutils.IsTypeStructOrStructPointer(objectType) {
							var outputZero O
							item.ResponseFunc(c, http.StatusOK, outputZero, fserr.New("领域对象不是结构或结构指针"))
							return
						}

						obj := reflect.New(reflectutils.PointerTypeElem(objectType)).Interface()
						err := request.AssignDTOToDomainObject(itemDTO, obj)
						if err != nil {
							var outputZero O
							item.ResponseFunc(c, http.StatusOK, outputZero, err)
							return
						}

						domainObjects = append(domainObjects, obj)
					}
				}
			}

			// 执行业务函数
			statusCode := http.StatusOK

			i := binder.i
			if item.Infrastructure != nil {
				i = item.Infrastructure
			}

			outputModel, err := item.ServiceFunc(c, itemDTO, domainObjects, i)
			if err != nil {
				statusCode = fserr.ParseCode(err).HttpCode
			}

			item.ResponseFunc(c, statusCode, outputModel, err)
			return
		},
	}

	handlers = append(handlers, middlewares...)

	// 所有的函数加入到执行链中
	binder.router.AddRoute(item.Method, item.Path, handlers...)
}