2016-04-16 22:58:13 -07:00

98 lines
2.3 KiB
Go

package swagger
import (
"strings"
"github.com/go-resty/resty"
"errors"
)
type ApiClient struct {
}
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(source []string, containvalue string) bool {
for _, a := range source {
if strings.ToLower(a) == strings.ToLower(containvalue) {
return true
}
}
return false
}
func CallApiAsync(path string, method string,
postBody interface{},
headerParams map[string]string,
queryParams map[string]string,
formParams map[string]string,
fileParams map[string]string) (*resty.Response, error) {
request := prepareRequest(postBody, headerParams, queryParams, formParams, fileParams)
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, errors.New("Invalid method " + method)
}
func prepareRequest(postBody interface{},
headerParams map[string]string,
queryParams map[string]string,
formParams map[string]string,
fileParams map[string]string) *resty.Request {
request := resty.R()
request.SetBody(postBody)
// add header parameter, if any
request.SetHeaders(headerParams)
// add query parameter, if any
request.SetQueryParams(queryParams)
// add form parameter, if any
request.SetFormData(formParams)
// add file parameter, if any
request.SetFiles(fileParams)
return request
}