database

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

type CreateTableOption func(opts *CreateTableOptions)

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

type Database interface {
	Connect(ctx context.Context, cfg config.Database) error
	Exec(ctx context.Context, query string, args ...any) (int64, error)
	CreateTable(ctx context.Context, model any, opts ...CreateTableOption) error

	StateRepository
	SchemeCommenter

	driver.Pinger
	io.Closer
}

type PartitionBy int

type PostgreSQLContainer struct {
	testcontainers.Container
	Config PostgreSQLContainerConfig
	// contains filtered or unexported fields
}

type PostgreSQLContainerConfig struct {
	User     string
	Password string
	Database string
	Host     string
	Port     int
	Image    string
}

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

type SchemeCommenter interface {
	MakeTableComment(ctx context.Context, name string, comment string) error
	MakeColumnComment(ctx context.Context, tableName string, columnName string, comment string) error
}

type State struct {
	bun.BaseModel `gorm:"-" pg:"-" bun:"dipdup_state" json:"-" comment:"Indexer state table"`

	IndexName string    `gorm:"primaryKey" pg:",pk" bun:",pk" json:"index_name" comment:"Index name"`
	IndexType string    `json:"index_type" comment:"Index type"`
	Hash      string    `json:"hash" comment:"Current hash"`
	Timestamp time.Time `json:"timestamp" comment:"Current timestamp"`
	Level     uint64    `json:"level" comment:"Index level"`
	UpdatedAt int       `gorm:"autoUpdateTime" comment:"Last updated timestamp"`
	CreatedAt int       `gorm:"autoCreateTime" comment:"Created timestamp"`
	// contains filtered or unexported fields
}

type StateRepository interface {
	State(ctx context.Context, name string) (*State, error)
	UpdateState(sctx context.Context, tate *State) error
	CreateState(ctx context.Context, state *State) error
	DeleteState(ctx context.Context, state *State) error
}