9631958a82
- Refactor lfs request code - The original code uses `performRequest` function to create the request, uses a callback to modify the request, and then send the request. - Now it's replaced with `createRequest` that only creates request and `performRequest` that only sends the request. - Reuse `createRequest` and `performRequest` in `http_client.go` and `transferadapter.go` --------- Co-authored-by: wxiaoguang <wxiaoguang@gmail.com>
89 lines
2.2 KiB
Go
89 lines
2.2 KiB
Go
// Copyright 2021 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package lfs
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
|
|
"code.gitea.io/gitea/modules/json"
|
|
"code.gitea.io/gitea/modules/log"
|
|
)
|
|
|
|
// TransferAdapter represents an adapter for downloading/uploading LFS objects.
|
|
type TransferAdapter interface {
|
|
Name() string
|
|
Download(ctx context.Context, l *Link) (io.ReadCloser, error)
|
|
Upload(ctx context.Context, l *Link, p Pointer, r io.Reader) error
|
|
Verify(ctx context.Context, l *Link, p Pointer) error
|
|
}
|
|
|
|
// BasicTransferAdapter implements the "basic" adapter.
|
|
type BasicTransferAdapter struct {
|
|
client *http.Client
|
|
}
|
|
|
|
// Name returns the name of the adapter.
|
|
func (a *BasicTransferAdapter) Name() string {
|
|
return "basic"
|
|
}
|
|
|
|
// Download reads the download location and downloads the data.
|
|
func (a *BasicTransferAdapter) Download(ctx context.Context, l *Link) (io.ReadCloser, error) {
|
|
req, err := createRequest(ctx, http.MethodGet, l.Href, l.Header, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
resp, err := performRequest(ctx, a.client, req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return resp.Body, nil
|
|
}
|
|
|
|
// Upload sends the content to the LFS server.
|
|
func (a *BasicTransferAdapter) Upload(ctx context.Context, l *Link, p Pointer, r io.Reader) error {
|
|
req, err := createRequest(ctx, http.MethodPut, l.Href, l.Header, r)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if req.Header.Get("Content-Type") == "" {
|
|
req.Header.Set("Content-Type", "application/octet-stream")
|
|
}
|
|
if req.Header.Get("Transfer-Encoding") == "chunked" {
|
|
req.TransferEncoding = []string{"chunked"}
|
|
}
|
|
req.ContentLength = p.Size
|
|
|
|
res, err := performRequest(ctx, a.client, req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer res.Body.Close()
|
|
return nil
|
|
}
|
|
|
|
// Verify calls the verify handler on the LFS server
|
|
func (a *BasicTransferAdapter) Verify(ctx context.Context, l *Link, p Pointer) error {
|
|
b, err := json.Marshal(p)
|
|
if err != nil {
|
|
log.Error("Error encoding json: %v", err)
|
|
return err
|
|
}
|
|
|
|
req, err := createRequest(ctx, http.MethodPost, l.Href, l.Header, bytes.NewReader(b))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
req.Header.Set("Content-Type", MediaType)
|
|
res, err := performRequest(ctx, a.client, req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer res.Body.Close()
|
|
return nil
|
|
}
|