fix: a potential race condition

kafka4beam · Aug 29, 2024 · 02541fb · 02541fb
1 parent 7959fa3
commit 02541fb
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/wolff_producer.erl b/src/wolff_producer.erl
@@ -201,7 +201,13 @@ send_sync(Pid, Batch0, Timeout) ->
     Timeout ->
       deref_caller(Caller),
       erlang:demonitor(Mref, [flush]),
-      erlang:error(timeout)
+      receive
+        {Mref, Partition, BaseOffset} ->
+          {Partition, BaseOffset}
+      after
+        0 ->
+          erlang:error(timeout)
+      end
   end.
 
 init(St) ->