before write to output

getevo · Oct 19, 2022 · 731cefb · 731cefb
1 parent cc1fa4e
commit 731cefb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/evo.context.go b/evo.context.go
@@ -27,7 +27,7 @@ type Request struct {
 	CacheDuration time.Duration
 	Debug         bool
 	flashes       []flash
-	BeforeWrite   func(request *Request)
+	BeforeWrite   func(request *Request, body []byte) []byte
 }
 type flash struct {
 	Type    string `json:"type"`

diff --git a/fiber.context.go b/fiber.context.go
@@ -477,7 +477,7 @@ func (r *Request) Write(body interface{}) {
 		}, r.CacheDuration)
 	}
 	if r.BeforeWrite != nil {
-		r.BeforeWrite(r)
+		data = r.BeforeWrite(r, data)
 	}
 	r.Context.Context().Response.SetBody(data)
 }