github.com/qiniu/go-sdk/v7@v7.27.0/storagev2/apis/api_set_bucket_image.go (about)

     1  // THIS FILE IS GENERATED BY api-generator, DO NOT EDIT DIRECTLY!
     2  
     3  package apis
     4  
     5  import (
     6  	"context"
     7  	"encoding/base64"
     8  	"net/http"
     9  	"strings"
    10  	"time"
    11  
    12  	auth "github.com/qiniu/go-sdk/v7/auth"
    13  	uplog "github.com/qiniu/go-sdk/v7/internal/uplog"
    14  	setbucketimage "github.com/qiniu/go-sdk/v7/storagev2/apis/set_bucket_image"
    15  	errors "github.com/qiniu/go-sdk/v7/storagev2/errors"
    16  	httpclient "github.com/qiniu/go-sdk/v7/storagev2/http_client"
    17  	region "github.com/qiniu/go-sdk/v7/storagev2/region"
    18  	uptoken "github.com/qiniu/go-sdk/v7/storagev2/uptoken"
    19  )
    20  
    21  type innerSetBucketImageRequest setbucketimage.Request
    22  
    23  func (pp *innerSetBucketImageRequest) getBucketName(ctx context.Context) (string, error) {
    24  	return pp.Bucket, nil
    25  }
    26  
    27  func (path *innerSetBucketImageRequest) buildPath() ([]string, error) {
    28  	allSegments := make([]string, 0, 5)
    29  	if path.Bucket != "" {
    30  		allSegments = append(allSegments, path.Bucket)
    31  	} else {
    32  		return nil, errors.MissingRequiredFieldError{Name: "Bucket"}
    33  	}
    34  	if path.Url != "" {
    35  		allSegments = append(allSegments, "from", base64.URLEncoding.EncodeToString([]byte(path.Url)))
    36  	} else {
    37  		return nil, errors.MissingRequiredFieldError{Name: "Url"}
    38  	}
    39  	if path.Host != "" {
    40  		allSegments = append(allSegments, "host", path.Host)
    41  	}
    42  	return allSegments, nil
    43  }
    44  
    45  type (
    46  	SetBucketImageRequest  = setbucketimage.Request
    47  	SetBucketImageResponse = setbucketimage.Response
    48  )
    49  
    50  // 设置源站镜像回源
    51  func (storage *Storage) SetBucketImage(ctx context.Context, request *SetBucketImageRequest, options *Options) (*SetBucketImageResponse, error) {
    52  	if options == nil {
    53  		options = &Options{}
    54  	}
    55  	innerRequest := (*innerSetBucketImageRequest)(request)
    56  	serviceNames := []region.ServiceName{region.ServiceBucket}
    57  	if innerRequest.Credentials == nil && storage.client.GetCredentials() == nil {
    58  		return nil, errors.MissingRequiredFieldError{Name: "Credentials"}
    59  	}
    60  	pathSegments := make([]string, 0, 6)
    61  	pathSegments = append(pathSegments, "image")
    62  	if segments, err := innerRequest.buildPath(); err != nil {
    63  		return nil, err
    64  	} else {
    65  		pathSegments = append(pathSegments, segments...)
    66  	}
    67  	path := "/" + strings.Join(pathSegments, "/")
    68  	var rawQuery string
    69  	headers := http.Header{}
    70  	bucketName := options.OverwrittenBucketName
    71  	if bucketName == "" {
    72  		var err error
    73  		if bucketName, err = innerRequest.getBucketName(ctx); err != nil {
    74  			return nil, err
    75  		}
    76  	}
    77  	uplogInterceptor, err := uplog.NewRequestUplog("setBucketImage", bucketName, "", func() (string, error) {
    78  		credentials := innerRequest.Credentials
    79  		if credentials == nil {
    80  			credentials = storage.client.GetCredentials()
    81  		}
    82  		putPolicy, err := uptoken.NewPutPolicy(bucketName, time.Now().Add(time.Hour))
    83  		if err != nil {
    84  			return "", err
    85  		}
    86  		return uptoken.NewSigner(putPolicy, credentials).GetUpToken(ctx)
    87  	})
    88  	if err != nil {
    89  		return nil, err
    90  	}
    91  	req := httpclient.Request{Method: "POST", ServiceNames: serviceNames, Path: path, RawQuery: rawQuery, Endpoints: options.OverwrittenEndpoints, Region: options.OverwrittenRegion, Interceptors: []httpclient.Interceptor{uplogInterceptor}, Header: headers, AuthType: auth.TokenQiniu, Credentials: innerRequest.Credentials, OnRequestProgress: options.OnRequestProgress}
    92  	if options.OverwrittenEndpoints == nil && options.OverwrittenRegion == nil && storage.client.GetRegions() == nil {
    93  		bucketHosts := httpclient.DefaultBucketHosts()
    94  		if options.OverwrittenBucketHosts != nil {
    95  			req.Endpoints = options.OverwrittenBucketHosts
    96  		} else {
    97  			req.Endpoints = bucketHosts
    98  		}
    99  	}
   100  	resp, err := storage.client.Do(ctx, &req)
   101  	if err != nil {
   102  		return nil, err
   103  	}
   104  	return &SetBucketImageResponse{}, resp.Body.Close()
   105  }