Ta-Ching ChenandGitHub
a645a1e197
The function implementations of controller client package are inconsistent. This PR lets functions reuse the functions that already implemented and able to set additional headers to request.
132 lines
3.4 KiB
Go
132 lines
3.4 KiB
Go
/*
|
|
Copyright 2016 The Fission Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package client
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
ferror "github.com/fission/fission/pkg/error"
|
|
)
|
|
|
|
var (
|
|
DefaultRequestHeaders map[string]string
|
|
)
|
|
|
|
type (
|
|
Client struct {
|
|
Url string
|
|
Headers map[string]string
|
|
}
|
|
)
|
|
|
|
func MakeClient(serverUrl string) *Client {
|
|
return &Client{
|
|
Url: strings.TrimSuffix(serverUrl, "/"),
|
|
Headers: DefaultRequestHeaders,
|
|
}
|
|
}
|
|
|
|
func (c *Client) create(relativeUrl string, contentType string, payload []byte) (*http.Response, error) {
|
|
var reader io.Reader
|
|
if len(payload) > 0 {
|
|
reader = bytes.NewReader(payload)
|
|
}
|
|
return c.sendRequest(http.MethodPost, c.v2CrdUrl(relativeUrl), map[string]string{"Content-type": contentType}, reader)
|
|
}
|
|
|
|
func (c *Client) put(relativeUrl string, contentType string, payload []byte) (*http.Response, error) {
|
|
var reader io.Reader
|
|
if len(payload) > 0 {
|
|
reader = bytes.NewReader(payload)
|
|
}
|
|
return c.sendRequest(http.MethodPut, c.v2CrdUrl(relativeUrl), map[string]string{"Content-type": contentType}, reader)
|
|
}
|
|
|
|
func (c *Client) get(relativeUrl string) (*http.Response, error) {
|
|
return c.sendRequest(http.MethodGet, c.v2CrdUrl(relativeUrl), nil, nil)
|
|
}
|
|
|
|
func (c *Client) delete(relativeUrl string) error {
|
|
resp, err := c.sendRequest(http.MethodDelete, c.v2CrdUrl(relativeUrl), nil, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode != 200 {
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return errors.Wrap(err, "error deleting")
|
|
} else {
|
|
return errors.Errorf("failed to delete: %v", string(body))
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *Client) proxy(method string, relativeUrl string, payload []byte) (*http.Response, error) {
|
|
var reader io.Reader
|
|
if len(payload) > 0 {
|
|
reader = bytes.NewReader(payload)
|
|
}
|
|
return c.sendRequest(method, c.proxyUrl(relativeUrl), nil, reader)
|
|
}
|
|
|
|
func (c *Client) sendRequest(method string, relativeUrl string, headers map[string]string, reader io.Reader) (*http.Response, error) {
|
|
req, err := http.NewRequest(method, relativeUrl, reader)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, hs := range []map[string]string{headers, c.Headers} {
|
|
for k, v := range hs {
|
|
req.Header.Set(k, v)
|
|
}
|
|
}
|
|
return http.DefaultClient.Do(req)
|
|
}
|
|
|
|
func (c *Client) v2CrdUrl(relativeUrl string) string {
|
|
return c.Url + "/v2/" + strings.TrimPrefix(relativeUrl, "/")
|
|
}
|
|
|
|
func (c *Client) proxyUrl(relativeUrl string) string {
|
|
return c.Url + "/proxy/" + strings.TrimPrefix(relativeUrl, "/")
|
|
}
|
|
|
|
func (c *Client) handleResponse(resp *http.Response) ([]byte, error) {
|
|
if resp.StatusCode != 200 {
|
|
return nil, ferror.MakeErrorFromHTTP(resp)
|
|
}
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
return body, err
|
|
}
|
|
|
|
func (c *Client) handleCreateResponse(resp *http.Response) ([]byte, error) {
|
|
if resp.StatusCode != 201 {
|
|
return nil, ferror.MakeErrorFromHTTP(resp)
|
|
}
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
return body, err
|
|
}
|