Check enqueueing result

Signed-off-by: MyonKeminta <MyonKeminta@users.noreply.github.com>
MyonKeminta · Jul 5, 2024 · a533754 · a533754
1 parent d25a8c8
commit a533754
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/client/tso_stream.go b/client/tso_stream.go
@@ -279,10 +279,12 @@ func (s *tsoStream) getServerURL() string {
 func (s *tsoStream) processRequests(reqID uint64,
 	clusterID uint64, keyspaceID, keyspaceGroupID uint32, dcLocation string, count int64, batchStartTime time.Time,
 ) error {
-	s.pendingReqIDs.Push(batchedReq{
+	if !s.pendingReqIDs.Push(batchedReq{
 		reqID:     reqID,
 		startTime: time.Now(),
-	})
+	}) {
+		return errors.New("channel full")
+	}
 
 	if err := s.stream.Send(clusterID, keyspaceID, keyspaceGroupID, dcLocation, count); err != nil {
 		if err == io.EOF {