Iris在log中打印请求信息和返回结果 方便调试

Iris这个框架相关的资料太少,进行任何封装操作都只能点进去看源码进行操作,总结了下打印日志信息的方式,留给有需要的人

大致思路

1、将每次请求返回的数据保存在ctx的values中
2、自定义日志中间件,每次请求done之后调用中间件
3、在日志中间件中打印请求数据

将每次请求返回的数据保存在ctx的values中
import (
	"xiaoge/src/core"
)

// TestController 服务于 "/", "/ping" and "/hello" 路由。
type TestController struct {
	core.BaseController
}

// GetHello serves
// Method:   POST
// Resource: http://localhost:8080/test/hello
func (c *TestController) PostHello() interface{} {
	return GenResult(c.Ctx, core.GenSuccessData("测试返回第一条json"))
}

func (c *TestController) GetHello() interface{} {
	return GenResult(c.Ctx, core.GenSuccessData("测试返回第一条json"))
}

func (c *TestController) GetError() interface{} {
	i := 0
	return GenResult(c.Ctx, core.GenSuccessData(i/i))
}
/**
* @Description: 生成返回的json结果
* @Author: guoyzh
* @Date: 2019/10/23
 */
func GenResult(ctx iris.Context, result *core.ResponseBean) interface{} {
	// 将返回结果封装到ctx的value中 方便日志数据打印
	ctx.Values().Set(core.ResponseJsonKey, result)
	return result
}
自定义日志中间件,每次请求done之后调用中间件
app.Done(LoggerMiddleware)
在日志中间件中打印请求数据
/**
* @Description: LoggerMiddleware 日志中间件
* @Author: guoyzh
* @Date: 2019/10/23
 */
func LoggerMiddleware(ctx iris.Context) {
	method := ctx.Request().Method
	path := ctx.Request().URL.String()
	token := ctx.Request().Header.Get("token")

	// 如果是POST/PUT请求,并且内容类型为JSON,则读取内容体
	if method == http.MethodPost || method == http.MethodPut || method == http.MethodPatch {
	}
	params := ""
	body, err := ioutil.ReadAll(ctx.Request().Body)
	if err == nil {
		defer ctx.Request().Body.Close()
		buf := bytes.NewBuffer(body)
		ctx.Request().Body = ioutil.NopCloser(buf)
		params = string(body)
		if strings.Contains(params, "\r\n") {
			params = strings.ReplaceAll(params, "\r\n", "")
		}
		if strings.Contains(params, "\n") {
			params = strings.ReplaceAll(params, "\n", "")
		}
		params = strings.ReplaceAll(params, " ", "")
	}
	ctx.Next()

	// 下面是返回日志
	resStr := ""
	if v := ctx.Values().Get(core.ResponseJsonKey); v != nil {
		jsonRes, _ := json.Marshal(v)
		resStr = string(jsonRes)
	}
	// writer := ctx.ResponseWriter()

	divider := "────────────────────────────────────────────────────────────────────────────────────────────────────────────────"
	topDivider := "┌" + divider
	middleDivider := "├" + divider
	bottomDivider := "└" + divider
	outputStr :=
		"\n" + topDivider +
			"\n│ 时间:" + util.GetNowTimeStr() +
			"\n│ 请求地址:" + path + "\n" + middleDivider +
			"\n│ token:" + token +
			"\n│ 请求参数:" + params +
			"\n│ 返回数据:" + resStr + "\n" + bottomDivider
	fmt.Println(outputStr)
}

你可能感兴趣的:(Go)