.

vendor/github.com/valyala/fasthttp/userdata.go

@@ -0,0 +1,105 @@
+package fasthttp
+
+import (
+	"io"
+)
+
+type userDataKV struct {
+	key   any
+	value any
+}
+
+type userData []userDataKV
+
+func (d *userData) Set(key, value any) {
+	if b, ok := key.([]byte); ok {
+		key = string(b)
+	}
+	args := *d
+	n := len(args)
+	for i := 0; i < n; i++ {
+		kv := &args[i]
+		if kv.key == key {
+			kv.value = value
+			return
+		}
+	}
+
+	if value == nil {
+		return
+	}
+
+	c := cap(args)
+	if c > n {
+		args = args[:n+1]
+		kv := &args[n]
+		kv.key = key
+		kv.value = value
+		*d = args
+		return
+	}
+
+	kv := userDataKV{}
+	kv.key = key
+	kv.value = value
+	args = append(args, kv)
+	*d = args
+}
+
+func (d *userData) SetBytes(key []byte, value any) {
+	d.Set(key, value)
+}
+
+func (d *userData) Get(key any) any {
+	if b, ok := key.([]byte); ok {
+		key = b2s(b)
+	}
+	args := *d
+	n := len(args)
+	for i := 0; i < n; i++ {
+		kv := &args[i]
+		if kv.key == key {
+			return kv.value
+		}
+	}
+	return nil
+}
+
+func (d *userData) GetBytes(key []byte) any {
+	return d.Get(key)
+}
+
+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]
+}
+
+func (d *userData) Remove(key any) {
+	if b, ok := key.([]byte); ok {
+		key = b2s(b)
+	}
+	args := *d
+	n := len(args)
+	for i := 0; i < n; i++ {
+		kv := &args[i]
+		if kv.key == key {
+			n--
+			args[i], args[n] = args[n], args[i]
+			args[n].value = nil
+			args = args[:n]
+			*d = args
+			return
+		}
+	}
+}
+
+func (d *userData) RemoveBytes(key []byte) {
+	d.Remove(key)
+}