Fixes after review

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>
This commit is contained in:
Evgeniy Kulikov 2021-02-03 16:01:30 +03:00
parent 71999a796d
commit cbaf9e6142
No known key found for this signature in database
GPG key ID: BF6AEE0A2A699BF2
5 changed files with 30 additions and 83 deletions

View file

@ -77,13 +77,4 @@ Peers preset:
HTTP_GW_PEERS_[N]_ADDRESS = string HTTP_GW_PEERS_[N]_ADDRESS = string
HTTP_GW_PEERS_[N]_WEIGHT = 0..1 (float) HTTP_GW_PEERS_[N]_WEIGHT = 0..1 (float)
Upload Header Table:
HTTP_GW_UPLOADER_HEADER_[N]_KEY = string - HTTP Header attribute name prefixed with `X-Attribute-`
HTTP_GW_UPLOADER_HEADER_[N]_VAL = string - NeoFS Object attribute mapping
# By default we had next headers:
- FileName - to set object filename attribute
- Timestamp - to set object timestamp attribute
``` ```

4
app.go
View file

@ -26,8 +26,6 @@ type (
cfg *viper.Viper cfg *viper.Viper
key *ecdsa.PrivateKey key *ecdsa.PrivateKey
hdr HeaderFilter
wlog logger.Logger wlog logger.Logger
web *fasthttp.Server web *fasthttp.Server
@ -78,8 +76,6 @@ func newApp(ctx context.Context, opt ...Option) App {
opt[i](a) opt[i](a)
} }
a.hdr = newHeaderFilter(a.log, a.cfg)
a.enableDefaultTimestamp = a.cfg.GetBool(cfgUploaderHeaderEnableDefaultTimestamp) a.enableDefaultTimestamp = a.cfg.GetBool(cfgUploaderHeaderEnableDefaultTimestamp)
a.wlog = logger.GRPC(a.log) a.wlog = logger.GRPC(a.log)

View file

@ -2,63 +2,33 @@ package main
import ( import (
"bytes" "bytes"
"strconv"
"strings"
"github.com/nspcc-dev/neofs-api-go/pkg/object"
"github.com/spf13/viper"
"github.com/valyala/fasthttp" "github.com/valyala/fasthttp"
"go.uber.org/zap" "go.uber.org/zap"
) )
type ( const (
HeaderFilter interface { userAttributeHeaderPrefix = "X-Attribute-"
Filter(header *fasthttp.RequestHeader) map[string]string neofsAttributeHeaderPrefix = "NEOFS-"
}
headerFilter struct { systemAttributePrefix = "__NEOFS__"
logger *zap.Logger
mapping map[string]string
}
) )
const userAttributeHeaderPrefix = "X-Attribute-" func systemTranslator(key []byte) []byte {
// replace `NEOFS-` with `__NEOFS__`
key = bytes.Replace(key, []byte(neofsAttributeHeaderPrefix), []byte(systemAttributePrefix), 1)
func newHeaderFilter(l *zap.Logger, v *viper.Viper) HeaderFilter { // replace `-` with `_`
filter := &headerFilter{ key = bytes.ReplaceAll(key, []byte("-"), []byte("_"))
logger: l,
mapping: make(map[string]string), // replace with uppercase
return bytes.ToUpper(key)
} }
for i := 0; ; i++ { func filterHeaders(l *zap.Logger, header *fasthttp.RequestHeader) map[string]string {
index := strconv.Itoa(i)
key := strings.Join([]string{cfgUploaderHeader, index, cfgUploaderHeaderKey}, ".")
rep := strings.Join([]string{cfgUploaderHeader, index, cfgUploaderHeaderVal}, ".")
keyValue := v.GetString(key)
repValue := v.GetString(rep)
if keyValue == "" || repValue == "" {
break
}
filter.mapping[keyValue] = repValue
l.Debug("load upload header table value",
zap.String("key", keyValue),
zap.String("val", repValue))
}
// Default values
filter.mapping[object.AttributeFileName] = object.AttributeFileName
filter.mapping[object.AttributeTimestamp] = object.AttributeTimestamp
return filter
}
func (h *headerFilter) Filter(header *fasthttp.RequestHeader) map[string]string {
result := make(map[string]string) result := make(map[string]string)
prefix := []byte(userAttributeHeaderPrefix) prefix := []byte(userAttributeHeaderPrefix)
system := []byte(neofsAttributeHeaderPrefix)
header.VisitAll(func(key, val []byte) { header.VisitAll(func(key, val []byte) {
// checks that key and val not empty // checks that key and val not empty
@ -71,27 +41,27 @@ func (h *headerFilter) Filter(header *fasthttp.RequestHeader) map[string]string
return return
} }
// checks that after removing attribute prefix we had not empty key // removing attribute prefix
if key = bytes.TrimPrefix(key, prefix); len(key) == 0 { 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 return
} }
// checks mapping table and if we found record store it // make string representation of key / val
// at resulting hashmap k, v := string(key), string(val)
if name, ok := h.mapping[string(key)]; ok {
result[name] = string(val)
h.logger.Debug("add attribute to result object", result[k] = v
zap.String("key", name),
zap.String("val", string(val)))
return l.Debug("add attribute to result object",
} zap.String("key", k),
zap.String("val", v))
// otherwise inform that attribute will be ignored
h.logger.Debug("ignore attribute",
zap.String("key", string(key)),
zap.String("val", string(val)))
}) })
return result return result

View file

@ -54,9 +54,6 @@ const (
cfgLoggerSamplingThereafter = "logger.sampling.thereafter" cfgLoggerSamplingThereafter = "logger.sampling.thereafter"
// Uploader Header // Uploader Header
cfgUploaderHeader = "uploader_header"
cfgUploaderHeaderKey = "key"
cfgUploaderHeaderVal = "val"
cfgUploaderHeaderEnableDefaultTimestamp = "upload_header.use_default_timestamp" cfgUploaderHeaderEnableDefaultTimestamp = "upload_header.use_default_timestamp"
// Peers // Peers
@ -187,13 +184,6 @@ func settings() *viper.Viper {
fmt.Printf("%s_%s_[N]_ADDRESS = string\n", Prefix, strings.ToUpper(cfgPeers)) fmt.Printf("%s_%s_[N]_ADDRESS = string\n", Prefix, strings.ToUpper(cfgPeers))
fmt.Printf("%s_%s_[N]_WEIGHT = 0..1 (float)\n", Prefix, strings.ToUpper(cfgPeers)) fmt.Printf("%s_%s_[N]_WEIGHT = 0..1 (float)\n", Prefix, strings.ToUpper(cfgPeers))
fmt.Println()
fmt.Println("Upload Header Table:")
fmt.Println()
fmt.Printf("%s_%s_[N]_%s = string\n", Prefix, strings.ToUpper(cfgUploaderHeader), strings.ToUpper(cfgUploaderHeaderKey))
fmt.Printf("%s_%s_[N]_%s = string\n", Prefix, strings.ToUpper(cfgUploaderHeader), strings.ToUpper(cfgUploaderHeaderVal))
os.Exit(0) os.Exit(0)
case version != nil && *version: case version != nil && *version:
fmt.Printf("NeoFS HTTP Gateway %s (%s)\n", Version, Build) fmt.Printf("NeoFS HTTP Gateway %s (%s)\n", Version, Build)

View file

@ -103,7 +103,7 @@ func (a *app) upload(c *fasthttp.RequestCtx) {
} }
} }
filtered := a.hdr.Filter(&c.Request.Header) filtered := filterHeaders(a.log, &c.Request.Header)
attributes := make([]*object.Attribute, 0, len(filtered)) attributes := make([]*object.Attribute, 0, len(filtered))
// prepares attributes from filtered headers // prepares attributes from filtered headers