storage

Storage 包

对象存储统一封装，支持阿里云 OSS、腾讯云 COS、AWS S3。

安装

go get github.com/tsopia/go-kit/storage

快速开始

package main

import (
    "context"
    "log"
    "os"
    "strings"

    "github.com/tsopia/go-kit/storage"
)

func main() {
    // 初始化
    cfg := &storage.Config{
        Type:            storage.TypeOSS,
        Bucket:          "my-bucket",
        Region:          "cn-hangzhou",
        AccessKeyID:     os.Getenv("OSS_ACCESS_KEY_ID"),
        AccessKeySecret: os.Getenv("OSS_ACCESS_KEY_SECRET"),
    }

    if err := storage.Configure(cfg); err != nil {
        log.Fatal(err)
    }

    ctx := context.Background()

    // 上传文件
    data := strings.NewReader("hello world")
    if err := storage.Upload(ctx, "test/hello.txt", data); err != nil {
        log.Fatal(err)
    }

    // 生成临时访问链接
    url, err := storage.SignedURL(ctx, "test/hello.txt", 30*60)
    if err != nil {
        log.Fatal(err)
    }
    log.Println("URL:", url)
}

配置说明

API 文档

上传

err := storage.Upload(ctx, "key", reader,
    storage.WithContentType("application/json"),
    storage.WithMetadata("author", "alice"),
)

下载

reader, err := storage.Download(ctx, "key")
if err != nil {
    return err
}
defer reader.Close()

删除

err := storage.Delete(ctx, "key")

预签名 URL

url, err := storage.SignedURL(ctx, "key", 30*60)

分片上传

// 初始化
upload, err := storage.InitMultipart(ctx, "large.zip")

// 上传分片
part1, err := storage.UploadPart(ctx, upload.UploadID, 1, reader1)
part2, err := storage.UploadPart(ctx, upload.UploadID, 2, reader2)

// 完成
err = storage.CompleteMultipart(ctx, upload.UploadID, []*storage.PartInfo{part1, part2})

多客户端

如果需要使用多个不同的存储配置：

// 获取配置的客户端
client := storage.GetClient()

// 或者使用指定客户端
storage.UploadWithClient(ctx, client, "key", reader)