-
Notifications
You must be signed in to change notification settings - Fork 1.1k
/
upload.go
56 lines (46 loc) · 953 Bytes
/
upload.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
package upload
import (
"context"
io "io"
"net/url"
"github.com/moby/buildkit/session"
"github.com/pkg/errors"
"google.golang.org/grpc/metadata"
)
const (
keyPath = "urlpath"
keyHost = "urlhost"
)
func New(ctx context.Context, c session.Caller, url *url.URL) (*Upload, error) {
opts := map[string][]string{
keyPath: {url.Path},
keyHost: {url.Host},
}
client := NewUploadClient(c.Conn())
ctx = metadata.NewOutgoingContext(ctx, opts)
cc, err := client.Pull(ctx)
if err != nil {
return nil, errors.WithStack(err)
}
return &Upload{cc: cc}, nil
}
type Upload struct {
cc Upload_PullClient
}
func (u *Upload) WriteTo(w io.Writer) (int64, error) {
var n int64
for {
var bm BytesMessage
if err := u.cc.RecvMsg(&bm); err != nil {
if err == io.EOF {
return n, nil
}
return n, errors.WithStack(err)
}
nn, err := w.Write(bm.Data)
n += int64(nn)
if err != nil {
return n, errors.WithStack(err)
}
}
}