Call Close on user values stored via RequestCtx.SetUserValue if these values implement io.Closer

1 files changed, 12 insertions, 0 deletions

diff --git a/userdata.go b/userdata.go
index ecc343e..a99ec5d 100644
--- a/userdata.go
+++ b/userdata.go
@@ -1,5 +1,9 @@
 package fasthttp
 
+import (
+	"io"
+)
+
 type userDataKV struct {
 	key   []byte
 	value interface{}
@@ -55,5 +59,13 @@ func (d *userData) GetBytes(key []byte) interface{} {
 }
 
 func (d *userData) Reset() {
+	args := *d
+	n := len(args)
+	for i := 0; i < n; i++ {
+		v := args[i].value
+		if vc, ok := v.(io.Closer); ok {
+			vc.Close()
+		}
+	}
 	*d = (*d)[:0]
 }