frostfs-http-gw/receive.go

171 lines
4.1 KiB
Go
Raw Normal View History

package main
import (
"context"
"io"
"net/http"
"path"
"strconv"
"time"
2020-03-31 08:37:10 +00:00
"github.com/nspcc-dev/neofs-api-go/object"
"github.com/nspcc-dev/neofs-api-go/refs"
"github.com/nspcc-dev/neofs-api-go/service"
2020-04-22 10:34:48 +00:00
"github.com/pkg/errors"
"github.com/valyala/fasthttp"
"go.uber.org/zap"
2020-02-25 10:31:20 +00:00
"google.golang.org/grpc"
2020-04-22 10:34:48 +00:00
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
)
func (a *app) receiveFile(c *fasthttp.RequestCtx) {
var (
err error
cid refs.CID
oid refs.ObjectID
start = time.Now()
con *grpc.ClientConn
cli object.Service_GetClient
ctx context.Context
sCID, _ = c.UserValue("cid").(string)
sOID, _ = c.UserValue("oid").(string)
)
log := a.log.With(
2020-02-25 10:31:20 +00:00
// zap.String("node", con.Target()),
zap.String("cid", sCID),
zap.String("oid", sOID))
if err = cid.Parse(sCID); err != nil {
log.Error("wrong container id", zap.Error(err))
c.Error("wrong container id", fasthttp.StatusBadRequest)
return
} else if err = oid.Parse(sOID); err != nil {
log.Error("wrong object id", zap.Error(err))
c.Error("wrong object id", fasthttp.StatusBadRequest)
return
}
2020-02-25 10:31:20 +00:00
{ // try to connect or throw http error:
2020-03-31 08:37:10 +00:00
ctx, cancel := context.WithTimeout(c, a.reqTimeout)
2020-02-25 10:31:20 +00:00
defer cancel()
if con, err = a.pool.getConnection(ctx); err != nil {
log.Error("getConnection timeout", zap.Error(err))
c.Error("could not get alive connection", fasthttp.StatusBadRequest)
return
2020-02-25 10:31:20 +00:00
}
}
2020-03-31 08:37:10 +00:00
ctx, cancel := context.WithTimeout(c, a.reqTimeout)
defer cancel()
2020-02-25 10:31:20 +00:00
log = log.With(zap.String("node", con.Target()))
defer func() {
if err != nil {
return
}
2020-02-27 09:51:11 +00:00
log.Info("object sent to client", zap.Stringer("elapsed", time.Since(start)))
}()
req := &object.GetRequest{Address: refs.Address{ObjectID: oid, CID: cid}}
req.SetTTL(service.SingleForwardingTTL)
2020-06-03 03:30:58 +00:00
if err = service.SignDataWithSessionToken(a.key, req); err != nil {
log.Error("could not sign request", zap.Error(err))
c.Error("could not sign request", fasthttp.StatusBadRequest)
return
}
if cli, err = object.NewServiceClient(con).Get(ctx, req); err != nil {
log.Error("could not prepare connection", zap.Error(err))
c.Error("could not prepare connection", fasthttp.StatusBadRequest)
return
} else if err = receiveObject(c, cli); err != nil {
log.Error("could not receive object",
zap.Stringer("elapsed", time.Since(start)),
zap.Error(err))
2020-04-22 10:34:48 +00:00
var (
msg = errors.Wrap(err, "could not receive object").Error()
code = fasthttp.StatusBadRequest
)
if st, ok := status.FromError(errors.Cause(err)); ok && st != nil {
if st.Code() == codes.NotFound {
code = fasthttp.StatusNotFound
}
msg = st.Message()
}
2020-04-22 10:34:48 +00:00
c.Error(msg, code)
return
}
}
func receiveObject(c *fasthttp.RequestCtx, cli object.Service_GetClient) error {
var (
typ string
put = c.Request.URI().QueryArgs().GetBool("download")
)
for {
resp, err := cli.Recv()
if err != nil {
if err == io.EOF {
break
}
return err
}
switch o := resp.R.(type) {
case *object.GetResponse_Object:
obj := o.Object
c.Response.Header.Set("Content-Length", strconv.FormatUint(obj.SystemHeader.PayloadLength, 10))
c.Response.Header.Set("x-object-id", obj.SystemHeader.ID.String())
c.Response.Header.Set("x-owner-id", obj.SystemHeader.OwnerID.String())
c.Response.Header.Set("x-container-id", obj.SystemHeader.CID.String())
for i := range obj.Headers {
if hdr := obj.Headers[i].GetUserHeader(); hdr != nil {
c.Response.Header.Set("x-"+hdr.Key, hdr.Value)
if hdr.Key == object.FilenameHeader && put {
// NOTE: we use path.Base because hdr.Value can be something like `/path/to/filename.ext`
c.Response.Header.Set("Content-Disposition", "attachment; filename="+path.Base(hdr.Value))
}
}
}
2020-02-29 13:40:54 +00:00
if len(obj.Payload) > 0 {
typ = http.DetectContentType(obj.Payload)
}
if _, err = c.Write(obj.Payload); err != nil {
return err
}
case *object.GetResponse_Chunk:
if typ == "" {
typ = http.DetectContentType(o.Chunk)
}
if _, err = c.Write(o.Chunk); err != nil {
return err
}
}
}
c.SetContentType(typ)
return nil
}