Skip to content

Commit

Permalink
feat: add cluster getPeerClient() func
Browse files Browse the repository at this point in the history
  • Loading branch information
stream1080 committed Oct 1, 2023
1 parent 7f0878e commit 3d868e0
Showing 1 changed file with 26 additions and 0 deletions.
26 changes: 26 additions & 0 deletions cluster/com.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
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
}

0 comments on commit 3d868e0

Please sign in to comment.