add: eof var

Author: pekopekopekopayo

lib/ld-eventsource/client.rb

@@ -119,6 +119,7 @@ def initialize(uri,
       @uri = URI(uri)
       @stopped = Concurrent::AtomicBoolean.new(false)
       @retry_enabled = retry_enabled
+      @eof = false
 
       @headers = headers.clone
       @connect_timeout = connect_timeout
@@ -265,6 +266,7 @@ def run_stream
           # connected but before @cxn was set. Checking the variable again is a bit clunky but avoids that.
           return if @stopped.value
           read_stream(resp) unless resp.nil?
+          return close if @eof
         rescue => e
           # When we deliberately close the connection, it will usually trigger an exception. The exact type
           # of exception depends on the specific Ruby runtime. But @stopped will always be set in this case.
@@ -344,7 +346,12 @@ def read_stream(cxn)
               # For historical reasons, we rethrow this as our own type
               raise Errors::ReadTimeoutError.new(@read_timeout)
             end
-            break if data.nil?
+
+            if data.nil?
+              @eof = true
+              break
+            end
+
             gen.yield data
           end
         end