in sample/encryption_client.go [57:157]
func main() {
flag.Parse()
if len(bucketName) == 0 {
flag.PrintDefaults()
log.Fatalf("invalid parameters, bucket name required")
}
if len(region) == 0 {
flag.PrintDefaults()
log.Fatalf("invalid parameters, region required")
}
if len(objectName) == 0 {
flag.PrintDefaults()
log.Fatalf("invalid parameters, object name required")
}
cfg := oss.LoadDefaultConfig().
WithCredentialsProvider(credentials.NewEnvironmentVariableCredentialsProvider()).
WithRegion(region)
client := oss.NewClient(cfg)
mc, err := crypto.CreateMasterRsa(map[string]string{"tag": "value"}, rsaPublicKey, rsaPrivateKey)
if err != nil {
log.Fatalf("failed to create master rsa %v", err)
}
eclient, err := oss.NewEncryptionClient(client, mc)
// case 1: Simple Upload
putObjRequest := &oss.PutObjectRequest{
Bucket: oss.Ptr(bucketName),
Key: oss.Ptr(objectName),
Body: strings.NewReader("hi, simple put object"),
}
putObjRequestResult, err := eclient.PutObject(context.TODO(), putObjRequest)
if err != nil {
log.Fatalf("failed to put object with encryption client %v", err)
}
log.Printf("put object with encryption client result:%#v\n", putObjRequestResult)
// case 2: Mutlipart Upload
initRequest := &oss.InitiateMultipartUploadRequest{
Bucket: oss.Ptr(bucketName),
Key: oss.Ptr(objectName),
}
initResult, err := eclient.InitiateMultipartUpload(context.TODO(), initRequest)
if err != nil {
log.Fatalf("failed to initiate multi part upload %v", err)
}
var wg sync.WaitGroup
var parts oss.UploadParts
count := 3
body := randStr(400000)
reader := strings.NewReader(body)
bufReader := bufio.NewReader(reader)
content, _ := io.ReadAll(bufReader)
partSize := len(body) / count
var mu sync.Mutex
for i := 0; i < count; i++ {
wg.Add(1)
go func(partNumber int, partSize int, i int) {
defer wg.Done()
partRequest := &oss.UploadPartRequest{
Bucket: oss.Ptr(bucketName),
Key: oss.Ptr(objectName),
PartNumber: int32(partNumber),
UploadId: oss.Ptr(*initResult.UploadId),
Body: strings.NewReader(string(content[i*partSize : (i+1)*partSize])),
CSEMultiPartContext: initResult.CSEMultiPartContext,
}
partResult, err := eclient.UploadPart(context.TODO(), partRequest)
if err != nil {
log.Fatalf("failed to upload part %d: %v", partNumber, err)
}
part := oss.UploadPart{
PartNumber: partRequest.PartNumber,
ETag: partResult.ETag,
}
mu.Lock()
parts = append(parts, part)
mu.Unlock()
}(i+1, partSize, i)
}
wg.Wait()
sort.Sort(parts)
request := &oss.CompleteMultipartUploadRequest{
Bucket: oss.Ptr(bucketName),
Key: oss.Ptr(objectName),
UploadId: oss.Ptr(*initResult.UploadId),
CompleteMultipartUpload: &oss.CompleteMultipartUpload{
Parts: parts,
},
}
result, err := eclient.CompleteMultipartUpload(context.TODO(), request)
if err != nil {
log.Fatalf("failed to complete multipart upload %v", err)
}
log.Printf("complete multipart upload result:%#v\n", result)
}