connection: fix stream id managing

1 files changed, 4 insertions, 2 deletions

diff --git a/lib/plum/server/connection.rb b/lib/plum/server/connection.rb
index c12ef00..450dcf6 100644
--- a/lib/plum/server/connection.rb
+++ b/lib/plum/server/connection.rb
@@ -11,8 +11,10 @@ module Plum
     # Reserves a new stream to server push.
     # @param args [Hash] The argument to pass to Stram.new.
     def reserve_stream(**args)
-      next_id = @max_even_stream_id + 2
-      stream = new_stream(next_id, state: :reserved_local, **args)
+      next_id = @max_stream_id + (@max_stream_id.odd? ? 1 : 2)
+      stream = stream(next_id)
+      stream.set_state(:reserved_local)
+      stream.update_dependency(**args)
       stream
     end