db

type ContractConfigCollection interface {
	SetContractConfig(
		ctx context.Context,
		config *model.JobContractConfig,
	) error

	GetContractConfig(
		ctx context.Context,
	) (*model.JobContractConfig, error)
}

type DBService interface {
	JobCollection

	DBName() string
	Client() *mongo.Client
	Connection() dbconn.Conn
	String() string
	Close()
}

type ExternalGorm interface {
	CreateJob(ctx context.Context, job *model.Job) error
	LoadJobs(ctx context.Context) ([]*model.Job, error)
	DeleteJob(ctx context.Context, jobID string) error

	Client() *gorm.DB
	Connection() (*sql.DB, error)
	String() string
}

type JobCollection interface {
	UpsertJob(
		ctx context.Context,
		spec *model.Job,
	) error

	DeleteJob(
		ctx context.Context,
		jobID model.ID,
	) error

	ListJobs(
		ctx context.Context,
		cursor *model.Cursor,
	) ([]*model.Job, error)
}

type JobDBService interface {
	PersistentStateCollection
	ContractConfigCollection
	PendingTransmissionCollection
	PeerAnnouncementCollection

	JobID() model.ID
	DBName() string
	Client() *mongo.Client
	Close()
}

type PeerAnnouncementCollection interface {
	UpsertAnnouncement(
		ctx context.Context,
		ann *model.JobPeerAnnouncement,
	) error

	ListAnnouncements(
		ctx context.Context,
		peerIDs []string,
		cursor *model.Cursor,
	) ([]*model.JobPeerAnnouncement, error)
}

type PendingTransmissionCollection interface {
	InsertPendingTranmission(
		ctx context.Context,
		pendingTx *model.JobPendingTransmission,
	) error

	ListPendingTransmissions(
		ctx context.Context,
		configDigest model.ID,
		cursor *model.Cursor,
	) ([]*model.JobPendingTransmission, error)

	DeletePendingTransmission(
		ctx context.Context,
		reportTimestamp model.ReportTimestamp,
	) error

	DeletePendingTransmissionsOlderThan(
		ctx context.Context,
		timestamp time.Time,
	) error
}

type PersistentStateCollection interface {
	SetPersistentState(
		ctx context.Context,
		state *model.JobPersistentState,
	) error

	GetPersistentState(
		ctx context.Context,
		configDigest model.ID,
	) (*model.JobPersistentState, error)
}