dilithium

type Closer interface {
	Close()
}

type ForwardingTable struct {
	llrb.Tree
	sync.RWMutex
}

type ForwardingTableEntry struct {
	MaxKey int
	Shard  Shard
}

type PhysicalShard struct {
	sync.RWMutex
	// contains filtered or unexported fields
}

type Pool struct {
	// Dial is an application supplied function for creating new connections.
	Dial func(url string) (Closer, error)

	// TestOnBorrow is an optional application supplied function for checking
	// the health of an idle connection before the connection is used again by
	// the application. Argument t is the time that the connection was returned
	// to the pool. If the function returns an error, then the connection is
	// closed.
	TestOnBorrow func(c Closer, t time.Time) error

	// Maximum number of idle connections in the pool.
	MaxIdle int

	// Close connections after remaining idle for this duration. If the value
	// is zero, then idle connections are not closed. Applications should set
	// the timeout to a value less than the server's timeout.
	IdleTimeout time.Duration
	// contains filtered or unexported fields
}

type Query struct {
	Method string
	Arg    QueryArg
	Reply  interface{}
	// contains filtered or unexported fields
}

type QueryArg interface {
	ShardKey() int
}

type ReplicateShard struct {
	sync.RWMutex
	// contains filtered or unexported fields
}

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

type Shard interface {
	// Get the parent of the shard. nil if the Shard is the root.
	Parent() Shard
	Children() []Shard
	SetParent(s Shard)
	AddChild(s Shard)
	// RemoveChild removes the child shard with ID() == id.
	// The receiver is responsible for calling Destroy() on the child shard.
	RemoveChild(id string)
	Query(q *Query) error
	// ID returns the shard ID. It must uniquely identify the shard.
	ID() string
	// Setup is called to setup the shard with the specified config.
	Setup(config map[string]interface{}) error
	// Config returns a JSON-serializable config that could be passed to Setup to recreate this shard.
	Config() map[string]interface{}
	// Destroy is called when the shard is removed from the tree. The shard must call Destroy on child shards.
	Destroy()
	// The shard must be locked before unmarshalling the configuration into it.
	sync.Locker
}

type ShardConfig struct {
	Type     string                 `json:"type"`
	Config   map[string]interface{} `json:"config"`
	Children []ShardConfig          `json:"children"`
}

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