/
stream.go
63 lines (51 loc) · 1.6 KB
/
stream.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
57
58
59
60
61
62
63
// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.
package metaclient
import (
"time"
"storj.io/common/pb"
)
// MutableStream is for manipulating stream information.
type MutableStream struct {
info Object
dynamic bool
dynamicMetadata SerializableMeta
dynamicExpires time.Time
}
// SerializableMeta is an interface for getting pb.SerializableMeta.
type SerializableMeta interface {
Metadata() ([]byte, error)
}
// BucketName returns streams bucket name.
func (stream *MutableStream) BucketName() string { return stream.info.Bucket.Name }
// Path returns streams path.
func (stream *MutableStream) Path() string { return stream.info.Path }
// Info returns object info about the stream.
func (stream *MutableStream) Info() Object { return stream.info }
// Expires returns stream expiration time.
func (stream *MutableStream) Expires() time.Time {
if stream.dynamic {
return stream.dynamicExpires
}
return stream.info.Expires
}
// Metadata returns metadata associated with the stream.
func (stream *MutableStream) Metadata() ([]byte, error) {
if stream.dynamic {
return stream.dynamicMetadata.Metadata()
}
if stream.info.ContentType != "" {
if stream.info.Metadata == nil {
stream.info.Metadata = make(map[string]string)
stream.info.Metadata[contentTypeKey] = stream.info.ContentType
} else if _, found := stream.info.Metadata[contentTypeKey]; !found {
stream.info.Metadata[contentTypeKey] = stream.info.ContentType
}
}
if stream.info.Metadata == nil {
return []byte{}, nil
}
return pb.Marshal(&pb.SerializableMeta{
UserDefined: stream.info.Metadata,
})
}