email-tracker/external/duckdb/third_party/tpce-tool/include/utilities/threading.h

/*
 * Legal Notice
 *
 * This document and associated source code (the "Work") is a part of a
 * benchmark specification maintained by the TPC.
 *
 * The TPC reserves all right, title, and interest to the Work as provided
 * under U.S. and international laws, including without limitation all patent
 * and trademark rights therein.
 *
 * No Warranty
 *
 * 1.1 TO THE MAXIMUM EXTENT PERMITTED BY APPLICABLE LAW, THE INFORMATION
 *     CONTAINED HEREIN IS PROVIDED "AS IS" AND WITH ALL FAULTS, AND THE
 *     AUTHORS AND DEVELOPERS OF THE WORK HEREBY DISCLAIM ALL OTHER
 *     WARRANTIES AND CONDITIONS, EITHER EXPRESS, IMPLIED OR STATUTORY,
 *     INCLUDING, BUT NOT LIMITED TO, ANY (IF ANY) IMPLIED WARRANTIES,
 *     DUTIES OR CONDITIONS OF MERCHANTABILITY, OF FITNESS FOR A PARTICULAR
 *     PURPOSE, OF ACCURACY OR COMPLETENESS OF RESPONSES, OF RESULTS, OF
 *     WORKMANLIKE EFFORT, OF LACK OF VIRUSES, AND OF LACK OF NEGLIGENCE.
 *     ALSO, THERE IS NO WARRANTY OR CONDITION OF TITLE, QUIET ENJOYMENT,
 *     QUIET POSSESSION, CORRESPONDENCE TO DESCRIPTION OR NON-INFRINGEMENT
 *     WITH REGARD TO THE WORK.
 * 1.2 IN NO EVENT WILL ANY AUTHOR OR DEVELOPER OF THE WORK BE LIABLE TO
 *     ANY OTHER PARTY FOR ANY DAMAGES, INCLUDING BUT NOT LIMITED TO THE
 *     COST OF PROCURING SUBSTITUTE GOODS OR SERVICES, LOST PROFITS, LOSS
 *     OF USE, LOSS OF DATA, OR ANY INCIDENTAL, CONSEQUENTIAL, DIRECT,
 *     INDIRECT, OR SPECIAL DAMAGES WHETHER UNDER CONTRACT, TORT, WARRANTY,
 *     OR OTHERWISE, ARISING IN ANY WAY OUT OF THIS OR ANY OTHER AGREEMENT
 *     RELATING TO THE WORK, WHETHER OR NOT SUCH AUTHOR OR DEVELOPER HAD
 *     ADVANCE NOTICE OF THE POSSIBILITY OF SUCH DAMAGES.
 *
 * Contributors
 * - Chris Chan-nui, Matt Emmerton
 */

#ifndef THREADING_H_INCLUDED
#define THREADING_H_INCLUDED

#include <memory>
#include <stdexcept>

#ifndef WIN32
#include <sstream>
#include <cstring>
#endif

#include "EGenStandardTypes.h"

namespace TPCE {

// Base class to provide a run() method for objects which can be threaded.
// This is required because under pthreads we have to provide an interface
// through a C ABI call, which we can't do with templated classes.
class ThreadBase {
public:
	virtual ~ThreadBase();
	virtual void invoke() = 0;
};

// Call the run() method of passed argument.  Always returns NULL.
#ifdef WIN32
DWORD WINAPI start_thread(LPVOID arg);
#else
extern "C" void *start_thread(void *arg);
#endif

// Template to wrap around a class that has a ThreadBase::run() method and
// spawn it in a thread of its own.
template <typename T> class Thread : public ThreadBase {
private:
	std::unique_ptr<T> obj_;
	TThread tid_;
#ifndef WIN32
	TThreadAttr attr_;
#endif
	int stacksize_;

public:
	Thread(std::unique_ptr<T> throbj)
	    : obj_(throbj), tid_()
#ifndef WIN32
	      ,
	      attr_()
#endif
	      ,
	      stacksize_(0) {
#ifndef WIN32
		pthread_attr_init(&attr_);
#endif
	}
	Thread(std::unique_ptr<T> throbj, int stacksize)
	    : obj_(throbj), tid_()
#ifndef WIN32
	      ,
	      attr_()
#endif
	      ,
	      stacksize_(stacksize) {
#ifndef WIN32
		pthread_attr_init(&attr_);
#endif
	}
	T *obj() {
		return obj_.get();
	}
	void invoke() {
		obj_->run(this);
	}
	void start();
	void stop();
};

//////////////////////////////////////////////////////////
// Windows Implementation
//////////////////////////////////////////////////////////

#ifdef WIN32

template <typename T> void Thread<T>::start() {
	tid_ = CreateThread(NULL, stacksize_, start_thread, this, NULL, NULL);
	if (tid_ == NULL) {
		std::ostringstream strm;
		strm << "CreateThread error: " << GetLastError();
		throw std::runtime_error(strm.str());
	}
}

template <typename T> void Thread<T>::stop() {
	DWORD rc = WaitForSingleObject(tid_, INFINITE);
	if (rc != 0) {
		std::ostringstream strm;
		strm << "WaitForSingleObject error: " << GetLastError();
		throw std::runtime_error(strm.str());
	}
}

//////////////////////////////////////////////////////////
// Non-Windows (pthread) Implementation
//////////////////////////////////////////////////////////

#else

template <typename T> void Thread<T>::start() {
	int rc = 0;

	if (stacksize_ > 0) {
		rc = pthread_attr_setstacksize(&attr_, stacksize_);
		if (rc != 0) {
			std::ostringstream strm;
			strm << "pthread_attr_setstacksize error: " << strerror(rc) << "(" << rc << ")";
			throw std::runtime_error(strm.str());
		}
	}

	rc = pthread_create(&tid_, &attr_, start_thread, this);
	if (rc != 0) {
		std::ostringstream strm;
		strm << "pthread_create error: " << strerror(rc) << "(" << rc << ")";
		throw std::runtime_error(strm.str());
	}
}

template <typename T> void Thread<T>::stop() {
	int rc = pthread_join(tid_, NULL);
	if (rc != 0) {
		std::ostringstream strm;
		strm << "pthread_join error: " << strerror(rc) << "(" << rc << ")";
		throw std::runtime_error(strm.str());
	}
}

#endif

} // namespace TPCE

#endif // THREADING_H_INCLUDED