最近更新时间:2024-09-24 18:48:23
Go SDK支持在函数中传递context参数用于取消请求。若您需要传入context参数,则需要在原方法名后添加WithContext。
以PUT上传为例,其函数名为PutObject,若您需要传入context参数,则需要调用PutObjectWithContext函数,该函数的第一个参数用于传递context,其他参数均与PutObject相同。
以下代码为PutObject和PutObjectWithContext的函数签名:
// PutObject 的函数签名
func (c *S3) PutObject(input *PutObjectInput) (*PutObjectOutput, error)
// PutObjectWithContext 的函数签名
func (c *S3) PutObjectWithContext(ctx aws.Context, input *PutObjectInput) (*PutObjectOutput, error)
其他函数的命名及参数规则均与此相同。
以下代码用于上传文件,若60s内未完成则取消请求:
package main
import (
"context"
"fmt"
"github.com/ks3sdklib/aws-sdk-go/aws"
"github.com/ks3sdklib/aws-sdk-go/aws/awsutil"
"github.com/ks3sdklib/aws-sdk-go/aws/credentials"
"github.com/ks3sdklib/aws-sdk-go/service/s3"
"os"
"time"
)
func main() {
// 创建访问凭证,请将<AccessKeyID>与<SecretAccessKey>替换成真正的值
cre := credentials.NewStaticCredentials("<AccessKeyID>", "<SecretAccessKey>", "")
// 创建S3Client,更多配置项请查看Go-SDK初始化文档
client := s3.New(&aws.Config{
Credentials: cre, // 访问凭证
Region: "BEIJING", // 填写您的Region
Endpoint: "ks3-cn-beijing.ksyuncs.com", // 填写您的Endpoint
})
// 填写存储空间名称
bucket := "<bucket_name>"
// 填写对象的key
key := "<object_key>"
// 打开文件
fd, err := os.Open("/Users/test/demo.txt")
if err != nil {
panic(err)
}
defer fd.Close()
// 创建context,设置超时时间为60s
ctx, cancelFunc := context.WithTimeout(context.Background(), time.Second*60)
// 延迟调用,60s后若请求未完成则取消请求
defer cancelFunc()
// 上传文件流,传递context参数
resp, err := client.PutObjectWithContext(ctx, &s3.PutObjectInput{
Bucket: aws.String(bucket), // 存储空间名称,必填
Key: aws.String(key), // 对象的Key,必填、
Body: fd, // 要上传的文件,必填
})
if err != nil {
panic(err)
}
fmt.Println("结果:\n", awsutil.StringValue(resp))
}
以下代码用于上传文件,若60s内未完成则主动取消请求:
package main
import (
"context"
"fmt"
"github.com/ks3sdklib/aws-sdk-go/aws"
"github.com/ks3sdklib/aws-sdk-go/aws/awsutil"
"github.com/ks3sdklib/aws-sdk-go/aws/credentials"
"github.com/ks3sdklib/aws-sdk-go/service/s3"
"os"
"sync"
"time"
)
func main() {
// 创建访问凭证,请将<AccessKeyID>与<SecretAccessKey>替换成真正的值
cre := credentials.NewStaticCredentials("<AccessKeyID>", "<SecretAccessKey>", "")
// 创建S3Client,更多配置项请查看Go-SDK初始化文档
client := s3.New(&aws.Config{
Credentials: cre, // 访问凭证
Region: "BEIJING", // 填写您的Region
Endpoint: "ks3-cn-beijing.ksyuncs.com", // 填写您的Endpoint
})
// 填写存储空间名称
bucket := "<bucket_name>"
// 填写对象的key
key := "<object_key>"
// 打开文件
fd, err := os.Open("/Users/test/demo.txt")
if err != nil {
panic(err)
}
defer fd.Close()
var wg sync.WaitGroup
wg.Add(1)
// 创建context
ctx, cancelFunc := context.WithCancel(context.Background())
// 该协程用于上传文件
go func() {
// 上传文件流,传递context参数
resp, err := client.PutObjectWithContext(ctx, &s3.PutObjectInput{
Bucket: aws.String(bucket), // 存储空间名称,必填
Key: aws.String(key), // 对象的Key,必填、
Body: fd, // 要上传的文件,必填
})
if err != nil {
panic(err)
}
fmt.Println("结果:\n", awsutil.StringValue(resp))
wg.Done()
}()
// 该协程用于在60s后调用cancelFunc方法取消请求
go func() {
time.Sleep(time.Second * 60)
cancelFunc()
}()
wg.Wait()
}
纯净模式