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)
}