interfaces

type Client interface {
	// GetGRPCConn returns a configured GRPC connection
	GetGRPCConn() *grpc.ClientConn
	// NotifyShutdown sends a notification to the channel when connection has errors
	NotifyShutdown(ctx context.Context, stopChannel chan<- struct{}, waitForStateChange time.Duration)
}

type ConnectionManager interface {
	Starter
	Stopper
}

type Controller interface {
	Starter
	Stopper
	// Save information about saved jetdrops
	SetJetDropData(pulse types.Pulse, jetID string)
}

type JetDropsExtractor interface {
	Starter
	Stopper
	// GetJetDrops stores JetDrop data in the main JetDrop channel
	GetJetDrops(ctx context.Context) <-chan *types.PlatformPulseData
	// LoadJetDrops loads JetDrop data between pulse numbers: (fromPulseNumber, toPulseNumber]
	LoadJetDrops(ctx context.Context, fromPulseNumber int64, toPulseNumber int64) error
}

type Processor interface {
	Starter
	Stopper
}

type PulseExtractor interface {
	// GetCurrentPulse returns the current Pulse number
	GetCurrentPulse(ctx context.Context) (uint32, error)
	// GetNextFinalizedPulse requests full pulse info
	GetNextFinalizedPulse(ctx context.Context, p int64) (*exporter.FullPulse, error)
}

type Starter interface {
	// Start starts the main thread
	Start(ctx context.Context) error
}

type Stopper interface {
	// Stops stops the main thread
	Stop(ctx context.Context) error
}

type Storage interface {
	StorageSetter
	StorageFetcher
	StorageExporterFetcher
}

type StorageAPIFetcher interface {
	// GetRecord returns record with provided reference from db.
	GetRecord(ref models.Reference) (models.Record, error)
	// GetPulse returns pulse with provided pulse number from db.
	GetPulse(pulseNumber int64) (models.Pulse, error)
	// GetPulse returns pulses from db.
	GetPulses(fromPulse *int64, timestampLte, timestampGte, pulseNumberLte, pulseNumberLt, pulseNumberGte, pulseNumberGt *int64, sortByAsc bool, limit, offset int) ([]models.Pulse, int, error)
	// GetJetDropsWithParams returns jetDrops for provided pulse with limit and offset.
	GetJetDropsWithParams(pulse models.Pulse, fromJetDropID *models.JetDropID, limit int, offset int) ([]models.JetDrop, int, error)
	// GetJetDropByID returns JetDrop by JetDropID, with slices of previous jetDrops and next jetDrops.
	GetJetDropByID(id models.JetDropID) (models.JetDrop, []models.JetDrop, []models.JetDrop, error)
	// GetJetDropsByJetID returns jetDrops for provided jetID sorting and filtering by pulseNumber.
	GetJetDropsByJetID(jetID string, pulseNumberLte, pulseNumberLt, pulseNumberGte, pulseNumberGt *int64, limit int, sortByPnAsc bool) ([]models.JetDrop, int, error)
	// GetLifeline returns records for provided object reference, ordered by desc by pulse number and order fields.
	GetLifeline(objRef []byte, fromIndex *string, pulseNumberLt, pulseNumberGt, timestampLte, timestampGte *int64, limit, offset int, sortByIndexAsc bool) ([]models.Record, int, error)
	// GetRecordsByJetDrop returns records for provided jet drop, ordered by order field.
	GetRecordsByJetDrop(jetDropID models.JetDropID, fromIndex, recordType *string, limit, offset int) ([]models.Record, int, error)
	// GetNextSavedPulse returns first pulse with pulse number bigger then fromPulseNumber from db.
	GetNextSavedPulse(fromPulseNumber models.Pulse, completedOnly bool) (models.Pulse, error)
	// GetJetDrops returns jetDrops for provided pulse from db.
	GetJetDrops(pulse models.Pulse) ([]models.JetDrop, error)
}

type StorageExporterFetcher interface {
	GetNextCompletePulseFilterByPrototypeReference(prevPulse int64, prototypes [][]byte) (models.Pulse, error)
}

type StorageFetcher interface {
	// GetIncompletePulses returns pulses that are not complete from db.
	GetIncompletePulses() ([]models.Pulse, error)
	// GetSequentialPulse returns max pulse that have is_sequential as true from db.
	GetSequentialPulse() (models.Pulse, error)
	// GetPulseByPrev returns pulse with provided prev pulse number from db.
	GetPulseByPrev(prevPulse models.Pulse) (models.Pulse, error)
	// GetNextSavedPulse returns first pulse with pulse number bigger then fromPulseNumber from db.
	GetNextSavedPulse(fromPulseNumber models.Pulse, completedOnly bool) (models.Pulse, error)
	// GetJetDrops returns jetDrops for provided pulse from db.
	GetJetDrops(pulse models.Pulse) ([]models.JetDrop, error)
}

type StorageSetter interface {
	// SaveJetDropData saves provided jetDrop and records to db in one transaction.
	// increase jet_drop_amount and record_amount
	SaveJetDropData(jetDrop models.JetDrop, records []models.Record, pulseNumber int64) error
	// SavePulse saves provided pulse to db.
	SavePulse(pulse models.Pulse) error
	// CompletePulse update pulse with provided number to completeness in db.
	CompletePulse(pulseNumber int64) error
	// SequencePulse update pulse with provided number to sequential in db.
	SequencePulse(pulseNumber int64) error
}

type Transformer interface {
	Starter
	Stopper
	// GetJetDropsChannel returns the channel where canonical data will be stored
	GetJetDropsChannel() <-chan *types.JetDrop
}