database

type Batch interface {
	GetBatch() any
	Len() int
	Queue(query string, arguments ...any)
}

type ConnConfig struct {
	VpcDirectConnection bool
	Host                string
	Port                int32
	DBName              string
	User                string
	Password            string
	MaxConn             int32
	IsLocalEnv          bool
}

type DbShard string

type DefaultResultSet struct {
	Rows     []Row
	RowsScan []RowScan
}

type InstanceManager interface {
	GetDbConnPool() (any, error)
	GetConnFromPool(ctx context.Context) (any, error)
	CloseDbConnPool()
	GetConnectionConfig() ConnConfig
	Query(ctx context.Context, lockId int64, query string, args ...any) (ResultSet, error)
	QueryAndClose(ctx context.Context, lockId int64, query string, args ...any) (ResultSet, error)
	QueryAndProcess(ctx context.Context, lockId int64, processCallback func(row Row, rowScan RowScan) error, query string, args ...any) error
	Exec(ctx context.Context, lockId int64, execQuery string, args ...any) (int64, error)
	TxBegin(ctx context.Context, lockId int64) (Transaction, error)
	ExecTransactionalTask(ctx context.Context, lockId int64, task func(ctx context.Context, tx Transaction) error) error
	AcquireDistributedLock(ctx context.Context, lockId int64) (int64, error)
	ReleaseDistributedLock(ctx context.Context, lockId int64) error
	ExecTaskWithDistributedLock(ctx context.Context, lockId int64, task func(ctx context.Context) error) error
}

type MasterReplicaManager struct {
	DbMaster  InstanceManager
	DbReplica InstanceManager
}

type PreparedStatement struct {
	Name  string
	Query string
}

type PreparedStatementsMap struct {
	DbPrepStmMap map[DbShard][]PreparedStatement
}

type ResultSet interface {
	GetRow(rowIdx int) (Row, error)
	GetRows() []Row
	GetRowScan(rowIdx int) (RowScan, error)
	GetRowsScan() []RowScan
	Close()
}

type Row []any

type RowScan interface {
	Scan(dest ...any) error
}

type ShardManager struct {
	DbShardMap map[DbShard]MasterReplicaManager
}

type Transaction interface {
	TxCommit(ctx context.Context, lockId int64) error
	TxRollback(ctx context.Context, lockId int64)
	TxQuery(ctx context.Context, query string, args ...any) (ResultSet, error)
	TxQueryAndClose(ctx context.Context, query string, args ...any) (ResultSet, error)
	TxExec(ctx context.Context, query string, args ...any) (int64, error)
	TxExecBatch(ctx context.Context, batch Batch) (int64, error)
}