mirror of
https://github.com/ceph/go-ceph
synced 2024-12-12 01:16:24 +00:00
115 lines
2.5 KiB
Go
115 lines
2.5 KiB
Go
package admin
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
"net/url"
|
|
"time"
|
|
|
|
"errors"
|
|
|
|
"github.com/aws/aws-sdk-go/aws/credentials"
|
|
v4 "github.com/aws/aws-sdk-go/aws/signer/v4"
|
|
)
|
|
|
|
const (
|
|
authRegion = "default"
|
|
service = "s3"
|
|
connectionTimeout = time.Second * 3
|
|
)
|
|
|
|
var (
|
|
errNoEndpoint = errors.New("endpoint not set")
|
|
errNoAccessKey = errors.New("access key not set")
|
|
errNoSecretKey = errors.New("secret key not set")
|
|
)
|
|
|
|
// HTTPClient interface that conforms to that of the http package's Client.
|
|
type HTTPClient interface {
|
|
Do(req *http.Request) (*http.Response, error)
|
|
}
|
|
|
|
// API struct for New Client
|
|
type API struct {
|
|
AccessKey string
|
|
SecretKey string
|
|
Endpoint string
|
|
HTTPClient HTTPClient
|
|
}
|
|
|
|
// New returns client for Ceph RGW
|
|
func New(endpoint, accessKey, secretKey string, httpClient HTTPClient) (*API, error) {
|
|
// validate endpoint
|
|
if endpoint == "" {
|
|
return nil, errNoEndpoint
|
|
}
|
|
|
|
// validate access key
|
|
if accessKey == "" {
|
|
return nil, errNoAccessKey
|
|
}
|
|
|
|
// validate secret key
|
|
if secretKey == "" {
|
|
return nil, errNoSecretKey
|
|
}
|
|
|
|
// If no client is passed initialize it
|
|
if httpClient == nil {
|
|
httpClient = &http.Client{Timeout: connectionTimeout}
|
|
}
|
|
|
|
return &API{
|
|
Endpoint: endpoint,
|
|
AccessKey: accessKey,
|
|
SecretKey: secretKey,
|
|
HTTPClient: httpClient,
|
|
}, nil
|
|
}
|
|
|
|
// call makes request to the RGW Admin Ops API
|
|
func (api *API) call(ctx context.Context, httpMethod, path string, args url.Values) (body []byte, err error) {
|
|
// Build request
|
|
request, err := http.NewRequestWithContext(ctx, httpMethod, buildQueryPath(api.Endpoint, path, args.Encode()), nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Build S3 authentication
|
|
cred := credentials.NewStaticCredentials(api.AccessKey, api.SecretKey, "")
|
|
signer := v4.NewSigner(cred)
|
|
// This was present in https://github.com/IrekFasikhov/go-rgwadmin/ but it seems that the lib works without it
|
|
// Let's keep it here just in case something shows up
|
|
// signer.DisableRequestBodyOverwrite = true
|
|
|
|
// Sign in S3
|
|
_, err = signer.Sign(request, nil, service, authRegion, time.Now())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Send HTTP request
|
|
resp, err := api.HTTPClient.Do(request)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
// Decode HTTP response
|
|
decodedResponse, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
resp.Body = io.NopCloser(bytes.NewBuffer(decodedResponse))
|
|
|
|
// Handle error in response
|
|
if resp.StatusCode >= 300 {
|
|
return nil, handleStatusError(decodedResponse)
|
|
}
|
|
|
|
return decodedResponse, nil
|
|
}
|