mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-11 09:21:49 +01:00
765aba2c1c
propagate context from all the API calls and log request ID, account ID and peer ID --------- Co-authored-by: Zoltan Papp <zoltan.pmail@gmail.com>
58 lines
1.6 KiB
Go
58 lines
1.6 KiB
Go
package activity
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
)
|
|
|
|
// Store provides an interface to store or stream events.
|
|
type Store interface {
|
|
// Save an event in the store
|
|
Save(ctx context.Context, event *Event) (*Event, error)
|
|
// Get returns "limit" number of events from the "offset" index ordered descending or ascending by a timestamp
|
|
Get(ctx context.Context, accountID string, offset, limit int, descending bool) ([]*Event, error)
|
|
// Close the sink flushing events if necessary
|
|
Close(ctx context.Context) error
|
|
}
|
|
|
|
// InMemoryEventStore implements the Store interface storing data in-memory
|
|
type InMemoryEventStore struct {
|
|
mu sync.Mutex
|
|
nextID uint64
|
|
events []*Event
|
|
}
|
|
|
|
// Save sets the Event.ID to 1
|
|
func (store *InMemoryEventStore) Save(_ context.Context, event *Event) (*Event, error) {
|
|
store.mu.Lock()
|
|
defer store.mu.Unlock()
|
|
if store.events == nil {
|
|
store.events = make([]*Event, 0)
|
|
}
|
|
event.ID = store.nextID
|
|
store.nextID++
|
|
store.events = append(store.events, event)
|
|
return event, nil
|
|
}
|
|
|
|
// Get returns a list of ALL events that belong to the given accountID without taking offset, limit and order into consideration
|
|
func (store *InMemoryEventStore) Get(_ context.Context, accountID string, offset, limit int, descending bool) ([]*Event, error) {
|
|
store.mu.Lock()
|
|
defer store.mu.Unlock()
|
|
events := make([]*Event, 0)
|
|
for _, event := range store.events {
|
|
if event.AccountID == accountID {
|
|
events = append(events, event)
|
|
}
|
|
}
|
|
return events, nil
|
|
}
|
|
|
|
// Close cleans up the event list
|
|
func (store *InMemoryEventStore) Close(_ context.Context) error {
|
|
store.mu.Lock()
|
|
defer store.mu.Unlock()
|
|
store.events = make([]*Event, 0)
|
|
return nil
|
|
}
|