cbaf9e6142
After discussion, we decided to simplify attribute translation for now. Full translation requires support for UTF-8 values encoded according to RFC 8187/7230, this can be clarified and implemented in further PRs. 1. Remove translation tables 2. Use simple translation rules only, for now ignoring UTF-8 keys and values - `X-Attribute-NEOFS-` prefixed headers considered well-known system attributes - System Attribute key is uppercased - System Attribute key's `-` translates to `_` - `X-Attribute-` prefixed headers considered regular object attributes - Normal attribute key is a result of http header prefix trimming - Value string should be left as is, for now ``` HTTP: X-Attribute-NEOFS-Expiration-Epoch: 123 X-Attribute-FileName: cat.jpg NeoFS: __NEOFS__EXPIRATION_EPOCH: "123" FileName: "cat.jpg" ``` Signed-off-by: Evgeniy Kulikov <kim@nspcc.ru>
68 lines
1.4 KiB
Go
68 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
|
|
"github.com/valyala/fasthttp"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
const (
|
|
userAttributeHeaderPrefix = "X-Attribute-"
|
|
neofsAttributeHeaderPrefix = "NEOFS-"
|
|
|
|
systemAttributePrefix = "__NEOFS__"
|
|
)
|
|
|
|
func systemTranslator(key []byte) []byte {
|
|
// replace `NEOFS-` with `__NEOFS__`
|
|
key = bytes.Replace(key, []byte(neofsAttributeHeaderPrefix), []byte(systemAttributePrefix), 1)
|
|
|
|
// replace `-` with `_`
|
|
key = bytes.ReplaceAll(key, []byte("-"), []byte("_"))
|
|
|
|
// replace with uppercase
|
|
return bytes.ToUpper(key)
|
|
}
|
|
|
|
func filterHeaders(l *zap.Logger, header *fasthttp.RequestHeader) map[string]string {
|
|
result := make(map[string]string)
|
|
prefix := []byte(userAttributeHeaderPrefix)
|
|
system := []byte(neofsAttributeHeaderPrefix)
|
|
|
|
header.VisitAll(func(key, val []byte) {
|
|
// checks that key and val not empty
|
|
if len(key) == 0 || len(val) == 0 {
|
|
return
|
|
}
|
|
|
|
// checks that key has attribute prefix
|
|
if !bytes.HasPrefix(key, prefix) {
|
|
return
|
|
}
|
|
|
|
// removing attribute prefix
|
|
key = bytes.TrimPrefix(key, prefix)
|
|
|
|
// checks that it's a system NeoFS header
|
|
if bytes.HasPrefix(key, system) {
|
|
key = systemTranslator(key)
|
|
}
|
|
|
|
// checks that attribute key not empty
|
|
if len(key) == 0 {
|
|
return
|
|
}
|
|
|
|
// make string representation of key / val
|
|
k, v := string(key), string(val)
|
|
|
|
result[k] = v
|
|
|
|
l.Debug("add attribute to result object",
|
|
zap.String("key", k),
|
|
zap.String("val", v))
|
|
})
|
|
|
|
return result
|
|
}
|