package main

import (
	"fmt"
	"gitee.com/liujinsuo/tool/toolhttp"
	"net/http"
)

func main() {
	httpt()
}
func httpm(handler http.Handler) http.Handler {
	return http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
		fmt.Println(1)
		w := toolhttp.NewResponseWriter(writer)
		handler.ServeHTTP(w, request)
		fmt.Println("body=", w.GetBody().String(), "code=", w.GetCode())

		w.CleanBody()              //清空响应body
		w.Write([]byte("1111111")) //重写响应body

		w.ResponseWriter.Write(w.GetBody().Bytes())
		w.ResponseWriter.WriteHeader(w.GetCode())
	})
}
func httpt() {
	http.Handle("/", httpm(http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
		fmt.Println(2)
		writer.WriteHeader(http.StatusOK)
		writer.Write([]byte("2222222222"))
	})))

	http.ListenAndServe(":8080", http.DefaultServeMux)
}