在中间件中重写内容长度

哈喽！大家好，很高兴又见面了，我是golang学习网的一名作者，今天由我给大家带来一篇《在中间件中重写内容长度》，本文主要会讲到等等知识点，希望大家一起学习进步，也欢迎大家关注、点赞、收藏、转发! 下面就一起来看看吧！

下面的代码重写了一些查询的http正文响应。

但是，它无法更新“内容长度”标头字段，它始终保持相同的原始值。

如何更新 http 响应的内容长度标头字段？

type writereplacer struct {
    http.responsewriter
    search  []byte
    replace func(*http.request) string
    buf     []byte
    r       *http.request
    dir     string
}

func (w *writereplacer) write(in []byte) (int, error) {
    if w.buf == nil {
        w.buf = []byte{}
    }
    w.buf = append(w.buf, in...)
    n := len(in)
    if index := bytes.lastindex(w.buf, w.search); index > -1 {
        var r []byte
        if w.dir == "before" {
            g := []byte(w.replace(w.r))
            n += len(g)
            r = append(g, w.buf[index:]...)
            w.buf = append(w.buf[:index], r...)
        } else {
            g := []byte(w.replace(w.r))
            n += len(g)
            r = append(r, w.buf[:index+len(w.search)]...)
            r = append(r, g...)
            r = append(r, w.buf[index:]...)
            w.buf = r
        }
    }
    return n, nil
}

func (w *writereplacer) flush() {
    w.responsewriter.header().set("content-length", fmt.sprint(len(w.buf)))
    w.responsewriter.write(w.buf[:])
    w.buf = w.buf[:0]
}

func insertafter(h http.handler, path string, search []byte, replace func(*http.request) string) http.handler {
    return http.handlerfunc(func(w http.responsewriter, r *http.request) {
        if r.url.path == path {
            w.header().del("content-length")
            w = &writereplacer{responsewriter: w, search: search, replace: replace, r: r, dir: "after"}
            defer w.(http.flusher).flush()
        }
        h.servehttp(w, r)
    })
}

func insertbefore(h http.handler, path string, search []byte, replace func(*http.request) string) http.handler {
    return http.handlerfunc(func(w http.responsewriter, r *http.request) {
        if r.url.path == path {
            w.header().del("content-length")
            w = &writereplacer{responsewriter: w, search: search, replace: replace, r: r, dir: "before"}
            defer w.(http.flusher).flush()
        }
        h.servehttp(w, r)
    })
}

我收到两个错误消息，第一个来自 nginx

2019/03/08 05:58:37 [error] 31194#0: *19 
upstream prematurely closed connection while reading upstream, 
client: 82.21.18.16, server: buycoffee.online, request: 
"get / http/1.1", upstream: "http://127.0.0.1:8081/", host: "buycoffee.online"

第二个来自curl

curl: (18) transfer closed with 6237 bytes remaining to read

解决方案

正如 mkopriva 所建议的，技巧是重写 responsewriter.writeheader。

我相信这与 write 可能在第一次调用时调用 writeheader 的事实有关，从那里我没有将指令放在正确的位置。

为了防止进一步的困难，我更喜欢使用分块传输。

代码更改为：

func (w *writeReplacer) WriteHeader(statusCode int) {
    w.Header().Del("Content-length")
    w.Header().Set("Transfer-Encoding", "chunked")
    w.ResponseWriter.WriteHeader(statusCode)
}

理论要掌握，实操不能落！以上关于《在中间件中重写内容长度》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！