Optimize lease get state for proxy (#27)

proxy/pool.go

@@ -0,0 +1,24 @@
+package proxy
+
+import (
+	"sync"
+)
+
+// ===============================
+// pool of lease get state
+// ===============================
+
+var leaseGetStatePool = sync.Pool{
+	New: func() any {
+		return &leaseGetState{}
+	},
+}
+
+func putLeaseGetState(s *leaseGetState) {
+	*s = leaseGetState{}
+	leaseGetStatePool.Put(s)
+}
+
+func getLeaseGetState() *leaseGetState {
+	return leaseGetStatePool.Get().(*leaseGetState)
+}

proxy/pool_test.go

@@ -0,0 +1,20 @@
+package proxy
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestLeaseGetStatePool(t *testing.T) {
+	t.Run("normal", func(t *testing.T) {
+		s := getLeaseGetState()
+		assert.Equal(t, &leaseGetState{}, s)
+
+		s.serverID = 123
+
+		putLeaseGetState(s)
+		s = getLeaseGetState()
+		assert.Equal(t, &leaseGetState{}, s)
+	})
+}

proxy/proxy.go

@@ -239,7 +239,12 @@ func (s *leaseGetState) nextFunc() {
 func (s *leaseGetState) Result() (memproxy.LeaseGetResponse, error) {
 	s.pipe.sess.Execute()
 	s.pipe.selector.Reset()
-	return s.resp, s.err
+
+	resp, err := s.resp, s.err
+
+	putLeaseGetState(s)
+
+	return resp, err
 }
 
 // LeaseGet ...
@@ -251,7 +256,8 @@ func (p *Pipeline) LeaseGet(
 	pipe := p.getRoutePipeline(serverID)
 	fn := pipe.LeaseGet(key, options)
 
-	state := &leaseGetState{
+	state := getLeaseGetState()
+	*state = leaseGetState{
 		pipe:     p,
 		serverID: serverID,
 		key:      key,