Files
codeocean/lib/docker_container_pool.rb
2020-04-14 14:23:35 +02:00

61 lines
2.3 KiB
Ruby

require 'concurrent/future'
require 'concurrent/timer_task'
class DockerContainerPool
def self.config
#TODO: Why erb?
@config ||= CodeOcean::Config.new(:docker).read(erb: true)[:pool]
end
def self.create_container(execution_environment)
Rails.logger.info('trying to create container for execution environment: ' + execution_environment.to_s)
container = DockerClient.create_container(execution_environment)
container.status = 'available' # FIXME: String vs Symbol usage?
#Rails.logger.debug('created container ' + container.to_s + ' for execution environment ' + execution_environment.to_s)
container
rescue StandardError => e
Raven.extra_context({container: container.inspect, execution_environment: execution_environment.inspect, config: config.inspect})
Raven.capture_exception(e)
nil
end
def self.return_container(container, execution_environment)
Faraday.get(config[:location] + "/docker_container_pool/return_container/" + container.id)
rescue StandardError => e
Raven.extra_context({container: container.inspect, execution_environment: execution_environment.inspect, config: config.inspect})
Raven.capture_exception(e)
nil
end
def self.get_container(execution_environment)
# if pooling is active, do pooling, otherwise just create an container and return it
if config[:active]
begin
container_id = JSON.parse(Faraday.get(config[:location] + "/docker_container_pool/get_container/" + execution_environment.id.to_s).body)['id']
Docker::Container.get(container_id) unless container_id.blank?
rescue StandardError => e
Raven.extra_context({container_id: container_id.inspect, execution_environment: execution_environment.inspect, config: config.inspect})
Raven.capture_exception(e)
nil
end
else
create_container(execution_environment)
end
end
def self.destroy_container(container)
Faraday.get(config[:location] + "/docker_container_pool/destroy_container/" + container.id)
end
def self.quantities
response = JSON.parse(Faraday.get(config[:location] + "/docker_container_pool/quantities").body)
response.transform_keys(&:to_i)
end
def self.dump_info
JSON.parse(Faraday.get(config[:location] + "/docker_container_pool/dump_info").body)
end
end