activity.go

package gosnowth

import (
	"context"
	"fmt"
)

// RebuildActivityRequest values represent a request to rebuild activity
// tracking data.
type RebuildActivityRequest struct {
	UUID   string `json:"check_uuid"`
	Metric string `json:"metric_name"`
}

// RebuildActivity rebuilds IRONdb activity tracking data for a list of metrics.
func (sc *SnowthClient) RebuildActivity(node *SnowthNode,
	rebuildRequest []RebuildActivityRequest,
) (*IRONdbPutResponse, error) {
	return sc.RebuildActivityContext(context.Background(), node, rebuildRequest)
}

// RebuildActivityContext is the context aware version of RebuildActivity.
func (sc *SnowthClient) RebuildActivityContext(ctx context.Context,
	node *SnowthNode,
	rebuildRequest []RebuildActivityRequest,
) (*IRONdbPutResponse, error) {
	data, err := encodeJSON(rebuildRequest)
	if err != nil {
		return nil, err
	}

	body, _, err := sc.DoRequestContext(ctx, node, "POST",
		"/surrogate/activity_rebuild", data, nil)
	if err != nil {
		return nil, err
	}

	r := &IRONdbPutResponse{}
	if err := decodeJSON(body, &r); err != nil {
		return nil, fmt.Errorf("unable to decode IRONdb response: %w", err)
	}

	return r, nil
}