-
Notifications
You must be signed in to change notification settings - Fork 51
/
client.rb
61 lines (49 loc) · 1.28 KB
/
client.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
require 'set'
require 'uri'
require 'yaml'
module Einhorn
class Client
# Keep this in this file so client can be loaded entirely
# standalone by user code.
module Transport
ParseError = defined?(Psych::SyntaxError) ? Psych::SyntaxError : ArgumentError
def self.send_message(socket, message)
line = serialize_message(message)
socket.write(line)
end
def self.receive_message(socket)
line = socket.readline
deserialize_message(line)
end
def self.serialize_message(message)
serialized = YAML.dump(message)
escaped = URI.escape(serialized, "%\n")
escaped + "\n"
end
def self.deserialize_message(line)
serialized = URI.unescape(line)
YAML.load(serialized)
end
end
def self.for_path(path_to_socket)
socket = UNIXSocket.open(path_to_socket)
self.new(socket)
end
def self.for_fd(fileno)
socket = UNIXSocket.for_fd(fileno)
self.new(socket)
end
def initialize(socket)
@socket = socket
end
def send_command(command_hash)
Transport.send_message(@socket, command_hash)
end
def receive_message
Transport.receive_message(@socket)
end
def close
@socket.close
end
end
end