io-stream 0.7.0 → 0.8.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c48d464c539d56862e04437f88dee2997e4393951efca83cf622a23bac9e4b75
-  data.tar.gz: 610277102a9233e5d470daabd4f12917d429dc6a141cf52897ff95af3a06de95
+  metadata.gz: 5f73e60fdaf4001c88fcfaa4ccb29bfc8ee90d2fc3ffc81408c8673b414038ae
+  data.tar.gz: 933183cda89e7d555239ba15a752dadccc938628d6197675b3bedb8cfeb60f43
 SHA512:
-  metadata.gz: 5761c359f16364169757dca90c4f41bfb87269ab3219b6a565b0081aafb3146f2ff6b11e77a95bebd478b31cbc702752e486c484666536f286e4ccdb4a2c6520
-  data.tar.gz: 6e44c1ce99d3741c6c9e8f65976f5664d287586e5a704749466a70992dcfcf5bac55ab84e1429cc5a22417af889ee7938af55ec63081309b3522629a9e05721b
+  metadata.gz: 732318c46b27af3ecabb0c99361b60de5ae3c2733013261ae3089fbaa0f4f2f542e70ee10c410227262a8d137a59d5f1e36ba9419307e801fefd0cfebb412ce5
+  data.tar.gz: 5ae80afa0d558011183648ce9d34c02dc52e2b8f0cf24bde53a319d1cba3dd188fa8705a477baa8c7fcd5afeb6ee487087e72c1a75d39c80566ea9c378d4e7b1

data/lib/io/stream/buffered.rb

@@ -106,23 +106,26 @@ module IO::Stream
 			end
 		end
 		
-		def syswrite(buffer)
-			# This fails due to re-entrancy issues with a concurrent call to `sysclose`.
-			# return @io.write(buffer)
-			
-			while true
-				result = @io.write_nonblock(buffer, exception: false)
-				
-				case result
-				when :wait_readable
-					@io.wait_readable(@io.timeout) or raise ::IO::TimeoutError, "read timeout"
-				when :wait_writable
-					@io.wait_writable(@io.timeout) or raise ::IO::TimeoutError, "write timeout"
-				else
-					if result == buffer.bytesize
-						return
+		if RUBY_VERSION >= "3.3"
+			def syswrite(buffer)
+				return @io.write(buffer)
+			end
+		else
+			def syswrite(buffer)
+				while true
+					result = @io.write_nonblock(buffer, exception: false)
+					
+					case result
+					when :wait_readable
+						@io.wait_readable(@io.timeout) or raise ::IO::TimeoutError, "read timeout"
+					when :wait_writable
+						@io.wait_writable(@io.timeout) or raise ::IO::TimeoutError, "write timeout"
 					else
-						buffer = buffer.byteslice(result, buffer.bytesize)
+						if result == buffer.bytesize
+							return
+						else
+							buffer = buffer.byteslice(result, buffer.bytesize)
+						end
 					end
 				end
 			end

data/lib/io/stream/generic.rb

@@ -35,7 +35,7 @@ module IO::Stream
 			return if closed?
 			
 			begin
-				flush
+				self.flush
 			rescue
 				# We really can't do anything here unless we want #close to raise exceptions.
 			ensure

data/lib/io/stream/readable.rb

@@ -116,17 +116,28 @@ module IO::Stream
 		# @parameter offset [Integer] The offset to start searching from.
 		# @parameter limit [Integer | Nil] The maximum number of bytes to read while searching.
 		# @returns [Integer | Nil] The index of the pattern, or nil if not found.
-		private def index_of(pattern, offset, limit)
+		private def index_of(pattern, offset, limit, discard = false)
 			# We don't want to split on the pattern, so we subtract the size of the pattern.
 			split_offset = pattern.bytesize - 1
-
+			
 			until index = @read_buffer.index(pattern, offset)
 				offset = @read_buffer.bytesize - split_offset
 				
 				offset = 0 if offset < 0
 				
-				return nil if limit and offset >= limit
-				return nil unless fill_read_buffer
+				if limit and offset >= limit
+					return nil
+				end
+				
+				unless fill_read_buffer
+					return nil
+				end
+				
+				if discard
+					# If we are discarding, we should consume the read buffer up to the offset:
+					consume_read_buffer(offset)
+					offset = 0
+				end
 			end
 			
 			return index
@@ -136,7 +147,8 @@ module IO::Stream
 		# @parameter pattern [String] The pattern to match.
 		# @parameter offset [Integer] The offset to start searching from.
 		# @parameter limit [Integer] The maximum number of bytes to read, including the pattern (even if chomped).
-		# @returns [String | Nil] The contents of the stream up until the pattern, which is consumed but not returned.
+		# @parameter chomp [Boolean] Whether to remove the pattern from the returned data.
+		# @returns [String | Nil] The contents of the stream up until the pattern, or nil if the pattern was not found. 
 		def read_until(pattern, offset = 0, limit: nil, chomp: true)
 			if index = index_of(pattern, offset, limit)
 				return nil if limit and index >= limit
@@ -149,6 +161,28 @@ module IO::Stream
 			end
 		end
 		
+		# Efficiently discard data from the stream until encountering pattern.
+		# @parameter pattern [String] The pattern to match.
+		# @parameter offset [Integer] The offset to start searching from.
+		# @parameter limit [Integer] The maximum number of bytes to read, including the pattern.
+		# @returns [String | Nil] The contents of the stream up until the pattern, or nil if the pattern was not found.
+		def discard_until(pattern, offset = 0, limit: nil)
+			if index = index_of(pattern, offset, limit, true)
+				@read_buffer.freeze
+				
+				if limit and index >= limit
+					@read_buffer = @read_buffer.byteslice(limit, @read_buffer.bytesize)
+					
+					return nil
+				end
+				
+				matched = @read_buffer.byteslice(0, index+pattern.bytesize)
+				@read_buffer = @read_buffer.byteslice(index+pattern.bytesize, @read_buffer.bytesize)
+				
+				return matched
+			end
+		end
+		
 		# Peek at data in the buffer without consuming it.
 		# @parameter size [Integer | Nil] The number of bytes to peek at. If nil, peek at all available data.
 		# @returns [String] The data in the buffer without consuming it.

data/lib/io/stream/version.rb

@@ -4,5 +4,5 @@
 # Copyright, 2023-2024, by Samuel Williams.
 
 module IO::Stream
-	VERSION = "0.7.0"
+	VERSION = "0.8.0"
 end

data/readme.md

@@ -12,6 +12,11 @@ Please see the [project documentation](https://socketry.github.io/io-stream) for
 
 Please see the [project releases](https://socketry.github.io/io-streamreleases/index) for all releases.
 
+### v0.8.0
+
+  - On Ruby v3.3+, use `IO#write` directly instead of `IO#write_nonblock`, for better performance.
+  - Introduce support for `Readable#discard_until` method to discard data until a specific pattern is found.
+
 ### v0.7.0
 
   - Split stream functionality into separate `Readable` and `Writable` modules for better modularity and composition.
@@ -57,11 +62,6 @@ Please see the [project releases](https://socketry.github.io/io-streamreleases/i
 
   - Add support for timeouts with compatibility shims for various IO types.
 
-### v0.2.0
-
-  - Prefer `write_nonblock` in `syswrite` implementation for better non-blocking behavior.
-  - Add test cases for crash scenarios.
-
 ## See Also
 
   - [async-io](https://github.com/socketry/async-io) — Where this implementation originally came from.

data/releases.md

@@ -1,5 +1,10 @@
 # Releases
 
+## v0.8.0
+
+  - On Ruby v3.3+, use `IO#write` directly instead of `IO#write_nonblock`, for better performance.
+  - Introduce support for `Readable#discard_until` method to discard data until a specific pattern is found.
+
 ## v0.7.0
 
   - Split stream functionality into separate `Readable` and `Writable` modules for better modularity and composition.

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: io-stream
 version: !ruby/object:Gem::Version
-  version: 0.7.0
+  version: 0.8.0
 platform: ruby
 authors:
 - Samuel Williams