mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-25 07:30:22 -04:00
165 lines
3.4 KiB
Go
165 lines
3.4 KiB
Go
![]() |
/*
|
||
|
* Echo Service
|
||
|
*
|
||
|
* Echo Service API consists of a single service which returns a message.
|
||
|
*
|
||
|
* OpenAPI spec version: version not set
|
||
|
*
|
||
|
* Generated by: https://github.com/swagger-api/swagger-codegen.git
|
||
|
*/
|
||
|
|
||
|
package echo
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"fmt"
|
||
|
"path/filepath"
|
||
|
"reflect"
|
||
|
"strings"
|
||
|
"net/url"
|
||
|
"io/ioutil"
|
||
|
"github.com/go-resty/resty"
|
||
|
)
|
||
|
|
||
|
type APIClient struct {
|
||
|
config *Configuration
|
||
|
}
|
||
|
|
||
|
func (c *APIClient) SelectHeaderContentType(contentTypes []string) string {
|
||
|
|
||
|
if len(contentTypes) == 0 {
|
||
|
return ""
|
||
|
}
|
||
|
if contains(contentTypes, "application/json") {
|
||
|
return "application/json"
|
||
|
}
|
||
|
return contentTypes[0] // use the first content type specified in 'consumes'
|
||
|
}
|
||
|
|
||
|
func (c *APIClient) SelectHeaderAccept(accepts []string) string {
|
||
|
|
||
|
if len(accepts) == 0 {
|
||
|
return ""
|
||
|
}
|
||
|
if contains(accepts, "application/json") {
|
||
|
return "application/json"
|
||
|
}
|
||
|
return strings.Join(accepts, ",")
|
||
|
}
|
||
|
|
||
|
func contains(haystack []string, needle string) bool {
|
||
|
for _, a := range haystack {
|
||
|
if strings.ToLower(a) == strings.ToLower(needle) {
|
||
|
return true
|
||
|
}
|
||
|
}
|
||
|
return false
|
||
|
}
|
||
|
|
||
|
func (c *APIClient) CallAPI(path string, method string,
|
||
|
postBody interface{},
|
||
|
headerParams map[string]string,
|
||
|
queryParams url.Values,
|
||
|
formParams map[string]string,
|
||
|
fileName string,
|
||
|
fileBytes []byte) (*resty.Response, error) {
|
||
|
|
||
|
rClient := c.prepareClient()
|
||
|
request := c.prepareRequest(rClient, postBody, headerParams, queryParams, formParams, fileName, fileBytes)
|
||
|
|
||
|
switch strings.ToUpper(method) {
|
||
|
case "GET":
|
||
|
response, err := request.Get(path)
|
||
|
return response, err
|
||
|
case "POST":
|
||
|
response, err := request.Post(path)
|
||
|
return response, err
|
||
|
case "PUT":
|
||
|
response, err := request.Put(path)
|
||
|
return response, err
|
||
|
case "PATCH":
|
||
|
response, err := request.Patch(path)
|
||
|
return response, err
|
||
|
case "DELETE":
|
||
|
response, err := request.Delete(path)
|
||
|
return response, err
|
||
|
}
|
||
|
|
||
|
return nil, fmt.Errorf("invalid method %v", method)
|
||
|
}
|
||
|
|
||
|
func (c *APIClient) ParameterToString(obj interface{}, collectionFormat string) string {
|
||
|
delimiter := ""
|
||
|
switch collectionFormat {
|
||
|
case "pipes":
|
||
|
delimiter = "|"
|
||
|
case "ssv":
|
||
|
delimiter = " "
|
||
|
case "tsv":
|
||
|
delimiter = "\t"
|
||
|
case "csv":
|
||
|
delimiter = ","
|
||
|
}
|
||
|
|
||
|
if reflect.TypeOf(obj).Kind() == reflect.Slice {
|
||
|
return strings.Trim(strings.Replace(fmt.Sprint(obj), " ", delimiter, -1), "[]")
|
||
|
}
|
||
|
|
||
|
return fmt.Sprintf("%v", obj)
|
||
|
}
|
||
|
|
||
|
func (c *APIClient) prepareClient() *resty.Client {
|
||
|
|
||
|
rClient := resty.New()
|
||
|
|
||
|
rClient.SetDebug(c.config.Debug)
|
||
|
if c.config.Transport != nil {
|
||
|
rClient.SetTransport(c.config.Transport)
|
||
|
}
|
||
|
|
||
|
if c.config.Timeout != nil {
|
||
|
rClient.SetTimeout(*c.config.Timeout)
|
||
|
}
|
||
|
rClient.SetLogger(ioutil.Discard)
|
||
|
return rClient
|
||
|
}
|
||
|
|
||
|
func (c *APIClient) prepareRequest(
|
||
|
rClient *resty.Client,
|
||
|
postBody interface{},
|
||
|
headerParams map[string]string,
|
||
|
queryParams url.Values,
|
||
|
formParams map[string]string,
|
||
|
fileName string,
|
||
|
fileBytes []byte) *resty.Request {
|
||
|
|
||
|
|
||
|
request := rClient.R()
|
||
|
request.SetBody(postBody)
|
||
|
|
||
|
if c.config.UserAgent != "" {
|
||
|
request.SetHeader("User-Agent", c.config.UserAgent)
|
||
|
}
|
||
|
|
||
|
// add header parameter, if any
|
||
|
if len(headerParams) > 0 {
|
||
|
request.SetHeaders(headerParams)
|
||
|
}
|
||
|
|
||
|
// add query parameter, if any
|
||
|
if len(queryParams) > 0 {
|
||
|
request.SetMultiValueQueryParams(queryParams)
|
||
|
}
|
||
|
|
||
|
// add form parameter, if any
|
||
|
if len(formParams) > 0 {
|
||
|
request.SetFormData(formParams)
|
||
|
}
|
||
|
|
||
|
if len(fileBytes) > 0 && fileName != "" {
|
||
|
_, fileNm := filepath.Split(fileName)
|
||
|
request.SetFileReader("file", fileNm, bytes.NewReader(fileBytes))
|
||
|
}
|
||
|
return request
|
||
|
}
|