| /* |
| |
| 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" |
| "errors" |
| "fmt" |
| "net" |
| "net/http" |
| |
| "github.com/apache/trafficcontrol/lib/go-tc" |
| ) |
| |
| const ( |
| API_TYPES = apiBase + "/types" |
| ) |
| |
| // CreateType creates a Type. There should be a very good reason for doing this. |
| func (to *Session) CreateType(typ tc.Type) (tc.Alerts, ReqInf, error) { |
| |
| var remoteAddr net.Addr |
| reqBody, err := json.Marshal(typ) |
| reqInf := ReqInf{CacheHitStatus: CacheHitStatusMiss, RemoteAddr: remoteAddr} |
| if err != nil { |
| return tc.Alerts{}, reqInf, err |
| } |
| resp, remoteAddr, err := to.request(http.MethodPost, API_TYPES, reqBody) |
| if err != nil { |
| return tc.Alerts{}, reqInf, err |
| } |
| defer resp.Body.Close() |
| var alerts tc.Alerts |
| err = json.NewDecoder(resp.Body).Decode(&alerts) |
| return alerts, reqInf, nil |
| } |
| |
| // UpdateTypeByID updates a Type by ID. |
| func (to *Session) UpdateTypeByID(id int, typ tc.Type) (tc.Alerts, ReqInf, error) { |
| |
| var remoteAddr net.Addr |
| reqBody, err := json.Marshal(typ) |
| reqInf := ReqInf{CacheHitStatus: CacheHitStatusMiss, RemoteAddr: remoteAddr} |
| if err != nil { |
| return tc.Alerts{}, reqInf, err |
| } |
| route := fmt.Sprintf("%s/%d", API_TYPES, id) |
| resp, remoteAddr, err := to.request(http.MethodPut, route, reqBody) |
| if err != nil { |
| return tc.Alerts{}, reqInf, err |
| } |
| defer resp.Body.Close() |
| var alerts tc.Alerts |
| err = json.NewDecoder(resp.Body).Decode(&alerts) |
| return alerts, reqInf, nil |
| } |
| |
| // GetTypes returns a list of Types. If a 'useInTable' parameter is passed, the returned Types |
| // are restricted to those with that exact 'useInTable' property. Only exactly 1 or exactly 0 |
| // 'useInTable' parameters may be passed; passing more will result in an error being returned. |
| func (to *Session) GetTypes(useInTable ...string) ([]tc.Type, ReqInf, error) { |
| if len(useInTable) > 1 { |
| return nil, ReqInf{}, errors.New("Please pass in a single value for the 'useInTable' parameter") |
| } |
| |
| resp, remoteAddr, err := to.request("GET", API_TYPES, nil) |
| reqInf := ReqInf{CacheHitStatus: CacheHitStatusMiss, RemoteAddr: remoteAddr} |
| if err != nil { |
| return nil, reqInf, err |
| } |
| defer resp.Body.Close() |
| |
| var data tc.TypesResponse |
| if err := json.NewDecoder(resp.Body).Decode(&data); err != nil { |
| return nil, reqInf, err |
| } |
| |
| var types []tc.Type |
| for _, d := range data.Response { |
| if useInTable != nil { |
| if d.UseInTable == useInTable[0] { |
| types = append(types, d) |
| } |
| } else { |
| types = append(types, d) |
| } |
| } |
| |
| return types, reqInf, nil |
| } |
| |
| // GetTypeByID GETs a Type by the Type ID. |
| func (to *Session) GetTypeByID(id int) ([]tc.Type, ReqInf, error) { |
| route := fmt.Sprintf("%s?id=%d", API_TYPES, id) |
| resp, remoteAddr, err := to.request(http.MethodGet, route, nil) |
| reqInf := ReqInf{CacheHitStatus: CacheHitStatusMiss, RemoteAddr: remoteAddr} |
| if err != nil { |
| return nil, reqInf, err |
| } |
| defer resp.Body.Close() |
| |
| var data tc.TypesResponse |
| if err := json.NewDecoder(resp.Body).Decode(&data); err != nil { |
| return nil, reqInf, err |
| } |
| |
| return data.Response, reqInf, nil |
| } |
| |
| // GetTypeByName GET a Type by the Type name. |
| func (to *Session) GetTypeByName(name string) ([]tc.Type, ReqInf, error) { |
| url := fmt.Sprintf("%s?name=%s", API_TYPES, name) |
| resp, remoteAddr, err := to.request(http.MethodGet, url, nil) |
| reqInf := ReqInf{CacheHitStatus: CacheHitStatusMiss, RemoteAddr: remoteAddr} |
| if err != nil { |
| return nil, reqInf, err |
| } |
| defer resp.Body.Close() |
| |
| var data tc.TypesResponse |
| if err := json.NewDecoder(resp.Body).Decode(&data); err != nil { |
| return nil, reqInf, err |
| } |
| |
| return data.Response, reqInf, nil |
| } |
| |
| // DeleteTypeByID DELETEs a Type by ID. |
| func (to *Session) DeleteTypeByID(id int) (tc.Alerts, ReqInf, error) { |
| route := fmt.Sprintf("%s/%d", API_TYPES, id) |
| resp, remoteAddr, err := to.request(http.MethodDelete, route, nil) |
| reqInf := ReqInf{CacheHitStatus: CacheHitStatusMiss, RemoteAddr: remoteAddr} |
| if err != nil { |
| return tc.Alerts{}, reqInf, err |
| } |
| defer resp.Body.Close() |
| var alerts tc.Alerts |
| err = json.NewDecoder(resp.Body).Decode(&alerts) |
| return alerts, reqInf, nil |
| } |