mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-30 06:16:37 +08:00
ryan
45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package storage
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"time"
|
|
|
|
"github.com/Rain-kl/Wavelet/pkg/httppool"
|
|
)
|
|
|
|
func getHTTPObject(ctx context.Context, baseURL, key string) (*Object, error) {
|
|
objectURL, err := url.JoinPath(baseURL, key)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("build CDN object URL: %w", err)
|
|
}
|
|
request, err := http.NewRequestWithContext(ctx, http.MethodGet, objectURL, nil)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("create CDN request: %w", err)
|
|
}
|
|
const cdnRequestTimeout = 30 * time.Second
|
|
client := httppool.NewClient(cdnRequestTimeout)
|
|
response, err := client.Do(request)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("get CDN object: %w", err)
|
|
}
|
|
if response.StatusCode != http.StatusOK {
|
|
_ = response.Body.Close()
|
|
return nil, fmt.Errorf("get CDN object: unexpected status %d", response.StatusCode)
|
|
}
|
|
contentType := response.Header.Get("Content-Type")
|
|
if contentType == "" {
|
|
contentType = defaultContentType
|
|
}
|
|
return &Object{
|
|
Body: response.Body,
|
|
ContentLength: response.ContentLength,
|
|
ContentType: contentType,
|
|
}, nil
|
|
}
|