httpHandler.go

package webapi

import (
	"github.com/farseer-go/fs/asyncLocal"
	"github.com/farseer-go/fs/container"
	"github.com/farseer-go/fs/flog"
	"github.com/farseer-go/fs/trace"
	"github.com/farseer-go/webapi/context"
	"net/http"
)

func HttpHandler(route *context.HttpRoute) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		// 解析报文、组装httpContext
		httpContext := context.NewHttpContext(route, w, r)
		// 创建链路追踪上下文
		trackContext := container.Resolve[trace.IManager]().EntryWebApi(httpContext.URI.Host, httpContext.URI.Url, httpContext.Method, httpContext.ContentType, httpContext.Header.ToMap(), httpContext.URI.GetRealIp())
		// 记录出入参
		defer func() {
			trackContext.SetBody(httpContext.Request.BodyString, httpContext.Response.GetHttpCode(), string(httpContext.Response.BodyBytes))
			trackContext.End(httpContext.Exception)
		}()
		httpContext.Data.Set("Trace", trackContext)

		// 设置到routine，可用于任意子函数获取
		context.RoutineHttpContext.Set(httpContext)
		// 执行第一个中间件
		route.HttpMiddleware.Invoke(httpContext)
		// 记录异常
		if httpContext.Exception != nil {
			_ = flog.Errorf("[%s]%s 发生错误：%s", httpContext.Method, httpContext.URI.Url, httpContext.Exception.Error())
		}
		asyncLocal.Release()
	}
}