golang 框架中如何使用云存储进行文件上传

go get -u cloud.google.com/go/storage

import (
    "context"
    "fmt"
    "io"

    "cloud.google.com/go/storage"
)

func main() {
    ctx := context.Background()

    // 创建存储客户端
    client, err := storage.NewClient(ctx)
    if err != nil {
        fmt.Printf("Failed to create client: %v", err)
        return
    }
    defer client.Close()

    // 打开存储桶
    bucket := client.Bucket("your-bucket-name")
}

func uploadFile(w io.Writer, bucket, object string, filename string) error {
    ctx := context.Background()

    file, err := os.Open(filename)
    if err != nil {
        fmt.Fprintf(w, "Failed to open file: %v", err)
        return err
    }
    defer file.Close()

    ctx, cancel := context.WithTimeout(ctx, time.Second*30)
    defer cancel()

    obj := bucket.Object(object)
    wc := obj.NewWriter(ctx)
    if _, err := io.Copy(wc, file); err != nil {
        return fmt.Errorf("Failed to upload file: %v", err)
    }
    if err := wc.Close(); err != nil {
        return fmt.Errorf("Failed to close writer: %v", err)
    }

    fmt.Fprintf(w, "Uploaded %v to gs://%v/%v\n", filename, bucket, object)
    return nil
}

package main

import (
    "context"
    "fmt"
    "log"
    "net/http"

    "cloud.google.com/go/storage"
)

// form中的field名要与表单中文件上传input的name(要有"name"属性)相对应
const (
    formFile  = "file"
    bucketName = "your-bucket-name"
)

func main() {
    http.HandleFunc("/", handler)
    log.Printf("Server running at http://localhost:8080")
    http.ListenAndServe(":8080", nil)
}

// handler接收表单并上传文件
func handler(w http.ResponseWriter, r *http.Request) {

    if err := r.ParseMultipartForm(10 << 20); err != nil {
        log.Printf("Failed to parse multipart form: %v", err)
        http.Error(w, http.StatusText(http.StatusBadRequest), http.StatusBadRequest)
        return
    }

    file, header, err := r.FormFile(formFile)
    if err != nil {
        log.Printf("Failed to get form file: %v", err)
        http.Error(w, http.StatusText(http.StatusBadRequest), http.StatusBadRequest)
        return
    }
    defer file.Close()

    ctx := context.Background()

    // 创建新的存储客户端
    client, err := storage.NewClient(ctx)
    if err != nil {
        log.Printf("Failed to create storage client: %v", err)
        http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
        return
    }
    defer client.Close()

    // 打开存储桶
    bucket := client.Bucket(bucketName)

    // 创建对象
    obj := bucket.Object(header.Filename)

    // 使用指定的对象名上传文件
    wc := obj.NewWriter(ctx)
    if _, err = io.Copy(wc, file); err != nil {
        log.Printf("Failed to upload the file: %v", err)
        http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
        return
    }
    if err := wc.Close(); err != nil {
        log.Printf("Failed to close writer: %v", err)
        http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
        return
    }

    fmt.Fprintf(w, "File uploaded successfully to %v with the name %v", bucketName, header.Filename)
}