Optimizing internal rpc

Author: dobyte

internal/transporter/internal/client/client.go

@@ -37,6 +37,7 @@ func NewClient(opts *Options) *Client {
 	c.opts = opts
 	c.connections = make([]*Conn, 0, defaultConnNum)
 	c.disorderlyQueue = make(chan *chWrite, 10240)
+	c.closed.Store(true)
 	c.pool = sync.Pool{New: func() any { return &chWrite{} }}
 
 	return c
@@ -74,6 +75,8 @@ func (c *Client) Establish() error {
 		return err
 	}
 
+	c.closed.Store(false)
+
 	return nil
 }
 

internal/transporter/internal/client/conn.go

@@ -127,10 +127,11 @@ func (c *Conn) handshake(conn net.Conn) error {
 
 	c.pending.store(seq, call)
 
-	defer close(call)
-
 	if _, err := conn.Write(buf.Bytes()); err != nil {
+		close(call)
+
 		c.pending.delete(seq)
+
 		return err
 	}