should be it

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

@@ -0,0 +1,177 @@
+/*
+ * 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