Merge pull request #760 from twmb/753

kgo: add AllowRebalance and CloseAllowingRebalance to GroupTransactSession
twmb · Oct 15, 2024 · fd084a5 · fd084a5
2 parents a6d66a9 + 4affe8e
commit fd084a5
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/pkg/kgo/txn.go b/pkg/kgo/txn.go
@@ -159,6 +159,18 @@ func (s *GroupTransactSession) Close() {
 	s.cl.Close()
 }
 
+// AllowRebalance is a wrapper around Client.AllowRebalance, with the exact
+// same semantics. Refer to that function's documentation.
+func (s *GroupTransactSession) AllowRebalance() {
+	s.cl.AllowRebalance()
+}
+
+// CloseAllowingRebalance is a wrapper around Client.CloseAllowingRebalance,
+// with the exact same semantics. Refer to that function's documentation.
+func (s *GroupTransactSession) CloseAllowingRebalance() {
+	s.cl.CloseAllowingRebalance()
+}
+
 // PollFetches is a wrapper around Client.PollFetches, with the exact same
 // semantics. Refer to that function's documentation.
 //