github.com/qiniu/go-sdk/v7@v7.27.0/storagev2/apis/api_batch_ops.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  	"net/url"
     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  	batchops "github.com/qiniu/go-sdk/v7/storagev2/apis/batch_ops"
    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 innerBatchOpsRequest batchops.Request
    22  
    23  func (form *innerBatchOpsRequest) build() (url.Values, error) {
    24  	formValues := make(url.Values)
    25  	if len(form.Operations) > 0 {
    26  		for _, value := range form.Operations {
    27  			formValues.Add("op", value)
    28  		}
    29  	} else {
    30  		return nil, errors.MissingRequiredFieldError{Name: "Operations"}
    31  	}
    32  	return formValues, nil
    33  }
    34  
    35  func (request *innerBatchOpsRequest) getAccessKey(ctx context.Context) (string, error) {
    36  	if request.Credentials != nil {
    37  		if credentials, err := request.Credentials.Get(ctx); err != nil {
    38  			return "", err
    39  		} else {
    40  			return credentials.AccessKey, nil
    41  		}
    42  	}
    43  	return "", nil
    44  }
    45  
    46  type (
    47  	BatchOpsRequest  = batchops.Request
    48  	BatchOpsResponse = batchops.Response
    49  )
    50  
    51  // 批量操作意指在单一请求中执行多次（最大限制1000次） 查询元信息、修改元信息、移动、复制、删除、修改状态、修改存储类型、修改生命周期和解冻操作，极大提高对象管理效率。其中，解冻操作仅针对归档存储文件有效
    52  func (storage *Storage) BatchOps(ctx context.Context, request *BatchOpsRequest, options *Options) (*BatchOpsResponse, error) {
    53  	if options == nil {
    54  		options = &Options{}
    55  	}
    56  	innerRequest := (*innerBatchOpsRequest)(request)
    57  	serviceNames := []region.ServiceName{region.ServiceRs}
    58  	if innerRequest.Credentials == nil && storage.client.GetCredentials() == nil {
    59  		return nil, errors.MissingRequiredFieldError{Name: "Credentials"}
    60  	}
    61  	pathSegments := make([]string, 0, 1)
    62  	pathSegments = append(pathSegments, "batch")
    63  	path := "/" + strings.Join(pathSegments, "/")
    64  	var rawQuery string
    65  	headers := http.Header{}
    66  	body, err := innerRequest.build()
    67  	if err != nil {
    68  		return nil, err
    69  	}
    70  	bucketName := options.OverwrittenBucketName
    71  	uplogInterceptor, err := uplog.NewRequestUplog("batchOps", bucketName, "", func() (string, error) {
    72  		credentials := innerRequest.Credentials
    73  		if credentials == nil {
    74  			credentials = storage.client.GetCredentials()
    75  		}
    76  		putPolicy, err := uptoken.NewPutPolicy(bucketName, time.Now().Add(time.Hour))
    77  		if err != nil {
    78  			return "", err
    79  		}
    80  		return uptoken.NewSigner(putPolicy, credentials).GetUpToken(ctx)
    81  	})
    82  	if err != nil {
    83  		return nil, err
    84  	}
    85  	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, BufferResponse: true, RequestBody: httpclient.GetFormRequestBody(body), OnRequestProgress: options.OnRequestProgress}
    86  	if options.OverwrittenEndpoints == nil && options.OverwrittenRegion == nil && storage.client.GetRegions() == nil {
    87  		bucketHosts := httpclient.DefaultBucketHosts()
    88  		if bucketName != "" {
    89  			query := storage.client.GetBucketQuery()
    90  			if query == nil {
    91  				if options.OverwrittenBucketHosts != nil {
    92  					if bucketHosts, err = options.OverwrittenBucketHosts.GetEndpoints(ctx); err != nil {
    93  						return nil, err
    94  					}
    95  				}
    96  				queryOptions := region.BucketRegionsQueryOptions{UseInsecureProtocol: storage.client.UseInsecureProtocol(), AccelerateUploading: storage.client.AccelerateUploadingEnabled(), HostFreezeDuration: storage.client.GetHostFreezeDuration(), Resolver: storage.client.GetResolver(), Chooser: storage.client.GetChooser(), BeforeResolve: storage.client.GetBeforeResolveCallback(), AfterResolve: storage.client.GetAfterResolveCallback(), ResolveError: storage.client.GetResolveErrorCallback(), BeforeBackoff: storage.client.GetBeforeBackoffCallback(), AfterBackoff: storage.client.GetAfterBackoffCallback(), BeforeRequest: storage.client.GetBeforeRequestCallback(), AfterResponse: storage.client.GetAfterResponseCallback()}
    97  				if hostRetryConfig := storage.client.GetHostRetryConfig(); hostRetryConfig != nil {
    98  					queryOptions.RetryMax = hostRetryConfig.RetryMax
    99  					queryOptions.Backoff = hostRetryConfig.Backoff
   100  				}
   101  				if query, err = region.NewBucketRegionsQuery(bucketHosts, &queryOptions); err != nil {
   102  					return nil, err
   103  				}
   104  			}
   105  			if query != nil {
   106  				var accessKey string
   107  				var err error
   108  				if accessKey, err = innerRequest.getAccessKey(ctx); err != nil {
   109  					return nil, err
   110  				}
   111  				if accessKey == "" {
   112  					if credentialsProvider := storage.client.GetCredentials(); credentialsProvider != nil {
   113  						if creds, err := credentialsProvider.Get(ctx); err != nil {
   114  							return nil, err
   115  						} else if creds != nil {
   116  							accessKey = creds.AccessKey
   117  						}
   118  					}
   119  				}
   120  				if accessKey != "" {
   121  					req.Region = query.Query(accessKey, bucketName)
   122  				}
   123  			}
   124  		}
   125  	}
   126  	var respBody BatchOpsResponse
   127  	if err := storage.client.DoAndAcceptJSON(ctx, &req, &respBody); err != nil {
   128  		return nil, err
   129  	}
   130  	return &respBody, nil
   131  }