hcq/influx-cli
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: improve codegen to support influx query rewrite (#60)

Browse Source 
* build: add goimports to fmt target to remove unused imports
* feat: update codegen template to support returning raw response body
* feat: add support for gunzip-ing response bodies
* refactor: remove unused piece from codegen return values
This commit is contained in:
Daniel Moran
2021-05-05 10:30:22 -04:00
committed by GitHub
parent 7eca7c0bb9
commit f95668ada6
28 changed files with 836 additions and 519 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Makefile
View File

@ -30,7 +30,11 @@ openapi:
./etc/generate-openapi.sh ./etc/generate-openapi.sh
fmt: $(SOURCES_NO_VENDOR) fmt: $(SOURCES_NO_VENDOR)
# Format everything, but the import-format doesn't match our desired pattern.
gofmt -w -s $^ gofmt -w -s $^
# Remove unused imports.
go run golang.org/x/tools/cmd/goimports -w $^
# Format imports.
go run github.com/daixiang0/gci -w $^ go run github.com/daixiang0/gci -w $^
bin/$(GOOS)/influx: $(SOURCES) bin/$(GOOS)/influx: $(SOURCES)

1
go.mod
View File

@ -15,6 +15,7 @@ require (
github.com/stretchr/testify v1.7.0 github.com/stretchr/testify v1.7.0
github.com/urfave/cli/v2 v2.3.0 github.com/urfave/cli/v2 v2.3.0
golang.org/x/text v0.3.3 golang.org/x/text v0.3.3
golang.org/x/tools v0.1.0
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 // indirect gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 // indirect
honnef.co/go/tools v0.1.3 honnef.co/go/tools v0.1.3
) )

189
internal/api/api_bucket_schemas.gen.go
View File

@ -11,8 +11,9 @@
package api package api
import ( import (
"bytes" _gzip "compress/gzip"
_context "context" _context "context"
_io "io"
_ioutil "io/ioutil" _ioutil "io/ioutil"
_nethttp "net/http" _nethttp "net/http"
_neturl "net/url" _neturl "net/url"
@ -38,7 +39,7 @@ type BucketSchemasApi interface {
* CreateMeasurementSchemaExecute executes the request * CreateMeasurementSchemaExecute executes the request
* @return MeasurementSchema * @return MeasurementSchema
*/ */
CreateMeasurementSchemaExecute(r ApiCreateMeasurementSchemaRequest) (MeasurementSchema, *_nethttp.Response, error) CreateMeasurementSchemaExecute(r ApiCreateMeasurementSchemaRequest) (MeasurementSchema, error)
/* /*
* GetMeasurementSchema Fetch schema information for a measurement * GetMeasurementSchema Fetch schema information for a measurement
@ -53,7 +54,7 @@ type BucketSchemasApi interface {
* GetMeasurementSchemaExecute executes the request * GetMeasurementSchemaExecute executes the request
* @return MeasurementSchema * @return MeasurementSchema
*/ */
GetMeasurementSchemaExecute(r ApiGetMeasurementSchemaRequest) (MeasurementSchema, *_nethttp.Response, error) GetMeasurementSchemaExecute(r ApiGetMeasurementSchemaRequest) (MeasurementSchema, error)
/* /*
* GetMeasurementSchemas Retrieve a list of measurement schemas defined for this bucket * GetMeasurementSchemas Retrieve a list of measurement schemas defined for this bucket
@ -67,7 +68,7 @@ type BucketSchemasApi interface {
* GetMeasurementSchemasExecute executes the request * GetMeasurementSchemasExecute executes the request
* @return MeasurementSchemaList * @return MeasurementSchemaList
*/ */
GetMeasurementSchemasExecute(r ApiGetMeasurementSchemasRequest) (MeasurementSchemaList, *_nethttp.Response, error) GetMeasurementSchemasExecute(r ApiGetMeasurementSchemasRequest) (MeasurementSchemaList, error)
/* /*
* UpdateMeasurementSchema Update existing measurement schema * UpdateMeasurementSchema Update existing measurement schema
@ -82,7 +83,23 @@ type BucketSchemasApi interface {
* UpdateMeasurementSchemaExecute executes the request * UpdateMeasurementSchemaExecute executes the request
* @return MeasurementSchema * @return MeasurementSchema
*/ */
UpdateMeasurementSchemaExecute(r ApiUpdateMeasurementSchemaRequest) (MeasurementSchema, *_nethttp.Response, error) UpdateMeasurementSchemaExecute(r ApiUpdateMeasurementSchemaRequest) (MeasurementSchema, error)
}
// bucketSchemasApiGzipReadCloser supports streaming gzip response-bodies directly from the server.
type bucketSchemasApiGzipReadCloser struct {
underlying _io.ReadCloser
gzip _io.ReadCloser
}
func (gzrc *bucketSchemasApiGzipReadCloser) Read(p []byte) (int, error) {
return gzrc.gzip.Read(p)
}
func (gzrc *bucketSchemasApiGzipReadCloser) Close() error {
if err := gzrc.gzip.Close(); err != nil {
return err
}
return gzrc.underlying.Close()
} }
// BucketSchemasApiService BucketSchemasApi service // BucketSchemasApiService BucketSchemasApi service
@ -129,7 +146,7 @@ func (r ApiCreateMeasurementSchemaRequest) GetMeasurementSchemaCreateRequest() *
return r.measurementSchemaCreateRequest return r.measurementSchemaCreateRequest
} }
func (r ApiCreateMeasurementSchemaRequest) Execute() (MeasurementSchema, *_nethttp.Response, error) { func (r ApiCreateMeasurementSchemaRequest) Execute() (MeasurementSchema, error) {
return r.ApiService.CreateMeasurementSchemaExecute(r) return r.ApiService.CreateMeasurementSchemaExecute(r)
} }
@ -151,7 +168,7 @@ func (a *BucketSchemasApiService) CreateMeasurementSchema(ctx _context.Context,
* Execute executes the request * Execute executes the request
* @return MeasurementSchema * @return MeasurementSchema
*/ */
func (a *BucketSchemasApiService) CreateMeasurementSchemaExecute(r ApiCreateMeasurementSchemaRequest) (MeasurementSchema, *_nethttp.Response, error) { func (a *BucketSchemasApiService) CreateMeasurementSchemaExecute(r ApiCreateMeasurementSchemaRequest) (MeasurementSchema, error) {
var ( var (
localVarHTTPMethod = _nethttp.MethodPost localVarHTTPMethod = _nethttp.MethodPost
localVarPostBody interface{} localVarPostBody interface{}
@ -163,7 +180,7 @@ func (a *BucketSchemasApiService) CreateMeasurementSchemaExecute(r ApiCreateMeas
localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "BucketSchemasApiService.CreateMeasurementSchema") localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "BucketSchemasApiService.CreateMeasurementSchema")
if err != nil { if err != nil {
return localVarReturnValue, nil, GenericOpenAPIError{error: err.Error()} return localVarReturnValue, GenericOpenAPIError{error: err.Error()}
} }
localVarPath := localBasePath + "/buckets/{bucketID}/schema/measurements" localVarPath := localBasePath + "/buckets/{bucketID}/schema/measurements"
@ -200,22 +217,30 @@ func (a *BucketSchemasApiService) CreateMeasurementSchemaExecute(r ApiCreateMeas
localVarPostBody = r.measurementSchemaCreateRequest localVarPostBody = r.measurementSchemaCreateRequest
req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes) req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes)
if err != nil { if err != nil {
return localVarReturnValue, nil, err return localVarReturnValue, err
} }
localVarHTTPResponse, err := a.client.callAPI(req) localVarHTTPResponse, err := a.client.callAPI(req)
if err != nil || localVarHTTPResponse == nil { if err != nil || localVarHTTPResponse == nil {
return localVarReturnValue, localVarHTTPResponse, err return localVarReturnValue, err
} }
localVarBody, err := _ioutil.ReadAll(localVarHTTPResponse.Body) var body _io.ReadCloser = localVarHTTPResponse.Body
localVarHTTPResponse.Body.Close() if localVarHTTPResponse.Header.Get("Content-Encoding") == "gzip" {
localVarHTTPResponse.Body = _ioutil.NopCloser(bytes.NewBuffer(localVarBody)) gzr, err := _gzip.NewReader(body)
if err != nil { if err != nil {
return localVarReturnValue, localVarHTTPResponse, err body.Close()
return localVarReturnValue, err
}
body = &bucketSchemasApiGzipReadCloser{underlying: body, gzip: gzr}
} }
if localVarHTTPResponse.StatusCode >= 300 { if localVarHTTPResponse.StatusCode >= 300 {
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: localVarHTTPResponse.Status, error: localVarHTTPResponse.Status,
@ -225,23 +250,28 @@ func (a *BucketSchemasApiService) CreateMeasurementSchemaExecute(r ApiCreateMeas
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
newErr.model = &v newErr.model = &v
} }
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: err.Error(), error: err.Error(),
} }
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
return localVarReturnValue, localVarHTTPResponse, nil return localVarReturnValue, nil
} }
type ApiGetMeasurementSchemaRequest struct { type ApiGetMeasurementSchemaRequest struct {
@ -285,7 +315,7 @@ func (r ApiGetMeasurementSchemaRequest) GetOrgID() *string {
return r.orgID return r.orgID
} }
func (r ApiGetMeasurementSchemaRequest) Execute() (MeasurementSchema, *_nethttp.Response, error) { func (r ApiGetMeasurementSchemaRequest) Execute() (MeasurementSchema, error) {
return r.ApiService.GetMeasurementSchemaExecute(r) return r.ApiService.GetMeasurementSchemaExecute(r)
} }
@ -309,7 +339,7 @@ func (a *BucketSchemasApiService) GetMeasurementSchema(ctx _context.Context, buc
* Execute executes the request * Execute executes the request
* @return MeasurementSchema * @return MeasurementSchema
*/ */
func (a *BucketSchemasApiService) GetMeasurementSchemaExecute(r ApiGetMeasurementSchemaRequest) (MeasurementSchema, *_nethttp.Response, error) { func (a *BucketSchemasApiService) GetMeasurementSchemaExecute(r ApiGetMeasurementSchemaRequest) (MeasurementSchema, error) {
var ( var (
localVarHTTPMethod = _nethttp.MethodGet localVarHTTPMethod = _nethttp.MethodGet
localVarPostBody interface{} localVarPostBody interface{}
@ -321,7 +351,7 @@ func (a *BucketSchemasApiService) GetMeasurementSchemaExecute(r ApiGetMeasuremen
localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "BucketSchemasApiService.GetMeasurementSchema") localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "BucketSchemasApiService.GetMeasurementSchema")
if err != nil { if err != nil {
return localVarReturnValue, nil, GenericOpenAPIError{error: err.Error()} return localVarReturnValue, GenericOpenAPIError{error: err.Error()}
} }
localVarPath := localBasePath + "/buckets/{bucketID}/schema/measurements/{measurementID}" localVarPath := localBasePath + "/buckets/{bucketID}/schema/measurements/{measurementID}"
@ -357,39 +387,52 @@ func (a *BucketSchemasApiService) GetMeasurementSchemaExecute(r ApiGetMeasuremen
} }
req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes) req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes)
if err != nil { if err != nil {
return localVarReturnValue, nil, err return localVarReturnValue, err
} }
localVarHTTPResponse, err := a.client.callAPI(req) localVarHTTPResponse, err := a.client.callAPI(req)
if err != nil || localVarHTTPResponse == nil { if err != nil || localVarHTTPResponse == nil {
return localVarReturnValue, localVarHTTPResponse, err return localVarReturnValue, err
} }
localVarBody, err := _ioutil.ReadAll(localVarHTTPResponse.Body) var body _io.ReadCloser = localVarHTTPResponse.Body
localVarHTTPResponse.Body.Close() if localVarHTTPResponse.Header.Get("Content-Encoding") == "gzip" {
localVarHTTPResponse.Body = _ioutil.NopCloser(bytes.NewBuffer(localVarBody)) gzr, err := _gzip.NewReader(body)
if err != nil { if err != nil {
return localVarReturnValue, localVarHTTPResponse, err body.Close()
return localVarReturnValue, err
}
body = &bucketSchemasApiGzipReadCloser{underlying: body, gzip: gzr}
} }
if localVarHTTPResponse.StatusCode >= 300 { if localVarHTTPResponse.StatusCode >= 300 {
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: localVarHTTPResponse.Status, error: localVarHTTPResponse.Status,
} }
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: err.Error(), error: err.Error(),
} }
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
return localVarReturnValue, localVarHTTPResponse, nil return localVarReturnValue, nil
} }
type ApiGetMeasurementSchemasRequest struct { type ApiGetMeasurementSchemasRequest struct {
@ -433,7 +476,7 @@ func (r ApiGetMeasurementSchemasRequest) GetName() *string {
return r.name return r.name
} }
func (r ApiGetMeasurementSchemasRequest) Execute() (MeasurementSchemaList, *_nethttp.Response, error) { func (r ApiGetMeasurementSchemasRequest) Execute() (MeasurementSchemaList, error) {
return r.ApiService.GetMeasurementSchemasExecute(r) return r.ApiService.GetMeasurementSchemasExecute(r)
} }
@ -455,7 +498,7 @@ func (a *BucketSchemasApiService) GetMeasurementSchemas(ctx _context.Context, bu
* Execute executes the request * Execute executes the request
* @return MeasurementSchemaList * @return MeasurementSchemaList
*/ */
func (a *BucketSchemasApiService) GetMeasurementSchemasExecute(r ApiGetMeasurementSchemasRequest) (MeasurementSchemaList, *_nethttp.Response, error) { func (a *BucketSchemasApiService) GetMeasurementSchemasExecute(r ApiGetMeasurementSchemasRequest) (MeasurementSchemaList, error) {
var ( var (
localVarHTTPMethod = _nethttp.MethodGet localVarHTTPMethod = _nethttp.MethodGet
localVarPostBody interface{} localVarPostBody interface{}
@ -467,7 +510,7 @@ func (a *BucketSchemasApiService) GetMeasurementSchemasExecute(r ApiGetMeasureme
localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "BucketSchemasApiService.GetMeasurementSchemas") localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "BucketSchemasApiService.GetMeasurementSchemas")
if err != nil { if err != nil {
return localVarReturnValue, nil, GenericOpenAPIError{error: err.Error()} return localVarReturnValue, GenericOpenAPIError{error: err.Error()}
} }
localVarPath := localBasePath + "/buckets/{bucketID}/schema/measurements" localVarPath := localBasePath + "/buckets/{bucketID}/schema/measurements"
@ -505,22 +548,30 @@ func (a *BucketSchemasApiService) GetMeasurementSchemasExecute(r ApiGetMeasureme
} }
req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes) req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes)
if err != nil { if err != nil {
return localVarReturnValue, nil, err return localVarReturnValue, err
} }
localVarHTTPResponse, err := a.client.callAPI(req) localVarHTTPResponse, err := a.client.callAPI(req)
if err != nil || localVarHTTPResponse == nil { if err != nil || localVarHTTPResponse == nil {
return localVarReturnValue, localVarHTTPResponse, err return localVarReturnValue, err
} }
localVarBody, err := _ioutil.ReadAll(localVarHTTPResponse.Body) var body _io.ReadCloser = localVarHTTPResponse.Body
localVarHTTPResponse.Body.Close() if localVarHTTPResponse.Header.Get("Content-Encoding") == "gzip" {
localVarHTTPResponse.Body = _ioutil.NopCloser(bytes.NewBuffer(localVarBody)) gzr, err := _gzip.NewReader(body)
if err != nil { if err != nil {
return localVarReturnValue, localVarHTTPResponse, err body.Close()
return localVarReturnValue, err
}
body = &bucketSchemasApiGzipReadCloser{underlying: body, gzip: gzr}
} }
if localVarHTTPResponse.StatusCode >= 300 { if localVarHTTPResponse.StatusCode >= 300 {
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: localVarHTTPResponse.Status, error: localVarHTTPResponse.Status,
@ -530,23 +581,28 @@ func (a *BucketSchemasApiService) GetMeasurementSchemasExecute(r ApiGetMeasureme
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
newErr.model = &v newErr.model = &v
} }
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: err.Error(), error: err.Error(),
} }
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
return localVarReturnValue, localVarHTTPResponse, nil return localVarReturnValue, nil
} }
type ApiUpdateMeasurementSchemaRequest struct { type ApiUpdateMeasurementSchemaRequest struct {
@ -599,7 +655,7 @@ func (r ApiUpdateMeasurementSchemaRequest) GetMeasurementSchemaUpdateRequest() *
return r.measurementSchemaUpdateRequest return r.measurementSchemaUpdateRequest
} }
func (r ApiUpdateMeasurementSchemaRequest) Execute() (MeasurementSchema, *_nethttp.Response, error) { func (r ApiUpdateMeasurementSchemaRequest) Execute() (MeasurementSchema, error) {
return r.ApiService.UpdateMeasurementSchemaExecute(r) return r.ApiService.UpdateMeasurementSchemaExecute(r)
} }
@ -623,7 +679,7 @@ func (a *BucketSchemasApiService) UpdateMeasurementSchema(ctx _context.Context,
* Execute executes the request * Execute executes the request
* @return MeasurementSchema * @return MeasurementSchema
*/ */
func (a *BucketSchemasApiService) UpdateMeasurementSchemaExecute(r ApiUpdateMeasurementSchemaRequest) (MeasurementSchema, *_nethttp.Response, error) { func (a *BucketSchemasApiService) UpdateMeasurementSchemaExecute(r ApiUpdateMeasurementSchemaRequest) (MeasurementSchema, error) {
var ( var (
localVarHTTPMethod = _nethttp.MethodPatch localVarHTTPMethod = _nethttp.MethodPatch
localVarPostBody interface{} localVarPostBody interface{}
@ -635,7 +691,7 @@ func (a *BucketSchemasApiService) UpdateMeasurementSchemaExecute(r ApiUpdateMeas
localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "BucketSchemasApiService.UpdateMeasurementSchema") localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "BucketSchemasApiService.UpdateMeasurementSchema")
if err != nil { if err != nil {
return localVarReturnValue, nil, GenericOpenAPIError{error: err.Error()} return localVarReturnValue, GenericOpenAPIError{error: err.Error()}
} }
localVarPath := localBasePath + "/buckets/{bucketID}/schema/measurements/{measurementID}" localVarPath := localBasePath + "/buckets/{bucketID}/schema/measurements/{measurementID}"
@ -673,22 +729,30 @@ func (a *BucketSchemasApiService) UpdateMeasurementSchemaExecute(r ApiUpdateMeas
localVarPostBody = r.measurementSchemaUpdateRequest localVarPostBody = r.measurementSchemaUpdateRequest
req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes) req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes)
if err != nil { if err != nil {
return localVarReturnValue, nil, err return localVarReturnValue, err
} }
localVarHTTPResponse, err := a.client.callAPI(req) localVarHTTPResponse, err := a.client.callAPI(req)
if err != nil || localVarHTTPResponse == nil { if err != nil || localVarHTTPResponse == nil {
return localVarReturnValue, localVarHTTPResponse, err return localVarReturnValue, err
} }
localVarBody, err := _ioutil.ReadAll(localVarHTTPResponse.Body) var body _io.ReadCloser = localVarHTTPResponse.Body
localVarHTTPResponse.Body.Close() if localVarHTTPResponse.Header.Get("Content-Encoding") == "gzip" {
localVarHTTPResponse.Body = _ioutil.NopCloser(bytes.NewBuffer(localVarBody)) gzr, err := _gzip.NewReader(body)
if err != nil { if err != nil {
return localVarReturnValue, localVarHTTPResponse, err body.Close()
return localVarReturnValue, err
}
body = &bucketSchemasApiGzipReadCloser{underlying: body, gzip: gzr}
} }
if localVarHTTPResponse.StatusCode >= 300 { if localVarHTTPResponse.StatusCode >= 300 {
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: localVarHTTPResponse.Status, error: localVarHTTPResponse.Status,
@ -698,21 +762,26 @@ func (a *BucketSchemasApiService) UpdateMeasurementSchemaExecute(r ApiUpdateMeas
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
newErr.model = &v newErr.model = &v
} }
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: err.Error(), error: err.Error(),
} }
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
return localVarReturnValue, localVarHTTPResponse, nil return localVarReturnValue, nil
} }

239
internal/api/api_buckets.gen.go
View File

@ -11,8 +11,9 @@
package api package api
import ( import (
"bytes" _gzip "compress/gzip"
_context "context" _context "context"
_io "io"
_ioutil "io/ioutil" _ioutil "io/ioutil"
_nethttp "net/http" _nethttp "net/http"
_neturl "net/url" _neturl "net/url"
@ -37,7 +38,7 @@ type BucketsApi interface {
/* /*
* DeleteBucketsIDExecute executes the request * DeleteBucketsIDExecute executes the request
*/ */
DeleteBucketsIDExecute(r ApiDeleteBucketsIDRequest) (*_nethttp.Response, error) DeleteBucketsIDExecute(r ApiDeleteBucketsIDRequest) error
/* /*
* GetBuckets List all buckets * GetBuckets List all buckets
@ -50,7 +51,7 @@ type BucketsApi interface {
* GetBucketsExecute executes the request * GetBucketsExecute executes the request
* @return Buckets * @return Buckets
*/ */
GetBucketsExecute(r ApiGetBucketsRequest) (Buckets, *_nethttp.Response, error) GetBucketsExecute(r ApiGetBucketsRequest) (Buckets, error)
/* /*
* GetBucketsID Retrieve a bucket * GetBucketsID Retrieve a bucket
@ -64,7 +65,7 @@ type BucketsApi interface {
* GetBucketsIDExecute executes the request * GetBucketsIDExecute executes the request
* @return Bucket * @return Bucket
*/ */
GetBucketsIDExecute(r ApiGetBucketsIDRequest) (Bucket, *_nethttp.Response, error) GetBucketsIDExecute(r ApiGetBucketsIDRequest) (Bucket, error)
/* /*
* PatchBucketsID Update a bucket * PatchBucketsID Update a bucket
@ -78,7 +79,7 @@ type BucketsApi interface {
* PatchBucketsIDExecute executes the request * PatchBucketsIDExecute executes the request
* @return Bucket * @return Bucket
*/ */
PatchBucketsIDExecute(r ApiPatchBucketsIDRequest) (Bucket, *_nethttp.Response, error) PatchBucketsIDExecute(r ApiPatchBucketsIDRequest) (Bucket, error)
/* /*
* PostBuckets Create a bucket * PostBuckets Create a bucket
@ -91,7 +92,23 @@ type BucketsApi interface {
* PostBucketsExecute executes the request * PostBucketsExecute executes the request
* @return Bucket * @return Bucket
*/ */
PostBucketsExecute(r ApiPostBucketsRequest) (Bucket, *_nethttp.Response, error) PostBucketsExecute(r ApiPostBucketsRequest) (Bucket, error)
}
// bucketsApiGzipReadCloser supports streaming gzip response-bodies directly from the server.
type bucketsApiGzipReadCloser struct {
underlying _io.ReadCloser
gzip _io.ReadCloser
}
func (gzrc *bucketsApiGzipReadCloser) Read(p []byte) (int, error) {
return gzrc.gzip.Read(p)
}
func (gzrc *bucketsApiGzipReadCloser) Close() error {
if err := gzrc.gzip.Close(); err != nil {
return err
}
return gzrc.underlying.Close()
} }
// BucketsApiService BucketsApi service // BucketsApiService BucketsApi service
@ -120,7 +137,7 @@ func (r ApiDeleteBucketsIDRequest) GetZapTraceSpan() *string {
return r.zapTraceSpan return r.zapTraceSpan
} }
func (r ApiDeleteBucketsIDRequest) Execute() (*_nethttp.Response, error) { func (r ApiDeleteBucketsIDRequest) Execute() error {
return r.ApiService.DeleteBucketsIDExecute(r) return r.ApiService.DeleteBucketsIDExecute(r)
} }
@ -141,7 +158,7 @@ func (a *BucketsApiService) DeleteBucketsID(ctx _context.Context, bucketID strin
/* /*
* Execute executes the request * Execute executes the request
*/ */
func (a *BucketsApiService) DeleteBucketsIDExecute(r ApiDeleteBucketsIDRequest) (*_nethttp.Response, error) { func (a *BucketsApiService) DeleteBucketsIDExecute(r ApiDeleteBucketsIDRequest) error {
var ( var (
localVarHTTPMethod = _nethttp.MethodDelete localVarHTTPMethod = _nethttp.MethodDelete
localVarPostBody interface{} localVarPostBody interface{}
@ -152,7 +169,7 @@ func (a *BucketsApiService) DeleteBucketsIDExecute(r ApiDeleteBucketsIDRequest)
localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "BucketsApiService.DeleteBucketsID") localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "BucketsApiService.DeleteBucketsID")
if err != nil { if err != nil {
return nil, GenericOpenAPIError{error: err.Error()} return GenericOpenAPIError{error: err.Error()}
} }
localVarPath := localBasePath + "/buckets/{bucketID}" localVarPath := localBasePath + "/buckets/{bucketID}"
@ -184,22 +201,30 @@ func (a *BucketsApiService) DeleteBucketsIDExecute(r ApiDeleteBucketsIDRequest)
} }
req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes) req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes)
if err != nil { if err != nil {
return nil, err return err
} }
localVarHTTPResponse, err := a.client.callAPI(req) localVarHTTPResponse, err := a.client.callAPI(req)
if err != nil || localVarHTTPResponse == nil { if err != nil || localVarHTTPResponse == nil {
return localVarHTTPResponse, err return err
} }
localVarBody, err := _ioutil.ReadAll(localVarHTTPResponse.Body) var body _io.ReadCloser = localVarHTTPResponse.Body
localVarHTTPResponse.Body.Close() if localVarHTTPResponse.Header.Get("Content-Encoding") == "gzip" {
localVarHTTPResponse.Body = _ioutil.NopCloser(bytes.NewBuffer(localVarBody)) gzr, err := _gzip.NewReader(body)
if err != nil { if err != nil {
return localVarHTTPResponse, err body.Close()
return err
}
body = &bucketsApiGzipReadCloser{underlying: body, gzip: gzr}
} }
if localVarHTTPResponse.StatusCode >= 300 { if localVarHTTPResponse.StatusCode >= 300 {
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return err
}
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: localVarHTTPResponse.Status, error: localVarHTTPResponse.Status,
@ -209,22 +234,22 @@ func (a *BucketsApiService) DeleteBucketsIDExecute(r ApiDeleteBucketsIDRequest)
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarHTTPResponse, newErr return newErr
} }
newErr.model = &v newErr.model = &v
return localVarHTTPResponse, newErr return newErr
} }
var v Error var v Error
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarHTTPResponse, newErr return newErr
} }
newErr.model = &v newErr.model = &v
return localVarHTTPResponse, newErr return newErr
} }
return localVarHTTPResponse, nil return nil
} }
type ApiGetBucketsRequest struct { type ApiGetBucketsRequest struct {
@ -304,7 +329,7 @@ func (r ApiGetBucketsRequest) GetId() *string {
return r.id return r.id
} }
func (r ApiGetBucketsRequest) Execute() (Buckets, *_nethttp.Response, error) { func (r ApiGetBucketsRequest) Execute() (Buckets, error) {
return r.ApiService.GetBucketsExecute(r) return r.ApiService.GetBucketsExecute(r)
} }
@ -324,7 +349,7 @@ func (a *BucketsApiService) GetBuckets(ctx _context.Context) ApiGetBucketsReques
* Execute executes the request * Execute executes the request
* @return Buckets * @return Buckets
*/ */
func (a *BucketsApiService) GetBucketsExecute(r ApiGetBucketsRequest) (Buckets, *_nethttp.Response, error) { func (a *BucketsApiService) GetBucketsExecute(r ApiGetBucketsRequest) (Buckets, error) {
var ( var (
localVarHTTPMethod = _nethttp.MethodGet localVarHTTPMethod = _nethttp.MethodGet
localVarPostBody interface{} localVarPostBody interface{}
@ -336,7 +361,7 @@ func (a *BucketsApiService) GetBucketsExecute(r ApiGetBucketsRequest) (Buckets,
localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "BucketsApiService.GetBuckets") localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "BucketsApiService.GetBuckets")
if err != nil { if err != nil {
return localVarReturnValue, nil, GenericOpenAPIError{error: err.Error()} return localVarReturnValue, GenericOpenAPIError{error: err.Error()}
} }
localVarPath := localBasePath + "/buckets" localVarPath := localBasePath + "/buckets"
@ -388,22 +413,30 @@ func (a *BucketsApiService) GetBucketsExecute(r ApiGetBucketsRequest) (Buckets,
} }
req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes) req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes)
if err != nil { if err != nil {
return localVarReturnValue, nil, err return localVarReturnValue, err
} }
localVarHTTPResponse, err := a.client.callAPI(req) localVarHTTPResponse, err := a.client.callAPI(req)
if err != nil || localVarHTTPResponse == nil { if err != nil || localVarHTTPResponse == nil {
return localVarReturnValue, localVarHTTPResponse, err return localVarReturnValue, err
} }
localVarBody, err := _ioutil.ReadAll(localVarHTTPResponse.Body) var body _io.ReadCloser = localVarHTTPResponse.Body
localVarHTTPResponse.Body.Close() if localVarHTTPResponse.Header.Get("Content-Encoding") == "gzip" {
localVarHTTPResponse.Body = _ioutil.NopCloser(bytes.NewBuffer(localVarBody)) gzr, err := _gzip.NewReader(body)
if err != nil { if err != nil {
return localVarReturnValue, localVarHTTPResponse, err body.Close()
return localVarReturnValue, err
}
body = &bucketsApiGzipReadCloser{underlying: body, gzip: gzr}
} }
if localVarHTTPResponse.StatusCode >= 300 { if localVarHTTPResponse.StatusCode >= 300 {
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: localVarHTTPResponse.Status, error: localVarHTTPResponse.Status,
@ -412,22 +445,27 @@ func (a *BucketsApiService) GetBucketsExecute(r ApiGetBucketsRequest) (Buckets,
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
newErr.model = &v newErr.model = &v
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: err.Error(), error: err.Error(),
} }
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
return localVarReturnValue, localVarHTTPResponse, nil return localVarReturnValue, nil
} }
type ApiGetBucketsIDRequest struct { type ApiGetBucketsIDRequest struct {
@ -453,7 +491,7 @@ func (r ApiGetBucketsIDRequest) GetZapTraceSpan() *string {
return r.zapTraceSpan return r.zapTraceSpan
} }
func (r ApiGetBucketsIDRequest) Execute() (Bucket, *_nethttp.Response, error) { func (r ApiGetBucketsIDRequest) Execute() (Bucket, error) {
return r.ApiService.GetBucketsIDExecute(r) return r.ApiService.GetBucketsIDExecute(r)
} }
@ -475,7 +513,7 @@ func (a *BucketsApiService) GetBucketsID(ctx _context.Context, bucketID string)
* Execute executes the request * Execute executes the request
* @return Bucket * @return Bucket
*/ */
func (a *BucketsApiService) GetBucketsIDExecute(r ApiGetBucketsIDRequest) (Bucket, *_nethttp.Response, error) { func (a *BucketsApiService) GetBucketsIDExecute(r ApiGetBucketsIDRequest) (Bucket, error) {
var ( var (
localVarHTTPMethod = _nethttp.MethodGet localVarHTTPMethod = _nethttp.MethodGet
localVarPostBody interface{} localVarPostBody interface{}
@ -487,7 +525,7 @@ func (a *BucketsApiService) GetBucketsIDExecute(r ApiGetBucketsIDRequest) (Bucke
localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "BucketsApiService.GetBucketsID") localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "BucketsApiService.GetBucketsID")
if err != nil { if err != nil {
return localVarReturnValue, nil, GenericOpenAPIError{error: err.Error()} return localVarReturnValue, GenericOpenAPIError{error: err.Error()}
} }
localVarPath := localBasePath + "/buckets/{bucketID}" localVarPath := localBasePath + "/buckets/{bucketID}"
@ -519,22 +557,30 @@ func (a *BucketsApiService) GetBucketsIDExecute(r ApiGetBucketsIDRequest) (Bucke
} }
req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes) req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes)
if err != nil { if err != nil {
return localVarReturnValue, nil, err return localVarReturnValue, err
} }
localVarHTTPResponse, err := a.client.callAPI(req) localVarHTTPResponse, err := a.client.callAPI(req)
if err != nil || localVarHTTPResponse == nil { if err != nil || localVarHTTPResponse == nil {
return localVarReturnValue, localVarHTTPResponse, err return localVarReturnValue, err
} }
localVarBody, err := _ioutil.ReadAll(localVarHTTPResponse.Body) var body _io.ReadCloser = localVarHTTPResponse.Body
localVarHTTPResponse.Body.Close() if localVarHTTPResponse.Header.Get("Content-Encoding") == "gzip" {
localVarHTTPResponse.Body = _ioutil.NopCloser(bytes.NewBuffer(localVarBody)) gzr, err := _gzip.NewReader(body)
if err != nil { if err != nil {
return localVarReturnValue, localVarHTTPResponse, err body.Close()
return localVarReturnValue, err
}
body = &bucketsApiGzipReadCloser{underlying: body, gzip: gzr}
} }
if localVarHTTPResponse.StatusCode >= 300 { if localVarHTTPResponse.StatusCode >= 300 {
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: localVarHTTPResponse.Status, error: localVarHTTPResponse.Status,
@ -543,22 +589,27 @@ func (a *BucketsApiService) GetBucketsIDExecute(r ApiGetBucketsIDRequest) (Bucke
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
newErr.model = &v newErr.model = &v
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: err.Error(), error: err.Error(),
} }
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
return localVarReturnValue, localVarHTTPResponse, nil return localVarReturnValue, nil
} }
type ApiPatchBucketsIDRequest struct { type ApiPatchBucketsIDRequest struct {
@ -593,7 +644,7 @@ func (r ApiPatchBucketsIDRequest) GetZapTraceSpan() *string {
return r.zapTraceSpan return r.zapTraceSpan
} }
func (r ApiPatchBucketsIDRequest) Execute() (Bucket, *_nethttp.Response, error) { func (r ApiPatchBucketsIDRequest) Execute() (Bucket, error) {
return r.ApiService.PatchBucketsIDExecute(r) return r.ApiService.PatchBucketsIDExecute(r)
} }
@ -615,7 +666,7 @@ func (a *BucketsApiService) PatchBucketsID(ctx _context.Context, bucketID string
* Execute executes the request * Execute executes the request
* @return Bucket * @return Bucket
*/ */
func (a *BucketsApiService) PatchBucketsIDExecute(r ApiPatchBucketsIDRequest) (Bucket, *_nethttp.Response, error) { func (a *BucketsApiService) PatchBucketsIDExecute(r ApiPatchBucketsIDRequest) (Bucket, error) {
var ( var (
localVarHTTPMethod = _nethttp.MethodPatch localVarHTTPMethod = _nethttp.MethodPatch
localVarPostBody interface{} localVarPostBody interface{}
@ -627,7 +678,7 @@ func (a *BucketsApiService) PatchBucketsIDExecute(r ApiPatchBucketsIDRequest) (B
localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "BucketsApiService.PatchBucketsID") localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "BucketsApiService.PatchBucketsID")
if err != nil { if err != nil {
return localVarReturnValue, nil, GenericOpenAPIError{error: err.Error()} return localVarReturnValue, GenericOpenAPIError{error: err.Error()}
} }
localVarPath := localBasePath + "/buckets/{bucketID}" localVarPath := localBasePath + "/buckets/{bucketID}"
@ -637,7 +688,7 @@ func (a *BucketsApiService) PatchBucketsIDExecute(r ApiPatchBucketsIDRequest) (B
localVarQueryParams := _neturl.Values{} localVarQueryParams := _neturl.Values{}
localVarFormParams := _neturl.Values{} localVarFormParams := _neturl.Values{}
if r.patchBucketRequest == nil { if r.patchBucketRequest == nil {
return localVarReturnValue, nil, reportError("patchBucketRequest is required and must be specified") return localVarReturnValue, reportError("patchBucketRequest is required and must be specified")
} }
// to determine the Content-Type header // to determine the Content-Type header
@ -664,22 +715,30 @@ func (a *BucketsApiService) PatchBucketsIDExecute(r ApiPatchBucketsIDRequest) (B
localVarPostBody = r.patchBucketRequest localVarPostBody = r.patchBucketRequest
req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes) req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes)
if err != nil { if err != nil {
return localVarReturnValue, nil, err return localVarReturnValue, err
} }
localVarHTTPResponse, err := a.client.callAPI(req) localVarHTTPResponse, err := a.client.callAPI(req)
if err != nil || localVarHTTPResponse == nil { if err != nil || localVarHTTPResponse == nil {
return localVarReturnValue, localVarHTTPResponse, err return localVarReturnValue, err
} }
localVarBody, err := _ioutil.ReadAll(localVarHTTPResponse.Body) var body _io.ReadCloser = localVarHTTPResponse.Body
localVarHTTPResponse.Body.Close() if localVarHTTPResponse.Header.Get("Content-Encoding") == "gzip" {
localVarHTTPResponse.Body = _ioutil.NopCloser(bytes.NewBuffer(localVarBody)) gzr, err := _gzip.NewReader(body)
if err != nil { if err != nil {
return localVarReturnValue, localVarHTTPResponse, err body.Close()
return localVarReturnValue, err
}
body = &bucketsApiGzipReadCloser{underlying: body, gzip: gzr}
} }
if localVarHTTPResponse.StatusCode >= 300 { if localVarHTTPResponse.StatusCode >= 300 {
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: localVarHTTPResponse.Status, error: localVarHTTPResponse.Status,
@ -688,22 +747,27 @@ func (a *BucketsApiService) PatchBucketsIDExecute(r ApiPatchBucketsIDRequest) (B
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
newErr.model = &v newErr.model = &v
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: err.Error(), error: err.Error(),
} }
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
return localVarReturnValue, localVarHTTPResponse, nil return localVarReturnValue, nil
} }
type ApiPostBucketsRequest struct { type ApiPostBucketsRequest struct {
@ -729,7 +793,7 @@ func (r ApiPostBucketsRequest) GetZapTraceSpan() *string {
return r.zapTraceSpan return r.zapTraceSpan
} }
func (r ApiPostBucketsRequest) Execute() (Bucket, *_nethttp.Response, error) { func (r ApiPostBucketsRequest) Execute() (Bucket, error) {
return r.ApiService.PostBucketsExecute(r) return r.ApiService.PostBucketsExecute(r)
} }
@ -749,7 +813,7 @@ func (a *BucketsApiService) PostBuckets(ctx _context.Context) ApiPostBucketsRequ
* Execute executes the request * Execute executes the request
* @return Bucket * @return Bucket
*/ */
func (a *BucketsApiService) PostBucketsExecute(r ApiPostBucketsRequest) (Bucket, *_nethttp.Response, error) { func (a *BucketsApiService) PostBucketsExecute(r ApiPostBucketsRequest) (Bucket, error) {
var ( var (
localVarHTTPMethod = _nethttp.MethodPost localVarHTTPMethod = _nethttp.MethodPost
localVarPostBody interface{} localVarPostBody interface{}
@ -761,7 +825,7 @@ func (a *BucketsApiService) PostBucketsExecute(r ApiPostBucketsRequest) (Bucket,
localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "BucketsApiService.PostBuckets") localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "BucketsApiService.PostBuckets")
if err != nil { if err != nil {
return localVarReturnValue, nil, GenericOpenAPIError{error: err.Error()} return localVarReturnValue, GenericOpenAPIError{error: err.Error()}
} }
localVarPath := localBasePath + "/buckets" localVarPath := localBasePath + "/buckets"
@ -770,7 +834,7 @@ func (a *BucketsApiService) PostBucketsExecute(r ApiPostBucketsRequest) (Bucket,
localVarQueryParams := _neturl.Values{} localVarQueryParams := _neturl.Values{}
localVarFormParams := _neturl.Values{} localVarFormParams := _neturl.Values{}
if r.postBucketRequest == nil { if r.postBucketRequest == nil {
return localVarReturnValue, nil, reportError("postBucketRequest is required and must be specified") return localVarReturnValue, reportError("postBucketRequest is required and must be specified")
} }
// to determine the Content-Type header // to determine the Content-Type header
@ -797,22 +861,30 @@ func (a *BucketsApiService) PostBucketsExecute(r ApiPostBucketsRequest) (Bucket,
localVarPostBody = r.postBucketRequest localVarPostBody = r.postBucketRequest
req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes) req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes)
if err != nil { if err != nil {
return localVarReturnValue, nil, err return localVarReturnValue, err
} }
localVarHTTPResponse, err := a.client.callAPI(req) localVarHTTPResponse, err := a.client.callAPI(req)
if err != nil || localVarHTTPResponse == nil { if err != nil || localVarHTTPResponse == nil {
return localVarReturnValue, localVarHTTPResponse, err return localVarReturnValue, err
} }
localVarBody, err := _ioutil.ReadAll(localVarHTTPResponse.Body) var body _io.ReadCloser = localVarHTTPResponse.Body
localVarHTTPResponse.Body.Close() if localVarHTTPResponse.Header.Get("Content-Encoding") == "gzip" {
localVarHTTPResponse.Body = _ioutil.NopCloser(bytes.NewBuffer(localVarBody)) gzr, err := _gzip.NewReader(body)
if err != nil { if err != nil {
return localVarReturnValue, localVarHTTPResponse, err body.Close()
return localVarReturnValue, err
}
body = &bucketsApiGzipReadCloser{underlying: body, gzip: gzr}
} }
if localVarHTTPResponse.StatusCode >= 300 { if localVarHTTPResponse.StatusCode >= 300 {
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: localVarHTTPResponse.Status, error: localVarHTTPResponse.Status,
@ -822,29 +894,34 @@ func (a *BucketsApiService) PostBucketsExecute(r ApiPostBucketsRequest) (Bucket,
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
newErr.model = &v newErr.model = &v
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
var v Error var v Error
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
newErr.model = &v newErr.model = &v
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: err.Error(), error: err.Error(),
} }
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
return localVarReturnValue, localVarHTTPResponse, nil return localVarReturnValue, nil
} }

66
internal/api/api_health.gen.go
View File

@ -11,8 +11,9 @@
package api package api
import ( import (
"bytes" _gzip "compress/gzip"
_context "context" _context "context"
_io "io"
_ioutil "io/ioutil" _ioutil "io/ioutil"
_nethttp "net/http" _nethttp "net/http"
_neturl "net/url" _neturl "net/url"
@ -36,7 +37,23 @@ type HealthApi interface {
* GetHealthExecute executes the request * GetHealthExecute executes the request
* @return HealthCheck * @return HealthCheck
*/ */
GetHealthExecute(r ApiGetHealthRequest) (HealthCheck, *_nethttp.Response, error) GetHealthExecute(r ApiGetHealthRequest) (HealthCheck, error)
}
// healthApiGzipReadCloser supports streaming gzip response-bodies directly from the server.
type healthApiGzipReadCloser struct {
underlying _io.ReadCloser
gzip _io.ReadCloser
}
func (gzrc *healthApiGzipReadCloser) Read(p []byte) (int, error) {
return gzrc.gzip.Read(p)
}
func (gzrc *healthApiGzipReadCloser) Close() error {
if err := gzrc.gzip.Close(); err != nil {
return err
}
return gzrc.underlying.Close()
} }
// HealthApiService HealthApi service // HealthApiService HealthApi service
@ -56,7 +73,7 @@ func (r ApiGetHealthRequest) GetZapTraceSpan() *string {
return r.zapTraceSpan return r.zapTraceSpan
} }
func (r ApiGetHealthRequest) Execute() (HealthCheck, *_nethttp.Response, error) { func (r ApiGetHealthRequest) Execute() (HealthCheck, error) {
return r.ApiService.GetHealthExecute(r) return r.ApiService.GetHealthExecute(r)
} }
@ -76,7 +93,7 @@ func (a *HealthApiService) GetHealth(ctx _context.Context) ApiGetHealthRequest {
* Execute executes the request * Execute executes the request
* @return HealthCheck * @return HealthCheck
*/ */
func (a *HealthApiService) GetHealthExecute(r ApiGetHealthRequest) (HealthCheck, *_nethttp.Response, error) { func (a *HealthApiService) GetHealthExecute(r ApiGetHealthRequest) (HealthCheck, error) {
var ( var (
localVarHTTPMethod = _nethttp.MethodGet localVarHTTPMethod = _nethttp.MethodGet
localVarPostBody interface{} localVarPostBody interface{}
@ -88,7 +105,7 @@ func (a *HealthApiService) GetHealthExecute(r ApiGetHealthRequest) (HealthCheck,
localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "HealthApiService.GetHealth") localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "HealthApiService.GetHealth")
if err != nil { if err != nil {
return localVarReturnValue, nil, GenericOpenAPIError{error: err.Error()} return localVarReturnValue, GenericOpenAPIError{error: err.Error()}
} }
localVarPath := localBasePath + "/health" localVarPath := localBasePath + "/health"
@ -119,22 +136,30 @@ func (a *HealthApiService) GetHealthExecute(r ApiGetHealthRequest) (HealthCheck,
} }
req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes) req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes)
if err != nil { if err != nil {
return localVarReturnValue, nil, err return localVarReturnValue, err
} }
localVarHTTPResponse, err := a.client.callAPI(req) localVarHTTPResponse, err := a.client.callAPI(req)
if err != nil || localVarHTTPResponse == nil { if err != nil || localVarHTTPResponse == nil {
return localVarReturnValue, localVarHTTPResponse, err return localVarReturnValue, err
} }
localVarBody, err := _ioutil.ReadAll(localVarHTTPResponse.Body) var body _io.ReadCloser = localVarHTTPResponse.Body
localVarHTTPResponse.Body.Close() if localVarHTTPResponse.Header.Get("Content-Encoding") == "gzip" {
localVarHTTPResponse.Body = _ioutil.NopCloser(bytes.NewBuffer(localVarBody)) gzr, err := _gzip.NewReader(body)
if err != nil { if err != nil {
return localVarReturnValue, localVarHTTPResponse, err body.Close()
return localVarReturnValue, err
}
body = &healthApiGzipReadCloser{underlying: body, gzip: gzr}
} }
if localVarHTTPResponse.StatusCode >= 300 { if localVarHTTPResponse.StatusCode >= 300 {
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: localVarHTTPResponse.Status, error: localVarHTTPResponse.Status,
@ -144,29 +169,34 @@ func (a *HealthApiService) GetHealthExecute(r ApiGetHealthRequest) (HealthCheck,
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
newErr.model = &v newErr.model = &v
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
var v Error var v Error
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
newErr.model = &v newErr.model = &v
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: err.Error(), error: err.Error(),
} }
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
return localVarReturnValue, localVarHTTPResponse, nil return localVarReturnValue, nil
} }

107
internal/api/api_organizations.gen.go
View File

@ -11,8 +11,9 @@
package api package api
import ( import (
"bytes" _gzip "compress/gzip"
_context "context" _context "context"
_io "io"
_ioutil "io/ioutil" _ioutil "io/ioutil"
_nethttp "net/http" _nethttp "net/http"
_neturl "net/url" _neturl "net/url"
@ -36,7 +37,7 @@ type OrganizationsApi interface {
* GetOrgsExecute executes the request * GetOrgsExecute executes the request
* @return Organizations * @return Organizations
*/ */
GetOrgsExecute(r ApiGetOrgsRequest) (Organizations, *_nethttp.Response, error) GetOrgsExecute(r ApiGetOrgsRequest) (Organizations, error)
/* /*
* PostOrgs Create an organization * PostOrgs Create an organization
@ -49,7 +50,23 @@ type OrganizationsApi interface {
* PostOrgsExecute executes the request * PostOrgsExecute executes the request
* @return Organization * @return Organization
*/ */
PostOrgsExecute(r ApiPostOrgsRequest) (Organization, *_nethttp.Response, error) PostOrgsExecute(r ApiPostOrgsRequest) (Organization, error)
}
// organizationsApiGzipReadCloser supports streaming gzip response-bodies directly from the server.
type organizationsApiGzipReadCloser struct {
underlying _io.ReadCloser
gzip _io.ReadCloser
}
func (gzrc *organizationsApiGzipReadCloser) Read(p []byte) (int, error) {
return gzrc.gzip.Read(p)
}
func (gzrc *organizationsApiGzipReadCloser) Close() error {
if err := gzrc.gzip.Close(); err != nil {
return err
}
return gzrc.underlying.Close()
} }
// OrganizationsApiService OrganizationsApi service // OrganizationsApiService OrganizationsApi service
@ -123,7 +140,7 @@ func (r ApiGetOrgsRequest) GetUserID() *string {
return r.userID return r.userID
} }
func (r ApiGetOrgsRequest) Execute() (Organizations, *_nethttp.Response, error) { func (r ApiGetOrgsRequest) Execute() (Organizations, error) {
return r.ApiService.GetOrgsExecute(r) return r.ApiService.GetOrgsExecute(r)
} }
@ -143,7 +160,7 @@ func (a *OrganizationsApiService) GetOrgs(ctx _context.Context) ApiGetOrgsReques
* Execute executes the request * Execute executes the request
* @return Organizations * @return Organizations
*/ */
func (a *OrganizationsApiService) GetOrgsExecute(r ApiGetOrgsRequest) (Organizations, *_nethttp.Response, error) { func (a *OrganizationsApiService) GetOrgsExecute(r ApiGetOrgsRequest) (Organizations, error) {
var ( var (
localVarHTTPMethod = _nethttp.MethodGet localVarHTTPMethod = _nethttp.MethodGet
localVarPostBody interface{} localVarPostBody interface{}
@ -155,7 +172,7 @@ func (a *OrganizationsApiService) GetOrgsExecute(r ApiGetOrgsRequest) (Organizat
localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "OrganizationsApiService.GetOrgs") localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "OrganizationsApiService.GetOrgs")
if err != nil { if err != nil {
return localVarReturnValue, nil, GenericOpenAPIError{error: err.Error()} return localVarReturnValue, GenericOpenAPIError{error: err.Error()}
} }
localVarPath := localBasePath + "/orgs" localVarPath := localBasePath + "/orgs"
@ -204,22 +221,30 @@ func (a *OrganizationsApiService) GetOrgsExecute(r ApiGetOrgsRequest) (Organizat
} }
req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes) req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes)
if err != nil { if err != nil {
return localVarReturnValue, nil, err return localVarReturnValue, err
} }
localVarHTTPResponse, err := a.client.callAPI(req) localVarHTTPResponse, err := a.client.callAPI(req)
if err != nil || localVarHTTPResponse == nil { if err != nil || localVarHTTPResponse == nil {
return localVarReturnValue, localVarHTTPResponse, err return localVarReturnValue, err
} }
localVarBody, err := _ioutil.ReadAll(localVarHTTPResponse.Body) var body _io.ReadCloser = localVarHTTPResponse.Body
localVarHTTPResponse.Body.Close() if localVarHTTPResponse.Header.Get("Content-Encoding") == "gzip" {
localVarHTTPResponse.Body = _ioutil.NopCloser(bytes.NewBuffer(localVarBody)) gzr, err := _gzip.NewReader(body)
if err != nil { if err != nil {
return localVarReturnValue, localVarHTTPResponse, err body.Close()
return localVarReturnValue, err
}
body = &organizationsApiGzipReadCloser{underlying: body, gzip: gzr}
} }
if localVarHTTPResponse.StatusCode >= 300 { if localVarHTTPResponse.StatusCode >= 300 {
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: localVarHTTPResponse.Status, error: localVarHTTPResponse.Status,
@ -228,22 +253,27 @@ func (a *OrganizationsApiService) GetOrgsExecute(r ApiGetOrgsRequest) (Organizat
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
newErr.model = &v newErr.model = &v
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: err.Error(), error: err.Error(),
} }
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
return localVarReturnValue, localVarHTTPResponse, nil return localVarReturnValue, nil
} }
type ApiPostOrgsRequest struct { type ApiPostOrgsRequest struct {
@ -269,7 +299,7 @@ func (r ApiPostOrgsRequest) GetZapTraceSpan() *string {
return r.zapTraceSpan return r.zapTraceSpan
} }
func (r ApiPostOrgsRequest) Execute() (Organization, *_nethttp.Response, error) { func (r ApiPostOrgsRequest) Execute() (Organization, error) {
return r.ApiService.PostOrgsExecute(r) return r.ApiService.PostOrgsExecute(r)
} }
@ -289,7 +319,7 @@ func (a *OrganizationsApiService) PostOrgs(ctx _context.Context) ApiPostOrgsRequ
* Execute executes the request * Execute executes the request
* @return Organization * @return Organization
*/ */
func (a *OrganizationsApiService) PostOrgsExecute(r ApiPostOrgsRequest) (Organization, *_nethttp.Response, error) { func (a *OrganizationsApiService) PostOrgsExecute(r ApiPostOrgsRequest) (Organization, error) {
var ( var (
localVarHTTPMethod = _nethttp.MethodPost localVarHTTPMethod = _nethttp.MethodPost
localVarPostBody interface{} localVarPostBody interface{}
@ -301,7 +331,7 @@ func (a *OrganizationsApiService) PostOrgsExecute(r ApiPostOrgsRequest) (Organiz
localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "OrganizationsApiService.PostOrgs") localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "OrganizationsApiService.PostOrgs")
if err != nil { if err != nil {
return localVarReturnValue, nil, GenericOpenAPIError{error: err.Error()} return localVarReturnValue, GenericOpenAPIError{error: err.Error()}
} }
localVarPath := localBasePath + "/orgs" localVarPath := localBasePath + "/orgs"
@ -310,7 +340,7 @@ func (a *OrganizationsApiService) PostOrgsExecute(r ApiPostOrgsRequest) (Organiz
localVarQueryParams := _neturl.Values{} localVarQueryParams := _neturl.Values{}
localVarFormParams := _neturl.Values{} localVarFormParams := _neturl.Values{}
if r.organization == nil { if r.organization == nil {
return localVarReturnValue, nil, reportError("organization is required and must be specified") return localVarReturnValue, reportError("organization is required and must be specified")
} }
// to determine the Content-Type header // to determine the Content-Type header
@ -337,22 +367,30 @@ func (a *OrganizationsApiService) PostOrgsExecute(r ApiPostOrgsRequest) (Organiz
localVarPostBody = r.organization localVarPostBody = r.organization
req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes) req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes)
if err != nil { if err != nil {
return localVarReturnValue, nil, err return localVarReturnValue, err
} }
localVarHTTPResponse, err := a.client.callAPI(req) localVarHTTPResponse, err := a.client.callAPI(req)
if err != nil || localVarHTTPResponse == nil { if err != nil || localVarHTTPResponse == nil {
return localVarReturnValue, localVarHTTPResponse, err return localVarReturnValue, err
} }
localVarBody, err := _ioutil.ReadAll(localVarHTTPResponse.Body) var body _io.ReadCloser = localVarHTTPResponse.Body
localVarHTTPResponse.Body.Close() if localVarHTTPResponse.Header.Get("Content-Encoding") == "gzip" {
localVarHTTPResponse.Body = _ioutil.NopCloser(bytes.NewBuffer(localVarBody)) gzr, err := _gzip.NewReader(body)
if err != nil { if err != nil {
return localVarReturnValue, localVarHTTPResponse, err body.Close()
return localVarReturnValue, err
}
body = &organizationsApiGzipReadCloser{underlying: body, gzip: gzr}
} }
if localVarHTTPResponse.StatusCode >= 300 { if localVarHTTPResponse.StatusCode >= 300 {
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: localVarHTTPResponse.Status, error: localVarHTTPResponse.Status,
@ -361,20 +399,25 @@ func (a *OrganizationsApiService) PostOrgsExecute(r ApiPostOrgsRequest) (Organiz
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
newErr.model = &v newErr.model = &v
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: err.Error(), error: err.Error(),
} }
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
return localVarReturnValue, localVarHTTPResponse, nil return localVarReturnValue, nil
} }

105
internal/api/api_setup.gen.go
View File

@ -11,8 +11,9 @@
package api package api
import ( import (
"bytes" _gzip "compress/gzip"
_context "context" _context "context"
_io "io"
_ioutil "io/ioutil" _ioutil "io/ioutil"
_nethttp "net/http" _nethttp "net/http"
_neturl "net/url" _neturl "net/url"
@ -37,7 +38,7 @@ type SetupApi interface {
* GetSetupExecute executes the request * GetSetupExecute executes the request
* @return InlineResponse200 * @return InlineResponse200
*/ */
GetSetupExecute(r ApiGetSetupRequest) (InlineResponse200, *_nethttp.Response, error) GetSetupExecute(r ApiGetSetupRequest) (InlineResponse200, error)
/* /*
* PostSetup Set up initial user, org and bucket * PostSetup Set up initial user, org and bucket
@ -51,7 +52,23 @@ type SetupApi interface {
* PostSetupExecute executes the request * PostSetupExecute executes the request
* @return OnboardingResponse * @return OnboardingResponse
*/ */
PostSetupExecute(r ApiPostSetupRequest) (OnboardingResponse, *_nethttp.Response, error) PostSetupExecute(r ApiPostSetupRequest) (OnboardingResponse, error)
}
// setupApiGzipReadCloser supports streaming gzip response-bodies directly from the server.
type setupApiGzipReadCloser struct {
underlying _io.ReadCloser
gzip _io.ReadCloser
}
func (gzrc *setupApiGzipReadCloser) Read(p []byte) (int, error) {
return gzrc.gzip.Read(p)
}
func (gzrc *setupApiGzipReadCloser) Close() error {
if err := gzrc.gzip.Close(); err != nil {
return err
}
return gzrc.underlying.Close()
} }
// SetupApiService SetupApi service // SetupApiService SetupApi service
@ -71,7 +88,7 @@ func (r ApiGetSetupRequest) GetZapTraceSpan() *string {
return r.zapTraceSpan return r.zapTraceSpan
} }
func (r ApiGetSetupRequest) Execute() (InlineResponse200, *_nethttp.Response, error) { func (r ApiGetSetupRequest) Execute() (InlineResponse200, error) {
return r.ApiService.GetSetupExecute(r) return r.ApiService.GetSetupExecute(r)
} }
@ -92,7 +109,7 @@ func (a *SetupApiService) GetSetup(ctx _context.Context) ApiGetSetupRequest {
* Execute executes the request * Execute executes the request
* @return InlineResponse200 * @return InlineResponse200
*/ */
func (a *SetupApiService) GetSetupExecute(r ApiGetSetupRequest) (InlineResponse200, *_nethttp.Response, error) { func (a *SetupApiService) GetSetupExecute(r ApiGetSetupRequest) (InlineResponse200, error) {
var ( var (
localVarHTTPMethod = _nethttp.MethodGet localVarHTTPMethod = _nethttp.MethodGet
localVarPostBody interface{} localVarPostBody interface{}
@ -104,7 +121,7 @@ func (a *SetupApiService) GetSetupExecute(r ApiGetSetupRequest) (InlineResponse2
localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "SetupApiService.GetSetup") localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "SetupApiService.GetSetup")
if err != nil { if err != nil {
return localVarReturnValue, nil, GenericOpenAPIError{error: err.Error()} return localVarReturnValue, GenericOpenAPIError{error: err.Error()}
} }
localVarPath := localBasePath + "/setup" localVarPath := localBasePath + "/setup"
@ -135,39 +152,52 @@ func (a *SetupApiService) GetSetupExecute(r ApiGetSetupRequest) (InlineResponse2
} }
req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes) req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes)
if err != nil { if err != nil {
return localVarReturnValue, nil, err return localVarReturnValue, err
} }
localVarHTTPResponse, err := a.client.callAPI(req) localVarHTTPResponse, err := a.client.callAPI(req)
if err != nil || localVarHTTPResponse == nil { if err != nil || localVarHTTPResponse == nil {
return localVarReturnValue, localVarHTTPResponse, err return localVarReturnValue, err
} }
localVarBody, err := _ioutil.ReadAll(localVarHTTPResponse.Body) var body _io.ReadCloser = localVarHTTPResponse.Body
localVarHTTPResponse.Body.Close() if localVarHTTPResponse.Header.Get("Content-Encoding") == "gzip" {
localVarHTTPResponse.Body = _ioutil.NopCloser(bytes.NewBuffer(localVarBody)) gzr, err := _gzip.NewReader(body)
if err != nil { if err != nil {
return localVarReturnValue, localVarHTTPResponse, err body.Close()
return localVarReturnValue, err
}
body = &setupApiGzipReadCloser{underlying: body, gzip: gzr}
} }
if localVarHTTPResponse.StatusCode >= 300 { if localVarHTTPResponse.StatusCode >= 300 {
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: localVarHTTPResponse.Status, error: localVarHTTPResponse.Status,
} }
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: err.Error(), error: err.Error(),
} }
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
return localVarReturnValue, localVarHTTPResponse, nil return localVarReturnValue, nil
} }
type ApiPostSetupRequest struct { type ApiPostSetupRequest struct {
@ -193,7 +223,7 @@ func (r ApiPostSetupRequest) GetZapTraceSpan() *string {
return r.zapTraceSpan return r.zapTraceSpan
} }
func (r ApiPostSetupRequest) Execute() (OnboardingResponse, *_nethttp.Response, error) { func (r ApiPostSetupRequest) Execute() (OnboardingResponse, error) {
return r.ApiService.PostSetupExecute(r) return r.ApiService.PostSetupExecute(r)
} }
@ -214,7 +244,7 @@ func (a *SetupApiService) PostSetup(ctx _context.Context) ApiPostSetupRequest {
* Execute executes the request * Execute executes the request
* @return OnboardingResponse * @return OnboardingResponse
*/ */
func (a *SetupApiService) PostSetupExecute(r ApiPostSetupRequest) (OnboardingResponse, *_nethttp.Response, error) { func (a *SetupApiService) PostSetupExecute(r ApiPostSetupRequest) (OnboardingResponse, error) {
var ( var (
localVarHTTPMethod = _nethttp.MethodPost localVarHTTPMethod = _nethttp.MethodPost
localVarPostBody interface{} localVarPostBody interface{}
@ -226,7 +256,7 @@ func (a *SetupApiService) PostSetupExecute(r ApiPostSetupRequest) (OnboardingRes
localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "SetupApiService.PostSetup") localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "SetupApiService.PostSetup")
if err != nil { if err != nil {
return localVarReturnValue, nil, GenericOpenAPIError{error: err.Error()} return localVarReturnValue, GenericOpenAPIError{error: err.Error()}
} }
localVarPath := localBasePath + "/setup" localVarPath := localBasePath + "/setup"
@ -235,7 +265,7 @@ func (a *SetupApiService) PostSetupExecute(r ApiPostSetupRequest) (OnboardingRes
localVarQueryParams := _neturl.Values{} localVarQueryParams := _neturl.Values{}
localVarFormParams := _neturl.Values{} localVarFormParams := _neturl.Values{}
if r.onboardingRequest == nil { if r.onboardingRequest == nil {
return localVarReturnValue, nil, reportError("onboardingRequest is required and must be specified") return localVarReturnValue, reportError("onboardingRequest is required and must be specified")
} }
// to determine the Content-Type header // to determine the Content-Type header
@ -262,22 +292,30 @@ func (a *SetupApiService) PostSetupExecute(r ApiPostSetupRequest) (OnboardingRes
localVarPostBody = r.onboardingRequest localVarPostBody = r.onboardingRequest
req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes) req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes)
if err != nil { if err != nil {
return localVarReturnValue, nil, err return localVarReturnValue, err
} }
localVarHTTPResponse, err := a.client.callAPI(req) localVarHTTPResponse, err := a.client.callAPI(req)
if err != nil || localVarHTTPResponse == nil { if err != nil || localVarHTTPResponse == nil {
return localVarReturnValue, localVarHTTPResponse, err return localVarReturnValue, err
} }
localVarBody, err := _ioutil.ReadAll(localVarHTTPResponse.Body) var body _io.ReadCloser = localVarHTTPResponse.Body
localVarHTTPResponse.Body.Close() if localVarHTTPResponse.Header.Get("Content-Encoding") == "gzip" {
localVarHTTPResponse.Body = _ioutil.NopCloser(bytes.NewBuffer(localVarBody)) gzr, err := _gzip.NewReader(body)
if err != nil { if err != nil {
return localVarReturnValue, localVarHTTPResponse, err body.Close()
return localVarReturnValue, err
}
body = &setupApiGzipReadCloser{underlying: body, gzip: gzr}
} }
if localVarHTTPResponse.StatusCode >= 300 { if localVarHTTPResponse.StatusCode >= 300 {
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: localVarHTTPResponse.Status, error: localVarHTTPResponse.Status,
@ -286,20 +324,25 @@ func (a *SetupApiService) PostSetupExecute(r ApiPostSetupRequest) (OnboardingRes
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
newErr.model = &v newErr.model = &v
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return localVarReturnValue, err
}
err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: err.Error(), error: err.Error(),
} }
return localVarReturnValue, localVarHTTPResponse, newErr return localVarReturnValue, newErr
} }
return localVarReturnValue, localVarHTTPResponse, nil return localVarReturnValue, nil
} }

77
internal/api/api_write.gen.go
View File

@ -11,8 +11,9 @@
package api package api
import ( import (
"bytes" _gzip "compress/gzip"
_context "context" _context "context"
_io "io"
_ioutil "io/ioutil" _ioutil "io/ioutil"
_nethttp "net/http" _nethttp "net/http"
_neturl "net/url" _neturl "net/url"
@ -35,7 +36,23 @@ type WriteApi interface {
/* /*
* PostWriteExecute executes the request * PostWriteExecute executes the request
*/ */
PostWriteExecute(r ApiPostWriteRequest) (*_nethttp.Response, error) PostWriteExecute(r ApiPostWriteRequest) error
}
// writeApiGzipReadCloser supports streaming gzip response-bodies directly from the server.
type writeApiGzipReadCloser struct {
underlying _io.ReadCloser
gzip _io.ReadCloser
}
func (gzrc *writeApiGzipReadCloser) Read(p []byte) (int, error) {
return gzrc.gzip.Read(p)
}
func (gzrc *writeApiGzipReadCloser) Close() error {
if err := gzrc.gzip.Close(); err != nil {
return err
}
return gzrc.underlying.Close()
} }
// WriteApiService WriteApi service // WriteApiService WriteApi service
@ -136,7 +153,7 @@ func (r ApiPostWriteRequest) GetPrecision() *WritePrecision {
return r.precision return r.precision
} }
func (r ApiPostWriteRequest) Execute() (*_nethttp.Response, error) { func (r ApiPostWriteRequest) Execute() error {
return r.ApiService.PostWriteExecute(r) return r.ApiService.PostWriteExecute(r)
} }
@ -155,7 +172,7 @@ func (a *WriteApiService) PostWrite(ctx _context.Context) ApiPostWriteRequest {
/* /*
* Execute executes the request * Execute executes the request
*/ */
func (a *WriteApiService) PostWriteExecute(r ApiPostWriteRequest) (*_nethttp.Response, error) { func (a *WriteApiService) PostWriteExecute(r ApiPostWriteRequest) error {
var ( var (
localVarHTTPMethod = _nethttp.MethodPost localVarHTTPMethod = _nethttp.MethodPost
localVarPostBody interface{} localVarPostBody interface{}
@ -166,7 +183,7 @@ func (a *WriteApiService) PostWriteExecute(r ApiPostWriteRequest) (*_nethttp.Res
localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "WriteApiService.PostWrite") localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "WriteApiService.PostWrite")
if err != nil { if err != nil {
return nil, GenericOpenAPIError{error: err.Error()} return GenericOpenAPIError{error: err.Error()}
} }
localVarPath := localBasePath + "/write" localVarPath := localBasePath + "/write"
@ -175,13 +192,13 @@ func (a *WriteApiService) PostWriteExecute(r ApiPostWriteRequest) (*_nethttp.Res
localVarQueryParams := _neturl.Values{} localVarQueryParams := _neturl.Values{}
localVarFormParams := _neturl.Values{} localVarFormParams := _neturl.Values{}
if r.org == nil { if r.org == nil {
return nil, reportError("org is required and must be specified") return reportError("org is required and must be specified")
} }
if r.bucket == nil { if r.bucket == nil {
return nil, reportError("bucket is required and must be specified") return reportError("bucket is required and must be specified")
} }
if r.body == nil { if r.body == nil {
return nil, reportError("body is required and must be specified") return reportError("body is required and must be specified")
} }
localVarQueryParams.Add("org", parameterToString(*r.org, "")) localVarQueryParams.Add("org", parameterToString(*r.org, ""))
@ -228,22 +245,30 @@ func (a *WriteApiService) PostWriteExecute(r ApiPostWriteRequest) (*_nethttp.Res
localVarPostBody = r.body localVarPostBody = r.body
req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes) req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes)
if err != nil { if err != nil {
return nil, err return err
} }
localVarHTTPResponse, err := a.client.callAPI(req) localVarHTTPResponse, err := a.client.callAPI(req)
if err != nil || localVarHTTPResponse == nil { if err != nil || localVarHTTPResponse == nil {
return localVarHTTPResponse, err return err
} }
localVarBody, err := _ioutil.ReadAll(localVarHTTPResponse.Body) var body _io.ReadCloser = localVarHTTPResponse.Body
localVarHTTPResponse.Body.Close() if localVarHTTPResponse.Header.Get("Content-Encoding") == "gzip" {
localVarHTTPResponse.Body = _ioutil.NopCloser(bytes.NewBuffer(localVarBody)) gzr, err := _gzip.NewReader(body)
if err != nil { if err != nil {
return localVarHTTPResponse, err body.Close()
return err
}
body = &writeApiGzipReadCloser{underlying: body, gzip: gzr}
} }
if localVarHTTPResponse.StatusCode >= 300 { if localVarHTTPResponse.StatusCode >= 300 {
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return err
}
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: localVarHTTPResponse.Status, error: localVarHTTPResponse.Status,
@ -253,50 +278,50 @@ func (a *WriteApiService) PostWriteExecute(r ApiPostWriteRequest) (*_nethttp.Res
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarHTTPResponse, newErr return newErr
} }
newErr.model = &v newErr.model = &v
return localVarHTTPResponse, newErr return newErr
} }
if localVarHTTPResponse.StatusCode == 401 { if localVarHTTPResponse.StatusCode == 401 {
var v Error var v Error
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarHTTPResponse, newErr return newErr
} }
newErr.model = &v newErr.model = &v
return localVarHTTPResponse, newErr return newErr
} }
if localVarHTTPResponse.StatusCode == 403 { if localVarHTTPResponse.StatusCode == 403 {
var v Error var v Error
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarHTTPResponse, newErr return newErr
} }
newErr.model = &v newErr.model = &v
return localVarHTTPResponse, newErr return newErr
} }
if localVarHTTPResponse.StatusCode == 413 { if localVarHTTPResponse.StatusCode == 413 {
var v LineProtocolLengthError var v LineProtocolLengthError
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarHTTPResponse, newErr return newErr
} }
newErr.model = &v newErr.model = &v
return localVarHTTPResponse, newErr return newErr
} }
var v Error var v Error
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return localVarHTTPResponse, newErr return newErr
} }
newErr.model = &v newErr.model = &v
return localVarHTTPResponse, newErr return newErr
} }
return localVarHTTPResponse, nil return nil
} }

4
internal/api/templates/README.md
View File

@ -15,6 +15,10 @@ multiple locations.
* Add `GetX()` methods for each request parameter `X`, for use in unit tests * Add `GetX()` methods for each request parameter `X`, for use in unit tests
* Add checks for `isByteArray` to generate `[]byte` request fields instead of `*string` * Add checks for `isByteArray` to generate `[]byte` request fields instead of `*string`
* Update creation of `GenericOpenAPIError` to track sub-error models by reference * Update creation of `GenericOpenAPIError` to track sub-error models by reference
* Add checks for `isResponseBinary` to directly return the response-body-reader, instead of
pulling the entire body into memory and transforming it into an `*os.File`
* GUnzip response bodies when `Content-Encoding: gzip` is set
* Remove `*http.Response`s from the return values of generated operations
`client.mustache` `client.mustache`
* Removed use of `golang.org/x/oauth2` to avoid its heavy dependencies * Removed use of `golang.org/x/oauth2` to avoid its heavy dependencies

92
internal/api/templates/api.mustache
View File

@ -4,7 +4,9 @@ package {{packageName}}
{{#operations}} {{#operations}}
import ( import (
"bytes" "bytes"
_gzip "compress/gzip"
_context "context" _context "context"
_io "io"
_ioutil "io/ioutil" _ioutil "io/ioutil"
_nethttp "net/http" _nethttp "net/http"
_neturl "net/url" _neturl "net/url"
@ -36,11 +38,26 @@ type {{classname}} interface {
* {{nickname}}Execute executes the request{{#returnType}} * {{nickname}}Execute executes the request{{#returnType}}
* @return {{{.}}}{{/returnType}} * @return {{{.}}}{{/returnType}}
*/ */
{{nickname}}Execute(r {{#structPrefix}}{{&classname}}{{/structPrefix}}Api{{operationId}}Request) ({{#returnType}}{{{.}}}, {{/returnType}}*_nethttp.Response, error) {{nickname}}Execute(r {{#structPrefix}}{{&classname}}{{/structPrefix}}Api{{operationId}}Request) ({{#returnType}}{{#isResponseBinary}}_io.ReadCloser{{/isResponseBinary}}{{^isResponseBinary}}{{{returnType}}}{{/isResponseBinary}}, {{/returnType}}error)
{{/operation}} {{/operation}}
} }
{{/generateInterfaces}} {{/generateInterfaces}}
// {{#lambda.camelcase}}{{classname}}{{/lambda.camelcase}}GzipReadCloser supports streaming gzip response-bodies directly from the server.
type {{#lambda.camelcase}}{{classname}}{{/lambda.camelcase}}GzipReadCloser struct {
underlying _io.ReadCloser
gzip _io.ReadCloser
}
func (gzrc *{{#lambda.camelcase}}{{classname}}{{/lambda.camelcase}}GzipReadCloser) Read(p []byte) (int, error) {
return gzrc.gzip.Read(p)
}
func (gzrc *{{#lambda.camelcase}}{{classname}}{{/lambda.camelcase}}GzipReadCloser) Close() error {
if err := gzrc.gzip.Close(); err != nil {
return err
}
return gzrc.underlying.Close()
}
// {{classname}}Service {{classname}} service // {{classname}}Service {{classname}} service
type {{classname}}Service service type {{classname}}Service service
{{#operation}} {{#operation}}
@ -61,12 +78,12 @@ func (r {{#structPrefix}}{{&classname}}{{/structPrefix}}Api{{operationId}}Reques
return r return r
} }
func (r {{#structPrefix}}{{&classname}}{{/structPrefix}}Api{{operationId}}Request) Get{{vendorExtensions.x-export-param-name}}() {{#isByteArray}}[]byte{{/isByteArray}}{{^isByteArray}}{{^isPathParam}}*{{/isPathParam}}{{{dataType}}}{{/isByteArray}} { func (r {{#structPrefix}}{{&classname}}{{/structPrefix}}Api{{operationId}}Request) Get{{vendorExtensions.x-export-param-name}}() {{#isByteArray}}[]byte{{/isByteArray}}{{^isByteArray}}{{^isPathParam}}*{{/isPathParam}}{{{dataType}}}{{/isByteArray}} {
return r.{{paramName}} return r.{{paramName}}
} }
{{/allParams}} {{/allParams}}
func (r {{#structPrefix}}{{&classname}}{{/structPrefix}}Api{{operationId}}Request) Execute() ({{#returnType}}{{{.}}}, {{/returnType}}*_nethttp.Response, error) { func (r {{#structPrefix}}{{&classname}}{{/structPrefix}}Api{{operationId}}Request) Execute() ({{#returnType}}{{#isResponseBinary}}_io.ReadCloser{{/isResponseBinary}}{{^isResponseBinary}}{{{returnType}}}{{/isResponseBinary}}, {{/returnType}}error) {
return r.ApiService.{{nickname}}Execute(r) return r.ApiService.{{nickname}}Execute(r)
} }
@ -93,7 +110,7 @@ func (a *{{{classname}}}Service) {{{nickname}}}(ctx _context.Context{{#pathParam
* Execute executes the request{{#returnType}} * Execute executes the request{{#returnType}}
* @return {{{.}}}{{/returnType}} * @return {{{.}}}{{/returnType}}
*/ */
func (a *{{{classname}}}Service) {{nickname}}Execute(r {{#structPrefix}}{{&classname}}{{/structPrefix}}Api{{operationId}}Request) ({{#returnType}}{{{.}}}, {{/returnType}}*_nethttp.Response, error) { func (a *{{{classname}}}Service) {{nickname}}Execute(r {{#structPrefix}}{{&classname}}{{/structPrefix}}Api{{operationId}}Request) ({{#returnType}}{{#isResponseBinary}}_io.ReadCloser{{/isResponseBinary}}{{^isResponseBinary}}{{{.}}}{{/isResponseBinary}}, {{/returnType}}error) {
var ( var (
localVarHTTPMethod = _nethttp.Method{{httpMethod}} localVarHTTPMethod = _nethttp.Method{{httpMethod}}
localVarPostBody interface{} localVarPostBody interface{}
@ -101,13 +118,18 @@ func (a *{{{classname}}}Service) {{nickname}}Execute(r {{#structPrefix}}{{&class
localVarFileName string localVarFileName string
localVarFileBytes []byte localVarFileBytes []byte
{{#returnType}} {{#returnType}}
localVarReturnValue {{{.}}} {{#isResponseBinary}}
localVarReturnValue _io.ReadCloser
{{/isResponseBinary}}
{{^isResponseBinary}}
localVarReturnValue {{{returnType}}}
{{/isResponseBinary}}
{{/returnType}} {{/returnType}}
) )
localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "{{{classname}}}Service.{{{nickname}}}") localBasePath, err := a.client.cfg.ServerURLWithContext(r.ctx, "{{{classname}}}Service.{{{nickname}}}")
if err != nil { if err != nil {
return {{#returnType}}localVarReturnValue, {{/returnType}}nil, GenericOpenAPIError{error: err.Error()} return {{#returnType}}localVarReturnValue, {{/returnType}}GenericOpenAPIError{error: err.Error()}
} }
localVarPath := localBasePath + "{{{path}}}"{{#pathParams}} localVarPath := localBasePath + "{{{path}}}"{{#pathParams}}
@ -120,27 +142,27 @@ func (a *{{{classname}}}Service) {{nickname}}Execute(r {{#structPrefix}}{{&class
{{#required}} {{#required}}
{{^isPathParam}} {{^isPathParam}}
if r.{{paramName}} == nil { if r.{{paramName}} == nil {
return {{#returnType}}localVarReturnValue, {{/returnType}}nil, reportError("{{paramName}} is required and must be specified") return {{#returnType}}localVarReturnValue, {{/returnType}}reportError("{{paramName}} is required and must be specified")
} }
{{/isPathParam}} {{/isPathParam}}
{{#minItems}} {{#minItems}}
if len({{^isPathParam}}*{{/isPathParam}}r.{{paramName}}) < {{minItems}} { if len({{^isPathParam}}*{{/isPathParam}}r.{{paramName}}) < {{minItems}} {
return {{#returnType}}localVarReturnValue, {{/returnType}}nil, reportError("{{paramName}} must have at least {{minItems}} elements") return {{#returnType}}localVarReturnValue, {{/returnType}}reportError("{{paramName}} must have at least {{minItems}} elements")
} }
{{/minItems}} {{/minItems}}
{{#maxItems}} {{#maxItems}}
if len({{^isPathParam}}*{{/isPathParam}}r.{{paramName}}) > {{maxItems}} { if len({{^isPathParam}}*{{/isPathParam}}r.{{paramName}}) > {{maxItems}} {
return {{#returnType}}localVarReturnValue, {{/returnType}}nil, reportError("{{paramName}} must have less than {{maxItems}} elements") return {{#returnType}}localVarReturnValue, {{/returnType}}reportError("{{paramName}} must have less than {{maxItems}} elements")
} }
{{/maxItems}} {{/maxItems}}
{{#minLength}} {{#minLength}}
if strlen({{^isPathParam}}*{{/isPathParam}}r.{{paramName}}) < {{minLength}} { if strlen({{^isPathParam}}*{{/isPathParam}}r.{{paramName}}) < {{minLength}} {
return {{#returnType}}localVarReturnValue, {{/returnType}}nil, reportError("{{paramName}} must have at least {{minLength}} elements") return {{#returnType}}localVarReturnValue, {{/returnType}}reportError("{{paramName}} must have at least {{minLength}} elements")
} }
{{/minLength}} {{/minLength}}
{{#maxLength}} {{#maxLength}}
if strlen({{^isPathParam}}*{{/isPathParam}}r.{{paramName}}) > {{maxLength}} { if strlen({{^isPathParam}}*{{/isPathParam}}r.{{paramName}}) > {{maxLength}} {
return {{#returnType}}localVarReturnValue, {{/returnType}}nil, reportError("{{paramName}} must have less than {{maxLength}} elements") return {{#returnType}}localVarReturnValue, {{/returnType}}reportError("{{paramName}} must have less than {{maxLength}} elements")
} }
{{/maxLength}} {{/maxLength}}
{{#minimum}} {{#minimum}}
@ -151,7 +173,7 @@ func (a *{{{classname}}}Service) {{nickname}}Execute(r {{#structPrefix}}{{&class
{{^isString}} {{^isString}}
if {{^isPathParam}}*{{/isPathParam}}r.{{paramName}} < {{minimum}} { if {{^isPathParam}}*{{/isPathParam}}r.{{paramName}} < {{minimum}} {
{{/isString}} {{/isString}}
return {{#returnType}}localVarReturnValue, {{/returnType}}nil, reportError("{{paramName}} must be greater than {{minimum}}") return {{#returnType}}localVarReturnValue, {{/returnType}}reportError("{{paramName}} must be greater than {{minimum}}")
} }
{{/minimum}} {{/minimum}}
{{#maximum}} {{#maximum}}
@ -162,7 +184,7 @@ func (a *{{{classname}}}Service) {{nickname}}Execute(r {{#structPrefix}}{{&class
{{^isString}} {{^isString}}
if {{^isPathParam}}*{{/isPathParam}}r.{{paramName}} > {{maximum}} { if {{^isPathParam}}*{{/isPathParam}}r.{{paramName}} > {{maximum}} {
{{/isString}} {{/isString}}
return {{#returnType}}localVarReturnValue, {{/returnType}}nil, reportError("{{paramName}} must be less than {{maximum}}") return {{#returnType}}localVarReturnValue, {{/returnType}}reportError("{{paramName}} must be less than {{maximum}}")
} }
{{/maximum}} {{/maximum}}
{{/required}} {{/required}}
@ -265,7 +287,7 @@ func (a *{{{classname}}}Service) {{nickname}}Execute(r {{#structPrefix}}{{&class
if r.{{paramName}} != nil { if r.{{paramName}} != nil {
paramJson, err := parameterToJson(*r.{{paramName}}) paramJson, err := parameterToJson(*r.{{paramName}})
if err != nil { if err != nil {
return {{#returnType}}localVarReturnValue, {{/returnType}}nil, err return {{#returnType}}localVarReturnValue, {{/returnType}}err
} }
localVarFormParams.Add("{{baseName}}", paramJson) localVarFormParams.Add("{{baseName}}", paramJson)
} }
@ -320,22 +342,30 @@ func (a *{{{classname}}}Service) {{nickname}}Execute(r {{#structPrefix}}{{&class
{{/authMethods}} {{/authMethods}}
req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes) req, err := a.client.prepareRequest(r.ctx, localVarPath, localVarHTTPMethod, localVarPostBody, localVarHeaderParams, localVarQueryParams, localVarFormParams, localVarFormFileName, localVarFileName, localVarFileBytes)
if err != nil { if err != nil {
return {{#returnType}}localVarReturnValue, {{/returnType}}nil, err return {{#returnType}}localVarReturnValue, {{/returnType}}err
} }
localVarHTTPResponse, err := a.client.callAPI(req) localVarHTTPResponse, err := a.client.callAPI(req)
if err != nil || localVarHTTPResponse == nil { if err != nil || localVarHTTPResponse == nil {
return {{#returnType}}localVarReturnValue, {{/returnType}}localVarHTTPResponse, err return {{#returnType}}localVarReturnValue, {{/returnType}}err
} }
localVarBody, err := _ioutil.ReadAll(localVarHTTPResponse.Body) var body _io.ReadCloser = localVarHTTPResponse.Body
localVarHTTPResponse.Body.Close() if localVarHTTPResponse.Header.Get("Content-Encoding") == "gzip" {
localVarHTTPResponse.Body = _ioutil.NopCloser(bytes.NewBuffer(localVarBody)) gzr, err := _gzip.NewReader(body)
if err != nil { if err != nil {
return {{#returnType}}localVarReturnValue, {{/returnType}}localVarHTTPResponse, err body.Close()
return {{#returnType}}localVarReturnValue, {{/returnType}}err
}
body = &{{#lambda.camelcase}}{{classname}}{{/lambda.camelcase}}GzipReadCloser{underlying: body, gzip: gzr}
} }
if localVarHTTPResponse.StatusCode >= 300 { if localVarHTTPResponse.StatusCode >= 300 {
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return {{#returnType}}localVarReturnValue, {{/returnType}}err
}
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: localVarHTTPResponse.Status, error: localVarHTTPResponse.Status,
@ -351,11 +381,11 @@ func (a *{{{classname}}}Service) {{nickname}}Execute(r {{#structPrefix}}{{&class
err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&v, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr.error = err.Error() newErr.error = err.Error()
return {{#returnType}}localVarReturnValue, {{/returnType}}localVarHTTPResponse, newErr return {{#returnType}}localVarReturnValue, {{/returnType}}newErr
} }
newErr.model = &v newErr.model = &v
{{^-last}} {{^-last}}
return {{#returnType}}localVarReturnValue, {{/returnType}}localVarHTTPResponse, newErr return {{#returnType}}localVarReturnValue, {{/returnType}}newErr
{{/-last}} {{/-last}}
{{^wildcard}} {{^wildcard}}
} }
@ -364,21 +394,31 @@ func (a *{{{classname}}}Service) {{nickname}}Execute(r {{#structPrefix}}{{&class
{{/is1xx}} {{/is1xx}}
{{/dataType}} {{/dataType}}
{{/responses}} {{/responses}}
return {{#returnType}}localVarReturnValue, {{/returnType}}localVarHTTPResponse, newErr return {{#returnType}}localVarReturnValue, {{/returnType}}newErr
} }
{{#returnType}} {{#returnType}}
{{#isResponseBinary}}
localVarReturnValue = body
{{/isResponseBinary}}
{{^isResponseBinary}}
localVarBody, err := _ioutil.ReadAll(body)
body.Close()
if err != nil {
return {{#returnType}}localVarReturnValue, {{/returnType}}err
}
err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type")) err = a.client.decode(&localVarReturnValue, localVarBody, localVarHTTPResponse.Header.Get("Content-Type"))
if err != nil { if err != nil {
newErr := GenericOpenAPIError{ newErr := GenericOpenAPIError{
body: localVarBody, body: localVarBody,
error: err.Error(), error: err.Error(),
} }
return {{#returnType}}localVarReturnValue, {{/returnType}}localVarHTTPResponse, newErr return {{#returnType}}localVarReturnValue, {{/returnType}}newErr
} }
{{/isResponseBinary}}
{{/returnType}} {{/returnType}}
return {{#returnType}}localVarReturnValue, {{/returnType}}localVarHTTPResponse, nil return {{#returnType}}localVarReturnValue, {{/returnType}}nil
} }
{{/operation}} {{/operation}}
{{/operations}} {{/operations}}

12
internal/bucket.go
View File

@ -77,7 +77,7 @@ func (c *CLI) BucketsCreate(ctx context.Context, clients *BucketsClients, params
if name == "" { if name == "" {
name = c.ActiveConfig.Org name = c.ActiveConfig.Org
} }
resp, _, err := clients.OrgApi.GetOrgs(ctx).Org(name).Execute() resp, err := clients.OrgApi.GetOrgs(ctx).Org(name).Execute()
if err != nil { if err != nil {
return fmt.Errorf("failed to lookup ID of org %q: %w", name, err) return fmt.Errorf("failed to lookup ID of org %q: %w", name, err)
} }
@ -88,7 +88,7 @@ func (c *CLI) BucketsCreate(ctx context.Context, clients *BucketsClients, params
reqBody.OrgID = orgs[0].GetId() reqBody.OrgID = orgs[0].GetId()
} }
bucket, _, err := clients.BucketApi.PostBuckets(ctx).PostBucketRequest(reqBody).Execute() bucket, err := clients.BucketApi.PostBuckets(ctx).PostBucketRequest(reqBody).Execute()
if err != nil { if err != nil {
return fmt.Errorf("failed to create bucket: %w", err) return fmt.Errorf("failed to create bucket: %w", err)
} }
@ -125,7 +125,7 @@ func (c *CLI) BucketsList(ctx context.Context, client api.BucketsApi, params *Bu
req = req.Id(params.ID) req = req.Id(params.ID)
} }
buckets, _, err := req.Execute() buckets, err := req.Execute()
if err != nil { if err != nil {
return fmt.Errorf("failed to list buckets: %w", err) return fmt.Errorf("failed to list buckets: %w", err)
} }
@ -172,7 +172,7 @@ func (c *CLI) BucketsUpdate(ctx context.Context, client api.BucketsApi, params *
reqBody.SetRetentionRules([]api.PatchRetentionRule{*patchRule}) reqBody.SetRetentionRules([]api.PatchRetentionRule{*patchRule})
} }
bucket, _, err := client.PatchBucketsID(ctx, params.ID).PatchBucketRequest(reqBody).Execute() bucket, err := client.PatchBucketsID(ctx, params.ID).PatchBucketRequest(reqBody).Execute()
if err != nil { if err != nil {
return fmt.Errorf("failed to update bucket %q: %w", params.ID, err) return fmt.Errorf("failed to update bucket %q: %w", params.ID, err)
} }
@ -217,7 +217,7 @@ func (c *CLI) BucketsDelete(ctx context.Context, client api.BucketsApi, params *
displayId = params.Name displayId = params.Name
} }
resp, _, err := getReq.Execute() resp, err := getReq.Execute()
if err != nil { if err != nil {
return fmt.Errorf("failed to find bucket %q: %w", displayId, err) return fmt.Errorf("failed to find bucket %q: %w", displayId, err)
} }
@ -227,7 +227,7 @@ func (c *CLI) BucketsDelete(ctx context.Context, client api.BucketsApi, params *
} }
bucket = buckets[0] bucket = buckets[0]
if _, err := client.DeleteBucketsID(ctx, bucket.GetId()).Execute(); err != nil { if err := client.DeleteBucketsID(ctx, bucket.GetId()).Execute(); err != nil {
return fmt.Errorf("failed to delete bucket %q: %w", displayId, err) return fmt.Errorf("failed to delete bucket %q: %w", displayId, err)
} }

241
internal/bucket_test.go
View File

@ -3,7 +3,6 @@ package internal_test
import ( import (
"context" "context"
"errors" "errors"
"net/http"
"strings" "strings"
"testing" "testing"
@ -22,8 +21,8 @@ func TestBucketsCreate(t *testing.T) {
name string name string
configOrgName string configOrgName string
params internal.BucketsCreateParams params internal.BucketsCreateParams
buildOrgLookupFn func(*testing.T) func(api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) buildOrgLookupFn func(*testing.T) func(api.ApiGetOrgsRequest) (api.Organizations, error)
buildBucketCreateFn func(*testing.T) func(api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) buildBucketCreateFn func(*testing.T) func(api.ApiPostBucketsRequest) (api.Bucket, error)
expectedStdoutPattern string expectedStdoutPattern string
expectedInErr string expectedInErr string
}{ }{
@ -33,13 +32,13 @@ func TestBucketsCreate(t *testing.T) {
OrgID: "123", OrgID: "123",
Name: "my-bucket", Name: "my-bucket",
}, },
buildOrgLookupFn: func(t *testing.T) func(api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) { buildOrgLookupFn: func(t *testing.T) func(api.ApiGetOrgsRequest) (api.Organizations, error) {
return func(api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) { return func(api.ApiGetOrgsRequest) (api.Organizations, error) {
return api.Organizations{}, nil, errors.New("unexpected org lookup call") return api.Organizations{}, errors.New("unexpected org lookup call")
} }
}, },
buildBucketCreateFn: func(t *testing.T) func(api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) { buildBucketCreateFn: func(t *testing.T) func(api.ApiPostBucketsRequest) (api.Bucket, error) {
return func(req api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) { return func(req api.ApiPostBucketsRequest) (api.Bucket, error) {
body := req.GetPostBucketRequest() body := req.GetPostBucketRequest()
require.NotNil(t, body) require.NotNil(t, body)
require.Equal(t, "123", body.OrgID) require.Equal(t, "123", body.OrgID)
@ -51,7 +50,7 @@ func TestBucketsCreate(t *testing.T) {
OrgID: api.PtrString("123"), OrgID: api.PtrString("123"),
Name: "my-bucket", Name: "my-bucket",
RetentionRules: nil, RetentionRules: nil,
}, nil, nil }, nil
} }
}, },
expectedStdoutPattern: "456\\s+my-bucket\\s+infinite\\s+n/a\\s+123", expectedStdoutPattern: "456\\s+my-bucket\\s+infinite\\s+n/a\\s+123",
@ -65,13 +64,13 @@ func TestBucketsCreate(t *testing.T) {
Retention: "24h", Retention: "24h",
ShardGroupDuration: "1h", ShardGroupDuration: "1h",
}, },
buildOrgLookupFn: func(t *testing.T) func(api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) { buildOrgLookupFn: func(t *testing.T) func(api.ApiGetOrgsRequest) (api.Organizations, error) {
return func(api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) { return func(api.ApiGetOrgsRequest) (api.Organizations, error) {
return api.Organizations{}, nil, errors.New("unexpected org lookup call") return api.Organizations{}, errors.New("unexpected org lookup call")
} }
}, },
buildBucketCreateFn: func(t *testing.T) func(api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) { buildBucketCreateFn: func(t *testing.T) func(api.ApiPostBucketsRequest) (api.Bucket, error) {
return func(req api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) { return func(req api.ApiPostBucketsRequest) (api.Bucket, error) {
body := req.GetPostBucketRequest() body := req.GetPostBucketRequest()
require.NotNil(t, body) require.NotNil(t, body)
require.Equal(t, "123", body.OrgID) require.Equal(t, "123", body.OrgID)
@ -86,7 +85,7 @@ func TestBucketsCreate(t *testing.T) {
OrgID: api.PtrString("123"), OrgID: api.PtrString("123"),
Name: "my-bucket", Name: "my-bucket",
RetentionRules: body.RetentionRules, RetentionRules: body.RetentionRules,
}, nil, nil }, nil
} }
}, },
expectedStdoutPattern: "456\\s+my-bucket\\s+24h0m0s\\s+1h0m0s\\s+123", expectedStdoutPattern: "456\\s+my-bucket\\s+24h0m0s\\s+1h0m0s\\s+123",
@ -98,13 +97,13 @@ func TestBucketsCreate(t *testing.T) {
Name: "my-bucket", Name: "my-bucket",
Retention: "24h", Retention: "24h",
}, },
buildOrgLookupFn: func(t *testing.T) func(api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) { buildOrgLookupFn: func(t *testing.T) func(api.ApiGetOrgsRequest) (api.Organizations, error) {
return func(api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) { return func(api.ApiGetOrgsRequest) (api.Organizations, error) {
return api.Organizations{}, nil, errors.New("unexpected org lookup call") return api.Organizations{}, errors.New("unexpected org lookup call")
} }
}, },
buildBucketCreateFn: func(t *testing.T) func(api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) { buildBucketCreateFn: func(t *testing.T) func(api.ApiPostBucketsRequest) (api.Bucket, error) {
return func(req api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) { return func(req api.ApiPostBucketsRequest) (api.Bucket, error) {
body := req.GetPostBucketRequest() body := req.GetPostBucketRequest()
require.NotNil(t, body) require.NotNil(t, body)
require.Equal(t, "123", body.OrgID) require.Equal(t, "123", body.OrgID)
@ -119,7 +118,7 @@ func TestBucketsCreate(t *testing.T) {
OrgID: api.PtrString("123"), OrgID: api.PtrString("123"),
Name: "my-bucket", Name: "my-bucket",
RetentionRules: body.RetentionRules, RetentionRules: body.RetentionRules,
}, nil, nil }, nil
} }
}, },
}, },
@ -131,13 +130,13 @@ func TestBucketsCreate(t *testing.T) {
Retention: "24h", Retention: "24h",
SchemaType: api.SCHEMATYPE_EXPLICIT, SchemaType: api.SCHEMATYPE_EXPLICIT,
}, },
buildOrgLookupFn: func(t *testing.T) func(api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) { buildOrgLookupFn: func(t *testing.T) func(api.ApiGetOrgsRequest) (api.Organizations, error) {
return func(api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) { return func(api.ApiGetOrgsRequest) (api.Organizations, error) {
return api.Organizations{}, nil, errors.New("unexpected org lookup call") return api.Organizations{}, errors.New("unexpected org lookup call")
} }
}, },
buildBucketCreateFn: func(t *testing.T) func(api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) { buildBucketCreateFn: func(t *testing.T) func(api.ApiPostBucketsRequest) (api.Bucket, error) {
return func(req api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) { return func(req api.ApiPostBucketsRequest) (api.Bucket, error) {
body := req.GetPostBucketRequest() body := req.GetPostBucketRequest()
require.NotNil(t, body) require.NotNil(t, body)
require.Equal(t, "123", body.OrgID) require.Equal(t, "123", body.OrgID)
@ -150,7 +149,7 @@ func TestBucketsCreate(t *testing.T) {
Name: "my-bucket", Name: "my-bucket",
RetentionRules: body.RetentionRules, RetentionRules: body.RetentionRules,
SchemaType: api.SCHEMATYPE_EXPLICIT.Ptr(), SchemaType: api.SCHEMATYPE_EXPLICIT.Ptr(),
}, nil, nil }, nil
} }
}, },
expectedStdoutPattern: `456\s+my-bucket\s+24h0m0s\s+n/a\s+123\s+explicit`, expectedStdoutPattern: `456\s+my-bucket\s+24h0m0s\s+n/a\s+123\s+explicit`,
@ -164,16 +163,16 @@ func TestBucketsCreate(t *testing.T) {
Retention: "24h", Retention: "24h",
ShardGroupDuration: "1h", ShardGroupDuration: "1h",
}, },
buildOrgLookupFn: func(t *testing.T) func(api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) { buildOrgLookupFn: func(t *testing.T) func(api.ApiGetOrgsRequest) (api.Organizations, error) {
return func(req api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) { return func(req api.ApiGetOrgsRequest) (api.Organizations, error) {
require.Equal(t, "my-org", *req.GetOrg()) require.Equal(t, "my-org", *req.GetOrg())
return api.Organizations{ return api.Organizations{
Orgs: &[]api.Organization{{Id: api.PtrString("123")}}, Orgs: &[]api.Organization{{Id: api.PtrString("123")}},
}, nil, nil }, nil
} }
}, },
buildBucketCreateFn: func(t *testing.T) func(api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) { buildBucketCreateFn: func(t *testing.T) func(api.ApiPostBucketsRequest) (api.Bucket, error) {
return func(req api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) { return func(req api.ApiPostBucketsRequest) (api.Bucket, error) {
body := req.GetPostBucketRequest() body := req.GetPostBucketRequest()
require.NotNil(t, body) require.NotNil(t, body)
require.Equal(t, "123", body.OrgID) require.Equal(t, "123", body.OrgID)
@ -188,7 +187,7 @@ func TestBucketsCreate(t *testing.T) {
OrgID: api.PtrString("123"), OrgID: api.PtrString("123"),
Name: "my-bucket", Name: "my-bucket",
RetentionRules: body.RetentionRules, RetentionRules: body.RetentionRules,
}, nil, nil }, nil
} }
}, },
expectedStdoutPattern: "456\\s+my-bucket\\s+24h0m0s\\s+1h0m0s\\s+123", expectedStdoutPattern: "456\\s+my-bucket\\s+24h0m0s\\s+1h0m0s\\s+123",
@ -202,16 +201,16 @@ func TestBucketsCreate(t *testing.T) {
ShardGroupDuration: "1h", ShardGroupDuration: "1h",
}, },
configOrgName: "my-org", configOrgName: "my-org",
buildOrgLookupFn: func(t *testing.T) func(api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) { buildOrgLookupFn: func(t *testing.T) func(api.ApiGetOrgsRequest) (api.Organizations, error) {
return func(req api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) { return func(req api.ApiGetOrgsRequest) (api.Organizations, error) {
require.Equal(t, "my-org", *req.GetOrg()) require.Equal(t, "my-org", *req.GetOrg())
return api.Organizations{ return api.Organizations{
Orgs: &[]api.Organization{{Id: api.PtrString("123")}}, Orgs: &[]api.Organization{{Id: api.PtrString("123")}},
}, nil, nil }, nil
} }
}, },
buildBucketCreateFn: func(t *testing.T) func(api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) { buildBucketCreateFn: func(t *testing.T) func(api.ApiPostBucketsRequest) (api.Bucket, error) {
return func(req api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) { return func(req api.ApiPostBucketsRequest) (api.Bucket, error) {
body := req.GetPostBucketRequest() body := req.GetPostBucketRequest()
require.NotNil(t, body) require.NotNil(t, body)
require.Equal(t, "123", body.OrgID) require.Equal(t, "123", body.OrgID)
@ -226,7 +225,7 @@ func TestBucketsCreate(t *testing.T) {
OrgID: api.PtrString("123"), OrgID: api.PtrString("123"),
Name: "my-bucket", Name: "my-bucket",
RetentionRules: body.RetentionRules, RetentionRules: body.RetentionRules,
}, nil, nil }, nil
} }
}, },
expectedStdoutPattern: "456\\s+my-bucket\\s+24h0m0s\\s+1h0m0s\\s+123", expectedStdoutPattern: "456\\s+my-bucket\\s+24h0m0s\\s+1h0m0s\\s+123",
@ -239,14 +238,14 @@ func TestBucketsCreate(t *testing.T) {
Retention: "24h", Retention: "24h",
ShardGroupDuration: "1h", ShardGroupDuration: "1h",
}, },
buildOrgLookupFn: func(t *testing.T) func(api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) { buildOrgLookupFn: func(t *testing.T) func(api.ApiGetOrgsRequest) (api.Organizations, error) {
return func(req api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) { return func(req api.ApiGetOrgsRequest) (api.Organizations, error) {
return api.Organizations{}, nil, errors.New("shouldn't be called") return api.Organizations{}, errors.New("shouldn't be called")
} }
}, },
buildBucketCreateFn: func(t *testing.T) func(api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) { buildBucketCreateFn: func(t *testing.T) func(api.ApiPostBucketsRequest) (api.Bucket, error) {
return func(req api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) { return func(req api.ApiPostBucketsRequest) (api.Bucket, error) {
return api.Bucket{}, nil, errors.New("shouldn't be called") return api.Bucket{}, errors.New("shouldn't be called")
} }
}, },
expectedInErr: "must specify org ID or org name", expectedInErr: "must specify org ID or org name",
@ -260,14 +259,14 @@ func TestBucketsCreate(t *testing.T) {
Retention: "24h", Retention: "24h",
ShardGroupDuration: "1h", ShardGroupDuration: "1h",
}, },
buildOrgLookupFn: func(t *testing.T) func(api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) { buildOrgLookupFn: func(t *testing.T) func(api.ApiGetOrgsRequest) (api.Organizations, error) {
return func(req api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) { return func(req api.ApiGetOrgsRequest) (api.Organizations, error) {
return api.Organizations{}, nil, nil return api.Organizations{}, nil
} }
}, },
buildBucketCreateFn: func(t *testing.T) func(api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) { buildBucketCreateFn: func(t *testing.T) func(api.ApiPostBucketsRequest) (api.Bucket, error) {
return func(req api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) { return func(req api.ApiPostBucketsRequest) (api.Bucket, error) {
return api.Bucket{}, nil, errors.New("shouldn't be called") return api.Bucket{}, errors.New("shouldn't be called")
} }
}, },
expectedInErr: "no organization found", expectedInErr: "no organization found",
@ -316,7 +315,7 @@ func TestBucketsList(t *testing.T) {
name string name string
configOrgName string configOrgName string
params internal.BucketsListParams params internal.BucketsListParams
buildBucketLookupFn func(*testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) buildBucketLookupFn func(*testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, error)
expectedStdoutPatterns []string expectedStdoutPatterns []string
expectedInErr string expectedInErr string
}{ }{
@ -326,8 +325,8 @@ func TestBucketsList(t *testing.T) {
ID: "123", ID: "123",
}, },
configOrgName: "my-default-org", configOrgName: "my-default-org",
buildBucketLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { buildBucketLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, error) {
return func(req api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { return func(req api.ApiGetBucketsRequest) (api.Buckets, error) {
require.Equal(t, "123", *req.GetId()) require.Equal(t, "123", *req.GetId())
require.Equal(t, "my-default-org", *req.GetOrg()) require.Equal(t, "my-default-org", *req.GetOrg())
require.Nil(t, req.GetName()) require.Nil(t, req.GetName())
@ -343,7 +342,7 @@ func TestBucketsList(t *testing.T) {
}, },
}, },
}, },
}, nil, nil }, nil
} }
}, },
expectedStdoutPatterns: []string{ expectedStdoutPatterns: []string{
@ -356,8 +355,8 @@ func TestBucketsList(t *testing.T) {
Name: "my-bucket", Name: "my-bucket",
}, },
configOrgName: "my-default-org", configOrgName: "my-default-org",
buildBucketLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { buildBucketLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, error) {
return func(req api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { return func(req api.ApiGetBucketsRequest) (api.Buckets, error) {
require.Equal(t, "my-bucket", *req.GetName()) require.Equal(t, "my-bucket", *req.GetName())
require.Equal(t, "my-default-org", *req.GetOrg()) require.Equal(t, "my-default-org", *req.GetOrg())
require.Nil(t, req.GetId()) require.Nil(t, req.GetId())
@ -373,7 +372,7 @@ func TestBucketsList(t *testing.T) {
}, },
}, },
}, },
}, nil, nil }, nil
} }
}, },
expectedStdoutPatterns: []string{ expectedStdoutPatterns: []string{
@ -386,13 +385,13 @@ func TestBucketsList(t *testing.T) {
OrgID: "456", OrgID: "456",
}, },
configOrgName: "my-default-org", configOrgName: "my-default-org",
buildBucketLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { buildBucketLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, error) {
return func(req api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { return func(req api.ApiGetBucketsRequest) (api.Buckets, error) {
require.Equal(t, "456", *req.GetOrgID()) require.Equal(t, "456", *req.GetOrgID())
require.Nil(t, req.GetId()) require.Nil(t, req.GetId())
require.Nil(t, req.GetOrg()) require.Nil(t, req.GetOrg())
require.Nil(t, req.GetOrg()) require.Nil(t, req.GetOrg())
return api.Buckets{}, nil, nil return api.Buckets{}, nil
} }
}, },
}, },
@ -402,8 +401,8 @@ func TestBucketsList(t *testing.T) {
OrgName: "my-org", OrgName: "my-org",
}, },
configOrgName: "my-default-org", configOrgName: "my-default-org",
buildBucketLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { buildBucketLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, error) {
return func(req api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { return func(req api.ApiGetBucketsRequest) (api.Buckets, error) {
require.Equal(t, "my-org", *req.GetOrg()) require.Equal(t, "my-org", *req.GetOrg())
require.Nil(t, req.GetId()) require.Nil(t, req.GetId())
require.Nil(t, req.GetName()) require.Nil(t, req.GetName())
@ -427,7 +426,7 @@ func TestBucketsList(t *testing.T) {
}, },
}, },
}, },
}, nil, nil }, nil
} }
}, },
expectedStdoutPatterns: []string{ expectedStdoutPatterns: []string{
@ -441,8 +440,8 @@ func TestBucketsList(t *testing.T) {
OrgName: "my-org", OrgName: "my-org",
}, },
configOrgName: "my-default-org", configOrgName: "my-default-org",
buildBucketLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { buildBucketLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, error) {
return func(req api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { return func(req api.ApiGetBucketsRequest) (api.Buckets, error) {
require.Equal(t, "my-org", *req.GetOrg()) require.Equal(t, "my-org", *req.GetOrg())
require.Nil(t, req.GetId()) require.Nil(t, req.GetId())
require.Nil(t, req.GetName()) require.Nil(t, req.GetName())
@ -476,7 +475,7 @@ func TestBucketsList(t *testing.T) {
SchemaType: api.SCHEMATYPE_EXPLICIT.Ptr(), SchemaType: api.SCHEMATYPE_EXPLICIT.Ptr(),
}, },
}, },
}, nil, nil }, nil
} }
}, },
expectedStdoutPatterns: []string{ expectedStdoutPatterns: []string{
@ -488,9 +487,9 @@ func TestBucketsList(t *testing.T) {
{ {
name: "no org specified", name: "no org specified",
expectedInErr: "must specify org ID or org name", expectedInErr: "must specify org ID or org name",
buildBucketLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { buildBucketLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, error) {
return func(api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { return func(api.ApiGetBucketsRequest) (api.Buckets, error) {
return api.Buckets{}, nil, errors.New("shouldn't be called") return api.Buckets{}, errors.New("shouldn't be called")
} }
}, },
}, },
@ -534,7 +533,7 @@ func TestBucketsUpdate(t *testing.T) {
var testCases = []struct { var testCases = []struct {
name string name string
params internal.BucketsUpdateParams params internal.BucketsUpdateParams
buildBucketUpdateFn func(*testing.T) func(api.ApiPatchBucketsIDRequest) (api.Bucket, *http.Response, error) buildBucketUpdateFn func(*testing.T) func(api.ApiPatchBucketsIDRequest) (api.Bucket, error)
expectedStdoutPattern string expectedStdoutPattern string
}{ }{
{ {
@ -543,8 +542,8 @@ func TestBucketsUpdate(t *testing.T) {
ID: "123", ID: "123",
Name: "cold-storage", Name: "cold-storage",
}, },
buildBucketUpdateFn: func(t *testing.T) func(api.ApiPatchBucketsIDRequest) (api.Bucket, *http.Response, error) { buildBucketUpdateFn: func(t *testing.T) func(api.ApiPatchBucketsIDRequest) (api.Bucket, error) {
return func(req api.ApiPatchBucketsIDRequest) (api.Bucket, *http.Response, error) { return func(req api.ApiPatchBucketsIDRequest) (api.Bucket, error) {
require.Equal(t, "123", req.GetBucketID()) require.Equal(t, "123", req.GetBucketID())
body := req.GetPatchBucketRequest() body := req.GetPatchBucketRequest()
require.Equal(t, "cold-storage", body.GetName()) require.Equal(t, "cold-storage", body.GetName())
@ -555,7 +554,7 @@ func TestBucketsUpdate(t *testing.T) {
Id: api.PtrString("123"), Id: api.PtrString("123"),
Name: "cold-storage", Name: "cold-storage",
OrgID: api.PtrString("456"), OrgID: api.PtrString("456"),
}, nil, nil }, nil
} }
}, },
expectedStdoutPattern: "123\\s+cold-storage\\s+infinite\\s+n/a\\s+456", expectedStdoutPattern: "123\\s+cold-storage\\s+infinite\\s+n/a\\s+456",
@ -566,8 +565,8 @@ func TestBucketsUpdate(t *testing.T) {
ID: "123", ID: "123",
Description: "a very useful description", Description: "a very useful description",
}, },
buildBucketUpdateFn: func(t *testing.T) func(api.ApiPatchBucketsIDRequest) (api.Bucket, *http.Response, error) { buildBucketUpdateFn: func(t *testing.T) func(api.ApiPatchBucketsIDRequest) (api.Bucket, error) {
return func(req api.ApiPatchBucketsIDRequest) (api.Bucket, *http.Response, error) { return func(req api.ApiPatchBucketsIDRequest) (api.Bucket, error) {
require.Equal(t, "123", req.GetBucketID()) require.Equal(t, "123", req.GetBucketID())
body := req.GetPatchBucketRequest() body := req.GetPatchBucketRequest()
require.Equal(t, "a very useful description", body.GetDescription()) require.Equal(t, "a very useful description", body.GetDescription())
@ -579,7 +578,7 @@ func TestBucketsUpdate(t *testing.T) {
Name: "my-bucket", Name: "my-bucket",
Description: api.PtrString("a very useful description"), Description: api.PtrString("a very useful description"),
OrgID: api.PtrString("456"), OrgID: api.PtrString("456"),
}, nil, nil }, nil
} }
}, },
expectedStdoutPattern: "123\\s+my-bucket\\s+infinite\\s+n/a\\s+456", expectedStdoutPattern: "123\\s+my-bucket\\s+infinite\\s+n/a\\s+456",
@ -590,8 +589,8 @@ func TestBucketsUpdate(t *testing.T) {
ID: "123", ID: "123",
Retention: "3w", Retention: "3w",
}, },
buildBucketUpdateFn: func(t *testing.T) func(api.ApiPatchBucketsIDRequest) (api.Bucket, *http.Response, error) { buildBucketUpdateFn: func(t *testing.T) func(api.ApiPatchBucketsIDRequest) (api.Bucket, error) {
return func(req api.ApiPatchBucketsIDRequest) (api.Bucket, *http.Response, error) { return func(req api.ApiPatchBucketsIDRequest) (api.Bucket, error) {
require.Equal(t, "123", req.GetBucketID()) require.Equal(t, "123", req.GetBucketID())
body := req.GetPatchBucketRequest() body := req.GetPatchBucketRequest()
require.Len(t, body.GetRetentionRules(), 1) require.Len(t, body.GetRetentionRules(), 1)
@ -608,7 +607,7 @@ func TestBucketsUpdate(t *testing.T) {
RetentionRules: []api.RetentionRule{ RetentionRules: []api.RetentionRule{
{EverySeconds: rule.GetEverySeconds()}, {EverySeconds: rule.GetEverySeconds()},
}, },
}, nil, nil }, nil
} }
}, },
expectedStdoutPattern: "123\\s+my-bucket\\s+504h0m0s\\s+n/a\\s+456", expectedStdoutPattern: "123\\s+my-bucket\\s+504h0m0s\\s+n/a\\s+456",
@ -619,8 +618,8 @@ func TestBucketsUpdate(t *testing.T) {
ID: "123", ID: "123",
ShardGroupDuration: "10h30m", ShardGroupDuration: "10h30m",
}, },
buildBucketUpdateFn: func(t *testing.T) func(api.ApiPatchBucketsIDRequest) (api.Bucket, *http.Response, error) { buildBucketUpdateFn: func(t *testing.T) func(api.ApiPatchBucketsIDRequest) (api.Bucket, error) {
return func(req api.ApiPatchBucketsIDRequest) (api.Bucket, *http.Response, error) { return func(req api.ApiPatchBucketsIDRequest) (api.Bucket, error) {
require.Equal(t, "123", req.GetBucketID()) require.Equal(t, "123", req.GetBucketID())
body := req.GetPatchBucketRequest() body := req.GetPatchBucketRequest()
require.Len(t, body.GetRetentionRules(), 1) require.Len(t, body.GetRetentionRules(), 1)
@ -637,7 +636,7 @@ func TestBucketsUpdate(t *testing.T) {
RetentionRules: []api.RetentionRule{ RetentionRules: []api.RetentionRule{
{ShardGroupDurationSeconds: rule.ShardGroupDurationSeconds}, {ShardGroupDurationSeconds: rule.ShardGroupDurationSeconds},
}, },
}, nil, nil }, nil
} }
}, },
expectedStdoutPattern: "123\\s+my-bucket\\s+infinite\\s+10h30m0s\\s+456", expectedStdoutPattern: "123\\s+my-bucket\\s+infinite\\s+10h30m0s\\s+456",
@ -675,8 +674,8 @@ func TestBucketsDelete(t *testing.T) {
name string name string
configOrgName string configOrgName string
params internal.BucketsDeleteParams params internal.BucketsDeleteParams
buildBucketsLookupFn func(*testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) buildBucketsLookupFn func(*testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, error)
buildBucketDeleteFn func(*testing.T) func(api.ApiDeleteBucketsIDRequest) (*http.Response, error) buildBucketDeleteFn func(*testing.T) func(api.ApiDeleteBucketsIDRequest) error
expectedStdoutPattern string expectedStdoutPattern string
expectedInErr string expectedInErr string
}{ }{
@ -686,8 +685,8 @@ func TestBucketsDelete(t *testing.T) {
params: internal.BucketsDeleteParams{ params: internal.BucketsDeleteParams{
ID: "123", ID: "123",
}, },
buildBucketsLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { buildBucketsLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, error) {
return func(req api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { return func(req api.ApiGetBucketsRequest) (api.Buckets, error) {
require.Equal(t, "123", *req.GetId()) require.Equal(t, "123", *req.GetId())
require.Nil(t, req.GetName()) require.Nil(t, req.GetName())
require.Nil(t, req.GetOrgID()) require.Nil(t, req.GetOrgID())
@ -704,13 +703,13 @@ func TestBucketsDelete(t *testing.T) {
}, },
}, },
}, },
}, nil, nil }, nil
} }
}, },
buildBucketDeleteFn: func(t *testing.T) func(api.ApiDeleteBucketsIDRequest) (*http.Response, error) { buildBucketDeleteFn: func(t *testing.T) func(api.ApiDeleteBucketsIDRequest) error {
return func(req api.ApiDeleteBucketsIDRequest) (*http.Response, error) { return func(req api.ApiDeleteBucketsIDRequest) error {
assert.Equal(t, "123", req.GetBucketID()) assert.Equal(t, "123", req.GetBucketID())
return nil, nil return nil
} }
}, },
expectedStdoutPattern: "123\\s+my-bucket\\s+1h0m0s\\s+n/a\\s+456\\s+implicit", expectedStdoutPattern: "123\\s+my-bucket\\s+1h0m0s\\s+n/a\\s+456\\s+implicit",
@ -722,8 +721,8 @@ func TestBucketsDelete(t *testing.T) {
Name: "my-bucket", Name: "my-bucket",
OrgID: "456", OrgID: "456",
}, },
buildBucketsLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { buildBucketsLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, error) {
return func(req api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { return func(req api.ApiGetBucketsRequest) (api.Buckets, error) {
require.Equal(t, "my-bucket", *req.GetName()) require.Equal(t, "my-bucket", *req.GetName())
require.Equal(t, "456", *req.GetOrgID()) require.Equal(t, "456", *req.GetOrgID())
require.Nil(t, req.GetId()) require.Nil(t, req.GetId())
@ -740,13 +739,13 @@ func TestBucketsDelete(t *testing.T) {
}, },
}, },
}, },
}, nil, nil }, nil
} }
}, },
buildBucketDeleteFn: func(t *testing.T) func(api.ApiDeleteBucketsIDRequest) (*http.Response, error) { buildBucketDeleteFn: func(t *testing.T) func(api.ApiDeleteBucketsIDRequest) error {
return func(req api.ApiDeleteBucketsIDRequest) (*http.Response, error) { return func(req api.ApiDeleteBucketsIDRequest) error {
assert.Equal(t, "123", req.GetBucketID()) assert.Equal(t, "123", req.GetBucketID())
return nil, nil return nil
} }
}, },
expectedStdoutPattern: "123\\s+my-bucket\\s+1h0m0s\\s+n/a\\s+456\\s+implicit", expectedStdoutPattern: "123\\s+my-bucket\\s+1h0m0s\\s+n/a\\s+456\\s+implicit",
@ -758,8 +757,8 @@ func TestBucketsDelete(t *testing.T) {
Name: "my-bucket", Name: "my-bucket",
OrgName: "my-org", OrgName: "my-org",
}, },
buildBucketsLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { buildBucketsLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, error) {
return func(req api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { return func(req api.ApiGetBucketsRequest) (api.Buckets, error) {
require.Equal(t, "my-bucket", *req.GetName()) require.Equal(t, "my-bucket", *req.GetName())
require.Equal(t, "my-org", *req.GetOrg()) require.Equal(t, "my-org", *req.GetOrg())
require.Nil(t, req.GetId()) require.Nil(t, req.GetId())
@ -776,13 +775,13 @@ func TestBucketsDelete(t *testing.T) {
}, },
}, },
}, },
}, nil, nil }, nil
} }
}, },
buildBucketDeleteFn: func(t *testing.T) func(api.ApiDeleteBucketsIDRequest) (*http.Response, error) { buildBucketDeleteFn: func(t *testing.T) func(api.ApiDeleteBucketsIDRequest) error {
return func(req api.ApiDeleteBucketsIDRequest) (*http.Response, error) { return func(req api.ApiDeleteBucketsIDRequest) error {
assert.Equal(t, "123", req.GetBucketID()) assert.Equal(t, "123", req.GetBucketID())
return nil, nil return nil
} }
}, },
expectedStdoutPattern: "123\\s+my-bucket\\s+1h0m0s\\s+n/a\\s+456\\s+implicit", expectedStdoutPattern: "123\\s+my-bucket\\s+1h0m0s\\s+n/a\\s+456\\s+implicit",
@ -793,8 +792,8 @@ func TestBucketsDelete(t *testing.T) {
params: internal.BucketsDeleteParams{ params: internal.BucketsDeleteParams{
Name: "my-bucket", Name: "my-bucket",
}, },
buildBucketsLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { buildBucketsLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, error) {
return func(req api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { return func(req api.ApiGetBucketsRequest) (api.Buckets, error) {
require.Equal(t, "my-bucket", *req.GetName()) require.Equal(t, "my-bucket", *req.GetName())
require.Equal(t, "my-default-org", *req.GetOrg()) require.Equal(t, "my-default-org", *req.GetOrg())
require.Nil(t, req.GetId()) require.Nil(t, req.GetId())
@ -811,13 +810,13 @@ func TestBucketsDelete(t *testing.T) {
}, },
}, },
}, },
}, nil, nil }, nil
} }
}, },
buildBucketDeleteFn: func(t *testing.T) func(api.ApiDeleteBucketsIDRequest) (*http.Response, error) { buildBucketDeleteFn: func(t *testing.T) func(api.ApiDeleteBucketsIDRequest) error {
return func(req api.ApiDeleteBucketsIDRequest) (*http.Response, error) { return func(req api.ApiDeleteBucketsIDRequest) error {
assert.Equal(t, "123", req.GetBucketID()) assert.Equal(t, "123", req.GetBucketID())
return nil, nil return nil
} }
}, },
expectedStdoutPattern: "123\\s+my-bucket\\s+1h0m0s\\s+n/a\\s+456\\s+implicit", expectedStdoutPattern: "123\\s+my-bucket\\s+1h0m0s\\s+n/a\\s+456\\s+implicit",
@ -827,14 +826,14 @@ func TestBucketsDelete(t *testing.T) {
params: internal.BucketsDeleteParams{ params: internal.BucketsDeleteParams{
Name: "my-bucket", Name: "my-bucket",
}, },
buildBucketsLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { buildBucketsLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, error) {
return func(api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { return func(api.ApiGetBucketsRequest) (api.Buckets, error) {
return api.Buckets{}, nil, errors.New("shouldn't be called") return api.Buckets{}, errors.New("shouldn't be called")
} }
}, },
buildBucketDeleteFn: func(t *testing.T) func(api.ApiDeleteBucketsIDRequest) (*http.Response, error) { buildBucketDeleteFn: func(t *testing.T) func(api.ApiDeleteBucketsIDRequest) error {
return func(api.ApiDeleteBucketsIDRequest) (*http.Response, error) { return func(api.ApiDeleteBucketsIDRequest) error {
return nil, errors.New("shouldn't be called") return errors.New("shouldn't be called")
} }
}, },
expectedInErr: "must specify org ID or org name", expectedInErr: "must specify org ID or org name",
@ -844,14 +843,14 @@ func TestBucketsDelete(t *testing.T) {
params: internal.BucketsDeleteParams{ params: internal.BucketsDeleteParams{
ID: "123", ID: "123",
}, },
buildBucketsLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { buildBucketsLookupFn: func(t *testing.T) func(api.ApiGetBucketsRequest) (api.Buckets, error) {
return func(api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { return func(api.ApiGetBucketsRequest) (api.Buckets, error) {
return api.Buckets{}, nil, nil return api.Buckets{}, nil
} }
}, },
buildBucketDeleteFn: func(t *testing.T) func(api.ApiDeleteBucketsIDRequest) (*http.Response, error) { buildBucketDeleteFn: func(t *testing.T) func(api.ApiDeleteBucketsIDRequest) error {
return func(api.ApiDeleteBucketsIDRequest) (*http.Response, error) { return func(api.ApiDeleteBucketsIDRequest) error {
return nil, errors.New("shouldn't be called") return errors.New("shouldn't be called")
} }
}, },
expectedInErr: "not found", expectedInErr: "not found",

10
internal/cmd/bucket_schema/client.go
View File

@ -24,7 +24,7 @@ type orgBucketID struct {
} }
func (c Client) resolveMeasurement(ctx context.Context, ids orgBucketID, name string) (string, error) { func (c Client) resolveMeasurement(ctx context.Context, ids orgBucketID, name string) (string, error) {
res, _, err := c.BucketSchemasApi. res, err := c.BucketSchemasApi.
GetMeasurementSchemas(ctx, ids.BucketID). GetMeasurementSchemas(ctx, ids.BucketID).
OrgID(ids.OrgID). OrgID(ids.OrgID).
Name(name). Name(name).
@ -62,7 +62,7 @@ func (c Client) resolveOrgBucketIds(ctx context.Context, params internal.OrgBuck
req = req.Org(c.CLI.ActiveConfig.Org) req = req.Org(c.CLI.ActiveConfig.Org)
} }
resp, _, err := req.Execute() resp, err := req.Execute()
if err != nil { if err != nil {
return nil, fmt.Errorf("failed to find bucket %q: %w", params.BucketName, err) return nil, fmt.Errorf("failed to find bucket %q: %w", params.BucketName, err)
} }
@ -120,7 +120,7 @@ func (c Client) Create(ctx context.Context, params CreateParams) error {
return err return err
} }
res, _, err := c.BucketSchemasApi. res, err := c.BucketSchemasApi.
CreateMeasurementSchema(ctx, ids.BucketID). CreateMeasurementSchema(ctx, ids.BucketID).
OrgID(ids.OrgID). OrgID(ids.OrgID).
MeasurementSchemaCreateRequest(api.MeasurementSchemaCreateRequest{ MeasurementSchemaCreateRequest(api.MeasurementSchemaCreateRequest{
@ -169,7 +169,7 @@ func (c Client) Update(ctx context.Context, params UpdateParams) error {
} }
} }
res, _, err := c.BucketSchemasApi. res, err := c.BucketSchemasApi.
UpdateMeasurementSchema(ctx, ids.BucketID, id). UpdateMeasurementSchema(ctx, ids.BucketID, id).
OrgID(ids.OrgID). OrgID(ids.OrgID).
MeasurementSchemaUpdateRequest(api.MeasurementSchemaUpdateRequest{ MeasurementSchemaUpdateRequest(api.MeasurementSchemaUpdateRequest{
@ -204,7 +204,7 @@ func (c Client) List(ctx context.Context, params ListParams) error {
req = req.Name(params.Name) req = req.Name(params.Name)
} }
res, _, err := req.Execute() res, err := req.Execute()
if err != nil { if err != nil {
return fmt.Errorf("failed to list measurement schemas: %w", err) return fmt.Errorf("failed to list measurement schemas: %w", err)
} }

14
internal/cmd/bucket_schema/client_test.go
View File

@ -112,7 +112,7 @@ func TestClient_Create(t *testing.T) {
GetBucketsExecute(tmock.MatchedBy(func(in api.ApiGetBucketsRequest) bool { GetBucketsExecute(tmock.MatchedBy(func(in api.ApiGetBucketsRequest) bool {
return cmp.Equal(in.GetOrg(), &a.params.OrgName) && cmp.Equal(in.GetName(), &a.params.BucketName) return cmp.Equal(in.GetOrg(), &a.params.OrgName) && cmp.Equal(in.GetName(), &a.params.BucketName)
})). })).
Return(api.Buckets{Buckets: &buckets}, nil, nil) Return(api.Buckets{Buckets: &buckets}, nil)
} }
} }
@ -150,7 +150,7 @@ func TestClient_Create(t *testing.T) {
Columns: a.cols, Columns: a.cols,
CreatedAt: createdAt, CreatedAt: createdAt,
UpdatedAt: createdAt, UpdatedAt: createdAt,
}, nil, nil) }, nil)
} }
} }
@ -354,7 +354,7 @@ func TestClient_Update(t *testing.T) {
return (in.GetOrg() != nil && *in.GetOrg() == a.params.OrgName) && return (in.GetOrg() != nil && *in.GetOrg() == a.params.OrgName) &&
(in.GetName() != nil && *in.GetName() == a.params.BucketName) (in.GetName() != nil && *in.GetName() == a.params.BucketName)
})). })).
Return(api.Buckets{Buckets: &buckets}, nil, nil) Return(api.Buckets{Buckets: &buckets}, nil)
} }
} }
@ -398,7 +398,7 @@ func TestClient_Update(t *testing.T) {
UpdatedAt: updatedAt, UpdatedAt: updatedAt,
}, },
}, },
}, nil, nil) }, nil)
} }
} }
@ -425,7 +425,7 @@ func TestClient_Update(t *testing.T) {
Columns: a.cols, Columns: a.cols,
CreatedAt: createdAt, CreatedAt: createdAt,
UpdatedAt: updatedAt, UpdatedAt: updatedAt,
}, nil, nil) }, nil)
} }
} }
@ -595,7 +595,7 @@ func TestClient_List(t *testing.T) {
return (in.GetOrg() != nil && *in.GetOrg() == a.params.OrgName) && return (in.GetOrg() != nil && *in.GetOrg() == a.params.OrgName) &&
(in.GetName() != nil && *in.GetName() == a.params.BucketName) (in.GetName() != nil && *in.GetName() == a.params.BucketName)
})). })).
Return(api.Buckets{Buckets: &buckets}, nil, nil) Return(api.Buckets{Buckets: &buckets}, nil)
} }
} }
@ -639,7 +639,7 @@ func TestClient_List(t *testing.T) {
UpdatedAt: updatedAt, UpdatedAt: updatedAt,
}, },
}, },
}, nil, nil) }, nil)
} }
} }

29
internal/mock/api_bucket_schemas.gen.go
View File

@ -6,7 +6,6 @@ package mock
import ( import (
context "context" context "context"
http "net/http"
reflect "reflect" reflect "reflect"
gomock "github.com/golang/mock/gomock" gomock "github.com/golang/mock/gomock"
@ -51,13 +50,12 @@ func (mr *MockBucketSchemasApiMockRecorder) CreateMeasurementSchema(arg0, arg1 i
} }
// CreateMeasurementSchemaExecute mocks base method. // CreateMeasurementSchemaExecute mocks base method.
func (m *MockBucketSchemasApi) CreateMeasurementSchemaExecute(arg0 api.ApiCreateMeasurementSchemaRequest) (api.MeasurementSchema, *http.Response, error) { func (m *MockBucketSchemasApi) CreateMeasurementSchemaExecute(arg0 api.ApiCreateMeasurementSchemaRequest) (api.MeasurementSchema, error) {
m.ctrl.T.Helper() m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "CreateMeasurementSchemaExecute", arg0) ret := m.ctrl.Call(m, "CreateMeasurementSchemaExecute", arg0)
ret0, _ := ret[0].(api.MeasurementSchema) ret0, _ := ret[0].(api.MeasurementSchema)
ret1, _ := ret[1].(*http.Response) ret1, _ := ret[1].(error)
ret2, _ := ret[2].(error) return ret0, ret1
return ret0, ret1, ret2
} }
// CreateMeasurementSchemaExecute indicates an expected call of CreateMeasurementSchemaExecute. // CreateMeasurementSchemaExecute indicates an expected call of CreateMeasurementSchemaExecute.
@ -81,13 +79,12 @@ func (mr *MockBucketSchemasApiMockRecorder) GetMeasurementSchema(arg0, arg1, arg
} }
// GetMeasurementSchemaExecute mocks base method. // GetMeasurementSchemaExecute mocks base method.
func (m *MockBucketSchemasApi) GetMeasurementSchemaExecute(arg0 api.ApiGetMeasurementSchemaRequest) (api.MeasurementSchema, *http.Response, error) { func (m *MockBucketSchemasApi) GetMeasurementSchemaExecute(arg0 api.ApiGetMeasurementSchemaRequest) (api.MeasurementSchema, error) {
m.ctrl.T.Helper() m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetMeasurementSchemaExecute", arg0) ret := m.ctrl.Call(m, "GetMeasurementSchemaExecute", arg0)
ret0, _ := ret[0].(api.MeasurementSchema) ret0, _ := ret[0].(api.MeasurementSchema)
ret1, _ := ret[1].(*http.Response) ret1, _ := ret[1].(error)
ret2, _ := ret[2].(error) return ret0, ret1
return ret0, ret1, ret2
} }
// GetMeasurementSchemaExecute indicates an expected call of GetMeasurementSchemaExecute. // GetMeasurementSchemaExecute indicates an expected call of GetMeasurementSchemaExecute.
@ -111,13 +108,12 @@ func (mr *MockBucketSchemasApiMockRecorder) GetMeasurementSchemas(arg0, arg1 int
} }
// GetMeasurementSchemasExecute mocks base method. // GetMeasurementSchemasExecute mocks base method.
func (m *MockBucketSchemasApi) GetMeasurementSchemasExecute(arg0 api.ApiGetMeasurementSchemasRequest) (api.MeasurementSchemaList, *http.Response, error) { func (m *MockBucketSchemasApi) GetMeasurementSchemasExecute(arg0 api.ApiGetMeasurementSchemasRequest) (api.MeasurementSchemaList, error) {
m.ctrl.T.Helper() m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetMeasurementSchemasExecute", arg0) ret := m.ctrl.Call(m, "GetMeasurementSchemasExecute", arg0)
ret0, _ := ret[0].(api.MeasurementSchemaList) ret0, _ := ret[0].(api.MeasurementSchemaList)
ret1, _ := ret[1].(*http.Response) ret1, _ := ret[1].(error)
ret2, _ := ret[2].(error) return ret0, ret1
return ret0, ret1, ret2
} }
// GetMeasurementSchemasExecute indicates an expected call of GetMeasurementSchemasExecute. // GetMeasurementSchemasExecute indicates an expected call of GetMeasurementSchemasExecute.
@ -141,13 +137,12 @@ func (mr *MockBucketSchemasApiMockRecorder) UpdateMeasurementSchema(arg0, arg1,
} }
// UpdateMeasurementSchemaExecute mocks base method. // UpdateMeasurementSchemaExecute mocks base method.
func (m *MockBucketSchemasApi) UpdateMeasurementSchemaExecute(arg0 api.ApiUpdateMeasurementSchemaRequest) (api.MeasurementSchema, *http.Response, error) { func (m *MockBucketSchemasApi) UpdateMeasurementSchemaExecute(arg0 api.ApiUpdateMeasurementSchemaRequest) (api.MeasurementSchema, error) {
m.ctrl.T.Helper() m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "UpdateMeasurementSchemaExecute", arg0) ret := m.ctrl.Call(m, "UpdateMeasurementSchemaExecute", arg0)
ret0, _ := ret[0].(api.MeasurementSchema) ret0, _ := ret[0].(api.MeasurementSchema)
ret1, _ := ret[1].(*http.Response) ret1, _ := ret[1].(error)
ret2, _ := ret[2].(error) return ret0, ret1
return ret0, ret1, ret2
} }
// UpdateMeasurementSchemaExecute indicates an expected call of UpdateMeasurementSchemaExecute. // UpdateMeasurementSchemaExecute indicates an expected call of UpdateMeasurementSchemaExecute.

36
internal/mock/api_buckets.gen.go
View File

@ -6,7 +6,6 @@ package mock
import ( import (
context "context" context "context"
http "net/http"
reflect "reflect" reflect "reflect"
gomock "github.com/golang/mock/gomock" gomock "github.com/golang/mock/gomock"
@ -51,12 +50,11 @@ func (mr *MockBucketsApiMockRecorder) DeleteBucketsID(arg0, arg1 interface{}) *g
} }
// DeleteBucketsIDExecute mocks base method. // DeleteBucketsIDExecute mocks base method.
func (m *MockBucketsApi) DeleteBucketsIDExecute(arg0 api.ApiDeleteBucketsIDRequest) (*http.Response, error) { func (m *MockBucketsApi) DeleteBucketsIDExecute(arg0 api.ApiDeleteBucketsIDRequest) error {
m.ctrl.T.Helper() m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "DeleteBucketsIDExecute", arg0) ret := m.ctrl.Call(m, "DeleteBucketsIDExecute", arg0)
ret0, _ := ret[0].(*http.Response) ret0, _ := ret[0].(error)
ret1, _ := ret[1].(error) return ret0
return ret0, ret1
} }
// DeleteBucketsIDExecute indicates an expected call of DeleteBucketsIDExecute. // DeleteBucketsIDExecute indicates an expected call of DeleteBucketsIDExecute.
@ -80,13 +78,12 @@ func (mr *MockBucketsApiMockRecorder) GetBuckets(arg0 interface{}) *gomock.Call
} }
// GetBucketsExecute mocks base method. // GetBucketsExecute mocks base method.
func (m *MockBucketsApi) GetBucketsExecute(arg0 api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { func (m *MockBucketsApi) GetBucketsExecute(arg0 api.ApiGetBucketsRequest) (api.Buckets, error) {
m.ctrl.T.Helper() m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetBucketsExecute", arg0) ret := m.ctrl.Call(m, "GetBucketsExecute", arg0)
ret0, _ := ret[0].(api.Buckets) ret0, _ := ret[0].(api.Buckets)
ret1, _ := ret[1].(*http.Response) ret1, _ := ret[1].(error)
ret2, _ := ret[2].(error) return ret0, ret1
return ret0, ret1, ret2
} }
// GetBucketsExecute indicates an expected call of GetBucketsExecute. // GetBucketsExecute indicates an expected call of GetBucketsExecute.
@ -110,13 +107,12 @@ func (mr *MockBucketsApiMockRecorder) GetBucketsID(arg0, arg1 interface{}) *gomo
} }
// GetBucketsIDExecute mocks base method. // GetBucketsIDExecute mocks base method.
func (m *MockBucketsApi) GetBucketsIDExecute(arg0 api.ApiGetBucketsIDRequest) (api.Bucket, *http.Response, error) { func (m *MockBucketsApi) GetBucketsIDExecute(arg0 api.ApiGetBucketsIDRequest) (api.Bucket, error) {
m.ctrl.T.Helper() m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetBucketsIDExecute", arg0) ret := m.ctrl.Call(m, "GetBucketsIDExecute", arg0)
ret0, _ := ret[0].(api.Bucket) ret0, _ := ret[0].(api.Bucket)
ret1, _ := ret[1].(*http.Response) ret1, _ := ret[1].(error)
ret2, _ := ret[2].(error) return ret0, ret1
return ret0, ret1, ret2
} }
// GetBucketsIDExecute indicates an expected call of GetBucketsIDExecute. // GetBucketsIDExecute indicates an expected call of GetBucketsIDExecute.
@ -140,13 +136,12 @@ func (mr *MockBucketsApiMockRecorder) PatchBucketsID(arg0, arg1 interface{}) *go
} }
// PatchBucketsIDExecute mocks base method. // PatchBucketsIDExecute mocks base method.
func (m *MockBucketsApi) PatchBucketsIDExecute(arg0 api.ApiPatchBucketsIDRequest) (api.Bucket, *http.Response, error) { func (m *MockBucketsApi) PatchBucketsIDExecute(arg0 api.ApiPatchBucketsIDRequest) (api.Bucket, error) {
m.ctrl.T.Helper() m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "PatchBucketsIDExecute", arg0) ret := m.ctrl.Call(m, "PatchBucketsIDExecute", arg0)
ret0, _ := ret[0].(api.Bucket) ret0, _ := ret[0].(api.Bucket)
ret1, _ := ret[1].(*http.Response) ret1, _ := ret[1].(error)
ret2, _ := ret[2].(error) return ret0, ret1
return ret0, ret1, ret2
} }
// PatchBucketsIDExecute indicates an expected call of PatchBucketsIDExecute. // PatchBucketsIDExecute indicates an expected call of PatchBucketsIDExecute.
@ -170,13 +165,12 @@ func (mr *MockBucketsApiMockRecorder) PostBuckets(arg0 interface{}) *gomock.Call
} }
// PostBucketsExecute mocks base method. // PostBucketsExecute mocks base method.
func (m *MockBucketsApi) PostBucketsExecute(arg0 api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) { func (m *MockBucketsApi) PostBucketsExecute(arg0 api.ApiPostBucketsRequest) (api.Bucket, error) {
m.ctrl.T.Helper() m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "PostBucketsExecute", arg0) ret := m.ctrl.Call(m, "PostBucketsExecute", arg0)
ret0, _ := ret[0].(api.Bucket) ret0, _ := ret[0].(api.Bucket)
ret1, _ := ret[1].(*http.Response) ret1, _ := ret[1].(error)
ret2, _ := ret[2].(error) return ret0, ret1
return ret0, ret1, ret2
} }
// PostBucketsExecute indicates an expected call of PostBucketsExecute. // PostBucketsExecute indicates an expected call of PostBucketsExecute.

21
internal/mock/api_buckets.go
View File

@ -2,7 +2,6 @@ package mock
import ( import (
"context" "context"
"net/http"
"github.com/influxdata/influx-cli/v2/internal/api" "github.com/influxdata/influx-cli/v2/internal/api"
) )
@ -10,18 +9,18 @@ import (
var _ api.BucketsApi = (*BucketsApi)(nil) var _ api.BucketsApi = (*BucketsApi)(nil)
type BucketsApi struct { type BucketsApi struct {
DeleteBucketsIDExecuteFn func(api.ApiDeleteBucketsIDRequest) (*http.Response, error) DeleteBucketsIDExecuteFn func(api.ApiDeleteBucketsIDRequest) error
GetBucketsExecuteFn func(api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) GetBucketsExecuteFn func(api.ApiGetBucketsRequest) (api.Buckets, error)
GetBucketsIDExecuteFn func(api.ApiGetBucketsIDRequest) (api.Bucket, *http.Response, error) GetBucketsIDExecuteFn func(api.ApiGetBucketsIDRequest) (api.Bucket, error)
PatchBucketsIDExecuteFn func(api.ApiPatchBucketsIDRequest) (api.Bucket, *http.Response, error) PatchBucketsIDExecuteFn func(api.ApiPatchBucketsIDRequest) (api.Bucket, error)
PostBucketsExecuteFn func(api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) PostBucketsExecuteFn func(api.ApiPostBucketsRequest) (api.Bucket, error)
} }
func (b *BucketsApi) DeleteBucketsID(_ context.Context, bucketID string) api.ApiDeleteBucketsIDRequest { func (b *BucketsApi) DeleteBucketsID(_ context.Context, bucketID string) api.ApiDeleteBucketsIDRequest {
return api.ApiDeleteBucketsIDRequest{ApiService: b}.BucketID(bucketID) return api.ApiDeleteBucketsIDRequest{ApiService: b}.BucketID(bucketID)
} }
func (b *BucketsApi) DeleteBucketsIDExecute(r api.ApiDeleteBucketsIDRequest) (*http.Response, error) { func (b *BucketsApi) DeleteBucketsIDExecute(r api.ApiDeleteBucketsIDRequest) error {
return b.DeleteBucketsIDExecuteFn(r) return b.DeleteBucketsIDExecuteFn(r)
} }
@ -29,7 +28,7 @@ func (b *BucketsApi) GetBuckets(context.Context) api.ApiGetBucketsRequest {
return api.ApiGetBucketsRequest{ApiService: b} return api.ApiGetBucketsRequest{ApiService: b}
} }
func (b *BucketsApi) GetBucketsExecute(r api.ApiGetBucketsRequest) (api.Buckets, *http.Response, error) { func (b *BucketsApi) GetBucketsExecute(r api.ApiGetBucketsRequest) (api.Buckets, error) {
return b.GetBucketsExecuteFn(r) return b.GetBucketsExecuteFn(r)
} }
@ -37,7 +36,7 @@ func (b *BucketsApi) GetBucketsID(_ context.Context, bucketID string) api.ApiGet
return api.ApiGetBucketsIDRequest{ApiService: b}.BucketID(bucketID) return api.ApiGetBucketsIDRequest{ApiService: b}.BucketID(bucketID)
} }
func (b *BucketsApi) GetBucketsIDExecute(r api.ApiGetBucketsIDRequest) (api.Bucket, *http.Response, error) { func (b *BucketsApi) GetBucketsIDExecute(r api.ApiGetBucketsIDRequest) (api.Bucket, error) {
return b.GetBucketsIDExecuteFn(r) return b.GetBucketsIDExecuteFn(r)
} }
@ -45,7 +44,7 @@ func (b *BucketsApi) PatchBucketsID(_ context.Context, bucketID string) api.ApiP
return api.ApiPatchBucketsIDRequest{ApiService: b}.BucketID(bucketID) return api.ApiPatchBucketsIDRequest{ApiService: b}.BucketID(bucketID)
} }
func (b *BucketsApi) PatchBucketsIDExecute(r api.ApiPatchBucketsIDRequest) (api.Bucket, *http.Response, error) { func (b *BucketsApi) PatchBucketsIDExecute(r api.ApiPatchBucketsIDRequest) (api.Bucket, error) {
return b.PatchBucketsIDExecuteFn(r) return b.PatchBucketsIDExecuteFn(r)
} }
@ -53,6 +52,6 @@ func (b *BucketsApi) PostBuckets(context.Context) api.ApiPostBucketsRequest {
return api.ApiPostBucketsRequest{ApiService: b} return api.ApiPostBucketsRequest{ApiService: b}
} }
func (b *BucketsApi) PostBucketsExecute(r api.ApiPostBucketsRequest) (api.Bucket, *http.Response, error) { func (b *BucketsApi) PostBucketsExecute(r api.ApiPostBucketsRequest) (api.Bucket, error) {
return b.PostBucketsExecuteFn(r) return b.PostBucketsExecuteFn(r)
} }

5
internal/mock/api_health.go
View File

@ -2,7 +2,6 @@ package mock
import ( import (
"context" "context"
"net/http"
"github.com/influxdata/influx-cli/v2/internal/api" "github.com/influxdata/influx-cli/v2/internal/api"
) )
@ -10,7 +9,7 @@ import (
var _ api.HealthApi = (*HealthApi)(nil) var _ api.HealthApi = (*HealthApi)(nil)
type HealthApi struct { type HealthApi struct {
GetHealthExecuteFn func(api.ApiGetHealthRequest) (api.HealthCheck, *http.Response, error) GetHealthExecuteFn func(api.ApiGetHealthRequest) (api.HealthCheck, error)
} }
func (p *HealthApi) GetHealth(context.Context) api.ApiGetHealthRequest { func (p *HealthApi) GetHealth(context.Context) api.ApiGetHealthRequest {
@ -19,6 +18,6 @@ func (p *HealthApi) GetHealth(context.Context) api.ApiGetHealthRequest {
} }
} }
func (p *HealthApi) GetHealthExecute(req api.ApiGetHealthRequest) (api.HealthCheck, *http.Response, error) { func (p *HealthApi) GetHealthExecute(req api.ApiGetHealthRequest) (api.HealthCheck, error) {
return p.GetHealthExecuteFn(req) return p.GetHealthExecuteFn(req)
} }

9
internal/mock/api_organizations.go
View File

@ -2,7 +2,6 @@ package mock
import ( import (
"context" "context"
"net/http"
"github.com/influxdata/influx-cli/v2/internal/api" "github.com/influxdata/influx-cli/v2/internal/api"
) )
@ -10,15 +9,15 @@ import (
var _ api.OrganizationsApi = (*OrganizationsApi)(nil) var _ api.OrganizationsApi = (*OrganizationsApi)(nil)
type OrganizationsApi struct { type OrganizationsApi struct {
GetOrgsExecuteFn func(api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) GetOrgsExecuteFn func(api.ApiGetOrgsRequest) (api.Organizations, error)
PostOrgsExecuteFn func(api.ApiPostOrgsRequest) (api.Organization, *http.Response, error) PostOrgsExecuteFn func(api.ApiPostOrgsRequest) (api.Organization, error)
} }
func (o *OrganizationsApi) GetOrgs(context.Context) api.ApiGetOrgsRequest { func (o *OrganizationsApi) GetOrgs(context.Context) api.ApiGetOrgsRequest {
return api.ApiGetOrgsRequest{ApiService: o} return api.ApiGetOrgsRequest{ApiService: o}
} }
func (o *OrganizationsApi) GetOrgsExecute(r api.ApiGetOrgsRequest) (api.Organizations, *http.Response, error) { func (o *OrganizationsApi) GetOrgsExecute(r api.ApiGetOrgsRequest) (api.Organizations, error) {
return o.GetOrgsExecuteFn(r) return o.GetOrgsExecuteFn(r)
} }
@ -26,6 +25,6 @@ func (o *OrganizationsApi) PostOrgs(context.Context) api.ApiPostOrgsRequest {
return api.ApiPostOrgsRequest{ApiService: o} return api.ApiPostOrgsRequest{ApiService: o}
} }
func (o *OrganizationsApi) PostOrgsExecute(r api.ApiPostOrgsRequest) (api.Organization, *http.Response, error) { func (o *OrganizationsApi) PostOrgsExecute(r api.ApiPostOrgsRequest) (api.Organization, error) {
return o.PostOrgsExecuteFn(r) return o.PostOrgsExecuteFn(r)
} }

9
internal/mock/api_setup.go
View File

@ -2,7 +2,6 @@ package mock
import ( import (
"context" "context"
"net/http"
"github.com/influxdata/influx-cli/v2/internal/api" "github.com/influxdata/influx-cli/v2/internal/api"
) )
@ -10,8 +9,8 @@ import (
var _ api.SetupApi = (*SetupApi)(nil) var _ api.SetupApi = (*SetupApi)(nil)
type SetupApi struct { type SetupApi struct {
GetSetupExecuteFn func(api.ApiGetSetupRequest) (api.InlineResponse200, *http.Response, error) GetSetupExecuteFn func(api.ApiGetSetupRequest) (api.InlineResponse200, error)
PostSetupExecuteFn func(api.ApiPostSetupRequest) (api.OnboardingResponse, *http.Response, error) PostSetupExecuteFn func(api.ApiPostSetupRequest) (api.OnboardingResponse, error)
} }
func (s *SetupApi) GetSetup(context.Context) api.ApiGetSetupRequest { func (s *SetupApi) GetSetup(context.Context) api.ApiGetSetupRequest {
@ -19,7 +18,7 @@ func (s *SetupApi) GetSetup(context.Context) api.ApiGetSetupRequest {
ApiService: s, ApiService: s,
} }
} }
func (s *SetupApi) GetSetupExecute(req api.ApiGetSetupRequest) (api.InlineResponse200, *http.Response, error) { func (s *SetupApi) GetSetupExecute(req api.ApiGetSetupRequest) (api.InlineResponse200, error) {
return s.GetSetupExecuteFn(req) return s.GetSetupExecuteFn(req)
} }
func (s *SetupApi) PostSetup(context.Context) api.ApiPostSetupRequest { func (s *SetupApi) PostSetup(context.Context) api.ApiPostSetupRequest {
@ -27,6 +26,6 @@ func (s *SetupApi) PostSetup(context.Context) api.ApiPostSetupRequest {
ApiService: s, ApiService: s,
} }
} }
func (s *SetupApi) PostSetupExecute(req api.ApiPostSetupRequest) (api.OnboardingResponse, *http.Response, error) { func (s *SetupApi) PostSetupExecute(req api.ApiPostSetupRequest) (api.OnboardingResponse, error) {
return s.PostSetupExecuteFn(req) return s.PostSetupExecuteFn(req)
} }

5
internal/mock/api_write.go
View File

@ -2,7 +2,6 @@ package mock
import ( import (
"context" "context"
"net/http"
"github.com/influxdata/influx-cli/v2/internal/api" "github.com/influxdata/influx-cli/v2/internal/api"
) )
@ -10,7 +9,7 @@ import (
var _ api.WriteApi = (*WriteApi)(nil) var _ api.WriteApi = (*WriteApi)(nil)
type WriteApi struct { type WriteApi struct {
PostWriteExecuteFn func(api.ApiPostWriteRequest) (*http.Response, error) PostWriteExecuteFn func(api.ApiPostWriteRequest) error
} }
func (w *WriteApi) PostWrite(context.Context) api.ApiPostWriteRequest { func (w *WriteApi) PostWrite(context.Context) api.ApiPostWriteRequest {
@ -18,6 +17,6 @@ func (w *WriteApi) PostWrite(context.Context) api.ApiPostWriteRequest {
ApiService: w, ApiService: w,
} }
} }
func (w *WriteApi) PostWriteExecute(req api.ApiPostWriteRequest) (*http.Response, error) { func (w *WriteApi) PostWriteExecute(req api.ApiPostWriteRequest) error {
return w.PostWriteExecuteFn(req) return w.PostWriteExecuteFn(req)
} }

2
internal/ping.go
View File

@ -8,7 +8,7 @@ import (
// Ping checks the health of a remote InfluxDB instance. // Ping checks the health of a remote InfluxDB instance.
func (c *CLI) Ping(ctx context.Context, client api.HealthApi) error { func (c *CLI) Ping(ctx context.Context, client api.HealthApi) error {
if _, _, err := client.GetHealth(ctx).Execute(); err != nil { if _, err := client.GetHealth(ctx).Execute(); err != nil {
return err return err
} }
_, err := c.StdIO.Write([]byte("OK\n")) _, err := c.StdIO.Write([]byte("OK\n"))

17
internal/ping_test.go
View File

@ -3,7 +3,6 @@ package internal_test
import ( import (
"context" "context"
"errors" "errors"
"net/http"
"testing" "testing"
"github.com/influxdata/influx-cli/v2/internal" "github.com/influxdata/influx-cli/v2/internal"
@ -16,8 +15,8 @@ func Test_PingSuccess(t *testing.T) {
t.Parallel() t.Parallel()
client := &mock.HealthApi{ client := &mock.HealthApi{
GetHealthExecuteFn: func(req api.ApiGetHealthRequest) (api.HealthCheck, *http.Response, error) { GetHealthExecuteFn: func(req api.ApiGetHealthRequest) (api.HealthCheck, error) {
return api.HealthCheck{Status: api.HEALTHCHECKSTATUS_PASS}, nil, nil return api.HealthCheck{Status: api.HEALTHCHECKSTATUS_PASS}, nil
}, },
} }
@ -33,8 +32,8 @@ func Test_PingFailedRequest(t *testing.T) {
e := "the internet is down" e := "the internet is down"
client := &mock.HealthApi{ client := &mock.HealthApi{
GetHealthExecuteFn: func(api.ApiGetHealthRequest) (api.HealthCheck, *http.Response, error) { GetHealthExecuteFn: func(api.ApiGetHealthRequest) (api.HealthCheck, error) {
return api.HealthCheck{}, nil, errors.New(e) return api.HealthCheck{}, errors.New(e)
}, },
} }
@ -49,8 +48,8 @@ func Test_PingFailedStatus(t *testing.T) {
e := "I broke" e := "I broke"
client := &mock.HealthApi{ client := &mock.HealthApi{
GetHealthExecuteFn: func(api.ApiGetHealthRequest) (api.HealthCheck, *http.Response, error) { GetHealthExecuteFn: func(api.ApiGetHealthRequest) (api.HealthCheck, error) {
return api.HealthCheck{}, nil, &api.HealthCheck{Status: api.HEALTHCHECKSTATUS_FAIL, Message: &e} return api.HealthCheck{}, &api.HealthCheck{Status: api.HEALTHCHECKSTATUS_FAIL, Message: &e}
}, },
} }
@ -65,8 +64,8 @@ func Test_PingFailedStatusNoMessage(t *testing.T) {
name := "foo" name := "foo"
client := &mock.HealthApi{ client := &mock.HealthApi{
GetHealthExecuteFn: func(api.ApiGetHealthRequest) (api.HealthCheck, *http.Response, error) { GetHealthExecuteFn: func(api.ApiGetHealthRequest) (api.HealthCheck, error) {
return api.HealthCheck{}, nil, &api.HealthCheck{Status: api.HEALTHCHECKSTATUS_FAIL, Name: name} return api.HealthCheck{}, &api.HealthCheck{Status: api.HEALTHCHECKSTATUS_FAIL, Name: name}
}, },
} }

4
internal/setup.go
View File

@ -35,7 +35,7 @@ const MinPasswordLen = 8
func (c *CLI) Setup(ctx context.Context, client api.SetupApi, params *SetupParams) error { func (c *CLI) Setup(ctx context.Context, client api.SetupApi, params *SetupParams) error {
// Check if setup is even allowed. // Check if setup is even allowed.
checkResp, _, err := client.GetSetup(ctx).Execute() checkResp, err := client.GetSetup(ctx).Execute()
if err != nil { if err != nil {
return fmt.Errorf("failed to check if already set up: %w", err) return fmt.Errorf("failed to check if already set up: %w", err)
} }
@ -54,7 +54,7 @@ func (c *CLI) Setup(ctx context.Context, client api.SetupApi, params *SetupParam
if err != nil { if err != nil {
return err return err
} }
resp, _, err := client.PostSetup(ctx).OnboardingRequest(setupBody).Execute() resp, err := client.PostSetup(ctx).OnboardingRequest(setupBody).Execute()
if err != nil { if err != nil {
return fmt.Errorf("failed to setup instance: %w", err) return fmt.Errorf("failed to setup instance: %w", err)
} }

41
internal/setup_test.go
View File

@ -4,7 +4,6 @@ import (
"context" "context"
"errors" "errors"
"fmt" "fmt"
"net/http"
"strconv" "strconv"
"strings" "strings"
"testing" "testing"
@ -21,8 +20,8 @@ func Test_SetupConfigNameCollision(t *testing.T) {
t.Parallel() t.Parallel()
client := &mock.SetupApi{ client := &mock.SetupApi{
GetSetupExecuteFn: func(api.ApiGetSetupRequest) (api.InlineResponse200, *http.Response, error) { GetSetupExecuteFn: func(api.ApiGetSetupRequest) (api.InlineResponse200, error) {
return api.InlineResponse200{Allowed: api.PtrBool(true)}, nil, nil return api.InlineResponse200{Allowed: api.PtrBool(true)}, nil
}, },
} }
@ -44,8 +43,8 @@ func Test_SetupConfigNameRequired(t *testing.T) {
t.Parallel() t.Parallel()
client := &mock.SetupApi{ client := &mock.SetupApi{
GetSetupExecuteFn: func(api.ApiGetSetupRequest) (api.InlineResponse200, *http.Response, error) { GetSetupExecuteFn: func(api.ApiGetSetupRequest) (api.InlineResponse200, error) {
return api.InlineResponse200{Allowed: api.PtrBool(true)}, nil, nil return api.InlineResponse200{Allowed: api.PtrBool(true)}, nil
}, },
} }
@ -65,8 +64,8 @@ func Test_SetupAlreadySetup(t *testing.T) {
t.Parallel() t.Parallel()
client := &mock.SetupApi{ client := &mock.SetupApi{
GetSetupExecuteFn: func(api.ApiGetSetupRequest) (api.InlineResponse200, *http.Response, error) { GetSetupExecuteFn: func(api.ApiGetSetupRequest) (api.InlineResponse200, error) {
return api.InlineResponse200{Allowed: api.PtrBool(false)}, nil, nil return api.InlineResponse200{Allowed: api.PtrBool(false)}, nil
}, },
} }
@ -87,8 +86,8 @@ func Test_SetupCheckFailed(t *testing.T) {
e := "oh no" e := "oh no"
client := &mock.SetupApi{ client := &mock.SetupApi{
GetSetupExecuteFn: func(api.ApiGetSetupRequest) (api.InlineResponse200, *http.Response, error) { GetSetupExecuteFn: func(api.ApiGetSetupRequest) (api.InlineResponse200, error) {
return api.InlineResponse200{}, nil, errors.New(e) return api.InlineResponse200{}, errors.New(e)
}, },
} }
@ -125,10 +124,10 @@ func Test_SetupSuccessNoninteractive(t *testing.T) {
Bucket: &api.Bucket{Name: params.Bucket}, Bucket: &api.Bucket{Name: params.Bucket},
} }
client := &mock.SetupApi{ client := &mock.SetupApi{
GetSetupExecuteFn: func(api.ApiGetSetupRequest) (api.InlineResponse200, *http.Response, error) { GetSetupExecuteFn: func(api.ApiGetSetupRequest) (api.InlineResponse200, error) {
return api.InlineResponse200{Allowed: api.PtrBool(true)}, nil, nil return api.InlineResponse200{Allowed: api.PtrBool(true)}, nil
}, },
PostSetupExecuteFn: func(req api.ApiPostSetupRequest) (api.OnboardingResponse, *http.Response, error) { PostSetupExecuteFn: func(req api.ApiPostSetupRequest) (api.OnboardingResponse, error) {
body := req.GetOnboardingRequest() body := req.GetOnboardingRequest()
require.Equal(t, params.Username, body.Username) require.Equal(t, params.Username, body.Username)
require.Equal(t, params.Password, *body.Password) require.Equal(t, params.Password, *body.Password)
@ -136,7 +135,7 @@ func Test_SetupSuccessNoninteractive(t *testing.T) {
require.Equal(t, params.Org, body.Org) require.Equal(t, params.Org, body.Org)
require.Equal(t, params.Bucket, body.Bucket) require.Equal(t, params.Bucket, body.Bucket)
require.Equal(t, retentionSecs, *body.RetentionPeriodSeconds) require.Equal(t, retentionSecs, *body.RetentionPeriodSeconds)
return resp, nil, nil return resp, nil
}, },
} }
@ -182,10 +181,10 @@ func Test_SetupSuccessInteractive(t *testing.T) {
Bucket: &api.Bucket{Name: bucket}, Bucket: &api.Bucket{Name: bucket},
} }
client := &mock.SetupApi{ client := &mock.SetupApi{
GetSetupExecuteFn: func(api.ApiGetSetupRequest) (api.InlineResponse200, *http.Response, error) { GetSetupExecuteFn: func(api.ApiGetSetupRequest) (api.InlineResponse200, error) {
return api.InlineResponse200{Allowed: api.PtrBool(true)}, nil, nil return api.InlineResponse200{Allowed: api.PtrBool(true)}, nil
}, },
PostSetupExecuteFn: func(req api.ApiPostSetupRequest) (api.OnboardingResponse, *http.Response, error) { PostSetupExecuteFn: func(req api.ApiPostSetupRequest) (api.OnboardingResponse, error) {
body := req.GetOnboardingRequest() body := req.GetOnboardingRequest()
require.Equal(t, username, body.Username) require.Equal(t, username, body.Username)
require.Equal(t, password, *body.Password) require.Equal(t, password, *body.Password)
@ -193,7 +192,7 @@ func Test_SetupSuccessInteractive(t *testing.T) {
require.Equal(t, org, body.Org) require.Equal(t, org, body.Org)
require.Equal(t, bucket, body.Bucket) require.Equal(t, bucket, body.Bucket)
require.Equal(t, retentionSecs, *body.RetentionPeriodSeconds) require.Equal(t, retentionSecs, *body.RetentionPeriodSeconds)
return resp, nil, nil return resp, nil
}, },
} }
@ -242,8 +241,8 @@ func Test_SetupPasswordParamToShort(t *testing.T) {
Force: false, Force: false,
} }
client := &mock.SetupApi{ client := &mock.SetupApi{
GetSetupExecuteFn: func(api.ApiGetSetupRequest) (api.InlineResponse200, *http.Response, error) { GetSetupExecuteFn: func(api.ApiGetSetupRequest) (api.InlineResponse200, error) {
return api.InlineResponse200{Allowed: api.PtrBool(true)}, nil, nil return api.InlineResponse200{Allowed: api.PtrBool(true)}, nil
}, },
} }
@ -273,8 +272,8 @@ func Test_SetupCancelAtConfirmation(t *testing.T) {
Force: false, Force: false,
} }
client := &mock.SetupApi{ client := &mock.SetupApi{
GetSetupExecuteFn: func(api.ApiGetSetupRequest) (api.InlineResponse200, *http.Response, error) { GetSetupExecuteFn: func(api.ApiGetSetupRequest) (api.InlineResponse200, error) {
return api.InlineResponse200{Allowed: api.PtrBool(true)}, nil, nil return api.InlineResponse200{Allowed: api.PtrBool(true)}, nil
}, },
} }

2
internal/write.go
View File

@ -69,7 +69,7 @@ func (c *CLI) Write(ctx context.Context, clients *WriteClients, params *WritePar
req = req.Org(c.ActiveConfig.Org) req = req.Org(c.ActiveConfig.Org)
} }
if _, err := req.Execute(); err != nil { if err := req.Execute(); err != nil {
return err return err
} }

13
internal/write_test.go
View File

@ -5,7 +5,6 @@ import (
"context" "context"
"io" "io"
"io/ioutil" "io/ioutil"
"net/http"
"strings" "strings"
"testing" "testing"
@ -69,7 +68,7 @@ func TestWriteByIDs(t *testing.T) {
var writtenLines []string var writtenLines []string
client := mock.WriteApi{ client := mock.WriteApi{
PostWriteExecuteFn: func(req api.ApiPostWriteRequest) (*http.Response, error) { PostWriteExecuteFn: func(req api.ApiPostWriteRequest) error {
// Make sure query params are set. // Make sure query params are set.
require.Equal(t, params.OrgID, *req.GetOrg()) require.Equal(t, params.OrgID, *req.GetOrg())
require.Equal(t, params.BucketID, *req.GetBucket()) require.Equal(t, params.BucketID, *req.GetBucket())
@ -78,7 +77,7 @@ func TestWriteByIDs(t *testing.T) {
// Make sure the body is properly marked for compression, and record what was sent. // Make sure the body is properly marked for compression, and record what was sent.
require.Equal(t, "gzip", *req.GetContentEncoding()) require.Equal(t, "gzip", *req.GetContentEncoding())
writtenLines = append(writtenLines, string(req.GetBody())) writtenLines = append(writtenLines, string(req.GetBody()))
return nil, nil return nil
}, },
} }
@ -113,7 +112,7 @@ func TestWriteByNames(t *testing.T) {
var writtenLines []string var writtenLines []string
client := mock.WriteApi{ client := mock.WriteApi{
PostWriteExecuteFn: func(req api.ApiPostWriteRequest) (*http.Response, error) { PostWriteExecuteFn: func(req api.ApiPostWriteRequest) error {
// Make sure query params are set. // Make sure query params are set.
require.Equal(t, params.OrgName, *req.GetOrg()) require.Equal(t, params.OrgName, *req.GetOrg())
require.Equal(t, params.BucketName, *req.GetBucket()) require.Equal(t, params.BucketName, *req.GetBucket())
@ -122,7 +121,7 @@ func TestWriteByNames(t *testing.T) {
// Make sure the body is properly marked for compression, and record what was sent. // Make sure the body is properly marked for compression, and record what was sent.
require.Equal(t, "gzip", *req.GetContentEncoding()) require.Equal(t, "gzip", *req.GetContentEncoding())
writtenLines = append(writtenLines, string(req.GetBody())) writtenLines = append(writtenLines, string(req.GetBody()))
return nil, nil return nil
}, },
} }
@ -156,7 +155,7 @@ func TestWriteOrgFromConfig(t *testing.T) {
var writtenLines []string var writtenLines []string
client := mock.WriteApi{ client := mock.WriteApi{
PostWriteExecuteFn: func(req api.ApiPostWriteRequest) (*http.Response, error) { PostWriteExecuteFn: func(req api.ApiPostWriteRequest) error {
// Make sure query params are set. // Make sure query params are set.
require.Equal(t, cli.ActiveConfig.Org, *req.GetOrg()) require.Equal(t, cli.ActiveConfig.Org, *req.GetOrg())
require.Equal(t, params.BucketName, *req.GetBucket()) require.Equal(t, params.BucketName, *req.GetBucket())
@ -165,7 +164,7 @@ func TestWriteOrgFromConfig(t *testing.T) {
// Make sure the body is properly marked for compression, and record what was sent. // Make sure the body is properly marked for compression, and record what was sent.
require.Equal(t, "gzip", *req.GetContentEncoding()) require.Equal(t, "gzip", *req.GetContentEncoding())
writtenLines = append(writtenLines, string(req.GetBody())) writtenLines = append(writtenLines, string(req.GetBody()))
return nil, nil return nil
}, },
} }

1
tools.go
View File

@ -11,5 +11,6 @@ package influxcli
import ( import (
_ "github.com/daixiang0/gci" _ "github.com/daixiang0/gci"
_ "golang.org/x/tools/cmd/goimports"
_ "honnef.co/go/tools/cmd/staticcheck" _ "honnef.co/go/tools/cmd/staticcheck"
) )