well-goknown/vendor/github.com/valyala/fasthttp/userdata.go

109 lines
1.5 KiB
Go
Raw Permalink Normal View History

package fasthttp
import (
"io"
)
type userDataKV struct {
2024-09-26 01:59:44 +00:00
key any
value any
}
type userData []userDataKV
2024-09-26 01:59:44 +00:00
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
}
2024-09-26 01:59:44 +00:00
func (d *userData) SetBytes(key []byte, value any) {
d.Set(key, value)
}
2024-09-26 01:59:44 +00:00
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
}
2024-09-26 01:59:44 +00:00
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()
}
2024-09-26 01:59:44 +00:00
(*d)[i].value = nil
(*d)[i].key = nil
}
*d = (*d)[:0]
}
2024-09-26 01:59:44 +00:00
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]
2024-09-26 01:59:44 +00:00
args[n].key = nil
args[n].value = nil
args = args[:n]
*d = args
return
}
}
}
func (d *userData) RemoveBytes(key []byte) {
d.Remove(key)
}