feat(aw): Insert and Recent over the events endpoint
This commit is contained in:
@@ -68,3 +68,57 @@ func hostname() string {
|
||||
}
|
||||
return "keel"
|
||||
}
|
||||
|
||||
type eventJSON struct {
|
||||
Timestamp time.Time `json:"timestamp"`
|
||||
Duration float64 `json:"duration"`
|
||||
Data map[string]any `json:"data"`
|
||||
}
|
||||
|
||||
// Insert posts a single event to the bucket. The AW events endpoint takes a
|
||||
// JSON array, so the event is wrapped in a one-element list.
|
||||
func (c *Client) Insert(ctx context.Context, bucketID string, e Event) error {
|
||||
body, _ := json.Marshal([]eventJSON{{Timestamp: e.Timestamp, Duration: e.Duration, Data: e.Data}})
|
||||
req, err := http.NewRequestWithContext(ctx, http.MethodPost,
|
||||
c.baseURL+"/api/0/buckets/"+bucketID+"/events", bytes.NewReader(body))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
resp, err := c.hc.Do(req)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
|
||||
return fmt.Errorf("aw: insert into %s: status %d", bucketID, resp.StatusCode)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// Recent returns up to limit events from the bucket, newest first (the AW
|
||||
// default ordering for the events endpoint).
|
||||
func (c *Client) Recent(ctx context.Context, bucketID string, limit int) ([]Event, error) {
|
||||
endpoint := fmt.Sprintf("%s/api/0/buckets/%s/events?limit=%d", c.baseURL, bucketID, limit)
|
||||
req, err := http.NewRequestWithContext(ctx, http.MethodGet, endpoint, nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
resp, err := c.hc.Do(req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
|
||||
return nil, fmt.Errorf("aw: read %s: status %d", bucketID, resp.StatusCode)
|
||||
}
|
||||
var raw []eventJSON
|
||||
if err := json.NewDecoder(resp.Body).Decode(&raw); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
out := make([]Event, len(raw))
|
||||
for i, e := range raw {
|
||||
out[i] = Event{Timestamp: e.Timestamp, Duration: e.Duration, Data: e.Data}
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user