persistence

type AnyMap map[string]any

type Db interface {
	// NewModelTx creates new ModelTx object
	NewModelTx() ModelTx
	// NewTx creates Tx object
	NewTx() Tx
}

type Index struct {
	ID        string    `db:"id"`
	Format    string    `db:"format"`
	Tags      AnyMap    `db:"tags"`
	CreatedAt time.Time `db:"created_at"`
}

type IndexRecord struct {
	ID      string `db:"id"`
	IndexID string `db:"index_id"`
	Segment string `db:"segment"`
	Vector  AnyMap `db:"vector"`
}

type ModelTx interface {
	Tx

	// CreateIndex creates index entry
	CreateIndex(index Index) (string, error)

	// CreateIndexRecord creates index record entry
	CreateIndexRecord(record IndexRecord) (string, error)
}

type QueryResult[T any] struct {
	Items []T
	Total int64
}

type SqlTestSuite struct {
	Dir string
	suite.Suite
	// contains filtered or unexported fields
}

type Tx interface {
	// MustBegin starts the transaction
	MustBegin()
	// MustBeginSerializable starts new transaction with Serializable isolation level
	MustBeginSerializable(ctx context.Context)
	// Commit commits the changes made within the transaction
	Commit() error
	// Rollback rolls the transaction back
	Rollback() error
	// ExecScript allows to execute the sql statements from the file provided
	ExecScript(sqlScript string) error
}