apps/core
1
0
Fork 0
mirror of https://github.com/datarhei/core.git synced 2025-10-11 19:00:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use buffer pool

Browse Source
This commit is contained in:
Ingo Oppermann
2024-10-09 16:29:55 +02:00
parent f97943b275
commit 05e4118e0c
7 changed files with 268 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
http/middleware/session/HTTP.go
View File

@@ -7,7 +7,7 @@ import (
"github.com/lithammer/shortuuid/v4"
)
func (h *handler) handleHTTP(c echo.Context, ctxuser string, data map[string]interface{}, next echo.HandlerFunc) error {
func (h *handler) handleHTTP(c echo.Context, _ string, data map[string]interface{}, next echo.HandlerFunc) error {
req := c.Request()
res := c.Response()
@@ -30,13 +30,13 @@ func (h *handler) handleHTTP(c echo.Context, ctxuser string, data map[string]int
id := shortuuid.New()
reader := req.Body
r := &fakeReader{
r := &bodysizeReader{
reader: req.Body,
}
req.Body = r
writer := res.Writer
w := &fakeWriter{
w := &bodysizeWriter{
ResponseWriter: res.Writer,
}
res.Writer = w
@@ -44,19 +44,21 @@ func (h *handler) handleHTTP(c echo.Context, ctxuser string, data map[string]int
h.httpCollector.RegisterAndActivate(id, "", location, referrer)
h.httpCollector.Extra(id, data)
defer h.httpCollector.Close(id)
defer func() {
buffer := h.bufferPool.Get()
req.Body = reader
h.httpCollector.Ingress(id, r.size+headerSize(req.Header))
}()
h.httpCollector.Ingress(id, r.size+headerSize(req.Header, buffer))
defer func() {
res.Writer = writer
h.httpCollector.Egress(id, w.size+headerSize(res.Header()))
h.httpCollector.Egress(id, w.size+headerSize(res.Header(), buffer))
data["code"] = res.Status
h.httpCollector.Extra(id, data)
h.httpCollector.Close(id)
h.bufferPool.Put(buffer)
}()
return next(c)