core

type Accessors interface {
	Shards() ShardContext
	Tenants() TenantContext
	DataNodes() DataNodeContext
	Tables() TableContext
	Schema() SchemaContext
	Setting() SettingContext
	Types() TypeContext
	Users() UserContext
	Pool() PoolContext
	Sequences() SequenceContext
}

type Colony interface {
	Accessors

	Query(string) (*frunk.QueryResponse, error)

	CoordinatorID() uint64
	IsLeader() bool
	Close()
	Addr() net.Addr
	Join(id, addr string) error
	JoinCluster() error
	LeaderID() (string, string, error)
	State() frunk.ClusterState

	Neighbors() ([]*frunk.Server, error)

	InitColony(config ColonyConfig, log timber.Logger) error
}

type ColonyConfig struct {
	DataDirectory         string
	JoinAddresses         []raft.Server
	Transport             TransportWrapper
	LocalPostgresAddress  string
	LocalPostgresPort     int32
	LocalPostgresUser     string
	LocalPostgresPassword string
	StartPool             bool
	AutoJoin              bool
}

type DataNodeContext interface {
	GetDataNode(id uint64) (DataNode, error)
	GetDataNodes() ([]DataNode, error)
	GetDataNodesForShard(uint64) ([]DataNode, error)
	GetDataNodeForDataNodeShard(uint64) (DataNode, error)
	GetRandomDataNodeShardID() (uint64, error)
	GetDataNodeShardIDs() ([]uint64, error)
	GetDataNodeShardIDsForShard(uint64) ([]uint64, error)
	NewDataNode(address string, port int32, user string, password string) (DataNode, error)
}

type DataNodePressure struct {
	DataNodeID uint64
	Shards     int
}

type FunctionContext interface {
}

type Listener interface {
	net.Listener
	Dial(address string, timeout time.Duration) (net.Conn, error)
}

type PoolConnection interface {
	Release()
	ID() uint64
	// contains filtered or unexported methods
}

type PoolContext interface {
	StartPool()
	GetConnectionForDataNodeShard(id uint64) (PoolConnection, error)
}

type SchemaContext interface {
	GetSchemas() ([]Schema, error)
	Exists(string) (bool, error)
	NewSchema(string) (Schema, error)
}

type SequenceContext interface {
	GetSequenceChunk(name string) (*frunk.SequenceChunkResponse, error)
}

type SettingContext interface {
	GetSetting(SettingKeyOptions) (Setting, bool, error)
	GetSettingValue(key SettingKeyOptions) (interface{}, bool, error)
}

type ShardContext interface {
	NewShard() (Shard, error)
	GetShards() ([]Shard, error)
	GetDataNodeShards() ([]DataNodeShard, error)
	GetWriteDataNodeShards(uint64) ([]DataNodeShard, error)
	BalanceOrphanShards() error
	GetDataNodesPressure(max int) ([]DataNodePressure, error)
	GetShardPressures(max int) ([]ShardPressure, error)
}

type ShardPressure struct {
	ShardID uint64
	Tenants int
}

type TableContext interface {
	NewTable(table Table, columns []Column) (Table, []Column, error)
	NextSequenceID(table Table, column Column) (uint64, error)
	GetTable(name string) (Table, bool, error)
	GetTables(...string) ([]Table, error)
	GetColumns(tableId uint64) ([]Column, error)
	GetPrimaryKeyColumnByName(name string) (Column, bool, error)
	GetSequenceColumnForTable(tableId uint64) (Column, bool, error)
	GetShardKeyColumnForTable(uint64) (Column, error)
	GetTablesInSchema(schema string, names ...string) ([]Table, error)
	GetTenantTable() (Table, bool, error)
	GetColumnFromTables(column string, tables []string) (Column, bool, error)
}

type TenantContext interface {
	GetTenants() ([]Tenant, error)
	GetTenant(uint64) (Tenant, error)
	NewTenants(tenantIds ...uint64) ([]Tenant, error)
}

type TransportWrapper interface {
	NormalTransport() net.Listener
	ForwardToRaft(net.Conn, error)
	ForwardToRpc(net.Conn, error)
	RaftTransport() Listener
	RpcTransport() Listener
	Port() int
	Addr() net.Addr
	Close()
}

type TypeContext interface {
	GetTypeByName(name string) (types.Type, bool, error)
	GetTypeByOid(oid types.OID) (types.Type, bool)
}

type UserContext interface {
	GetUsers() ([]User, error)
}