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
110
111
112
113
114
|
# frozen_string_literal: true
#
# Copyright, 2017, by Samuel G. D. Williams. <http://www.codeotaku.com>
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
require 'async/io/endpoint'
require 'async/io/stream'
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
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']
self.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
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
|
