mirror of https://github.com/lukechilds/node.git
Fedor Indutny
10 years ago
4 changed files with 248 additions and 5 deletions
@ -0,0 +1,166 @@ |
|||||
|
// Copyright Fedor Indutny and other Node contributors.
|
||||
|
//
|
||||
|
// Permission is hereby granted, free of charge, to any person obtaining a
|
||||
|
// copy of this software and associated documentation files (the
|
||||
|
// "Software"), to deal in the Software without restriction, including
|
||||
|
// without limitation the rights to use, copy, modify, merge, publish,
|
||||
|
// distribute, sublicense, and/or sell copies of the Software, and to permit
|
||||
|
// persons to whom the Software is furnished to do so, subject to the
|
||||
|
// following conditions:
|
||||
|
//
|
||||
|
// The above copyright notice and this permission notice shall be included
|
||||
|
// in all copies or substantial portions of the Software.
|
||||
|
//
|
||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
||||
|
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
||||
|
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
|
||||
|
// NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
|
||||
|
// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
|
||||
|
// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
|
||||
|
// USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||
|
|
||||
|
#include "node_v8_platform.h" |
||||
|
|
||||
|
#include "node.h" |
||||
|
#include "util.h" |
||||
|
#include "util-inl.h" |
||||
|
#include "uv.h" |
||||
|
#include "v8-platform.h" |
||||
|
|
||||
|
namespace node { |
||||
|
|
||||
|
using v8::Task; |
||||
|
using v8::Isolate; |
||||
|
|
||||
|
// The last task to encounter before killing the worker
|
||||
|
class StopTask : public Task { |
||||
|
public: |
||||
|
StopTask() {} |
||||
|
~StopTask() {} |
||||
|
|
||||
|
void Run() {} |
||||
|
}; |
||||
|
|
||||
|
static StopTask stop_task_; |
||||
|
|
||||
|
|
||||
|
Platform::Platform(unsigned int worker_count) : worker_count_(worker_count) { |
||||
|
workers_ = new uv_thread_t[worker_count_]; |
||||
|
|
||||
|
for (unsigned int i = 0; i < worker_count_; i++) { |
||||
|
int err; |
||||
|
|
||||
|
err = uv_thread_create(worker_at(i), WorkerBody, this); |
||||
|
CHECK_EQ(err, 0); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
|
||||
|
Platform::~Platform() { |
||||
|
// Push stop task
|
||||
|
for (unsigned int i = 0; i < worker_count(); i++) |
||||
|
global_queue()->Push(&stop_task_); |
||||
|
|
||||
|
// And wait for workers to exit
|
||||
|
for (unsigned int i = 0; i < worker_count(); i++) { |
||||
|
int err; |
||||
|
|
||||
|
err = uv_thread_join(worker_at(i)); |
||||
|
CHECK_EQ(err, 0); |
||||
|
} |
||||
|
delete[] workers_; |
||||
|
} |
||||
|
|
||||
|
|
||||
|
void Platform::CallOnBackgroundThread(Task* task, |
||||
|
ExpectedRuntime expected_runtime) { |
||||
|
global_queue()->Push(task); |
||||
|
} |
||||
|
|
||||
|
|
||||
|
void Platform::CallOnForegroundThread(Isolate* isolate, Task* task) { |
||||
|
// TODO(indutny): create per-isolate thread pool
|
||||
|
global_queue()->Push(task); |
||||
|
} |
||||
|
|
||||
|
|
||||
|
void Platform::WorkerBody(void* arg) { |
||||
|
Platform* p = static_cast<Platform*>(arg); |
||||
|
|
||||
|
for (;;) { |
||||
|
Task* task = p->global_queue()->Shift(); |
||||
|
if (task == &stop_task_) |
||||
|
break; |
||||
|
|
||||
|
task->Run(); |
||||
|
delete task; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
|
||||
|
TaskQueue::TaskQueue() { |
||||
|
int err; |
||||
|
|
||||
|
STATIC_ASSERT(kRingSize == (kRingSize & (~(kRingSize - 1)))); |
||||
|
|
||||
|
size_ = kRingSize; |
||||
|
ring_ = new Task*[size_]; |
||||
|
mask_ = size_ - 1; |
||||
|
read_off_ = 0; |
||||
|
write_off_ = 0; |
||||
|
|
||||
|
err = uv_sem_init(&sem_, 0); |
||||
|
CHECK_EQ(err, 0); |
||||
|
|
||||
|
err = uv_cond_init(&cond_); |
||||
|
CHECK_EQ(err, 0); |
||||
|
|
||||
|
err = uv_mutex_init(&mutex_); |
||||
|
CHECK_EQ(err, 0); |
||||
|
} |
||||
|
|
||||
|
|
||||
|
TaskQueue::~TaskQueue() { |
||||
|
CHECK_EQ(read_off_, write_off_); |
||||
|
|
||||
|
delete[] ring_; |
||||
|
ring_ = NULL; |
||||
|
uv_sem_destroy(&sem_); |
||||
|
uv_cond_destroy(&cond_); |
||||
|
uv_mutex_destroy(&mutex_); |
||||
|
} |
||||
|
|
||||
|
|
||||
|
void TaskQueue::Push(Task* task) { |
||||
|
uv_mutex_lock(&mutex_); |
||||
|
|
||||
|
// Wait for empty cell
|
||||
|
while (ring_[write_off_] != NULL) |
||||
|
uv_cond_wait(&cond_, &mutex_); |
||||
|
|
||||
|
ring_[write_off_] = task; |
||||
|
write_off_++; |
||||
|
write_off_ &= mask_; |
||||
|
uv_mutex_unlock(&mutex_); |
||||
|
|
||||
|
uv_sem_post(&sem_); |
||||
|
} |
||||
|
|
||||
|
|
||||
|
Task* TaskQueue::Shift() { |
||||
|
uv_sem_wait(&sem_); |
||||
|
|
||||
|
uv_mutex_lock(&mutex_); |
||||
|
Task* task = ring_[read_off_]; |
||||
|
ring_[read_off_] = NULL; |
||||
|
uv_cond_signal(&cond_); |
||||
|
|
||||
|
read_off_++; |
||||
|
read_off_ &= mask_; |
||||
|
uv_mutex_unlock(&mutex_); |
||||
|
|
||||
|
return task; |
||||
|
} |
||||
|
|
||||
|
|
||||
|
} // namespace node
|
@ -0,0 +1,74 @@ |
|||||
|
// Copyright Fedor Indutny and other Node contributors.
|
||||
|
//
|
||||
|
// Permission is hereby granted, free of charge, to any person obtaining a
|
||||
|
// copy of this software and associated documentation files (the
|
||||
|
// "Software"), to deal in the Software without restriction, including
|
||||
|
// without limitation the rights to use, copy, modify, merge, publish,
|
||||
|
// distribute, sublicense, and/or sell copies of the Software, and to permit
|
||||
|
// persons to whom the Software is furnished to do so, subject to the
|
||||
|
// following conditions:
|
||||
|
//
|
||||
|
// The above copyright notice and this permission notice shall be included
|
||||
|
// in all copies or substantial portions of the Software.
|
||||
|
//
|
||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
||||
|
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
||||
|
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
|
||||
|
// NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
|
||||
|
// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
|
||||
|
// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
|
||||
|
// USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||
|
|
||||
|
#ifndef SRC_NODE_V8_PLATFORM_H_ |
||||
|
#define SRC_NODE_V8_PLATFORM_H_ |
||||
|
|
||||
|
#include "uv.h" |
||||
|
#include "v8-platform.h" |
||||
|
|
||||
|
namespace node { |
||||
|
|
||||
|
class TaskQueue { |
||||
|
public: |
||||
|
TaskQueue(); |
||||
|
~TaskQueue(); |
||||
|
|
||||
|
void Push(v8::Task* task); |
||||
|
v8::Task* Shift(); |
||||
|
|
||||
|
private: |
||||
|
static const unsigned int kRingSize = 1024; |
||||
|
|
||||
|
uv_sem_t sem_; |
||||
|
uv_cond_t cond_; |
||||
|
uv_mutex_t mutex_; |
||||
|
v8::Task** ring_; |
||||
|
unsigned int size_; |
||||
|
unsigned int mask_; |
||||
|
unsigned int read_off_; |
||||
|
unsigned int write_off_; |
||||
|
}; |
||||
|
|
||||
|
class Platform : public v8::Platform { |
||||
|
public: |
||||
|
explicit Platform(unsigned int worker_count); |
||||
|
~Platform(); |
||||
|
|
||||
|
void CallOnBackgroundThread(v8::Task* task, |
||||
|
ExpectedRuntime expected_runtime); |
||||
|
void CallOnForegroundThread(v8::Isolate* isolate, v8::Task* task); |
||||
|
|
||||
|
protected: |
||||
|
static void WorkerBody(void* arg); |
||||
|
|
||||
|
inline TaskQueue* global_queue() { return &global_queue_; } |
||||
|
inline uv_thread_t* worker_at(unsigned int index) { return &workers_[index]; } |
||||
|
inline unsigned int worker_count() const { return worker_count_; } |
||||
|
|
||||
|
uv_thread_t* workers_; |
||||
|
unsigned int worker_count_; |
||||
|
TaskQueue global_queue_; |
||||
|
}; |
||||
|
|
||||
|
} // namespace node
|
||||
|
|
||||
|
#endif // SRC_NODE_V8_PLATFORM_H_
|
Loading…
Reference in new issue