// Copyright (c) 2018 Emil Dotchevski
// Copyright (c) 2018 Second Spectrum, Inc.

// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)

// This is a simple program that demonstrates the use of LEAF to transport e-objects between threads,
// without using exception handling. See capture_eh.cpp for the exception-handling variant.

#include <boost/leaf/capture_exception.hpp>
#include <boost/leaf/try.hpp>
#include <boost/leaf/throw.hpp>
#include <vector>
#include <string>
#include <future>
#include <iterator>
#include <iostream>
#include <algorithm>

namespace leaf = boost::leaf;

// Define several e-types.
struct e_thread_id { std::thread::id value; };
struct e_failure_info1 { std::string value; };
struct e_failure_info2 { int value; };

// A type that represents a successfully returned result from a task.
struct task_result { };

 // This is our task function. It produces objects of type task_result, but it may fail...
task_result task()
{
	bool succeed = (rand()%4) !=0; //...at random.
	if( succeed )
		return task_result { };
	else
		throw leaf::exception( std::exception(),
			e_thread_id{std::this_thread::get_id()},
			e_failure_info1{"info"},
			e_failure_info2{42} );
};

int main()
{
	int const task_count = 42;

	// Container to collect the generated std::future objects.
	std::vector<std::future<task_result>> fut;

	// Launch the tasks, but rather than launching the task function directly, we launch the
	// wrapper function returned by leaf::capture_result. It captures the specified error object
	// types and automatically transports them in the leaf::result<task_result> it returns.
	std::generate_n( std::inserter(fut,fut.end()), task_count,
		[ ]
		{
			return std::async(
				std::launch::async,
				leaf::capture_exception<e_thread_id, e_failure_info1, e_failure_info2>(task) );
		} );

	// Wait on the futures, get the task results, handle errors.
	for( auto & f : fut )
	{
		f.wait();

		leaf::try_(
			[&]
			{
				task_result r = f.get();

				// Success! Use r to access task_result.
				std::cout << "Success!" << std::endl;
				(void) r;

			},

			[ ]( e_failure_info1 const & v1, e_failure_info2 const & v2, e_thread_id const & tid )
			{
				std::cerr << "Error in thread " << tid.value << "! failure_info1: " << v1.value << ", failure_info2: " << v2.value << std::endl;
			},

			[ ]( leaf::error_info const & unmatched )
			{
				std::cerr <<
					"Unknown failure detected" << std::endl <<
					"Cryptic diagnostic information follows" << std::endl <<
					unmatched;
			} );
	}
}