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
|
#!/usr/bin/env ruby
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
require 'socket'
require 'avro'
MAIL_PROTOCOL_JSON = <<-EOS
{"namespace": "example.proto",
"protocol": "Mail",
"types": [
{"name": "Message", "type": "record",
"fields": [
{"name": "to", "type": "string"},
{"name": "from", "type": "string"},
{"name": "body", "type": "string"}
]
}
],
"messages": {
"send": {
"request": [{"name": "message", "type": "Message"}],
"response": "string"
},
"replay": {
"request": [],
"response": "string"
}
}
}
EOS
MAIL_PROTOCOL = Avro::Protocol.parse(MAIL_PROTOCOL_JSON)
class MailResponder < Avro::IPC::Responder
def initialize
super(MAIL_PROTOCOL)
end
def call(message, request)
if message.name == 'send'
request_content = request['message']
"Sent message to #{request_content['to']} from #{request_content['from']} with body #{request_content['body']}"
elsif message.name == 'replay'
'replay'
end
end
end
class RequestHandler
def initialize(address, port)
@ip_address = address
@port = port
end
def run
server = TCPServer.new(@ip_address, @port)
while (session = server.accept)
handle(session)
session.close
end
end
end
class MailHandler < RequestHandler
def handle(request)
responder = MailResponder.new()
transport = Avro::IPC::SocketTransport.new(request)
str = transport.read_framed_message
transport.write_framed_message(responder.respond(str))
end
end
if $0 == __FILE__
handler = MailHandler.new('localhost', 9090)
handler.run
end
|
