go/coordinator/internal/notification/notification_store.go

package notification

import (
	"context"

	"github.com/chroma/chroma-coordinator/internal/model"
)

type NotificationStore interface {
	GetAllPendingNotifications(ctx context.Context) (map[string][]model.Notification, error)
	GetNotifications(ctx context.Context, collecitonID string) ([]model.Notification, error)
	AddNotification(ctx context.Context, notification model.Notification) error
	RemoveNotifications(ctx context.Context, notifications []model.Notification) error
}