Reorder reservation removal to a more logical spot (861a8937) · Commits · olcf / container-builder

BuilderQueue/include/BuilderQueue.h

+2 −2

Original line number	Diff line number	Diff line
		@@ -10,8 +10,8 @@
		class BuilderQueue {
		public:
		explicit BuilderQueue(asio::io_service &io_service) : io_service(io_service),
		max_builders(1),
		max_available_builders(max_builders),
		max_builders(20),
		max_available_builders(5),
		pending_requests(0)
		{}

Client/main.cpp

+62 −30

Original line number	Diff line number	Diff line
		@@ -7,7 +7,7 @@
		#include <boost/asio.hpp>
		#include <iostream>
		#include "Messenger.h"
		#include <memory>


		namespace asio = boost::asio;
		using asio::ip::tcp;
		@@ -28,6 +28,34 @@ std::string queue_port() {
		}
		return std::string(env);
		}
		/*
		void display_spinner(asio::io_service &io_service, asio::yield_context &yield, bool display) {
		static bool should_display;
		static boost::asio::deadline_timer timer(io_service);

		should_display = display;

		auto expire_time = boost::posix_time::milliseconds(250);

		asio::spawn(io_service,
		[&](asio::yield_context yield) {
		while (should_display) {
		std::cout << "\b--" << std::flush;
		timer.expires_from_now(expire_time);
		timer.async_wait(yield);
		std::cout << "\b\\" << std::flush;
		timer.expires_from_now(expire_time);
		timer.async_wait(yield);
		std::cout << "\b\|" << std::flush;
		timer.expires_from_now(expire_time);
		timer.async_wait(yield);
		timer.expires_from_now(expire_time);
		std::cout << "\b/" << std::flush;
		timer.async_wait(yield);
		}
		});
		}
		*/

		int main(int argc, char *argv[]) {

		@@ -41,32 +69,38 @@ int main(int argc, char *argv[]) {
		std::string definition_path(argv[1]);
		std::string container_path(argv[2]);

		std::cout << "Attempting to connect to BuilderQueue: " << queue_host() << ":" << queue_port() << std::endl;

		asio::io_service io_service;

		asio::spawn(io_service,
		[&](asio::yield_context yield) {

		std::cout << "Attempting to connect to BuilderQueue: " << queue_host() << ":" << queue_port()
		<< std::endl;

		tcp::socket queue_socket(io_service);
		tcp::resolver queue_resolver(io_service);
		boost::system::error_code ec;
		asio::connect(queue_socket, queue_resolver.resolve({queue_host(), queue_port()}));

		asio::async_connect(queue_socket, queue_resolver.resolve({queue_host(), queue_port()}), yield);

		std::cout << "Connected to BuilderQueue: " << queue_host() << ":" << queue_port() << std::endl;

		Messenger queue_messenger(queue_socket);

		// Initiate a build request
		queue_messenger.send("checkout_builder_request");
		queue_messenger.async_send("checkout_builder_request", yield);

		// Receive an available builder
		auto builder = queue_messenger.receive_builder();
		// Wait on a builder from the queue
		auto builder = queue_messenger.async_receive_builder(yield);

		std::cout << "Attempting to connect to build host: " << builder.host << ":" << builder.port << std::endl;
		std::cout << "Attempting to connect to build host: " << builder.host << ":" << builder.port
		<< std::endl;

		// Block until the initial connection to the builder is made
		tcp::socket builder_socket(io_service);
		tcp::resolver builder_resolver(io_service);
		do {
		asio::connect(builder_socket, builder_resolver.resolve({builder.host, builder.port}), ec);
		asio::async_connect(builder_socket, builder_resolver.resolve({builder.host, builder.port}),
		yield[ec]);
		} while (ec != boost::system::errc::success);

		std::cout << "Connected to builder host: " << builder.host << ":" << builder.port << std::endl;
		@@ -74,14 +108,12 @@ int main(int argc, char *argv[]) {
		Messenger builder_messenger(builder_socket);

		// Once we're connected to the builder start the client process
		asio::spawn(io_service,
		[&](asio::yield_context yield) {
		std::cout << "Sending definition: " << definition_path << std::endl;

		// Send the definition file
		builder_messenger.async_send_file(definition_path, yield, true);

		std::cout << "Start reading builder output:" << std::endl;
		std::cout << "Start of Singularity builder output:" << std::endl;

		// Hide the cursor and disable buffering for cleaner output
		std::cout << "\e[?25l" << std::flush;
		@@ -94,11 +126,11 @@ int main(int argc, char *argv[]) {
		} while (!line.empty());

		// Read the container image
		std::cout << "Begin receive of container: " << container_path << std::endl;
		std::cout << "Sending finished container: " << container_path << std::endl;

		builder_messenger.async_receive_file(container_path, yield, true);

		std::cout << "End receive of container: " << container_path << std::endl;
		std::cout << "Container received: " << container_path << std::endl;

		// Inform the queue we're done
		queue_messenger.async_send(std::string("checkout_builder_complete"), yield);