kgo: avoid overflow on 32 bit systems

twmb · Oct 11, 2022 · 719c6f4 · 719c6f4
1 parent db5c159
commit 719c6f4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pkg/kgo/internal/sticky/help_test.go b/pkg/kgo/internal/sticky/help_test.go
@@ -1,6 +1,7 @@
 package sticky
 
 import (
+	"math"
 	"testing"
 
 	"github.com/twmb/franz-go/pkg/kmsg"
@@ -69,7 +70,7 @@ func partitionsForMember(member map[string][]int32) int {
 func testEqualDivvy(t *testing.T, plan Plan, expSticky int, input []GroupMember) {
 	t.Helper()
 
-	min := 1 << 31
+	min := math.MaxInt32
 	max := 0
 	var stickiness int
 	for member, topics := range plan {

diff --git a/pkg/kgo/producer.go b/pkg/kgo/producer.go
@@ -97,7 +97,7 @@ func (p *producer) init(cl *Client) {
 	p.cl = cl
 	p.topics = newTopicsPartitions()
 	p.unknownTopics = make(map[string]*unknownTopicProduces)
-	p.waitBuffer = make(chan struct{}, math.MaxInt64)
+	p.waitBuffer = make(chan struct{}, math.MaxInt32)
 	p.idVersion = -1
 	p.id.Store(&producerID{
 		id:    -1,