74 lines
2.7 KiB
Go
74 lines
2.7 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"iter"
|
|
"time"
|
|
|
|
"gitea.libretechconsulting.com/rmcguire/wingbits/pkg/types/readsb"
|
|
"gitea.libretechconsulting.com/rmcguire/wingbits/pkg/types/wingbits"
|
|
)
|
|
|
|
// Poll turns any fetch into a pull-based stream: an iter.Seq2[R, error] that
|
|
// yields a fresh result immediately, then once per interval, until ctx is
|
|
// cancelled or the consumer breaks out of the range. Each step yields exactly
|
|
// one of a value or the error from that poll.
|
|
//
|
|
// It is the engine behind every Poll* method, exported so the same cadence can
|
|
// be applied to a custom or composed fetch without re-implementing the loop.
|
|
// Because the iterator runs in the consumer's goroutine, breaking the range
|
|
// stops polling immediately — there is no background goroutine to leak.
|
|
func Poll[R any](ctx context.Context, interval time.Duration, fetch func(context.Context) (R, error)) iter.Seq2[R, error] {
|
|
return func(yield func(R, error) bool) {
|
|
tick := time.NewTicker(interval)
|
|
defer tick.Stop()
|
|
for {
|
|
if !yield(fetch(ctx)) {
|
|
return
|
|
}
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
case <-tick.C:
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// PollAircraft polls aircraft.json every interval, applying the given filters.
|
|
func (c *Client) PollAircraft(ctx context.Context, interval time.Duration, filters ...readsb.AircraftFilter) iter.Seq2[*readsb.AircraftReport, error] {
|
|
return Poll(ctx, interval, func(ctx context.Context) (*readsb.AircraftReport, error) {
|
|
return c.Aircraft(ctx, filters...)
|
|
})
|
|
}
|
|
|
|
// PollStats polls stats.json every interval.
|
|
func (c *Client) PollStats(ctx context.Context, interval time.Duration) iter.Seq2[*readsb.Stats, error] {
|
|
return Poll(ctx, interval, c.Stats)
|
|
}
|
|
|
|
// PollReceiver polls receiver.json every interval.
|
|
func (c *Client) PollReceiver(ctx context.Context, interval time.Duration) iter.Seq2[*readsb.Receiver, error] {
|
|
return Poll(ctx, interval, c.Receiver)
|
|
}
|
|
|
|
// PollOutline polls outline.json every interval.
|
|
func (c *Client) PollOutline(ctx context.Context, interval time.Duration) iter.Seq2[*readsb.Outline, error] {
|
|
return Poll(ctx, interval, c.Outline)
|
|
}
|
|
|
|
// PollDiagnostics polls /network/diagnostics every interval.
|
|
func (c *Client) PollDiagnostics(ctx context.Context, interval time.Duration) iter.Seq2[*wingbits.Diagnostics, error] {
|
|
return Poll(ctx, interval, c.Diagnostics)
|
|
}
|
|
|
|
// PollMetrics polls /metrics every interval.
|
|
func (c *Client) PollMetrics(ctx context.Context, interval time.Duration) iter.Seq2[*wingbits.Metrics, error] {
|
|
return Poll(ctx, interval, c.Metrics)
|
|
}
|
|
|
|
// PollStatus polls the Tailscale status every interval.
|
|
func (c *Client) PollStatus(ctx context.Context, interval time.Duration) iter.Seq2[*wingbits.TailscaleStatus, error] {
|
|
return Poll(ctx, interval, c.Status)
|
|
}
|