1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
#!/usr/bin/env ruby
# frozen_string_literal: true
# Released under the MIT License.
# Copyright, 2024, by Samuel Williams.
require "async"
require "async/http/client"
require "async/http/server"
require "async/http/endpoint"
require "protocol/http/body/streamable"
require "protocol/http/body/writable"
require "protocol/http/body/stream"
endpoint = Async::HTTP::Endpoint.parse("http://localhost:3000")
Async do
server = Async::HTTP::Server.for(endpoint) do |request|
output = Protocol::HTTP::Body::Streamable.response(request) do |stream|
# Simple echo server:
while chunk = stream.readpartial(1024)
$stderr.puts "Server chunk: #{chunk.inspect}"
stream.write(chunk)
$stderr.puts "Server waiting for next chunk..."
end
$stderr.puts "Server done reading request."
rescue EOFError
$stderr.puts "Server EOF."
# Ignore EOF errors.
ensure
$stderr.puts "Server closing stream."
stream.close
end
Protocol::HTTP::Response[200, {}, output]
end
server_task = Async{server.run}
client = Async::HTTP::Client.new(endpoint)
streamable = Protocol::HTTP::Body::Streamable.request do |stream|
stream.write("Hello, ")
stream.write("World!")
$stderr.puts "Client closing write..."
stream.close_write
$stderr.puts "Client reading response..."
while chunk = stream.readpartial(1024)
$stderr.puts "Client chunk: #{chunk.inspect}"
puts chunk
end
$stderr.puts "Client done reading response."
rescue EOFError
$stderr.puts "Client EOF."
# Ignore EOF errors.
ensure
$stderr.puts "Client closing stream: #{$!}"
stream.close
end
$stderr.puts "Client sending request..."
response = client.get("/", body: streamable)
$stderr.puts "Client received response and streaming it..."
streamable.stream(response.body)
ensure
server_task.stop
end
|
