Go 언어와 Redis를 사용하여 클라우드 스토리지 서비스를 구축하는 방법

package main

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

    "github.com/go-redis/redis"
)

func main() {
    client := redis.NewClient(&redis.Options{
        Addr:     "localhost:6379",
        Password: "", //填写自己的Redis密码
        DB:       0,
    })

    pong, err := client.Ping().Result()
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println("Connected to Redis:", pong)
}

func uploadFile(client *redis.Client) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        r.ParseMultipartForm(32 << 20)

        file, handler, err := r.FormFile("file")
        if err != nil {
            log.Println(err)
            http.Error(w, "Failed to upload file", http.StatusInternalServerError)
            return
        }

        defer file.Close()
        bytes, err := ioutil.ReadAll(file)
        if err != nil {
            log.Println(err)
            http.Error(w, "Failed to read file", http.StatusInternalServerError)
            return
        }

        fileID := uuid.NewString()
        err = client.Set(fileID, bytes, 0).Err()
        if err != nil {
            log.Println(err)
            http.Error(w, "Failed to save file", http.StatusInternalServerError)
            return
        }

        response := map[string]string{"fileID": fileID}
        jsonResponse, err := json.Marshal(response)
        if err != nil {
            log.Println(err)
            http.Error(w, "Failed to create JSON response", http.StatusInternalServerError)
            return
        }

        w.Header().Set("Content-Type", "application/json")
        w.Write(jsonResponse)
    }
}

func downloadFile(client *redis.Client) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        fileID := r.URL.Query().Get("fileID")

        bytes, err := client.Get(fileID).Bytes()
        if err != nil {
            log.Println(err)
            http.Error(w, "Failed to get file", http.StatusInternalServerError)
            return
        }

        w.Header().Set("Content-Type", "application/octet-stream")
        w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=%s", fileID))
        w.Write(bytes)
    }
}

func main() {
    ...
    http.HandleFunc("/upload", uploadFile(client))
    http.HandleFunc("/download", downloadFile(client))

    log.Fatal(http.ListenAndServe(":8080", nil))
}

go run main.go

curl -X POST -H "Content-Type: multipart/form-data" -F "file=@/path/to/file" http://localhost:8080/upload

curl -OJ http://localhost:8080/download?fileID=<fileID>