Merge pull request #56 from stream1080/feat/cluster

feat: add cluster com and databases

cluster/cluster_database.go

@@ -0,0 +1,55 @@
+package cluster
+
+import (
+	"context"
+	pool "github.com/jolestar/go-commons-pool/v2"
+
+	"github.com/stream1080/godis/config"
+	database2 "github.com/stream1080/godis/database"
+	"github.com/stream1080/godis/interface/database"
+	"github.com/stream1080/godis/interface/resp"
+	"github.com/stream1080/godis/lib/consistenthash"
+)
+
+type ClusterDatabases struct {
+	self     string
+	nodes    []string
+	peerPick *consistenthash.NodeMap
+	peerConn map[string]*pool.ObjectPool
+	db       database.Database
+}
+
+func MakeClusterDatabases() *ClusterDatabases {
+	cluster := &ClusterDatabases{
+		self:     config.Properties.Self,
+		db:       database2.NewStandaloneDatabase(),
+		peerPick: consistenthash.NewNodeMap(nil),
+		peerConn: make(map[string]*pool.ObjectPool),
+	}
+	nodes := make([]string, 0, len(config.Properties.Peers)+1)
+	for _, peer := range config.Properties.Peers {
+		nodes = append(nodes, peer)
+	}
+	nodes = append(nodes, config.Properties.Self)
+	cluster.peerPick.AddNode(nodes...)
+	ctx := context.Background()
+	for _, peer := range config.Properties.Peers {
+		cluster.peerConn[peer] = pool.NewObjectPoolWithDefaultConfig(ctx, &ConnFactory{
+			Peer: peer,
+		})
+	}
+	cluster.nodes = nodes
+	return cluster
+}
+
+func (cluster *ClusterDatabases) Exec(client resp.Connection, args [][]byte) resp.Reply {
+	panic("implement me")
+}
+
+func (cluster *ClusterDatabases) Close() {
+	panic("implement me")
+}
+
+func (cluster *ClusterDatabases) AfterClientClose(conn resp.Connection) {
+	panic("implement me")
+}

cluster/com.go

@@ -0,0 +1,35 @@
+package cluster
+
+import (
+	"context"
+	"errors"
+	"github.com/stream1080/godis/resp/client"
+)
+
+func (cluster *ClusterDatabases) getPeerClient(peer string) (*client.Client, error) {
+	pool, ok := cluster.peerConn[peer]
+	if !ok {
+		return nil, errors.New("conn not found")
+	}
+
+	object, err := pool.BorrowObject(context.Background())
+	if err != nil {
+		return nil, err
+	}
+
+	c, ok := object.(*client.Client)
+	if !ok {
+		return nil, errors.New("wrong type")
+	}
+
+	return c, nil
+}
+
+func (cluster *ClusterDatabases) returnPeerClient(peer string, client *client.Client) error {
+	pool, ok := cluster.peerConn[peer]
+	if !ok {
+		return errors.New("conn not found")
+	}
+
+	return pool.ReturnObject(context.Background(), client)
+}