575057acd3
Many functions in submission_controller.rb still are very long and have a high complexity. Because the logic for handling execution of submissions will probably move elsewhere (when switching to ActionCable), this is fine for now.
81 lines
1.6 KiB
Ruby
81 lines
1.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'faye/websocket/client'
|
|
require 'json_schemer'
|
|
|
|
class RunnerConnection
|
|
EVENTS = %i[start output exit stdout stderr].freeze
|
|
BACKEND_OUTPUT_SCHEMA = JSONSchemer.schema(JSON.parse(File.read('lib/runner/backend-output.schema.json')))
|
|
|
|
def initialize(url)
|
|
@socket = Faye::WebSocket::Client.new(url, [], ping: 5)
|
|
|
|
%i[open message error close].each do |event_type|
|
|
@socket.on(event_type) { |event| __send__(:"on_#{event_type}", event) }
|
|
end
|
|
|
|
EVENTS.each { |event_type| instance_variable_set(:"@#{event_type}_callback", ->(e) {}) }
|
|
@start_callback = -> {}
|
|
@exit_code = 0
|
|
end
|
|
|
|
def on(event, &block)
|
|
return unless EVENTS.include? event
|
|
|
|
instance_variable_set(:"@#{event}_callback", block)
|
|
end
|
|
|
|
def send(data)
|
|
@socket.send(encode(data))
|
|
end
|
|
|
|
private
|
|
|
|
def decode(event)
|
|
JSON.parse(event).deep_symbolize_keys
|
|
end
|
|
|
|
def encode(data)
|
|
data
|
|
end
|
|
|
|
def on_message(event)
|
|
return unless BACKEND_OUTPUT_SCHEMA.valid?(JSON.parse(event.data))
|
|
|
|
event = decode(event.data)
|
|
__send__("handle_#{event[:type]}", event)
|
|
end
|
|
|
|
def on_open(_event)
|
|
@start_callback.call
|
|
end
|
|
|
|
def on_error(event); end
|
|
|
|
def on_close(_event)
|
|
@exit_callback.call @exit_code
|
|
end
|
|
|
|
def handle_exit(event)
|
|
@exit_code = event[:data]
|
|
end
|
|
|
|
def handle_stdout(event)
|
|
@stdout_callback.call event[:data]
|
|
@output_callback.call event[:data]
|
|
end
|
|
|
|
def handle_stderr(event)
|
|
@stderr_callback.call event[:data]
|
|
@output_callback.call event[:data]
|
|
end
|
|
|
|
def handle_error(event) end
|
|
|
|
def handle_start(event) end
|
|
|
|
def handle_timeout(event)
|
|
# TODO: set the runner state
|
|
end
|
|
end
|