store

type IStore interface {
	AddRegistration(ctx context.Context, req *protos.RegisterRequest) error
	DeleteRegistration(ctx context.Context, req *protos.DeregisterRequest) error
	RecordRegistration(ctx context.Context, req *protos.RegisterRequest) error
	SeenRegistration(ctx context.Context, req *protos.RegisterRequest) bool
	GetPipelines(ctx context.Context) (map[string]*protos.Pipeline, error)
	GetPipeline(ctx context.Context, pipelineID string) (*protos.Pipeline, error)
	GetAudienceMappings(ctx context.Context) (map[*protos.Audience]*protos.PipelineConfigs, error)
	GetPipelineConfigsByAudience(ctx context.Context, aud *protos.Audience) (*protos.PipelineConfigs, error)
	GetLive(ctx context.Context) ([]*types.LiveEntry, error)
	CreatePipeline(ctx context.Context, pipeline *protos.Pipeline) error
	AddAudience(ctx context.Context, req *protos.NewAudienceRequest) error

	// CreateAudience creates a new audience in the store. This method differs
	// from AddAudience() in that it does not create any live entry.
	CreateAudience(ctx context.Context, aud *protos.Audience) error
	DeleteAudience(ctx context.Context, req *protos.DeleteAudienceRequest) error
	DeletePipeline(ctx context.Context, pipelineID string) error
	UpdatePipeline(ctx context.Context, pipeline *protos.Pipeline) error
	SetPauseResume(ctx context.Context, audience *protos.Audience, pipelineID string, pause bool) (bool, error)
	GetAudiences(ctx context.Context) ([]*protos.Audience, error)
	GetNotificationConfig(ctx context.Context, req *protos.GetNotificationRequest) (*protos.NotificationConfig, error)
	GetNotificationConfigs(ctx context.Context) (map[string]*protos.NotificationConfig, error)
	CreateNotificationConfig(ctx context.Context, req *protos.CreateNotificationRequest) error
	UpdateNotificationConfig(ctx context.Context, req *protos.UpdateNotificationRequest) error
	DeleteNotificationConfig(ctx context.Context, req *protos.DeleteNotificationRequest) error
	GetPipelineUsage(ctx context.Context) ([]*PipelineUsage, error)
	GetActivePipelineUsage(ctx context.Context, pipelineID string) ([]*PipelineUsage, error)
	GetActiveTailCommandsByService(ctx context.Context, serviceName string) ([]*protos.Command, error)
	AddActiveTailRequest(ctx context.Context, req *protos.TailRequest) (string, error) // Returns key that tail req is stored under

	// GetAudiencesBySessionID returns all audiences for a given session id
	// This is needed to inject an inferschema pipeline for each announced audience
	// to the session via a goroutine in internal.Register()
	GetAudiencesBySessionID(ctx context.Context, sessionID string) ([]*protos.Audience, error)

	GetAudiencesByService(ctx context.Context, serviceName string) ([]*protos.Audience, error)

	// WatchKeys will watch for key changes for given key pattern; every time key
	// is updated, it will send a message on the return channel.
	// WatchKeys launches a dedicated goroutine for the watch - it can be stopped
	// via the provided context.
	WatchKeys(quitCtx context.Context, key string) chan struct{}

	AddSchema(ctx context.Context, req *protos.SendSchemaRequest) error

	GetSchema(ctx context.Context, aud *protos.Audience) (*protos.Schema, error)

	// GetInstallID returns the unique ID for this server cluster.
	// If an ID has not been set yet, a new one is generated and returned
	GetInstallID(ctx context.Context) (string, error)

	// GetCreationDate returns the creation date of this server cluster. This is used
	// for sending server_timestamp_created_seconds metric to telemetry
	GetCreationDate(ctx context.Context) (int64, error)

	// SetCreationDate sets the creation date of this server cluster
	SetCreationDate(ctx context.Context, ts int64) error

	// IsPipelineAttachedAny returns if pipeline is attached to any audience. Used for telemetry tags
	IsPipelineAttachedAny(ctx context.Context, pipelineID string) bool

	// PauseTailRequest pauses a tail request by its ID
	PauseTailRequest(ctx context.Context, req *protos.PauseTailRequest) (*protos.TailRequest, error)

	// ResumeTailRequest resumes a tail request by its ID
	ResumeTailRequest(ctx context.Context, req *protos.ResumeTailRequest) (*protos.TailRequest, error)

	// GetTailRequestById returns a TailRequest by its ID
	GetTailRequestById(ctx context.Context, tailID string) (*protos.TailRequest, error)

	// SetPipelines saves pipelines as SetPipelinesConfig json to $audience key in store
	SetPipelines(ctx context.Context, req *protos.SetPipelinesRequest) error

	// GetSetPipelinesCommandsByService returns a slice of SetPipelines commands for a given service
	GetSetPipelinesCommandsByService(ctx context.Context, serviceName string) ([]*protos.Command, error)

	// GetSessionIDs returns a slice of ALL known session ids;
	// accepts an optional "node name" to filter by.
	GetSessionIDs(ctx context.Context, nodeName ...string) ([]string, error)

	// GetSessionIDsByAudience returns a slice of session IDs for a given audience;
	// accepts an optional "node name" to filter by.
	GetSessionIDsByAudience(ctx context.Context, aud *protos.Audience, nodeName ...string) ([]string, error)

	// GetSessionIDsByPipelineID returns a slice of session IDs that use a pipeline ID
	GetSessionIDsByPipelineID(ctx context.Context, pipelineID string) ([]string, error)

	// GetPipelinesByAudience will fetch pipeline configs and then use the ID to
	// fetch the actual pipeline from store + update the paused state.
	GetPipelinesByAudience(ctx context.Context, aud *protos.Audience) ([]*protos.Pipeline, error)

	// DeletePipelineConfig deletes a pipeline config for all audiences that use
	// given pipeline ID; it will return a list of audiences that the pipeline id
	// was used by.
	DeletePipelineConfig(ctx context.Context, pipelineID string) ([]*protos.Audience, error)

	// GetAudiencesByPipelineID returns a slice of audiences that use a pipeline ID
	GetAudiencesByPipelineID(ctx context.Context, pipelineID string) ([]*protos.Audience, error)

	// BEGIN Wasm-related methods
	GetAllWasm(ctx context.Context) ([]*shared.WasmModule, error)
	GetWasm(ctx context.Context, name, id string) (*shared.WasmModule, error)
	GetWasmByID(ctx context.Context, id string) (*shared.WasmModule, error)
	GetWasmByName(ctx context.Context, name string) (*shared.WasmModule, error)
	SetWasm(ctx context.Context, id, name string, wasm *shared.WasmModule) error
	SetWasmByName(ctx context.Context, name string, wasm *shared.WasmModule) error
	SetWasmByID(ctx context.Context, id string, wasm *shared.WasmModule) error
	DeleteWasm(ctx context.Context, id, name string) error
	DeleteWasmByID(ctx context.Context, id string) error
	DeleteWasmByName(ctx context.Context, name string) error

	// GetPipelinesByWasmID returns a slice of pipelines that use a given Wasm ID
	GetPipelinesByWasmID(ctx context.Context, wasmID string) ([]*protos.Pipeline, error)
}

type Option struct {
	// When set, the store will NOT overwrite an existing object. Applies to write().
	NoOverwrite bool

	// When set, the store will ensure that the object exists before performing
	// the operation. Applies to write() and delete(). NOTE: MustExist
	// and MustNotExist are mutually exclusive.
	MustExist bool

	// Optional TTL to set on the key. Applies to write().
	TTL time.Duration
}

type Options struct {
	Encryption   encryption.IEncryption
	RedisBackend *redis.Client
	ShutdownCtx  context.Context
	NodeName     string
	SessionTTL   time.Duration
	Telemetry    statsd.Statter
	InstallID    string
}

type PipelineUsage struct {
	PipelineId string
	Active     bool
	NodeName   string
	SessionId  string
	Audience   *protos.Audience
}

type Store struct {
	// contains filtered or unexported fields
}