forked from yunionio/cloudmux
/
objects.go
103 lines (92 loc) · 2.76 KB
/
objects.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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
// Copyright 2023 Yunion
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package volcengine
import (
"context"
"net/http"
"github.com/nyl1001/cloudmux/pkg/cloudprovider"
"github.com/nyl1001/pkg/errors"
"github.com/volcengine/ve-tos-golang-sdk/v2/tos"
"github.com/volcengine/ve-tos-golang-sdk/v2/tos/enum"
"yunion.io/x/log"
)
const (
OSS_META_HEADER = "x-oss-meta-"
)
type SObject struct {
bucket *SBucket
cloudprovider.SBaseCloudObject
}
func (obj *SObject) GetIBucket() cloudprovider.ICloudBucket {
return obj.bucket
}
func (obj *SObject) GetAcl() cloudprovider.TBucketACLType {
acl := cloudprovider.ACLPrivate
toscli, err := obj.bucket.region.GetTosClient()
if err != nil {
log.Errorf("Get Client %s", err)
return acl
}
result, err := toscli.GetObjectACL(context.Background(), &tos.GetObjectACLInput{Bucket: obj.bucket.Name, Key: obj.Key})
if err != nil {
log.Errorf("GetObjectAcl %s", err)
}
grants := result.Grants
return grantToCannedAcl(grants)
}
func (obj *SObject) SetAcl(aclStr cloudprovider.TBucketACLType) error {
toscli, err := obj.bucket.region.GetTosClient()
if err != nil {
return errors.Wrap(err, "GetTosClient")
}
input := &tos.PutObjectACLInput{Bucket: obj.bucket.Name, Key: obj.Key, ACL: enum.ACLType(aclStr)}
_, err = toscli.PutObjectACL(context.Background(), input)
if err != nil {
return errors.Wrapf(err, "PutObjectACL")
}
return nil
}
func (obj *SObject) GetMeta() http.Header {
if obj.Meta != nil {
return obj.Meta
}
toscli, err := obj.bucket.region.GetTosClient()
if err != nil {
log.Errorf("Get Client %s", err)
return nil
}
result, err := toscli.GetObjectV2(context.Background(), &tos.GetObjectV2Input{Bucket: obj.bucket.Name, Key: obj.Key})
if err != nil {
log.Errorf("Get Object error %s", err)
return nil
}
newHeader := http.Header{}
meta := result.GetObjectBasicOutput.ObjectMetaV2.Meta
for _, key := range meta.AllKeys() {
value, exist := meta.Get(key)
if !exist {
log.Errorf("Key missing in meta data %s", key)
} else {
newHeader.Add(key, value)
}
}
obj.Meta = cloudprovider.FetchMetaFromHttpHeader(
OSS_META_HEADER,
newHeader,
)
return obj.Meta
}
func (obj *SObject) SetMeta(ctx context.Context, meta http.Header) error {
return cloudprovider.ObjectSetMeta(ctx, obj.bucket, obj, meta)
}