Class: DBWrapper::DatabaseProxy

Inherits:
Object
  • Object
show all
Defined in:
lib/db_wrapper/database_proxy.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(host, port, database_host, database_port) ⇒ DatabaseProxy

Returns a new instance of DatabaseProxy.



7
8
9
10
11
12
13
 
# File 'lib/db_wrapper/database_proxy.rb', line 7

def initialize(host, port, database_host, database_port)
  @host = host
  @port = port
  @database_host = database_host
  @database_port = database_port
  @client_listeners = []
end

Instance Attribute Details

#database_host ⇒ Object (readonly)

Returns the value of attribute database_host.



4
5
6
 
# File 'lib/db_wrapper/database_proxy.rb', line 4

def database_host
  @database_host
end

#database_port ⇒ Object (readonly)

Returns the value of attribute database_port.



4
5
6
 
# File 'lib/db_wrapper/database_proxy.rb', line 4

def database_port
  @database_port
end

#host ⇒ Object (readonly)

Returns the value of attribute host.



4
5
6
 
# File 'lib/db_wrapper/database_proxy.rb', line 4

def host
  @host
end

#listener_server_socket ⇒ Object (readonly)

Returns the value of attribute listener_server_socket.



4
5
6
 
# File 'lib/db_wrapper/database_proxy.rb', line 4

def listener_server_socket
  @listener_server_socket
end

#port ⇒ Object (readonly)

Returns the value of attribute port.



4
5
6
 
# File 'lib/db_wrapper/database_proxy.rb', line 4

def port
  @port
end

#protocol ⇒ Object

Returns the value of attribute protocol.



5
6
7
 
# File 'lib/db_wrapper/database_proxy.rb', line 5

def protocol
  @protocol
end

Instance Method Details

#add_client_listener(client_listener) ⇒ Object 



15
16
17
 
# File 'lib/db_wrapper/database_proxy.rb', line 15

def add_client_listener(client_listener)
  @client_listeners << client_listener
end

#create_listener_server ⇒ Object 



38
39
40
41
42
43
44
45
46
47
48
49
50
51
 
# File 'lib/db_wrapper/database_proxy.rb', line 38

def create_listener_server
  listener_server = TCPServer.new(host, 0) #Creates on first free port it can find
  client_listeners_controller = ListenersController.new @client_listeners
  fork do
    Socket.accept_loop(listener_server) do |connection|
      begin
        while data = connection.readpartial(self.protocol.max_packet_size) do
          client_listeners_controller.call_listeners(self.protocol, data)
        end
      rescue Interrupt; end  #Thrown when you send a termination signal
    end          
  end
  listener_server      
end

#start! ⇒ Object 



19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
 
# File 'lib/db_wrapper/database_proxy.rb', line 19

def start!
  raise 'No protocol was given' if self.protocol.nil?
  listener_server_port = create_listener_server.addr[1]
  @listener_server_socket = TCPSocket.new(self.host, listener_server_port)
  database_proxy = self
  Proxy.start(host: @host, port: @port) do |conn|
    conn.server :database, host: database_proxy.database_host, port: database_proxy.database_port, relay_server: true

    conn.on_data do |data|
      database_proxy.listener_server_socket.write_nonblock(data)
      data
    end

    conn.on_finish do |server, name|
      unbind if server == :database
    end
  end
end