implement grpc weather

pkg/weather/state/state.go

@@ -0,0 +1,41 @@
+package state
+
+import (
+	"context"
+	"sync"
+
+	"go.opentelemetry.io/otel/metric"
+	"go.opentelemetry.io/otel/trace"
+
+	"gitea.libretechconsulting.com/rmcguire/go-app/pkg/otel"
+
+	"gitea.libretechconsulting.com/rmcguire/ambient-local-exporter/pkg/weather"
+)
+
+type WeatherState struct {
+	updates []*weather.WeatherUpdate
+	keep    int
+	ctx     context.Context
+	tracer  trace.Tracer
+	meter   metric.Meter
+	*sync.RWMutex
+}
+
+type Opts struct {
+	Ctx      context.Context
+	KeepLast int
+}
+
+func NewWeatherState(opts *Opts) *WeatherState {
+	if opts.KeepLast < 1 {
+		opts.KeepLast = 1
+	}
+
+	return &WeatherState{
+		updates: make([]*weather.WeatherUpdate, 0),
+		keep:    opts.KeepLast,
+		ctx:     opts.Ctx,
+		tracer:  otel.GetTracer(opts.Ctx, "weatherState"),
+		meter:   otel.GetMeter(opts.Ctx, "weatherState"),
+	}
+}

pkg/weather/state/state_get.go

@@ -0,0 +1,87 @@
+package state
+
+import (
+	"context"
+	"errors"
+
+	"go.opentelemetry.io/otel/attribute"
+	"go.opentelemetry.io/otel/codes"
+	"go.opentelemetry.io/otel/trace"
+
+	"gitea.libretechconsulting.com/rmcguire/ambient-local-exporter/pkg/weather"
+)
+
+// Returns last requested number of weather updates
+func (w *WeatherState) Get(ctx context.Context, last int) (
+	[]*weather.WeatherUpdate, error,
+) {
+	if last < 1 {
+		last = 1
+	}
+
+	ctx, span := w.tracer.Start(ctx, "getWeatherState")
+	span.SetAttributes(
+		attribute.Int("last", last),
+		attribute.Int("keep", w.keep),
+		attribute.Int("currentSize", w.Count()),
+	)
+	defer span.End()
+
+	updates, err := w.get(ctx, last)
+	if err != nil {
+		span.RecordError(err)
+		span.SetStatus(codes.Error, err.Error())
+	} else {
+		span.SetStatus(codes.Ok, "")
+	}
+
+	return updates, err
+}
+
+func (w *WeatherState) get(ctx context.Context, last int) (
+	[]*weather.WeatherUpdate, error,
+) {
+	span := trace.SpanFromContext(ctx)
+
+	w.RLock()
+	defer w.Unlock()
+
+	span.AddEvent("acquired lock on state cache")
+
+	updates := w.updates
+
+	if w.count() == 0 {
+		err := errors.New("no state to get")
+		span.RecordError(err)
+		return nil, err
+	} else if w.count() <= last {
+		span.RecordError(errors.New("requested more state than exists"))
+	} else {
+		updates = updates[len(updates)-last:]
+	}
+
+	span.SetAttributes(attribute.Int("retrieved", len(updates)))
+	span.SetStatus(codes.Ok, "")
+
+	return updates, nil
+}
+
+// Returns count of retained weather updates
+func (w *WeatherState) Count() int {
+	_, span := w.tracer.Start(w.ctx, "countWeatherState")
+	defer span.End()
+
+	count := w.count()
+
+	span.SetAttributes(attribute.Int("count", count))
+	span.SetStatus(codes.Ok, "")
+
+	return count
+}
+
+func (w *WeatherState) count() int {
+	w.RLock()
+	defer w.RUnlock()
+
+	return len(w.updates)
+}

pkg/weather/state/state_set.go

@@ -0,0 +1,35 @@
+package state
+
+import (
+	"context"
+
+	"go.opentelemetry.io/otel/attribute"
+	"go.opentelemetry.io/otel/trace"
+
+	"gitea.libretechconsulting.com/rmcguire/ambient-local-exporter/pkg/weather"
+)
+
+func (w *WeatherState) Set(ctx context.Context, u *weather.WeatherUpdate) error {
+	_, span := w.tracer.Start(ctx, "setWeatherState")
+	span.SetAttributes(
+		attribute.Int("countWeatherUpdates", w.Count()),
+		attribute.Int("keepUpdates", w.keep),
+	)
+	defer span.End()
+
+	return w.set(span, u)
+}
+
+func (w *WeatherState) set(span trace.Span, u *weather.WeatherUpdate) error {
+	w.Lock()
+	defer w.Unlock()
+
+	if len(w.updates) > w.keep {
+		w.updates = w.updates[1:]
+		span.AddEvent("trimmed state updates by 1")
+	}
+
+	w.updates = append(w.updates, u)
+	span.AddEvent("recorded weather state")
+	return nil
+}