package memdb import ( "context" "fmt" "github.com/influxdata/influxdb/chronograf" ) // Ensure KapacitorStore implements chronograf.ServersStore. var _ chronograf.ServersStore = &KapacitorStore{} // KapacitorStore implements the chronograf.ServersStore interface, and keeps // an in-memory Kapacitor according to startup configuration type KapacitorStore struct { Kapacitor *chronograf.Server } // All will return a slice containing a configured source func (store *KapacitorStore) All(ctx context.Context) ([]chronograf.Server, error) { if store.Kapacitor != nil { return []chronograf.Server{*store.Kapacitor}, nil } return nil, nil } // Add does not have any effect func (store *KapacitorStore) Add(ctx context.Context, kap chronograf.Server) (chronograf.Server, error) { return chronograf.Server{}, fmt.Errorf("in-memory KapacitorStore does not support adding a Kapacitor") } // Delete removes the in-memory configured Kapacitor if its ID matches what's provided func (store *KapacitorStore) Delete(ctx context.Context, kap chronograf.Server) error { if store.Kapacitor == nil || store.Kapacitor.ID != kap.ID { return fmt.Errorf("unable to find Kapacitor with id %d", kap.ID) } store.Kapacitor = nil return nil } // Get returns the in-memory Kapacitor if its ID matches what's provided func (store *KapacitorStore) Get(ctx context.Context, id int) (chronograf.Server, error) { if store.Kapacitor == nil || store.Kapacitor.ID != id { return chronograf.Server{}, fmt.Errorf("unable to find Kapacitor with id %d", id) } return *store.Kapacitor, nil } // Update overwrites the in-memory configured Kapacitor if its ID matches what's provided func (store *KapacitorStore) Update(ctx context.Context, kap chronograf.Server) error { if store.Kapacitor == nil || store.Kapacitor.ID != kap.ID { return fmt.Errorf("unable to find Kapacitor with id %d", kap.ID) } store.Kapacitor = &kap return nil }