Module: Bluepill::Socket

Extended by:
Socket
Included in:
Socket
Defined in:
lib/bluepill/socket.rb

Constant Summary collapse

RETRIES = 
5
@@current_retry = 
0

Instance Method Summary collapse

Instance Method Details

#client(base_dir, name, &b) ⇒ Object 



10
11
12
 
# File 'lib/bluepill/socket.rb', line 10

def client(base_dir, name, &b)
  UNIXSocket.open(socket_path(base_dir, name), &b)
end

#client_command(base_dir, name, command, timeout) ⇒ Object 



14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
 
# File 'lib/bluepill/socket.rb', line 14

def client_command(base_dir, name, command, timeout)
  client(base_dir, name) do |socket|
    Timeout.timeout(timeout) do
      socket.puts command
      Marshal.load(socket)
    end
  end
rescue EOFError, Timeout::Error
  @@current_retry += 1
  puts "Retry #{@@current_retry} of #{RETRIES}"
  if @@current_retry <= RETRIES
    client_command(base_dir, name, command, timeout)
  else
    abort("Socket Timeout: Server may not be responding")
  end
ensure
  @@current_retry = 0
end

#server(base_dir, name) ⇒ Object 



33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
 
# File 'lib/bluepill/socket.rb', line 33

def server(base_dir, name)
  socket_path = self.socket_path(base_dir, name)
  begin
    UNIXServer.open(socket_path)
  rescue Errno::EADDRINUSE
    # if sock file has been created.  test to see if there is a server
    begin
      UNIXSocket.open(socket_path)
    rescue Errno::ECONNREFUSED
      File.delete(socket_path)
      return UNIXServer.open(socket_path)
    else
      logger.err("Server is already running!")
      exit(7)
    end
  end
end

#socket_path(base_dir, name) ⇒ Object 



51
52
53
 
# File 'lib/bluepill/socket.rb', line 51

def socket_path(base_dir, name)
  File.join(base_dir, 'socks', name + ".sock")
end