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

     1  // THIS FILE IS GENERATED BY api-generator, DO NOT EDIT DIRECTLY!
     2  
     3  package apis
     4  
     5  import (
     6  	"context"
     7  	"net/http"
     8  	"strconv"
     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  	setbucketquota "github.com/qiniu/go-sdk/v7/storagev2/apis/set_bucket_quota"
    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 innerSetBucketQuotaRequest setbucketquota.Request
    22  
    23  func (pp *innerSetBucketQuotaRequest) getBucketName(ctx context.Context) (string, error) {
    24  	return pp.Bucket, nil
    25  }
    26  
    27  func (path *innerSetBucketQuotaRequest) 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.Size != 0 {
    35  		allSegments = append(allSegments, "size", strconv.FormatInt(path.Size, 10))
    36  	}
    37  	if path.Count != 0 {
    38  		allSegments = append(allSegments, "count", strconv.FormatInt(path.Count, 10))
    39  	}
    40  	return allSegments, nil
    41  }
    42  
    43  type (
    44  	SetBucketQuotaRequest  = setbucketquota.Request
    45  	SetBucketQuotaResponse = setbucketquota.Response
    46  )
    47  
    48  // 设置用户存储空间配额限制
    49  func (storage *Storage) SetBucketQuota(ctx context.Context, request *SetBucketQuotaRequest, options *Options) (*SetBucketQuotaResponse, error) {
    50  	if options == nil {
    51  		options = &Options{}
    52  	}
    53  	innerRequest := (*innerSetBucketQuotaRequest)(request)
    54  	serviceNames := []region.ServiceName{region.ServiceBucket}
    55  	if innerRequest.Credentials == nil && storage.client.GetCredentials() == nil {
    56  		return nil, errors.MissingRequiredFieldError{Name: "Credentials"}
    57  	}
    58  	pathSegments := make([]string, 0, 6)
    59  	pathSegments = append(pathSegments, "setbucketquota")
    60  	if segments, err := innerRequest.buildPath(); err != nil {
    61  		return nil, err
    62  	} else {
    63  		pathSegments = append(pathSegments, segments...)
    64  	}
    65  	path := "/" + strings.Join(pathSegments, "/")
    66  	var rawQuery string
    67  	headers := http.Header{}
    68  	bucketName := options.OverwrittenBucketName
    69  	if bucketName == "" {
    70  		var err error
    71  		if bucketName, err = innerRequest.getBucketName(ctx); err != nil {
    72  			return nil, err
    73  		}
    74  	}
    75  	uplogInterceptor, err := uplog.NewRequestUplog("setBucketQuota", bucketName, "", func() (string, error) {
    76  		credentials := innerRequest.Credentials
    77  		if credentials == nil {
    78  			credentials = storage.client.GetCredentials()
    79  		}
    80  		putPolicy, err := uptoken.NewPutPolicy(bucketName, time.Now().Add(time.Hour))
    81  		if err != nil {
    82  			return "", err
    83  		}
    84  		return uptoken.NewSigner(putPolicy, credentials).GetUpToken(ctx)
    85  	})
    86  	if err != nil {
    87  		return nil, err
    88  	}
    89  	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}
    90  	if options.OverwrittenEndpoints == nil && options.OverwrittenRegion == nil && storage.client.GetRegions() == nil {
    91  		bucketHosts := httpclient.DefaultBucketHosts()
    92  		if options.OverwrittenBucketHosts != nil {
    93  			req.Endpoints = options.OverwrittenBucketHosts
    94  		} else {
    95  			req.Endpoints = bucketHosts
    96  		}
    97  	}
    98  	resp, err := storage.client.Do(ctx, &req)
    99  	if err != nil {
   100  		return nil, err
   101  	}
   102  	return &SetBucketQuotaResponse{}, resp.Body.Close()
   103  }