Class: ProcessBot::ControlSocket
- Inherits:
-
Object
- Object
- ProcessBot::ControlSocket
- Defined in:
- lib/process_bot/control_socket.rb
Instance Attribute Summary collapse
-
#options ⇒ Object
readonly
Returns the value of attribute options.
-
#process ⇒ Object
readonly
Returns the value of attribute process.
-
#server ⇒ Object
readonly
Returns the value of attribute server.
Instance Method Summary collapse
-
#handle_client(client) ⇒ Object
rubocop:disable Metrics/AbcSize.
-
#initialize(options:, process:) ⇒ ControlSocket
constructor
A new instance of ControlSocket.
- #logger ⇒ Object
- #port ⇒ Object
- #run_client_loop ⇒ Object
- #start ⇒ Object
- #stop ⇒ Object
- #symbolize_keys(hash) ⇒ Object
Constructor Details
#initialize(options:, process:) ⇒ ControlSocket
Returns a new instance of ControlSocket.
6 7 8 9 |
# File 'lib/process_bot/control_socket.rb', line 6 def initialize(options:, process:) @options = @process = process end |
Instance Attribute Details
#options ⇒ Object (readonly)
Returns the value of attribute options.
4 5 6 |
# File 'lib/process_bot/control_socket.rb', line 4 def @options end |
#process ⇒ Object (readonly)
Returns the value of attribute process.
4 5 6 |
# File 'lib/process_bot/control_socket.rb', line 4 def process @process end |
#server ⇒ Object (readonly)
Returns the value of attribute server.
4 5 6 |
# File 'lib/process_bot/control_socket.rb', line 4 def server @server end |
Instance Method Details
#handle_client(client) ⇒ Object
rubocop:disable Metrics/AbcSize
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 |
# File 'lib/process_bot/control_socket.rb', line 42 def handle_client(client) # rubocop:disable Metrics/AbcSize loop do data = client.gets break if data.nil? # Client disconnected command = JSON.parse(data) command_type = command.fetch("command") if command_type == "graceful" || command_type == "stop" begin = if command["options"] symbolize_keys(command.fetch("options")) else {} end logger.logs "Command #{command_type} with options #{}" process.__send__(command_type, **) client.puts(JSON.generate(type: "success")) rescue => e # rubocop:disable Style/RescueStandardError logger.logs e., type: :stderr logger.logs e.backtrace, type: :stderr client.puts(JSON.generate(type: "error", message: e., backtrace: e.backtrace)) raise e end else client.puts(JSON.generate(type: "error", message: "Unknown command: #{command_type}", backtrace: Thread.current.backtrace)) end end end |
#logger ⇒ Object
11 12 13 |
# File 'lib/process_bot/control_socket.rb', line 11 def logger @logger ||= ProcessBot::Logger.new(options: ) end |
#port ⇒ Object
15 16 17 |
# File 'lib/process_bot/control_socket.rb', line 15 def port .fetch(:port).to_i end |
#run_client_loop ⇒ Object
32 33 34 35 36 37 38 39 40 |
# File 'lib/process_bot/control_socket.rb', line 32 def run_client_loop Thread.new do client = server.accept Thread.new do handle_client(client) end end end |
#start ⇒ Object
19 20 21 22 23 24 25 26 |
# File 'lib/process_bot/control_socket.rb', line 19 def start @server = TCPServer.new("localhost", port) run_client_loop logger.logs "TCPServer started" .events.call(:on_socket_opened, port: port) end |
#stop ⇒ Object
28 29 30 |
# File 'lib/process_bot/control_socket.rb', line 28 def stop server.close end |
#symbolize_keys(hash) ⇒ Object
76 77 78 79 80 81 82 83 84 85 |
# File 'lib/process_bot/control_socket.rb', line 76 def symbolize_keys(hash) new_hash = {} hash.each do |key, value| next if key == "port" new_hash[key.to_sym] = value end new_hash end |