From 60bb3c03a185da3faa1efeee235f7eef1d168818 Mon Sep 17 00:00:00 2001 From: Michael Yang Date: Thu, 2 Nov 2023 13:10:58 -0700 Subject: [PATCH] use http.Method --- api/client.go | 2 +- server/auth.go | 4 ++-- server/download.go | 4 ++-- server/images.go | 4 ++-- server/upload.go | 6 +++--- 5 files changed, 10 insertions(+), 10 deletions(-) diff --git a/api/client.go b/api/client.go index c89ff38f..974c08eb 100644 --- a/api/client.go +++ b/api/client.go @@ -72,7 +72,7 @@ func ClientFromEnvironment() (*Client, error) { }, } - mockRequest, err := http.NewRequest("HEAD", client.base.String(), nil) + mockRequest, err := http.NewRequest(http.MethodHead, client.base.String(), nil) if err != nil { return nil, err } diff --git a/server/auth.go b/server/auth.go index 6b2376f2..6c151ac3 100644 --- a/server/auth.go +++ b/server/auth.go @@ -91,7 +91,7 @@ func getAuthToken(ctx context.Context, redirData AuthRedirect) (string, error) { } s := SignatureData{ - Method: "GET", + Method: http.MethodGet, Path: redirectURL.String(), Data: nil, } @@ -103,7 +103,7 @@ func getAuthToken(ctx context.Context, redirData AuthRedirect) (string, error) { headers := make(http.Header) headers.Set("Authorization", sig) - resp, err := makeRequest(ctx, "GET", redirectURL, headers, nil, nil) + resp, err := makeRequest(ctx, http.MethodGet, redirectURL, headers, nil, nil) if err != nil { log.Printf("couldn't get token: %q", err) return "", err diff --git a/server/download.go b/server/download.go index be3eda7c..2ebd9737 100644 --- a/server/download.go +++ b/server/download.go @@ -89,7 +89,7 @@ func (b *blobDownload) Prepare(ctx context.Context, requestURL *url.URL, opts *R } if len(b.Parts) == 0 { - resp, err := makeRequest(ctx, "HEAD", requestURL, nil, nil, opts) + resp, err := makeRequest(ctx, http.MethodHead, requestURL, nil, nil, opts) if err != nil { return err } @@ -199,7 +199,7 @@ func (b *blobDownload) run(ctx context.Context, requestURL *url.URL, opts *Regis func (b *blobDownload) downloadChunk(ctx context.Context, requestURL *url.URL, w io.Writer, part *blobDownloadPart, opts *RegistryOptions) error { headers := make(http.Header) headers.Set("Range", fmt.Sprintf("bytes=%d-%d", part.StartsAt(), part.StopsAt()-1)) - resp, err := makeRequest(ctx, "GET", requestURL, headers, nil, opts) + resp, err := makeRequest(ctx, http.MethodGet, requestURL, headers, nil, opts) if err != nil { return err } diff --git a/server/images.go b/server/images.go index 6f72f0cf..358520bd 100644 --- a/server/images.go +++ b/server/images.go @@ -998,7 +998,7 @@ func PushModel(ctx context.Context, name string, regOpts *RegistryOptions, fn fu headers := make(http.Header) headers.Set("Content-Type", "application/vnd.docker.distribution.manifest.v2+json") - resp, err := makeRequestWithRetry(ctx, "PUT", requestURL, headers, bytes.NewReader(manifestJSON), regOpts) + resp, err := makeRequestWithRetry(ctx, http.MethodPut, requestURL, headers, bytes.NewReader(manifestJSON), regOpts) if err != nil { return err } @@ -1120,7 +1120,7 @@ func pullModelManifest(ctx context.Context, mp ModelPath, regOpts *RegistryOptio headers := make(http.Header) headers.Set("Accept", "application/vnd.docker.distribution.manifest.v2+json") - resp, err := makeRequest(ctx, "GET", requestURL, headers, nil, regOpts) + resp, err := makeRequest(ctx, http.MethodGet, requestURL, headers, nil, regOpts) if err != nil { log.Printf("couldn't get manifest: %v", err) return nil, err diff --git a/server/upload.go b/server/upload.go index df8e8cf3..56a762b3 100644 --- a/server/upload.go +++ b/server/upload.go @@ -67,7 +67,7 @@ func (b *blobUpload) Prepare(ctx context.Context, requestURL *url.URL, opts *Reg requestURL.RawQuery = values.Encode() } - resp, err := makeRequestWithRetry(ctx, "POST", requestURL, nil, nil, opts) + resp, err := makeRequestWithRetry(ctx, http.MethodPost, requestURL, nil, nil, opts) if err != nil { return err } @@ -187,7 +187,7 @@ func (b *blobUpload) Run(ctx context.Context, opts *RegistryOptions) { headers.Set("Content-Type", "application/octet-stream") headers.Set("Content-Length", "0") - resp, err := makeRequestWithRetry(ctx, "PUT", requestURL, headers, nil, opts) + resp, err := makeRequestWithRetry(ctx, http.MethodPut, requestURL, headers, nil, opts) if err != nil { b.err = err return @@ -334,7 +334,7 @@ func uploadBlob(ctx context.Context, mp ModelPath, layer *Layer, opts *RegistryO requestURL := mp.BaseURL() requestURL = requestURL.JoinPath("v2", mp.GetNamespaceRepository(), "blobs", layer.Digest) - resp, err := makeRequest(ctx, "HEAD", requestURL, nil, nil, opts) + resp, err := makeRequest(ctx, http.MethodHead, requestURL, nil, nil, opts) if err != nil { return err }