| /* |
| |
| 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_STATIC_DNS_ENTRIES = apiBase + "/staticdnsentries" |
| ) |
| |
| func staticDNSEntryIDs(to *Session, sdns *tc.StaticDNSEntry) error { |
| if sdns.CacheGroupID == 0 && sdns.CacheGroupName != "" { |
| p, _, err := to.GetCacheGroupNullableByName(sdns.CacheGroupName, nil) |
| if err != nil { |
| return err |
| } |
| if len(p) == 0 { |
| return errors.New("no CacheGroup named " + sdns.CacheGroupName) |
| } |
| if p[0].ID == nil { |
| return errors.New("CacheGroup named " + sdns.CacheGroupName + " has a nil ID") |
| } |
| sdns.CacheGroupID = *p[0].ID |
| } |
| |
| if sdns.DeliveryServiceID == 0 && sdns.DeliveryService != "" { |
| dses, _, err := to.GetDeliveryServiceByXMLIDNullable(sdns.DeliveryService, nil) |
| if err != nil { |
| return err |
| } |
| if len(dses) == 0 { |
| return errors.New("no deliveryservice with name " + sdns.DeliveryService) |
| } |
| if dses[0].ID == nil { |
| return errors.New("Deliveryservice with name " + sdns.DeliveryService + " has a nil ID") |
| } |
| sdns.DeliveryServiceID = *dses[0].ID |
| } |
| |
| if sdns.TypeID == 0 && sdns.Type != "" { |
| types, _, err := to.GetTypeByName(sdns.Type, nil) |
| if err != nil { |
| return err |
| } |
| if len(types) == 0 { |
| return errors.New("no type with name " + sdns.Type) |
| } |
| sdns.TypeID = types[0].ID |
| } |
| |
| return nil |
| } |
| |
| // CreateStaticDNSEntry creates a Static DNS Entry. |
| func (to *Session) CreateStaticDNSEntry(sdns tc.StaticDNSEntry) (tc.Alerts, ReqInf, error) { |
| // fill in missing IDs from names |
| staticDNSEntryIDs(to, &sdns) |
| var remoteAddr net.Addr |
| reqBody, err := json.Marshal(sdns) |
| reqInf := ReqInf{CacheHitStatus: CacheHitStatusMiss, RemoteAddr: remoteAddr} |
| if err != nil { |
| return tc.Alerts{}, reqInf, err |
| } |
| resp, remoteAddr, err := to.request(http.MethodPost, API_STATIC_DNS_ENTRIES, reqBody, nil) |
| 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 |
| } |
| |
| // UpdateStaticDNSEntryByID updates a Static DNS Entry by ID. |
| func (to *Session) UpdateStaticDNSEntryByID(id int, sdns tc.StaticDNSEntry) (tc.Alerts, ReqInf, int, error) { |
| // fill in missing IDs from names |
| staticDNSEntryIDs(to, &sdns) |
| var remoteAddr net.Addr |
| reqBody, err := json.Marshal(sdns) |
| reqInf := ReqInf{CacheHitStatus: CacheHitStatusMiss, RemoteAddr: remoteAddr} |
| if err != nil { |
| return tc.Alerts{}, reqInf, 0, err |
| } |
| route := fmt.Sprintf("%s?id=%d", API_STATIC_DNS_ENTRIES, id) |
| resp, remoteAddr, errClient := to.RawRequest(http.MethodPut, route, reqBody, nil) |
| 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 |
| } |
| |
| // GetStaticDNSEntries returns a list of Static DNS Entrys. |
| func (to *Session) GetStaticDNSEntries(header http.Header) ([]tc.StaticDNSEntry, ReqInf, error) { |
| resp, remoteAddr, err := to.request(http.MethodGet, API_STATIC_DNS_ENTRIES, nil, header) |
| reqInf := ReqInf{CacheHitStatus: CacheHitStatusMiss, RemoteAddr: remoteAddr} |
| if resp != nil { |
| reqInf.StatusCode = resp.StatusCode |
| if reqInf.StatusCode == http.StatusNotModified { |
| return []tc.StaticDNSEntry{}, reqInf, nil |
| } |
| } |
| if err != nil { |
| return nil, reqInf, err |
| } |
| defer resp.Body.Close() |
| |
| var data tc.StaticDNSEntriesResponse |
| err = json.NewDecoder(resp.Body).Decode(&data) |
| return data.Response, reqInf, nil |
| } |
| |
| // GetStaticDNSEntryByID GETs a Static DNS Entry by the Static DNS Entry's ID. |
| func (to *Session) GetStaticDNSEntryByID(id int, header http.Header) ([]tc.StaticDNSEntry, ReqInf, error) { |
| route := fmt.Sprintf("%s?id=%d", API_STATIC_DNS_ENTRIES, id) |
| resp, remoteAddr, err := to.request(http.MethodGet, route, nil, header) |
| reqInf := ReqInf{CacheHitStatus: CacheHitStatusMiss, RemoteAddr: remoteAddr} |
| if resp != nil { |
| reqInf.StatusCode = resp.StatusCode |
| if reqInf.StatusCode == http.StatusNotModified { |
| return []tc.StaticDNSEntry{}, reqInf, nil |
| } |
| } |
| if err != nil { |
| return nil, reqInf, err |
| } |
| defer resp.Body.Close() |
| |
| var data tc.StaticDNSEntriesResponse |
| if err := json.NewDecoder(resp.Body).Decode(&data); err != nil { |
| return nil, reqInf, err |
| } |
| |
| return data.Response, reqInf, nil |
| } |
| |
| // GetStaticDNSEntriesByHost GETs a Static DNS Entry by the Static DNS Entry's host. |
| func (to *Session) GetStaticDNSEntriesByHost(host string, header http.Header) ([]tc.StaticDNSEntry, ReqInf, error) { |
| url := fmt.Sprintf("%s?host=%s", API_STATIC_DNS_ENTRIES, host) |
| resp, remoteAddr, err := to.request(http.MethodGet, url, nil, header) |
| reqInf := ReqInf{CacheHitStatus: CacheHitStatusMiss, RemoteAddr: remoteAddr} |
| if resp != nil { |
| reqInf.StatusCode = resp.StatusCode |
| if reqInf.StatusCode == http.StatusNotModified { |
| return []tc.StaticDNSEntry{}, reqInf, nil |
| } |
| } |
| if err != nil { |
| return nil, reqInf, err |
| } |
| defer resp.Body.Close() |
| |
| var data tc.StaticDNSEntriesResponse |
| if err := json.NewDecoder(resp.Body).Decode(&data); err != nil { |
| return nil, reqInf, err |
| } |
| |
| return data.Response, reqInf, nil |
| } |
| |
| // DeleteStaticDNSEntryByID DELETEs a Static DNS Entry by ID. |
| func (to *Session) DeleteStaticDNSEntryByID(id int) (tc.Alerts, ReqInf, error) { |
| route := fmt.Sprintf("%s?id=%d", API_STATIC_DNS_ENTRIES, id) |
| resp, remoteAddr, err := to.request(http.MethodDelete, route, nil, 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 |
| } |