jpayne@69: // Copyright (c) 2013-2014 Sandstorm Development Group, Inc. and contributors
jpayne@69: // Licensed under the MIT License:
jpayne@69: //
jpayne@69: // Permission is hereby granted, free of charge, to any person obtaining a copy
jpayne@69: // of this software and associated documentation files (the "Software"), to deal
jpayne@69: // in the Software without restriction, including without limitation the rights
jpayne@69: // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
jpayne@69: // copies of the Software, and to permit persons to whom the Software is
jpayne@69: // furnished to do so, subject to the following conditions:
jpayne@69: //
jpayne@69: // The above copyright notice and this permission notice shall be included in
jpayne@69: // all copies or substantial portions of the Software.
jpayne@69: //
jpayne@69: // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
jpayne@69: // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
jpayne@69: // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
jpayne@69: // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
jpayne@69: // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
jpayne@69: // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
jpayne@69: // THE SOFTWARE.
jpayne@69: 
jpayne@69: #pragma once
jpayne@69: 
jpayne@69: #include "common.h"
jpayne@69: #include "function.h"
jpayne@69: #include "exception.h"
jpayne@69: 
jpayne@69: KJ_BEGIN_HEADER
jpayne@69: 
jpayne@69: namespace kj {
jpayne@69: 
jpayne@69: class Thread {
jpayne@69:   // A thread!  Pass a lambda to the constructor, and it runs in the thread.  The destructor joins
jpayne@69:   // the thread.  If the function throws an exception, it is rethrown from the thread's destructor
jpayne@69:   // (if not unwinding from another exception).
jpayne@69: 
jpayne@69: public:
jpayne@69:   explicit Thread(Function<void()> func);
jpayne@69:   KJ_DISALLOW_COPY_AND_MOVE(Thread);
jpayne@69: 
jpayne@69:   ~Thread() noexcept(false);
jpayne@69: 
jpayne@69: #if !_WIN32
jpayne@69:   void sendSignal(int signo);
jpayne@69:   // Send a Unix signal to the given thread, using pthread_kill or an equivalent.
jpayne@69: #endif
jpayne@69: 
jpayne@69:   void detach();
jpayne@69:   // Don't join the thread in ~Thread().
jpayne@69: 
jpayne@69: private:
jpayne@69:   struct ThreadState {
jpayne@69:     ThreadState(Function<void()> func);
jpayne@69: 
jpayne@69:     Function<void()> func;
jpayne@69:     Function<void(Function<void()>)> initializer;
jpayne@69:     kj::Maybe<kj::Exception> exception;
jpayne@69: 
jpayne@69:     unsigned int refcount;
jpayne@69:     // Owned by the parent thread and the child thread.
jpayne@69: 
jpayne@69:     void unref();
jpayne@69:   };
jpayne@69:   ThreadState* state;
jpayne@69: 
jpayne@69: #if _WIN32
jpayne@69:   void* threadHandle;
jpayne@69: #else
jpayne@69:   unsigned long long threadId;  // actually pthread_t
jpayne@69: #endif
jpayne@69:   bool detached = false;
jpayne@69: 
jpayne@69: #if _WIN32
jpayne@69:   static unsigned long __stdcall runThread(void* ptr);
jpayne@69: #else
jpayne@69:   static void* runThread(void* ptr);
jpayne@69: #endif
jpayne@69: };
jpayne@69: 
jpayne@69: }  // namespace kj
jpayne@69: 
jpayne@69: KJ_END_HEADER