-
-
Notifications
You must be signed in to change notification settings - Fork 45
/
server.rb
executable file
·109 lines (85 loc) · 2.73 KB
/
server.rb
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
# frozen_string_literal: true
# Released under the MIT License.
# Copyright, 2017-2024, by Samuel Williams.
# Copyright, 2019, by Brian Morearty.
require 'io/endpoint'
require 'protocol/http/middleware'
require 'traces/provider'
require_relative 'protocol'
module Async
module HTTP
class Server < ::Protocol::HTTP::Middleware
def self.for(*arguments, **options, &block)
self.new(block, *arguments, **options)
end
def initialize(app, endpoint, protocol: endpoint.protocol, scheme: endpoint.scheme)
super(app)
@endpoint = endpoint
@protocol = protocol
@scheme = scheme
end
def as_json(...)
{
endpoint: @endpoint.to_s,
protocol: @protocol,
scheme: @scheme,
}
end
def to_json(...)
as_json.to_json(...)
end
attr :endpoint
attr :protocol
attr :scheme
def accept(peer, address, task: Task.current)
connection = @protocol.server(peer)
Console.logger.debug(self) {"Incoming connnection from #{address.inspect} to #{@protocol}"}
connection.each do |request|
# We set the default scheme unless it was otherwise specified.
# https://tools.ietf.org/html/rfc7230#section-5.5
request.scheme ||= self.scheme
# This is a slight optimization to avoid having to get the address from the socket.
request.remote_address = address
# Console.logger.debug(self) {"Incoming request from #{address.inspect}: #{request.method} #{request.path}"}
# If this returns nil, we assume that the connection has been hijacked.
self.call(request)
end
ensure
connection&.close
end
def run
@endpoint.accept(&self.method(:accept))
end
Traces::Provider(self) do
def call(request)
if trace_parent = request.headers['traceparent']
Traces.trace_context = Traces::Context.parse(trace_parent.join, request.headers['tracestate'], remote: true)
end
attributes = {
'http.method': request.method,
'http.authority': request.authority,
'http.scheme': request.scheme,
'http.path': request.path,
'http.user_agent': request.headers['user-agent'],
}
if length = request.body&.length
attributes['http.request.length'] = length
end
if protocol = request.protocol
attributes['http.protocol'] = protocol
end
Traces.trace('async.http.server.call', resource: "#{request.method} #{request.path}", attributes: attributes) do |span|
super.tap do |response|
if status = response&.status
span['http.status_code'] = status
end
if length = response&.body&.length
span['http.response.length'] = length
end
end
end
end
end
end
end
end