Module: Peatio::Ranger

Defined in:
lib/peatio/ranger/events.rb,
lib/peatio/ranger/router.rb,
lib/peatio/ranger/connection.rb,
lib/peatio/ranger/web_socket.rb

Defined Under Namespace

Classes: Connection, Events, Router

Class Method Summary collapse

Class Method Details

.run(jwt_public_key, exchange_name, opts = {}) ⇒ Object 



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
 
# File 'lib/peatio/ranger/web_socket.rb', line 4

def self.run(jwt_public_key, exchange_name, opts={})
  host = opts[:ranger_host] || ENV["RANGER_HOST"] || "0.0.0.0"
  port = opts[:ranger_port] || ENV["RANGER_PORT"] || "8081"

  authenticator = Peatio::Auth::JWTAuthenticator.new(jwt_public_key)

  logger = Peatio::Logger.logger
  logger.info "Starting the server on port #{port}"

  client = Peatio::MQ::Client.new
  router = Peatio::Ranger::Router.new(opts[:registry])
  client.subscribe(exchange_name, &router.method(:on_message))

  if opts[:display_stats]
    EM.add_periodic_timer(opts[:stats_period]) do
      Peatio::Logger.logger.info { router.stats }
      Peatio::Logger.logger.debug { router.debug }
    end
  end

  EM::WebSocket.start(host: host, port: port, secure: false) do |socket|
    connection = Peatio::Ranger::Connection.new(router, socket, logger)
    socket.onopen do |hs|
      connection.handshake(authenticator, hs)
      router.on_connection_open(connection)
    end

    socket.onmessage do |msg|
      connection.handle(msg)
    end

    socket.onping do |value|
      logger.debug { "Received ping: #{value}" }
    end

    socket.onclose do
      logger.debug { "Websocket connection closed" }
      router.on_connection_close(connection)
    end

    socket.onerror do |e|
      logger.error { "WebSocket Error: #{e.message}\n" + e.backtrace.join("\n") }
    end
  end
end

.run!(jwt_public_key, exchange_name, opts = {}) ⇒ Object 



50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
 
# File 'lib/peatio/ranger/web_socket.rb', line 50

def self.run!(jwt_public_key, exchange_name, opts={})
  metrics_host = opts[:metrics_host] || ENV["METRICS_HOST"] || "0.0.0.0"
  metrics_port = opts[:metrics_port] || ENV["METRICS_PORT"] || "8082"

  EM.run do
    run(jwt_public_key, exchange_name, opts)

    if opts[:registry]
      thin = Rack::Handler.get("thin")
      thin.run(Peatio::Metrics::Server.app(opts[:registry]), Host: metrics_host, Port: metrics_port)
    end

    trap("INT") do
      puts "\nSIGINT received, stopping ranger..."
      EM.stop
    end
  end
end