golang-gin使用中间件处理文本-时间字符串格式

想象造物

已于 2024-08-27 09:09:33 修改

阅读量1.2k

点赞数 10

分类专栏：技术日志文章标签： gin golang 中间件

于 2024-08-27 09:08:41 首次发布

本文链接：https://blog.youkuaiyun.com/simulate_create/article/details/141586658

版权

技术日志专栏收录该内容

5 篇文章

订阅专栏

一、问题

面临的问题：需要对已有业务代码修改时间格式，原本的时间格式里面带了一个T。因为时间分散在各处，又没有统一的管理入口。这面临着大量的代码修改，于是想找一个没有那么大修改量的方式。

二、思路

思考了一下，两个地方是肯定会经过的，一个是JSON转化的地方，一个是中间件。

1. 在JSON序列化处修改。

如果能添加一个类似Python的encoder之类的东西，那么就可以解决所有的日期的序列化。可惜在一番寻找后，并没有发现这样的东西。其间发现gorm支持在tag上添加序列化，但发现gin是直接使用的原生的json模块，并没有走gorm的逻辑。而原生的json没有统一的修改序列化的地方。

2. 在中间件处修改。

这个是gin支持的中间件机制。之前有使用过。

3. monkey patch到time模块上，修改其序列化方法。

直接monkey patch修改原生time类型的序列化方法。这类方法比较脏，不确定是否可能有潜在风险。

三、日志

我上手尝试了一下中间件机制。发现可行就没有再继续验证其它方法。

1. 创建一个简单的REST API


package main
import (
    "regexp"
    "time"
    "github.com/gin-gonic/gin"
)
type StructA struct {
    FieldA string    `form:"field_a"  binding:"required"`
    Ctime  time.Time `form:"ctime" binding:"required"`
}
type StructB struct {
    // 嵌套结构体
    StructA
    FieldB string `form:"field_b"`
}
func GetB(c *gin.Context) {
    // 绑定表单数据至自定义结构体
    var b StructB
    b.Ctime = time.Now()
    c.JSON(200, b)
}
func main() {
    r := gin.Default()
    r.GET("/getb", GetB)
    r.Run()
}

2. 添加中间件


type copyWriter struct {
    gin.ResponseWriter
}
func (cw copyWriter) Write(b []byte) (int, error) {
    // time_re := regexp.MustCompile(`(\d{4}-\d{2}-\d{2})T(\d{2}:\d{2}:\d{2}\.\d+)([+-]\d{2}:\d{2})`)
    time_re := regexp.MustCompile(`(\d{4}-\d{2}-\d{2})T(\d{2}:\d{2}:\d{2}\.\d{2})\d*([+-]\d{2}:\d{2})`)
    // time_re := regexp.MustCompile(`(\d{4}-\d{2}-\d{2})T(\d{2}:\d{2}:\d{2})`)
    s := string(b)
    result := time_re.ReplaceAllString(s, "$1 $2")
    return cw.ResponseWriter.WriteString(result)
}

func main() {
    r := gin.Default()
    r.Use(func(ctx *gin.Context) {
        cw := &copyWriter{ResponseWriter: ctx.Writer}
        ctx.Writer = cw
        ctx.Next()
        // read data in buf
        // do something for data
        // write to ResponseWriter
    })
    r.GET("/getb", GetB)
    r.Run()
}

其中提供了几种不同输出时间格式的正则：


    // time_re := regexp.MustCompile(`(\d{4}-\d{2}-\d{2})T(\d{2}:\d{2}:\d{2}\.\d+)([+-]\d{2}:\d{2})`)
    time_re := regexp.MustCompile(`(\d{4}-\d{2}-\d{2})T(\d{2}:\d{2}:\d{2}\.\d{2})\d*([+-]\d{2}:\d{2})`)
    // time_re := regexp.MustCompile(`(\d{4}-\d{2}-\d{2})T(\d{2}:\d{2}:\d{2})`)

3. 输出结果如下：


{
    "FieldA": "",
    "Ctime": "2024-08-19 19:33:42.75",
    "FieldB": ""
}

还可以根据需要，在中间件中增加过滤条件。例如，在仅支持json格式的过滤，防止将文件下载等里的日期也替换掉。

4. 完整代码


package main
import (
    "regexp"
    "time"
    "github.com/gin-gonic/gin"
)
type StructA struct {
    FieldA string    `form:"field_a"  binding:"required"`
    Ctime  time.Time `form:"ctime" binding:"required"`
}
type StructB struct {
    // 嵌套结构体
    StructA
    FieldB string `form:"field_b"`
}
func GetB(c *gin.Context) {
    // 绑定表单数据至自定义结构体
    var b StructB
    b.Ctime = time.Now()
    c.JSON(200, b)
}
type copyWriter struct {
    gin.ResponseWriter
}
func (cw copyWriter) Write(b []byte) (int, error) {
    // time_re := regexp.MustCompile(`(\d{4}-\d{2}-\d{2})T(\d{2}:\d{2}:\d{2}\.\d+)([+-]\d{2}:\d{2})`)
    time_re := regexp.MustCompile(`(\d{4}-\d{2}-\d{2})T(\d{2}:\d{2}:\d{2}\.\d{2})\d*([+-]\d{2}:\d{2})`)
    // time_re := regexp.MustCompile(`(\d{4}-\d{2}-\d{2})T(\d{2}:\d{2}:\d{2})`)
    s := string(b)
    result := time_re.ReplaceAllString(s, "$1 $2")
    return cw.ResponseWriter.WriteString(result)
}
func main() {
    r := gin.Default()
    // 创建一个middleware函数，替换返回中的字符串
    r.Use(func(ctx *gin.Context) {
        cw := &copyWriter{ResponseWriter: ctx.Writer}
        ctx.Writer = cw
        ctx.Next()
        // read data in buf
        // do something for data
        // write to ResponseWriter
    })
    r.GET("/getb", GetB)
    r.Run()
}