before write to output

getevo · Oct 19, 2022 · 935e137 · 935e137
1 parent 3c48bb8
commit 935e137
Show file tree

Hide file tree

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

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