71 lines
1.7 KiB
Ruby
71 lines
1.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'runner_connection'
|
|
|
|
class Runner
|
|
BASE_URL = CodeOcean::Config.new(:code_ocean).read[:container_management][:url]
|
|
HEADERS = {"Content-Type" => "application/json"}
|
|
|
|
attr_accessor :waiting_time
|
|
|
|
def initialize(execution_environment, time_limit = nil)
|
|
url = "#{BASE_URL}/runners"
|
|
body = {execution_environment_id: execution_environment.id}
|
|
if time_limit
|
|
body[:time_limit] = time_limit
|
|
end
|
|
response = Faraday.post(url, body.to_json, HEADERS)
|
|
response = parse response
|
|
@id = response[:id]
|
|
end
|
|
|
|
def copy_files(files)
|
|
url = runner_url + "/files"
|
|
body = { files: files.map{ |filename, content| { filename: filename, content: content } } }
|
|
Faraday.post(url, body.to_json, HEADERS)
|
|
end
|
|
|
|
def copy_submission_files(submission)
|
|
files = {}
|
|
submission.collect_files.each do |file|
|
|
files[file.name_with_extension] = file.content
|
|
end
|
|
copy_files(files)
|
|
end
|
|
|
|
def execute_command(command)
|
|
url = runner_url + "/execute"
|
|
response = Faraday.post(url, {command: command}.to_json, HEADERS)
|
|
response = parse response
|
|
response
|
|
end
|
|
|
|
def execute_interactively(command)
|
|
starting_time = Time.now
|
|
websocket_url = execute_command(command)[:websocket_url]
|
|
EventMachine.run do
|
|
socket = RunnerConnection.new(websocket_url)
|
|
yield(self, socket) if block_given?
|
|
end
|
|
Time.now - starting_time # execution time
|
|
end
|
|
|
|
def destroy
|
|
Faraday.delete runner_url
|
|
end
|
|
|
|
def status
|
|
parse(Faraday.get(runner_url))[:status].to_sym
|
|
end
|
|
|
|
private
|
|
|
|
def runner_url
|
|
"#{BASE_URL}/runners/#{@id}"
|
|
end
|
|
|
|
def parse(response)
|
|
JSON.parse(response.body).deep_symbolize_keys
|
|
end
|
|
end
|