terraform/vendor/github.com/Ensighten/udnssdk/event.go

126 lines
3.4 KiB
Go

package udnssdk
import (
"fmt"
"log"
"net/http"
"time"
)
// EventsService manages Events
type EventsService struct {
client *Client
}
// EventInfoDTO wraps an event's info response
type EventInfoDTO struct {
ID string `json:"id"`
PoolRecord string `json:"poolRecord"`
EventType string `json:"type"`
Start time.Time `json:"start"`
Repeat string `json:"repeat"`
End time.Time `json:"end"`
Notify string `json:"notify"`
}
// EventInfoListDTO wraps a list of event info and list metadata, from an index request
type EventInfoListDTO struct {
Events []EventInfoDTO `json:"events"`
Queryinfo QueryInfo `json:"queryInfo"`
Resultinfo ResultInfo `json:"resultInfo"`
}
// EventKey collects the identifiers of an Event
type EventKey struct {
Zone string
Type string
Name string
GUID string
}
// RRSetKey generates the RRSetKey for the EventKey
func (p EventKey) RRSetKey() RRSetKey {
return RRSetKey{
Zone: p.Zone,
Type: p.Type,
Name: p.Name,
}
}
// URI generates the URI for a probe
func (p EventKey) URI() string {
return fmt.Sprintf("%s/%s", p.RRSetKey().EventsURI(), p.GUID)
}
// Select requests all events, using pagination and error handling
func (s *EventsService) Select(r RRSetKey, query string) ([]EventInfoDTO, error) {
// TODO: Sane Configuration for timeouts / retries
maxerrs := 5
waittime := 5 * time.Second
// init accumulators
pis := []EventInfoDTO{}
offset := 0
errcnt := 0
for {
reqEvents, ri, res, err := s.SelectWithOffset(r, query, offset)
if err != nil {
if res != nil && res.StatusCode >= 500 {
errcnt = errcnt + 1
if errcnt < maxerrs {
time.Sleep(waittime)
continue
}
}
return pis, err
}
log.Printf("ResultInfo: %+v\n", ri)
for _, pi := range reqEvents {
pis = append(pis, pi)
}
if ri.ReturnedCount+ri.Offset >= ri.TotalCount {
return pis, nil
}
offset = ri.ReturnedCount + ri.Offset
continue
}
}
// SelectWithOffset requests list of events by RRSetKey, query and offset, also returning list metadata, the actual response, or an error
func (s *EventsService) SelectWithOffset(r RRSetKey, query string, offset int) ([]EventInfoDTO, ResultInfo, *http.Response, error) {
var tld EventInfoListDTO
uri := r.EventsQueryURI(query, offset)
res, err := s.client.get(uri, &tld)
pis := []EventInfoDTO{}
for _, pi := range tld.Events {
pis = append(pis, pi)
}
return pis, tld.Resultinfo, res, err
}
// Find requests an event by name, type, zone & guid, also returning the actual response, or an error
func (s *EventsService) Find(e EventKey) (EventInfoDTO, *http.Response, error) {
var t EventInfoDTO
res, err := s.client.get(e.URI(), &t)
return t, res, err
}
// Create requests creation of an event by RRSetKey, with provided event-info, returning actual response or an error
func (s *EventsService) Create(r RRSetKey, ev EventInfoDTO) (*http.Response, error) {
return s.client.post(r.EventsURI(), ev, nil)
}
// Update requests update of an event by EventKey, withprovided event-info, returning the actual response or an error
func (s *EventsService) Update(e EventKey, ev EventInfoDTO) (*http.Response, error) {
return s.client.put(e.URI(), ev, nil)
}
// Delete requests deletion of an event by EventKey, returning the actual response or an error
func (s *EventsService) Delete(e EventKey) (*http.Response, error) {
return s.client.delete(e.URI(), nil)
}