ファイルのアップロードとダウンロード

var _ = API("upload_download", func() {
    Description("ファイルアップロードとダウンロードの例")
})

var _ = Service("updown", func() {
    Description("ファイルのアップロードとダウンロードを処理するサービス")

    // アップロードエンドポイント
    Method("upload", func() {
        Payload(func() {
            // アップロードに必要なヘッダーとパラメータを定義
            // content_typeはマルチパートフォームデータの解析に必要
            Attribute("content_type", String, "マルチパート境界を含むContent-Typeヘッダー")
            // dirはアップロードされたファイルを保存する場所を指定
            Attribute("dir", String, "アップロードディレクトリのパス")
        })

        HTTP(func() {
            POST("/upload/{*dir}")  // ディレクトリをURLパラメータとするPOSTエンドポイント
            Header("content_type:Content-Type")  // content_typeをContent-Typeヘッダーにマッピング
            SkipRequestBodyEncodeDecode()  // アップロード用のストリーミングを有効化
        })
    })

    // ダウンロードエンドポイント
    Method("download", func() {
        Payload(String)  // ダウンロードするファイル名
        
        Result(func() {
            // Content-Lengthヘッダーでファイルサイズを返す
            Attribute("length", Int64, "バイト単位のコンテンツ長")
            Required("length")
        })

        HTTP(func() {
            GET("/download/{*filename}")  // ファイル名をURLパラメータとするGETエンドポイント
            SkipResponseBodyEncodeDecode()  // ダウンロード用のストリーミングを有効化
            Response(func() {
                Header("length:Content-Length")  // lengthをContent-Lengthヘッダーにマッピング
            })
        })
    })
})

// serviceは、アップロード/ダウンロードサービスの設定を保持する構造体です
type service struct {
    dir string  // ファイルを保存するベースディレクトリ
}

// Upload実装は、マルチパートフォームデータを介してファイルアップロードを処理します
func (s *service) Upload(ctx context.Context, p *updown.UploadPayload, req io.ReadCloser) error {
    // 完了時にリクエストボディを必ずクローズ
    defer req.Close()

    // リクエストからマルチパートフォームデータを解析
    // これには境界パラメータを含むContent-Typeヘッダーが必要
    _, params, err := mime.ParseMediaType(p.ContentType)
    if err != nil {
        return err  // 無効なContent-Typeヘッダー
    }
    mr := multipart.NewReader(req, params["boundary"])

    // マルチパートフォーム内の各ファイルを処理
    for {
        part, err := mr.NextPart()
        if err == io.EOF {
            break  // これ以上ファイルがない
        }
        if err != nil {
            return err  // パート読み取りエラー
        }
        
        // 保存先ファイルを作成
        // ベースディレクトリとアップロードされたファイル名を結合
        dst := filepath.Join(s.dir, part.FileName())
        f, err := os.Create(dst)
        if err != nil {
            return err  // ファイル作成エラー
        }
        defer f.Close()  // 早期リターンの場合でもファイルを確実にクローズ

        // リクエストからディスクにファイル内容をストリーミング
        // io.Copyはストリーミングを効率的に処理
        if _, err := io.Copy(f, part); err != nil {
            return err  // ファイル書き込みエラー
        }
    }
    return nil
}

// Download実装は、ディスクからクライアントにファイルをストリーミングします
func (s *service) Download(ctx context.Context, filename string) (*updown.DownloadResult, io.ReadCloser, error) {
    // 完全なファイルパスを構築
    path := filepath.Join(s.dir, filename)
    
    // ファイル情報を取得（主にサイズのため）
    fi, err := os.Stat(path)
    if err != nil {
        return nil, nil, err  // ファイルが見つからないか他のエラー
    }

    // 読み取り用にファイルを開く
    f, err := os.Open(path)
    if err != nil {
        return nil, nil, err  // ファイルオープンエラー
    }

    // ファイルサイズとファイルリーダーを返す
    // 呼び出し元がリーダーをクローズする責任を持つ
    return &updown.DownloadResult{Length: fi.Size()}, f, nil
}

# まず、1つの端末でサーバーを起動
$ go run cmd/upload_download/main.go

# 別の端末でファイルをアップロード
# --streamフラグは、ファイルをディスクから直接ストリーミングするようCLIに指示
$ go run cmd/upload_download-cli/main.go updown upload \
    --stream /path/to/file.jpg \
    --dir uploads

# 以前にアップロードしたファイルをダウンロード
# 出力は新しいファイルにリダイレクト
$ go run cmd/upload_download-cli/main.go updown download file.jpg > downloaded.jpg

# ファイルをアップロード
$ curl -X POST -F "file=@/path/to/file.jpg" http://localhost:8080/upload/uploads

# ファイルをダウンロード
$ curl http://localhost:8080/download/file.jpg -o downloaded.jpg