diff --git a/Kafka/SocketSync.php b/Kafka/SocketSync.php index 6b0bafd1..d66bd6d9 100644 --- a/Kafka/SocketSync.php +++ b/Kafka/SocketSync.php @@ -14,6 +14,8 @@ namespace Kafka; +use Snowflake\Event; +use Snowflake\Snowflake; use Swoole\Coroutine\Client; /** @@ -342,6 +344,11 @@ class SocketSync $failedWriteAttempts = 0; $written = 0; $buflen = strlen($buf); + + + $event = Snowflake::app()->getEvent(); + $event->on(Event::EVENT_AFTER_REQUEST, [$this, 'close']); + while ($written < $buflen) { if ($buflen - $written > self::MAX_WRITE_BUFFER) { // write max buffer size