mbiryukova/frostfs-http-gw
1
0
Fork 0
forked from TrueCloudLab/frostfs-http-gw
Code Pull requests Activity

[#18] Add fetching owner variously on file uploading

Browse source 
Signed-off-by: Pavel Korotkov <pavel@nspcc.ru>
This commit is contained in:
Pavel Korotkov 2021-02-16 18:38:30 +03:00 committed by pkorotkov
parent 237c247ec4
commit f77c4e49dc
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
upload.go
View file

@ -1,6 +1,7 @@
package main
import (
"context"
"encoding/json"
"io"
"strconv"
@ -9,6 +10,7 @@ import (
sdk "github.com/nspcc-dev/cdn-sdk"
"github.com/nspcc-dev/neofs-api-go/pkg/container"
"github.com/nspcc-dev/neofs-api-go/pkg/object"
"github.com/nspcc-dev/neofs-api-go/pkg/owner"
"github.com/valyala/fasthttp"
"go.uber.org/zap"
)
@ -33,6 +35,14 @@ func (pr *putResponse) Encode(w io.Writer) error {
return enc.Encode(pr)
}
func (a *app) fetchOwner(ctx context.Context) *owner.ID {
if tkn, err := sdk.BearerToken(ctx); err == nil && tkn != nil {
return tkn.Issuer()
}
return a.cli.Owner()
}
func (a *app) upload(c *fasthttp.RequestCtx) {
var (
err error
@ -109,7 +119,7 @@ func (a *app) upload(c *fasthttp.RequestCtx) {
// prepares new object and fill it
raw := object.NewRaw()
raw.SetContainerID(cid)
raw.SetOwnerID(a.cli.Owner()) // should be various: from sdk / BearerToken
raw.SetOwnerID(a.fetchOwner(c))
raw.SetAttributes(attributes...)
// tries to put file into NeoFS or throw error