github.com/qiniu/go-sdk/v7@v7.27.0/storagev2/apis/api_set_buckets_mirror.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  	setbucketsmirror "github.com/qiniu/go-sdk/v7/storagev2/apis/set_buckets_mirror"
    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 innerSetBucketsMirrorRequest setbucketsmirror.Request
    22  
    23  func (pp *innerSetBucketsMirrorRequest) getBucketName(ctx context.Context) (string, error) {
    24  	return pp.Bucket, nil
    25  }
    26  
    27  func (path *innerSetBucketsMirrorRequest) 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.SrcSiteUrl != "" {
    35  		allSegments = append(allSegments, "from", base64.URLEncoding.EncodeToString([]byte(path.SrcSiteUrl)))
    36  	} else {
    37  		return nil, errors.MissingRequiredFieldError{Name: "SrcSiteUrl"}
    38  	}
    39  	if path.Host != "" {
    40  		allSegments = append(allSegments, "host", base64.URLEncoding.EncodeToString([]byte(path.Host)))
    41  	} else {
    42  		return nil, errors.MissingRequiredFieldError{Name: "Host"}
    43  	}
    44  	return allSegments, nil
    45  }
    46  
    47  type (
    48  	SetBucketsMirrorRequest  = setbucketsmirror.Request
    49  	SetBucketsMirrorResponse = setbucketsmirror.Response
    50  )
    51  
    52  // 设置存储空间的镜像源
    53  func (storage *Storage) SetBucketsMirror(ctx context.Context, request *SetBucketsMirrorRequest, options *Options) (*SetBucketsMirrorResponse, error) {
    54  	if options == nil {
    55  		options = &Options{}
    56  	}
    57  	innerRequest := (*innerSetBucketsMirrorRequest)(request)
    58  	serviceNames := []region.ServiceName{region.ServiceBucket}
    59  	if innerRequest.Credentials == nil && storage.client.GetCredentials() == nil {
    60  		return nil, errors.MissingRequiredFieldError{Name: "Credentials"}
    61  	}
    62  	pathSegments := make([]string, 0, 6)
    63  	pathSegments = append(pathSegments, "image")
    64  	if segments, err := innerRequest.buildPath(); err != nil {
    65  		return nil, err
    66  	} else {
    67  		pathSegments = append(pathSegments, segments...)
    68  	}
    69  	path := "/" + strings.Join(pathSegments, "/")
    70  	var rawQuery string
    71  	headers := http.Header{}
    72  	bucketName := options.OverwrittenBucketName
    73  	if bucketName == "" {
    74  		var err error
    75  		if bucketName, err = innerRequest.getBucketName(ctx); err != nil {
    76  			return nil, err
    77  		}
    78  	}
    79  	uplogInterceptor, err := uplog.NewRequestUplog("setBucketsMirror", bucketName, "", func() (string, error) {
    80  		credentials := innerRequest.Credentials
    81  		if credentials == nil {
    82  			credentials = storage.client.GetCredentials()
    83  		}
    84  		putPolicy, err := uptoken.NewPutPolicy(bucketName, time.Now().Add(time.Hour))
    85  		if err != nil {
    86  			return "", err
    87  		}
    88  		return uptoken.NewSigner(putPolicy, credentials).GetUpToken(ctx)
    89  	})
    90  	if err != nil {
    91  		return nil, err
    92  	}
    93  	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}
    94  	if options.OverwrittenEndpoints == nil && options.OverwrittenRegion == nil && storage.client.GetRegions() == nil {
    95  		bucketHosts := httpclient.DefaultBucketHosts()
    96  		if options.OverwrittenBucketHosts != nil {
    97  			req.Endpoints = options.OverwrittenBucketHosts
    98  		} else {
    99  			req.Endpoints = bucketHosts
   100  		}
   101  	}
   102  	resp, err := storage.client.Do(ctx, &req)
   103  	if err != nil {
   104  		return nil, err
   105  	}
   106  	return &SetBucketsMirrorResponse{}, resp.Body.Close()
   107  }