tp2 0.19.3 → 0.20.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e0b5eac74e3e54f7dfbad1af92a110aa755c3e5ebe18229879c734baaa24e8cf
-  data.tar.gz: ca1bf493c839517f11db5b0d346958c44bfd1aa194dbbd18e2300e9d843e8d9b
+  metadata.gz: f50f04dad197bd317793e48f2f46e90f87c0ddf0f65c98d99c57cc8e29a24f66
+  data.tar.gz: 287838481d4b0e8150c021368cf709543331b5d4b0e87bb37b335c750c8ae267
 SHA512:
-  metadata.gz: bab1d252cbe06291116e6ddbb8e973e8a0b0acefd055acd52290d371629dc85abb6a4f4f5339a13bdbf20b2f1647b53517f30695ea356b0a5f9dcc30d01cb360
-  data.tar.gz: 6c22dd2ebdb82c883f6805f2f0d456e76f5b86d84e57d2a1dfc3c8b66dbd2c36710e4f98d91607b6e73301390b3fbab078dda1e9a571cd3f832ed9e8b2efe2e7
+  metadata.gz: 1597b8856cdb6adf327652d5e161d7f8b9965358b9898ebbf2f8a5eeb5af994e3d9110b09093867ae651c9237dd4f3f31b10d7bb2a8d6e0854cb5379980d8e7d
+  data.tar.gz: 9f85ecb5c04107b8db550a7996429d39e39baab2efbcf3a355cfcb756ba6be45fb009a9da63ab2cf44fbda4bd2bec15b5187a65e867c97ac6d8b6aeeb7479832

data/.github/workflows/test.yml

@@ -8,7 +8,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest]
-        ruby: [3.4, 'head']
+        ruby: ['4.0', 'head']
 
     name: >-
       ${{matrix.os}}, ${{matrix.ruby}}

data/CHANGELOG.md

@@ -1,3 +1,13 @@
+# 0.20.0 2026-01-28
+
+- Update Qeweney
+- Fix lack of sendv (for kernels < 6.17)
+
+# 0.19.4 2026-01-26
+
+- Fix and make graceful_shutdown idempotent
+- Update dependencies
+
 # 0.19.3 2025-12-10
 
 - Update UringMachine

data/Gemfile

@@ -1,12 +1,12 @@
-source 'https://rubygems.org'
+source 'https://gem.coop'
 
 gemspec name: 'tp2'
 
 group :development do
-  gem 'listen',         '3.9.0'
-  gem 'minitest',       '5.25.4'
-  gem 'rake',           '13.2.1'
-  gem 'rake-compiler',  '1.2.9'
+  gem 'listen',         '3.10.0'
+  gem 'minitest',       '6.0.1'
+  gem 'rake',           '13.3.1'
+  gem 'rake-compiler',  '1.3.1'
   gem 'simplecov',      '0.22.0'
-  gem 'yard',           '0.9.37'
+  gem 'yard',           '0.9.38'
 end

data/examples/simple.rb

@@ -8,9 +8,19 @@ end
 
 require 'tp2'
 
+BODY = 'foobar' * 1000
 app = ->(req) {
-  req.respond('foobar', ':status' => Qeweney::Status::TEAPOT)
+  req.respond(BODY)
 }
 
-TP2.config(&app)
-TP2.run
+PORT = 1234
+
+machine = UM.new
+opts = {
+  banner: TP2::BANNER,
+}
+# opts[:logger] = TP2::Logger.new(machine, **opts)
+env = { bind: "127.0.0.1:#{PORT}" }.merge(opts)
+puts "Listening on #{env[:bind]}"
+server = TP2::Server.new(machine, env) { app&.call(it) }
+server.run

data/lib/tp2/connection.rb

@@ -153,25 +153,50 @@ module TP2
 
     SEND_FLAGS = UM::MSG_NOSIGNAL | UM::MSG_WAITALL
 
-    # Sends response including headers and body. Waits for the request to complete
-    # if not yet completed. The body is sent using chunked transfer encoding.
-    # @param request [Qeweney::Request] HTTP request
-    # @param body [String] response body
-    # @param headers
-    def respond(request, body, headers)
-      headers = @response_headers.merge(headers) if @response_headers
+    EMPTY_CHUNK = "0\r\n\r\n"
+    EMPTY_CHUNK_LEN = EMPTY_CHUNK.bytesize
 
-      formatted_headers = format_headers(headers, body)
-      @response_headers = headers
-      request&.tx_incr(formatted_headers.bytesize + (body ? body.bytesize : 0))
-      if body
-        buf = "#{formatted_headers}#{body.bytesize.to_s(16)}\r\n#{body}\r\n#{EMPTY_CHUNK}"
-        @machine.send(@fd, buf, buf.bytesize, SEND_FLAGS)
-      else
-        @machine.send(@fd, formatted_headers, formatted_headers.bytesize, SEND_FLAGS)
+    CHUNKED_ENCODING_POSTLUDE = "\r\n#{EMPTY_CHUNK}"
+
+    if UM.instance_methods.include?(:sendv)
+      # Sends response including headers and body. Waits for the request to complete
+      # if not yet completed. The body is sent using chunked transfer encoding.
+      # @param request [Qeweney::Request] HTTP request
+      # @param body [String] response body
+      # @param headers
+      def respond(request, body, headers)
+        headers = @response_headers.merge(headers) if @response_headers
+
+        formatted_headers = format_headers(headers, body)
+        @response_headers = headers
+        request&.tx_incr(formatted_headers.bytesize + (body ? body.bytesize : 0))
+        if body
+          chunk_prelude = "#{body.bytesize.to_s(16)}\r\n"
+          @machine.sendv(@fd, formatted_headers, chunk_prelude, body, CHUNKED_ENCODING_POSTLUDE)
+        else
+          @machine.send(@fd, formatted_headers, formatted_headers.bytesize, SEND_FLAGS)
+        end
+        @logger&.info(request: request, response_headers: headers) if request
+        @done = true
+      end
+    else
+      def respond(request, body, headers)
+        headers = @response_headers.merge(headers) if @response_headers
+
+        formatted_headers = format_headers(headers, body)
+        @response_headers = headers
+        request&.tx_incr(formatted_headers.bytesize + (body ? body.bytesize : 0))
+        if body
+          chunk_prelude = "#{body.bytesize.to_s(16)}\r\n"
+          buf = +''
+          buf << formatted_headers << chunk_prelude << body << CHUNKED_ENCODING_POSTLUDE
+          @machine.send(@fd, buf, SEND_FLAGS)
+        else
+          @machine.send(@fd, formatted_headers, formatted_headers.bytesize, SEND_FLAGS)
+        end
+        @logger&.info(request: request, response_headers: headers) if request
+        @done = true
       end
-      @logger&.info(request: request, response_headers: headers) if request
-      @done = true
     end
 
     # Sends response headers. If empty_response is truthy, the response status
@@ -187,9 +212,6 @@ module TP2
       @response_headers = headers
     end
 
-    EMPTY_CHUNK = "0\r\n\r\n"
-    EMPTY_CHUNK_LEN = EMPTY_CHUNK.bytesize
-
     # Sends a response body chunk. If no headers were sent, default headers are
     # sent using #send_headers. if the done option is true(thy), an empty chunk
     # will be sent to signal response completion to the client.

data/lib/tp2/server.rb

@@ -50,11 +50,15 @@ module TP2
 
     def run
       setup
-      @machine.join(*@accept_fibers)
+      @machine.await_fibers(@accept_fibers)
     rescue UM::Terminate
       graceful_shutdown
     end
 
+    def stop!
+      graceful_shutdown
+    end
+
     private
 
     def setup
@@ -69,7 +73,7 @@ module TP2
       setup_server_extensions
 
       # map fibers
-      @connection_fiber_map = {}
+      @connection_fibers = Set.new
     end
 
     def get_bind_entries
@@ -124,47 +128,52 @@ module TP2
     end
 
     def accept_incoming(listen_fd)
-      @machine.accept_each(listen_fd) do |fd|
-        conn = Connection.new(self, @machine, fd, @env, &@app)
-        f = @machine.spin(conn) do
-          it.run
-        ensure
-          @connection_fiber_map.delete(f)
-        end
-        @connection_fiber_map[f] = true
-      end
+      @machine.accept_each(listen_fd) { start_client_connection(it) }
     rescue UM::Terminate
       # terminated
-    rescue Exception => e
-      p e
-      p e.backtrace
-      exit
+    end
+
+    def start_client_connection(fd)
+      conn = Connection.new(self, @machine, fd, @env, &@app)
+      f = @machine.spin(conn) do
+        it.run
+      ensure
+        @connection_fibers.delete(f)
+      end
+      @connection_fibers << f
     end
 
     def close_all_server_fds
       @server_fds.each { @machine.close_async(it) }
     end
 
+    STOP = UM::Terminate.new
+
+    def stop_accept_fibers
+      @accept_fibers.each { @machine.schedule(it, STOP) if !it.done? }
+      @machine.await_fibers(@accept_fibers)
+    end
+
     def graceful_shutdown
       @env[:logger]&.info(message: 'Shutting down gracefully...')
 
       # stop listening
       close_all_server_fds
+      stop_accept_fibers
       @machine.snooze
 
-      return if @connection_fiber_map.empty?
+      return if @connection_fibers.empty?
 
       # sleep for a bit, let requests finish
       @machine.sleep(PENDING_REQUESTS_GRACE_PERIOD)
-      return if @connection_fiber_map.empty?
+      return if @connection_fibers.empty?
 
       # terminate pending fibers
-      pending = @connection_fiber_map.keys
-      signal = UM::Terminate.new
-      pending.each { @machine.schedule(it, signal) }
+      pending = @connection_fibers.to_a
+      pending.each { @machine.schedule(it, STOP) }
 
       @machine.timeout(PENDING_REQUESTS_TIMEOUT_PERIOD, UM::Terminate) do
-        @machine.join(*@connection_fiber_map.keys)
+        @machine.await_fibers(@connection_fibers)
       rescue UM::Terminate
         # timeout on waiting for adapters to finish running, do nothing
       end

data/lib/tp2/version.rb

@@ -1,3 +1,3 @@
 module TP2
-  VERSION = '0.19.3'
+  VERSION = '0.20.0'
 end

data/test/test_connection.rb

@@ -491,9 +491,9 @@ class ConnectionTest < Minitest::Test
       @machine.snooze
     end
 
-    content = IO.read(__FILE__)
+    content = IO.read(fn)
     file_size = content.bytesize
-    expected = "HTTP/1.1 200\r\nTransfer-Encoding: chunked\r\n\r\n6\r\nfoobar\r\n0\r\n\r\n"
+    expected = "HTTP/1.1 200\r\nTransfer-Encoding: chunked\r\n\r\n#{file_size.to_s(16)}\r\n#{content}\r\n0\r\n\r\n"
 
     assert_equal expected, response
   end
@@ -545,7 +545,7 @@ class ConnectionTest < Minitest::Test
     assert_equal expected, response
   end
 
-  def test_set_response_headers
+  def test_set_response_headers_1
     @hook = ->(req) {
       req.set_response_headers("Set-Cookie" => 'foo=bar')
       req.respond('foo')
@@ -569,7 +569,7 @@ class ConnectionTest < Minitest::Test
     assert_equal expected, response
   end
 
-  def test_set_response_headers
+  def test_set_response_headers_2
     @hook = ->(req) {
       req.set_response_headers("Set-Cookie" => 'foo=bar')
       req.set_response_headers("Foo" => 'bar')

data/test/test_server.rb

@@ -38,15 +38,13 @@ class ServerTest < Minitest::Test
 
   def run_server
     @server.run
-  rescue STOP
-    # ignore
   ensure
     @server_done = true
   end
 
   def teardown
     @machine.close(@client_fd) rescue nil
-    @machine.schedule(@f_server, STOP.new)
+    @server.stop!
     @machine.snooze until @server_done
   end
 
@@ -113,7 +111,7 @@ class ServerTest < Minitest::Test
     write_http_request "GET /foo HTTP/1.1\r\nServer: foo.com\r\n\r\nSCHMET /bar HTTP/1.1\r\n\r\n"
 
     @machine.sleep(0.01)
-    @machine.schedule(@f_server, UM::Terminate.new)
+    @server.stop!
     @machine.snooze
 
     response = read_client_side
@@ -282,9 +280,9 @@ class ServerTest < Minitest::Test
     @env[:logger] = TestLogger.new
     @app = ->(req) { reqs << req; req.respond('Hello, world!', {}) }
 
-    write_http_request "GET / HTTP/1.0\r\n\r\n"
+    write_http_request "GET / HTTP/1.1\r\n\r\n"
     response = read_client_side
-    expected = "HTTP/1.1 200\r\nContent-Length: 13\r\n\r\nHello, world!"
+    expected = "HTTP/1.1 200\r\nTransfer-Encoding: chunked\r\n\r\nd\r\nHello, world!\r\n0\r\n\r\n"
     assert_equal(expected, response)
 
     entries = @env[:logger].entries

data/tp2.gemspec

@@ -20,7 +20,7 @@ Gem::Specification.new do |s|
   s.required_ruby_version = '>= 3.4'
   s.executables   = ['tp2']
 
-  s.add_dependency 'uringmachine',    '~> 0.22.0'
-  s.add_dependency 'qeweney',         '~> 0.23'
+  s.add_dependency 'uringmachine',    '~> 0.23.1'
+  s.add_dependency 'qeweney',         '~> 0.25'
   s.add_dependency 'rack',            '~> 3.1.15'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tp2
 version: !ruby/object:Gem::Version
-  version: 0.19.3
+  version: 0.20.0
 platform: ruby
 authors:
 - Sharon Rosner
@@ -15,28 +15,28 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.22.0
+        version: 0.23.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.22.0
+        version: 0.23.1
 - !ruby/object:Gem::Dependency
   name: qeweney
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.23'
+        version: '0.25'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.23'
+        version: '0.25'
 - !ruby/object:Gem::Dependency
   name: rack
   requirement: !ruby/object:Gem::Requirement
@@ -111,7 +111,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 4.0.0
+rubygems_version: 4.0.3
 specification_version: 4
 summary: Experimental HTTP/1 server for UringMachine
 test_files: []