| /* |
| |
| 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 ( |
| "encoding/json" |
| "fmt" |
| "net" |
| "net/http" |
| "net/url" |
| |
| "github.com/apache/trafficcontrol/lib/go-tc" |
| ) |
| |
| const ( |
| API_v14_ROLES = "/api/1.4/roles" |
| ) |
| |
| // Create a Role |
| func (to *Session) CreateRole(region tc.Role) (tc.Alerts, ReqInf, int, error) { |
| |
| var remoteAddr net.Addr |
| reqBody, err := json.Marshal(region) |
| reqInf := ReqInf{CacheHitStatus: CacheHitStatusMiss, RemoteAddr: remoteAddr} |
| if err != nil { |
| return tc.Alerts{}, reqInf, 0, err |
| } |
| resp, remoteAddr, errClient := to.RawRequest(http.MethodPost, API_v14_ROLES, reqBody) |
| if resp != nil { |
| defer resp.Body.Close() |
| var alerts tc.Alerts |
| if err = json.NewDecoder(resp.Body).Decode(&alerts); err != nil { |
| return alerts, reqInf, resp.StatusCode, err |
| } |
| return alerts, reqInf, resp.StatusCode, errClient |
| } |
| return tc.Alerts{}, reqInf, 0, errClient |
| } |
| |
| // Update a Role by ID |
| func (to *Session) UpdateRoleByID(id int, region tc.Role) (tc.Alerts, ReqInf, int, error) { |
| |
| var remoteAddr net.Addr |
| reqBody, err := json.Marshal(region) |
| reqInf := ReqInf{CacheHitStatus: CacheHitStatusMiss, RemoteAddr: remoteAddr} |
| if err != nil { |
| return tc.Alerts{}, reqInf, 0, err |
| } |
| route := fmt.Sprintf("%s/?id=%d", API_v14_ROLES, id) |
| resp, remoteAddr, errClient := to.RawRequest(http.MethodPut, route, reqBody) |
| if resp != nil { |
| defer resp.Body.Close() |
| var alerts tc.Alerts |
| if err = json.NewDecoder(resp.Body).Decode(&alerts); err != nil { |
| return alerts, reqInf, resp.StatusCode, err |
| } |
| return alerts, reqInf, resp.StatusCode, errClient |
| } |
| return tc.Alerts{}, reqInf, 0, errClient |
| } |
| |
| // Returns a list of roles |
| func (to *Session) GetRoles() ([]tc.Role, ReqInf, int, error) { |
| resp, remoteAddr, errClient := to.RawRequest(http.MethodGet, API_v14_ROLES, nil) |
| reqInf := ReqInf{CacheHitStatus: CacheHitStatusMiss, RemoteAddr: remoteAddr} |
| if resp != nil { |
| defer resp.Body.Close() |
| |
| var data tc.RolesResponse |
| if err := json.NewDecoder(resp.Body).Decode(&data); err != nil { |
| return data.Response, reqInf, resp.StatusCode, err |
| } |
| return data.Response, reqInf, resp.StatusCode, errClient |
| } |
| return []tc.Role{}, reqInf, 0, errClient |
| } |
| |
| // GET a Role by the Role id |
| func (to *Session) GetRoleByID(id int) ([]tc.Role, ReqInf, int, error) { |
| route := fmt.Sprintf("%s/?id=%d", API_v14_ROLES, id) |
| resp, remoteAddr, errClient := to.RawRequest(http.MethodGet, route, nil) |
| reqInf := ReqInf{CacheHitStatus: CacheHitStatusMiss, RemoteAddr: remoteAddr} |
| if resp != nil { |
| defer resp.Body.Close() |
| |
| var data tc.RolesResponse |
| if err := json.NewDecoder(resp.Body).Decode(&data); err != nil { |
| return data.Response, reqInf, resp.StatusCode, err |
| } |
| return data.Response, reqInf, resp.StatusCode, errClient |
| } |
| return []tc.Role{}, reqInf, 0, errClient |
| } |
| |
| // GET a Role by the Role name |
| func (to *Session) GetRoleByName(name string) ([]tc.Role, ReqInf, int, error) { |
| route := fmt.Sprintf("%s?name=%s", API_v14_ROLES, url.QueryEscape(name)) |
| resp, remoteAddr, errClient := to.RawRequest(http.MethodGet, route, nil) |
| reqInf := ReqInf{CacheHitStatus: CacheHitStatusMiss, RemoteAddr: remoteAddr} |
| if resp != nil { |
| defer resp.Body.Close() |
| |
| var data tc.RolesResponse |
| if err := json.NewDecoder(resp.Body).Decode(&data); err != nil { |
| return data.Response, reqInf, resp.StatusCode, err |
| } |
| return data.Response, reqInf, resp.StatusCode, errClient |
| } |
| return []tc.Role{}, reqInf, 0, errClient |
| } |
| |
| // GetRoleByQueryParams gets a Role by the Role query parameters |
| func (to *Session) GetRoleByQueryParams(queryParams map[string]string) ([]tc.Role, ReqInf, int, error) { |
| route := fmt.Sprintf("%s?", API_v14_ROLES) |
| for param, val := range queryParams { |
| route += fmt.Sprintf("%s=%s&", url.QueryEscape(param), url.QueryEscape(val)) |
| } |
| resp, remoteAddr, errClient := to.RawRequest(http.MethodGet, route, nil) |
| reqInf := ReqInf{CacheHitStatus: CacheHitStatusMiss, RemoteAddr: remoteAddr} |
| if resp != nil { |
| defer resp.Body.Close() |
| |
| var data tc.RolesResponse |
| if err := json.NewDecoder(resp.Body).Decode(&data); err != nil { |
| return data.Response, reqInf, resp.StatusCode, err |
| } |
| return data.Response, reqInf, resp.StatusCode, errClient |
| } |
| return []tc.Role{}, reqInf, 0, errClient |
| } |
| |
| // DELETE a Role by ID |
| func (to *Session) DeleteRoleByID(id int) (tc.Alerts, ReqInf, int, error) { |
| route := fmt.Sprintf("%s/?id=%d", API_v14_ROLES, id) |
| resp, remoteAddr, errClient := to.RawRequest(http.MethodDelete, route, nil) |
| reqInf := ReqInf{CacheHitStatus: CacheHitStatusMiss, RemoteAddr: remoteAddr} |
| if resp != nil { |
| defer resp.Body.Close() |
| |
| var alerts tc.Alerts |
| if err := json.NewDecoder(resp.Body).Decode(&alerts); err != nil { |
| return alerts, reqInf, resp.StatusCode, err |
| } |
| return alerts, reqInf, resp.StatusCode, errClient |
| } |
| return tc.Alerts{}, reqInf, 0, errClient |
| } |