forked from LeenkxTeam/LNXRNT
Multithread
This commit is contained in:
69
Sources/data_map.h
Normal file
69
Sources/data_map.h
Normal file
@ -0,0 +1,69 @@
|
||||
#pragma once
|
||||
|
||||
#include <cstdint>
|
||||
#include <cstdio>
|
||||
|
||||
#ifdef _WIN32
|
||||
#include <Windows.h>
|
||||
#else
|
||||
#include <sys/mman.h>
|
||||
#include <unistd.h>
|
||||
#endif
|
||||
|
||||
struct DataMap {
|
||||
void* data = nullptr;
|
||||
size_t size = 0;
|
||||
|
||||
#ifdef _WIN32
|
||||
HANDLE map_handle = nullptr;
|
||||
#endif
|
||||
|
||||
bool map_from_reader(void* reader_data, size_t reader_size) {
|
||||
size = reader_size;
|
||||
if (size == 0) return true;
|
||||
#ifdef _WIN32
|
||||
HANDLE file_handle = reader_data;
|
||||
map_handle = CreateFileMappingA(file_handle, nullptr, PAGE_READONLY, 0, 0, nullptr);
|
||||
if (!map_handle) return false;
|
||||
data = MapViewOfFile(map_handle, FILE_MAP_READ, 0, 0, 0);
|
||||
if (!data) {
|
||||
CloseHandle(map_handle);
|
||||
map_handle = nullptr;
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
#else
|
||||
int fd = fileno((FILE*)reader_data);
|
||||
if (fd < 0) return false;
|
||||
data = mmap(nullptr, size, PROT_READ | PROT_WRITE, MAP_PRIVATE, fd, 0);
|
||||
if (data == MAP_FAILED) {
|
||||
data = nullptr;
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
#endif
|
||||
}
|
||||
|
||||
void close() {
|
||||
if (data) {
|
||||
#ifdef _WIN32
|
||||
UnmapViewOfFile(data);
|
||||
#else
|
||||
munmap(data, size);
|
||||
#endif
|
||||
data = nullptr;
|
||||
}
|
||||
#ifdef _WIN32
|
||||
if (map_handle) { CloseHandle(map_handle); map_handle = nullptr; }
|
||||
#endif
|
||||
size = 0;
|
||||
}
|
||||
|
||||
~DataMap() { close(); }
|
||||
};
|
||||
|
||||
inline void delete_map(void* ptr, size_t length, void* deleter_data) {
|
||||
DataMap* dm = static_cast<DataMap*>(deleter_data);
|
||||
if (dm) dm->close();
|
||||
delete dm;
|
||||
}
|
||||
@ -95,6 +95,7 @@ bool show_window = false;
|
||||
#endif
|
||||
|
||||
#include "async_engine.h"
|
||||
#include "data_map.h"
|
||||
|
||||
// TODO
|
||||
//#ifdef KINC_KONG
|
||||
@ -233,8 +234,14 @@ namespace {
|
||||
void handle_exception(TryCatch *try_catch) {
|
||||
MaybeLocal<Value> trace = try_catch->StackTrace(isolate->GetCurrentContext());
|
||||
if (trace.IsEmpty()) {
|
||||
String::Utf8Value stack_trace(isolate, try_catch->Message()->Get());
|
||||
write_stack_trace(*stack_trace);
|
||||
MaybeLocal<v8::Message> message = try_catch->Message();
|
||||
if (message.IsEmpty()) {
|
||||
write_stack_trace("uncaught exception (no message available)");
|
||||
}
|
||||
else {
|
||||
String::Utf8Value stack_trace(isolate, message.ToLocalChecked()->Get());
|
||||
write_stack_trace(*stack_trace);
|
||||
}
|
||||
}
|
||||
else {
|
||||
String::Utf8Value stack_trace(isolate, trace.ToLocalChecked());
|
||||
@ -1436,18 +1443,33 @@ namespace {
|
||||
Local<Function> cb = Local<Function>::Cast(args[1]);
|
||||
auto persistent_cb = std::make_shared<Global<Function>>(isolate, cb);
|
||||
|
||||
EngineManager::AsyncEngine::instance().execute_io([filepath, persistent_cb]() {
|
||||
(void)EngineManager::AsyncEngine::instance().execute_io([filepath, persistent_cb]() {
|
||||
kinc_file_reader_t reader;
|
||||
bool success = kinc_file_reader_open(&reader, filepath.c_str(), KINC_FILE_TYPE_ASSET);
|
||||
std::vector<uint8_t> bytes;
|
||||
if (success) {
|
||||
uint32_t size = (uint32_t)kinc_file_reader_size(&reader);
|
||||
bytes.resize(size);
|
||||
kinc_file_reader_read(&reader, bytes.data(), size);
|
||||
kinc_file_reader_close(&reader);
|
||||
bool success = false;
|
||||
bool reader_open = false;
|
||||
DataMap* dm = nullptr;
|
||||
|
||||
try {
|
||||
success = kinc_file_reader_open(&reader, filepath.c_str(), KINC_FILE_TYPE_ASSET);
|
||||
if (success) {
|
||||
reader_open = true;
|
||||
size_t size = kinc_file_reader_size(&reader);
|
||||
dm = new DataMap();
|
||||
if (!dm->map_from_reader(reader.data, size)) {
|
||||
delete dm;
|
||||
dm = nullptr;
|
||||
success = false;
|
||||
}
|
||||
kinc_file_reader_close(&reader);
|
||||
reader_open = false;
|
||||
}
|
||||
} catch (...) {
|
||||
success = false;
|
||||
if (dm) { dm->close(); delete dm; dm = nullptr; }
|
||||
if (reader_open) { kinc_file_reader_close(&reader); reader_open = false; }
|
||||
}
|
||||
|
||||
EngineManager::AsyncEngine::instance().push_event([persistent_cb, bytes = std::move(bytes), success]() {
|
||||
EngineManager::AsyncEngine::instance().push_event([persistent_cb, dm, success]() {
|
||||
Locker locker{isolate};
|
||||
Isolate::Scope isolate_scope(isolate);
|
||||
HandleScope handle_scope(isolate);
|
||||
@ -1456,12 +1478,20 @@ namespace {
|
||||
TryCatch try_catch(isolate);
|
||||
|
||||
Local<Function> cb = Local<Function>::New(isolate, *persistent_cb);
|
||||
if (success && !bytes.empty()) {
|
||||
Local<ArrayBuffer> buffer = ArrayBuffer::New(isolate, bytes.size());
|
||||
memcpy(buffer->GetBackingStore()->Data(), bytes.data(), bytes.size());
|
||||
if (success && dm) {
|
||||
Local<ArrayBuffer> buffer;
|
||||
if (dm->data) {
|
||||
std::unique_ptr<BackingStore> backing =
|
||||
ArrayBuffer::NewBackingStore(dm->data, dm->size, delete_map, dm);
|
||||
buffer = ArrayBuffer::New(isolate, std::move(backing));
|
||||
} else {
|
||||
dm->close(); delete dm;
|
||||
buffer = ArrayBuffer::New(isolate, 0);
|
||||
}
|
||||
Local<Value> argv[1] = { buffer };
|
||||
(void)cb->Call(context, context->Global(), 1, argv);
|
||||
} else {
|
||||
if (dm) { dm->close(); delete dm; }
|
||||
Local<Value> argv[1] = { Null(isolate) };
|
||||
(void)cb->Call(context, context->Global(), 1, argv);
|
||||
}
|
||||
@ -3030,6 +3060,7 @@ namespace {
|
||||
Local<Object> navigator = Object::New(isolate);
|
||||
navigator->Set(context, String::NewFromUtf8(isolate, "userAgent").ToLocalChecked(), String::NewFromUtf8(isolate, "RunT/1.0").ToLocalChecked());
|
||||
navigator->Set(context, String::NewFromUtf8(isolate, "platform").ToLocalChecked(), String::NewFromUtf8(isolate, "RunT").ToLocalChecked());
|
||||
navigator->Set(context, String::NewFromUtf8(isolate, "hardwareConcurrency").ToLocalChecked(), Integer::New(isolate, kinc_hardware_threads()));
|
||||
window->Set(context, String::NewFromUtf8(isolate, "navigator").ToLocalChecked(), navigator);
|
||||
|
||||
Local<Object> document = Object::New(isolate);
|
||||
@ -3128,6 +3159,7 @@ namespace {
|
||||
|
||||
#ifdef WITH_WORKER
|
||||
handle_worker_messages(isolate, global_context);
|
||||
MicrotasksScope::PerformCheckpoint(isolate);
|
||||
#endif
|
||||
|
||||
#ifdef WITH_NETWORKING
|
||||
@ -4038,6 +4070,14 @@ int kickstart(int argc, char **argv) {
|
||||
|
||||
start_runt(snapshot_found ? NULL : code);
|
||||
|
||||
// TODO: find and fix
|
||||
while (update_func.IsEmpty()) {
|
||||
handle_worker_messages(isolate, global_context);
|
||||
EngineManager::AsyncEngine::instance().process_events();
|
||||
MicrotasksScope::PerformCheckpoint(isolate);
|
||||
kinc_thread_sleep(1);
|
||||
}
|
||||
|
||||
#ifdef WITH_VIEWPORT
|
||||
if (viewport_server_mode) {
|
||||
kinc_log(KINC_LOG_LEVEL_INFO, "Initializing viewport server: %dx%d, shmem=%s",
|
||||
@ -4052,27 +4092,6 @@ int kickstart(int argc, char **argv) {
|
||||
|
||||
kinc_start();
|
||||
|
||||
// NOTE: WebSocket benchmark disabled for faster bootup
|
||||
// #ifdef WITH_NETWORKING
|
||||
// kinc_log(KINC_LOG_LEVEL_INFO, "=== WEBSOCKET BOOTUP BENCHMARK ===");
|
||||
// {
|
||||
// WebSocketBenchmark benchmark;
|
||||
// kinc_log(KINC_LOG_LEVEL_INFO, "Running quick WebSocket performance check...");
|
||||
// auto result = benchmark.runCollaborative3DTest(5, 10);
|
||||
// kinc_log(KINC_LOG_LEVEL_INFO, "Benchmark Results:");
|
||||
// kinc_log(KINC_LOG_LEVEL_INFO, "- Messages/sec: %llu", result.messagesPerSecond);
|
||||
// kinc_log(KINC_LOG_LEVEL_INFO, "- Avg Latency: %.1f μs", result.averageLatency);
|
||||
// kinc_log(KINC_LOG_LEVEL_INFO, "- CPU Usage: %.1f%%", result.avgCpuPercent);
|
||||
// kinc_log(KINC_LOG_LEVEL_INFO, "- Memory: %zu MB", result.peakMemoryMB);
|
||||
// if (result.messagesPerSecond > 100) {
|
||||
// kinc_log(KINC_LOG_LEVEL_INFO, "✓ WebSocket performance: GOOD");
|
||||
// } else {
|
||||
// kinc_log(KINC_LOG_LEVEL_WARNING, "⚠ WebSocket performance: LOW - check system load");
|
||||
// }
|
||||
// }
|
||||
// kinc_log(KINC_LOG_LEVEL_INFO, "=== BENCHMARK COMPLETE ===");
|
||||
// #endif
|
||||
|
||||
#ifdef WITH_AUDIO
|
||||
if (enable_audio) {
|
||||
kinc_a2_shutdown();
|
||||
|
||||
@ -562,16 +562,16 @@ void viewport_server_set_camera(float pos_x, float pos_y, float pos_z,
|
||||
SharedFramebufferHeader* header = (SharedFramebufferHeader*)g_viewport_state.shmem_ptr;
|
||||
if (!header) return;
|
||||
|
||||
header->krom_camera_pos[0] = pos_x;
|
||||
header->krom_camera_pos[1] = pos_y;
|
||||
header->krom_camera_pos[2] = pos_z;
|
||||
header->runt_camera_pos[0] = pos_x;
|
||||
header->runt_camera_pos[1] = pos_y;
|
||||
header->runt_camera_pos[2] = pos_z;
|
||||
|
||||
header->krom_camera_rot[0] = rot_x;
|
||||
header->krom_camera_rot[1] = rot_y;
|
||||
header->krom_camera_rot[2] = rot_z;
|
||||
header->krom_camera_rot[3] = rot_w;
|
||||
header->runt_camera_rot[0] = rot_x;
|
||||
header->runt_camera_rot[1] = rot_y;
|
||||
header->runt_camera_rot[2] = rot_z;
|
||||
header->runt_camera_rot[3] = rot_w;
|
||||
|
||||
header->krom_camera_dirty = 1;
|
||||
header->runt_camera_dirty = 1;
|
||||
}
|
||||
|
||||
bool viewport_server_read_input(InputEvent* event) {
|
||||
|
||||
@ -8,30 +8,42 @@
|
||||
#include <kinc/log.h>
|
||||
#include <kinc/threads/thread.h>
|
||||
#include <kinc/threads/mutex.h>
|
||||
#include <kinc/threads/event.h>
|
||||
#include <kinc/io/filereader.h>
|
||||
#include <kinc/system.h>
|
||||
|
||||
#include <v8-value-serializer.h>
|
||||
#include <v8-wasm.h>
|
||||
#include <v8-array-buffer.h>
|
||||
#include <atomic>
|
||||
#include <memory>
|
||||
|
||||
using namespace v8;
|
||||
|
||||
namespace {
|
||||
const uint32_t worker_data_slot = 0;
|
||||
|
||||
struct TransferData;
|
||||
|
||||
struct WorkerMessage {
|
||||
char* message;
|
||||
int length;
|
||||
uint8_t* data;
|
||||
size_t size;
|
||||
TransferData* transferData;
|
||||
};
|
||||
|
||||
struct MessageQueue {
|
||||
std::vector<WorkerMessage> messages;
|
||||
kinc_mutex_t messageMutex;
|
||||
kinc_event_t event;
|
||||
Global<Function> messageFunc;
|
||||
const char* label = "?";
|
||||
};
|
||||
|
||||
struct WorkerMessagePort {
|
||||
MessageQueue workerMessages;
|
||||
MessageQueue ownerMessages;
|
||||
|
||||
bool isTerminated = false;
|
||||
std::atomic<bool> isTerminated{false};
|
||||
};
|
||||
|
||||
struct WorkerData {
|
||||
@ -53,6 +65,7 @@ namespace {
|
||||
double interval;
|
||||
double nextCallTime;
|
||||
int id;
|
||||
bool isTimeout;
|
||||
};
|
||||
|
||||
struct IntervalFunctions {
|
||||
@ -60,6 +73,70 @@ namespace {
|
||||
int latestId;
|
||||
};
|
||||
|
||||
struct TransferData {
|
||||
std::vector<std::shared_ptr<BackingStore>> sharedArrayBuffers;
|
||||
std::vector<CompiledWasmModule> wasmModules;
|
||||
std::unique_ptr<SharedValueConveyor> sharedValueDispatch;
|
||||
};
|
||||
|
||||
class SerializerDelegate : public ValueSerializer::Delegate {
|
||||
public:
|
||||
TransferData* transferData;
|
||||
|
||||
SerializerDelegate(TransferData* data) : transferData(data) {}
|
||||
|
||||
void ThrowDataCloneError(Local<String> message) override {
|
||||
Isolate* isolate = Isolate::GetCurrent();
|
||||
isolate->ThrowException(message);
|
||||
}
|
||||
|
||||
Maybe<uint32_t> GetSharedArrayBufferId(Isolate* isolate, Local<SharedArrayBuffer> shared_array_buffer) override {
|
||||
auto backingStore = shared_array_buffer->GetBackingStore();
|
||||
uint32_t id = (uint32_t)transferData->sharedArrayBuffers.size();
|
||||
transferData->sharedArrayBuffers.push_back(backingStore);
|
||||
return Just(id);
|
||||
}
|
||||
|
||||
Maybe<uint32_t> GetWasmModuleTransferId(Isolate* isolate, Local<WasmModuleObject> module) override {
|
||||
CompiledWasmModule compiled = module->GetCompiledModule();
|
||||
uint32_t id = (uint32_t)transferData->wasmModules.size();
|
||||
transferData->wasmModules.push_back(compiled);
|
||||
return Just(id);
|
||||
}
|
||||
|
||||
bool AdoptSharedValueConveyor(Isolate* isolate, SharedValueConveyor&& conveyor) override {
|
||||
transferData->sharedValueDispatch = std::make_unique<SharedValueConveyor>(std::move(conveyor));
|
||||
return true;
|
||||
}
|
||||
};
|
||||
|
||||
class DeserializerDelegate : public ValueDeserializer::Delegate {
|
||||
public:
|
||||
TransferData* transferData;
|
||||
|
||||
DeserializerDelegate(TransferData* data) : transferData(data) {}
|
||||
|
||||
MaybeLocal<SharedArrayBuffer> GetSharedArrayBufferFromId(Isolate* isolate, uint32_t clone_id) override {
|
||||
if (clone_id >= transferData->sharedArrayBuffers.size()) {
|
||||
return MaybeLocal<SharedArrayBuffer>();
|
||||
}
|
||||
return SharedArrayBuffer::New(isolate, transferData->sharedArrayBuffers[clone_id]);
|
||||
}
|
||||
|
||||
MaybeLocal<WasmModuleObject> GetWasmModuleFromId(Isolate* isolate, uint32_t transfer_id) override {
|
||||
if (transfer_id >= transferData->wasmModules.size()) {
|
||||
return MaybeLocal<WasmModuleObject>();
|
||||
}
|
||||
return WasmModuleObject::FromCompiledModule(isolate, transferData->wasmModules[transfer_id]);
|
||||
}
|
||||
|
||||
const SharedValueConveyor* GetSharedValueConveyor(Isolate* isolate) override {
|
||||
return transferData->sharedValueDispatch.get();
|
||||
}
|
||||
};
|
||||
|
||||
void handle_exception(Isolate* isolate, TryCatch* try_catch);
|
||||
|
||||
void set_onmessage(MessageQueue* messageQueue, Isolate* isolate, Local<Value> callback) {
|
||||
if (callback->IsFunction()) {
|
||||
messageQueue->messageFunc.Reset(isolate, Local<Function>::Cast(callback));
|
||||
@ -73,20 +150,53 @@ namespace {
|
||||
}
|
||||
|
||||
void post_message(MessageQueue* messageQueue, Isolate* isolate, Local<Value> messageObject) {
|
||||
Local<String> result = JSON::Stringify(isolate->GetCurrentContext(), messageObject).ToLocalChecked();
|
||||
Local<Context> context = isolate->GetCurrentContext();
|
||||
TransferData* transferData = new TransferData();
|
||||
|
||||
SerializerDelegate delegate(transferData);
|
||||
ValueSerializer serializer(isolate, &delegate);
|
||||
serializer.WriteHeader();
|
||||
|
||||
TryCatch try_catch(isolate);
|
||||
Maybe<bool> writeResult = serializer.WriteValue(context, messageObject);
|
||||
if (writeResult.IsNothing()) {
|
||||
if (try_catch.HasCaught()) {
|
||||
isolate->ThrowException(try_catch.Exception());
|
||||
} else {
|
||||
isolate->ThrowException(String::NewFromUtf8(isolate, "postMessage: failed to serialize message").ToLocalChecked());
|
||||
}
|
||||
delete transferData;
|
||||
return;
|
||||
}
|
||||
|
||||
std::pair<uint8_t*, size_t> released = serializer.Release();
|
||||
|
||||
WorkerMessage message;
|
||||
message.length = result->Length();
|
||||
message.message = new char[message.length];
|
||||
result->WriteUtf8(isolate, message.message, message.length);
|
||||
message.data = released.first;
|
||||
message.size = released.second;
|
||||
message.transferData = transferData;
|
||||
|
||||
kinc_mutex_lock(&messageQueue->messageMutex);
|
||||
messageQueue->messages.push_back(message);
|
||||
kinc_mutex_unlock(&messageQueue->messageMutex);
|
||||
kinc_event_signal(&messageQueue->event);
|
||||
}
|
||||
|
||||
void handle_message_queue(Isolate* isolate, const Global<Context>& context, MessageQueue* messageQueue) {
|
||||
kinc_mutex_lock(&messageQueue->messageMutex);
|
||||
if (messageQueue->messages.empty()) {
|
||||
kinc_mutex_unlock(&messageQueue->messageMutex);
|
||||
return;
|
||||
}
|
||||
std::vector<WorkerMessage> messages;
|
||||
messages.swap(messageQueue->messages);
|
||||
kinc_mutex_unlock(&messageQueue->messageMutex);
|
||||
|
||||
if (messageQueue->messageFunc.IsEmpty()) {
|
||||
for (WorkerMessage& message : messages) {
|
||||
free(message.data);
|
||||
delete message.transferData;
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
@ -94,29 +204,63 @@ namespace {
|
||||
Local<Context> current_context = Local<Context>::New(isolate, context);
|
||||
Context::Scope context_scope(current_context);
|
||||
|
||||
kinc_mutex_lock(&messageQueue->messageMutex);
|
||||
for (WorkerMessage& message : messages) {
|
||||
DeserializerDelegate delegate(message.transferData);
|
||||
ValueDeserializer deserializer(isolate, message.data, message.size, &delegate);
|
||||
|
||||
Maybe<bool> headerResult = deserializer.ReadHeader(current_context);
|
||||
if (headerResult.IsNothing()) {
|
||||
free(message.data);
|
||||
delete message.transferData;
|
||||
continue;
|
||||
}
|
||||
|
||||
MaybeLocal<Value> value = deserializer.ReadValue(current_context);
|
||||
|
||||
free(message.data);
|
||||
|
||||
if (value.IsEmpty()) {
|
||||
delete message.transferData;
|
||||
continue;
|
||||
}
|
||||
|
||||
for (WorkerMessage& message : messageQueue->messages) {
|
||||
MaybeLocal<Value> value = JSON::Parse(current_context, String::NewFromUtf8(isolate, message.message, NewStringType::kNormal, message.length).ToLocalChecked());
|
||||
delete[] message.message;
|
||||
Local<Object> object = Object::New(isolate);
|
||||
object->Set(current_context, String::NewFromUtf8(isolate, "data").ToLocalChecked(), value.ToLocalChecked());
|
||||
Local<Value> args[] = { object };
|
||||
|
||||
Local<Function> message_func = messageQueue->messageFunc.Get(isolate);
|
||||
message_func->Call(current_context, current_context->Global(), 1, args);
|
||||
TryCatch try_catch(isolate);
|
||||
Local<Value> callResult;
|
||||
if (!message_func->Call(current_context, current_context->Global(), 1, args).ToLocal(&callResult)) {
|
||||
handle_exception(isolate, &try_catch);
|
||||
}
|
||||
delete message.transferData;
|
||||
}
|
||||
}
|
||||
|
||||
messageQueue->messages.clear();
|
||||
|
||||
kinc_mutex_unlock(&messageQueue->messageMutex);
|
||||
void worker_console_log(const FunctionCallbackInfo<Value>& args) {
|
||||
Isolate* isolate = args.GetIsolate();
|
||||
HandleScope scope(isolate);
|
||||
std::string output;
|
||||
for (int i = 0; i < args.Length(); ++i) {
|
||||
if (i > 0) output += " ";
|
||||
String::Utf8Value utf8(isolate, args[i]);
|
||||
output += *utf8;
|
||||
}
|
||||
kinc_log(KINC_LOG_LEVEL_INFO, "%s", output.c_str());
|
||||
}
|
||||
|
||||
void handle_exception(Isolate* isolate, TryCatch* try_catch) {
|
||||
MaybeLocal<Value> trace = try_catch->StackTrace(isolate->GetCurrentContext());
|
||||
if (trace.IsEmpty()) {
|
||||
v8::String::Utf8Value stack_trace(isolate, try_catch->Message()->Get());
|
||||
kinc_log(KINC_LOG_LEVEL_ERROR, "uncaught exception %s", *stack_trace);
|
||||
MaybeLocal<v8::Message> message = try_catch->Message();
|
||||
if (message.IsEmpty()) {
|
||||
kinc_log(KINC_LOG_LEVEL_ERROR, "uncaught exception (no message available)");
|
||||
}
|
||||
else {
|
||||
v8::String::Utf8Value stack_trace(isolate, message.ToLocalChecked()->Get());
|
||||
kinc_log(KINC_LOG_LEVEL_ERROR, "uncaught exception %s", *stack_trace);
|
||||
}
|
||||
}
|
||||
else {
|
||||
v8::String::Utf8Value stack_trace(isolate, trace.ToLocalChecked());
|
||||
@ -130,9 +274,6 @@ namespace {
|
||||
if (args.Length() < 1) {
|
||||
return;
|
||||
}
|
||||
if (args.Length() > 1) {
|
||||
kinc_log(KINC_LOG_LEVEL_WARNING, "RunT workers only support 1 argument for postMessage");
|
||||
}
|
||||
|
||||
Local<External> external = Local<External>::Cast(args.Data());
|
||||
WorkerMessagePort* messagePort = (WorkerMessagePort*)external->Value();
|
||||
@ -182,41 +323,60 @@ namespace {
|
||||
set_onmessage(&messagePort->ownerMessages, isolate, args[1]);
|
||||
}
|
||||
|
||||
void worker_set_interval(const FunctionCallbackInfo<Value>& args) {
|
||||
Isolate* isolate = args.GetIsolate();
|
||||
HandleScope scope(isolate);
|
||||
|
||||
Local<External> external = Local<External>::Cast(args.Data());
|
||||
IntervalFunctions* interval_functions = (IntervalFunctions*)external->Value();
|
||||
|
||||
int add_timer(IntervalFunctions* interval_functions, Isolate* isolate, const FunctionCallbackInfo<Value>& args, bool is_timeout) {
|
||||
if (args.Length() < 1 || !args[0]->IsFunction()) {
|
||||
isolate->ThrowException(String::NewFromUtf8(isolate, "Timer callback must be a function").ToLocalChecked());
|
||||
return -1;
|
||||
}
|
||||
Local<Function> function = Local<Function>::Cast(args[0]);
|
||||
|
||||
double frequency;
|
||||
double delay;
|
||||
if (args.Length() > 1) {
|
||||
frequency = Local<Number>::Cast(args[1])->Value();
|
||||
frequency /= 1000.0;
|
||||
if (!args[1]->IsNumber()) {
|
||||
isolate->ThrowException(String::NewFromUtf8(isolate, "Timer delay must be a number").ToLocalChecked());
|
||||
return -1;
|
||||
}
|
||||
delay = Local<Number>::Cast(args[1])->Value() / 1000.0;
|
||||
}
|
||||
else {
|
||||
frequency = 0.016;
|
||||
delay = is_timeout ? 0.0 : 0.016;
|
||||
}
|
||||
|
||||
interval_functions->functions.emplace_back();
|
||||
IntervalFunction& interval_function = interval_functions->functions[interval_functions->functions.size() - 1];
|
||||
IntervalFunction& interval_function = interval_functions->functions.back();
|
||||
interval_function.function.Reset(isolate, function);
|
||||
interval_function.interval = frequency;
|
||||
interval_function.nextCallTime = kinc_time() + frequency;
|
||||
interval_function.interval = delay;
|
||||
interval_function.nextCallTime = kinc_time() + delay;
|
||||
interval_function.id = interval_functions->latestId;
|
||||
interval_function.isTimeout = is_timeout;
|
||||
|
||||
interval_functions->latestId++;
|
||||
|
||||
args.GetReturnValue().Set(interval_function.id);
|
||||
return interval_function.id;
|
||||
}
|
||||
|
||||
void worker_clear_interval(const FunctionCallbackInfo<Value>& args) {
|
||||
void worker_set_interval(const FunctionCallbackInfo<Value>& args) {
|
||||
HandleScope scope(args.GetIsolate());
|
||||
IntervalFunctions* interval_functions = (IntervalFunctions*)Local<External>::Cast(args.Data())->Value();
|
||||
args.GetReturnValue().Set(add_timer(interval_functions, args.GetIsolate(), args, false));
|
||||
}
|
||||
|
||||
void worker_set_timeout(const FunctionCallbackInfo<Value>& args) {
|
||||
HandleScope scope(args.GetIsolate());
|
||||
IntervalFunctions* interval_functions = (IntervalFunctions*)Local<External>::Cast(args.Data())->Value();
|
||||
args.GetReturnValue().Set(add_timer(interval_functions, args.GetIsolate(), args, true));
|
||||
}
|
||||
|
||||
void worker_clear_timer(const FunctionCallbackInfo<Value>& args) {
|
||||
Isolate* isolate = args.GetIsolate();
|
||||
HandleScope scope(isolate);
|
||||
if (args.Length() < 1 || !args[0]->IsInt32()) {
|
||||
return;
|
||||
}
|
||||
|
||||
Local<External> external = Local<External>::Cast(args.Data());
|
||||
IntervalFunctions* intervalFunctions = (IntervalFunctions*)external->Value();
|
||||
|
||||
int id = Local<Int32>::Cast(args[0])->Value();
|
||||
int id = args[0]->Int32Value(isolate->GetCurrentContext()).ToChecked();
|
||||
|
||||
for (std::vector<IntervalFunction>::iterator it = intervalFunctions->functions.begin(); it != intervalFunctions->functions.end(); ++it) {
|
||||
if (it->id == id) {
|
||||
@ -232,8 +392,10 @@ namespace {
|
||||
WorkerMessagePort* message_port = worker_data->messagePort;
|
||||
|
||||
Isolate::CreateParams create_params;
|
||||
create_params.array_buffer_allocator = v8::ArrayBuffer::Allocator::NewDefaultAllocator();
|
||||
auto array_buffer_allocator = v8::ArrayBuffer::Allocator::NewDefaultAllocator();
|
||||
create_params.array_buffer_allocator = array_buffer_allocator;
|
||||
Isolate* isolate = Isolate::New(create_params);
|
||||
{
|
||||
Locker locker(isolate);
|
||||
Isolate::Scope isolate_scope(isolate);
|
||||
HandleScope handle_scope(isolate);
|
||||
@ -250,13 +412,34 @@ namespace {
|
||||
Local<External> extern_interval_functions = External::New(isolate, &interval_functions);
|
||||
|
||||
global->Set(String::NewFromUtf8(isolate, "setInterval").ToLocalChecked(), FunctionTemplate::New(isolate, worker_set_interval, extern_interval_functions));
|
||||
global->Set(String::NewFromUtf8(isolate, "clearInterval").ToLocalChecked(), FunctionTemplate::New(isolate, worker_clear_interval, extern_interval_functions));
|
||||
global->Set(String::NewFromUtf8(isolate, "clearInterval").ToLocalChecked(), FunctionTemplate::New(isolate, worker_clear_timer, extern_interval_functions));
|
||||
global->Set(String::NewFromUtf8(isolate, "setTimeout").ToLocalChecked(), FunctionTemplate::New(isolate, worker_set_timeout, extern_interval_functions));
|
||||
global->Set(String::NewFromUtf8(isolate, "clearTimeout").ToLocalChecked(), FunctionTemplate::New(isolate, worker_clear_timer, extern_interval_functions));
|
||||
|
||||
Local<ObjectTemplate> console_template = ObjectTemplate::New(isolate);
|
||||
console_template->Set(String::NewFromUtf8(isolate, "log").ToLocalChecked(), FunctionTemplate::New(isolate, worker_console_log));
|
||||
console_template->Set(String::NewFromUtf8(isolate, "warn").ToLocalChecked(), FunctionTemplate::New(isolate, worker_console_log));
|
||||
console_template->Set(String::NewFromUtf8(isolate, "error").ToLocalChecked(), FunctionTemplate::New(isolate, worker_console_log));
|
||||
console_template->Set(String::NewFromUtf8(isolate, "info").ToLocalChecked(), FunctionTemplate::New(isolate, worker_console_log));
|
||||
console_template->Set(String::NewFromUtf8(isolate, "debug").ToLocalChecked(), FunctionTemplate::New(isolate, worker_console_log));
|
||||
global->Set(String::NewFromUtf8(isolate, "console").ToLocalChecked(), console_template);
|
||||
|
||||
Local<Context> context = Context::New(isolate, nullptr, global);
|
||||
Global<Context> global_context;
|
||||
global_context.Reset(isolate, context);
|
||||
Context::Scope context_scope(context);
|
||||
|
||||
Local<Object> global_obj = context->Global();
|
||||
global_obj->Set(context, String::NewFromUtf8(isolate, "self").ToLocalChecked(), global_obj).ToChecked();
|
||||
global_obj->Set(context, String::NewFromUtf8(isolate, "WorkerGlobalScope").ToLocalChecked(), global_obj).ToChecked();
|
||||
global_obj->Set(context, String::NewFromUtf8(isolate, "globalThis").ToLocalChecked(), global_obj).ToChecked();
|
||||
|
||||
Local<Object> navigator = Object::New(isolate);
|
||||
navigator->Set(context, String::NewFromUtf8(isolate, "hardwareConcurrency").ToLocalChecked(), Integer::New(isolate, kinc_hardware_threads())).ToChecked();
|
||||
navigator->Set(context, String::NewFromUtf8(isolate, "userAgent").ToLocalChecked(), String::NewFromUtf8(isolate, "RunT/1.0").ToLocalChecked()).ToChecked();
|
||||
navigator->Set(context, String::NewFromUtf8(isolate, "language").ToLocalChecked(), String::NewFromUtf8(isolate, "en-US").ToLocalChecked()).ToChecked();
|
||||
global_obj->Set(context, String::NewFromUtf8(isolate, "navigator").ToLocalChecked(), navigator).ToChecked();
|
||||
|
||||
bind_worker_class(isolate, global_context);
|
||||
|
||||
kinc_file_reader_t reader;
|
||||
@ -274,46 +457,105 @@ namespace {
|
||||
Local<String> source = String::NewFromUtf8(isolate, code, NewStringType::kNormal).ToLocalChecked();
|
||||
|
||||
TryCatch try_catch(isolate);
|
||||
Local<Script> compiled_script = Script::Compile(isolate->GetCurrentContext(), source).ToLocalChecked();
|
||||
|
||||
Local<Value> result;
|
||||
if (!compiled_script->Run(context).ToLocal(&result)) {
|
||||
Local<Script> compiled_script;
|
||||
if (!Script::Compile(isolate->GetCurrentContext(), source).ToLocal(&compiled_script)) {
|
||||
handle_exception(isolate, &try_catch);
|
||||
}
|
||||
else {
|
||||
Local<Value> result;
|
||||
if (!compiled_script->Run(context).ToLocal(&result)) {
|
||||
handle_exception(isolate, &try_catch);
|
||||
}
|
||||
MicrotasksScope::PerformCheckpoint(isolate);
|
||||
}
|
||||
|
||||
while (!message_port->isTerminated) {
|
||||
std::vector<int> due_ids;
|
||||
due_ids.reserve(8);
|
||||
while (!message_port->isTerminated.load()) {
|
||||
HandleScope loop_scope(isolate);
|
||||
double time = kinc_time();
|
||||
for (int i = 0; i < interval_functions.functions.size(); ++i) {
|
||||
if (interval_functions.functions[i].nextCallTime <= time) {
|
||||
TryCatch try_catch(isolate);
|
||||
Local<v8::Function> func = Local<v8::Function>::New(isolate, interval_functions.functions[i].function);
|
||||
Local<Value> result;
|
||||
bool had_work = false;
|
||||
due_ids.clear();
|
||||
for (const IntervalFunction& fn : interval_functions.functions) {
|
||||
if (fn.nextCallTime <= time) {
|
||||
due_ids.push_back(fn.id);
|
||||
}
|
||||
}
|
||||
|
||||
if (!func->Call(context, context->Global(), 0, NULL).ToLocal(&result)) {
|
||||
handle_exception(isolate, &try_catch);
|
||||
for (int due_id : due_ids) {
|
||||
int idx = -1;
|
||||
for (int j = 0; j < (int)interval_functions.functions.size(); ++j) {
|
||||
if (interval_functions.functions[j].id == due_id) {
|
||||
idx = j;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (idx < 0) {
|
||||
continue;
|
||||
}
|
||||
|
||||
interval_functions.functions[i].nextCallTime = time + interval_functions.functions[i].interval;
|
||||
had_work = true;
|
||||
IntervalFunction& fn = interval_functions.functions[idx];
|
||||
Local<v8::Function> func = Local<v8::Function>::New(isolate, fn.function);
|
||||
|
||||
if (fn.isTimeout) {
|
||||
fn.function.Reset();
|
||||
interval_functions.functions.erase(interval_functions.functions.begin() + idx);
|
||||
}
|
||||
else {
|
||||
fn.nextCallTime = time + fn.interval;
|
||||
}
|
||||
|
||||
TryCatch try_catch(isolate);
|
||||
Local<Value> result;
|
||||
if (!func->Call(context, context->Global(), 0, NULL).ToLocal(&result)) {
|
||||
handle_exception(isolate, &try_catch);
|
||||
}
|
||||
}
|
||||
|
||||
handle_message_queue(isolate, global_context, &message_port->ownerMessages);
|
||||
MicrotasksScope::PerformCheckpoint(isolate);
|
||||
|
||||
handle_worker_messages(isolate, global_context);
|
||||
if (!had_work && !message_port->isTerminated.load()) {
|
||||
double timeout = 1.0;
|
||||
for (const IntervalFunction& fn : interval_functions.functions) {
|
||||
double dt = fn.nextCallTime - time;
|
||||
if (dt < timeout) {
|
||||
timeout = dt;
|
||||
}
|
||||
}
|
||||
if (timeout < 0.0) {
|
||||
timeout = 0.0;
|
||||
}
|
||||
kinc_event_try_to_wait(&message_port->ownerMessages.event, timeout);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
ContextData* context_data = (ContextData*)isolate->GetData(worker_data_slot);
|
||||
|
||||
for (int i = 0; i < context_data->workers.size(); ++i) {
|
||||
WorkerMessagePort* workerPort = context_data->workers[i].workerMessagePort;
|
||||
workerPort->isTerminated = true;
|
||||
workerPort->isTerminated.store(true);
|
||||
|
||||
|
||||
kinc_event_signal(&workerPort->ownerMessages.event);
|
||||
|
||||
kinc_thread_wait_and_destroy(context_data->workers[i].workerThread);
|
||||
|
||||
delete context_data->workers[i].workerThread;
|
||||
kinc_mutex_destroy(&context_data->workers[i].workerMessagePort->ownerMessages.messageMutex);
|
||||
kinc_mutex_destroy(&context_data->workers[i].workerMessagePort->workerMessages.messageMutex);
|
||||
for (auto& msg : workerPort->ownerMessages.messages) {
|
||||
free(msg.data);
|
||||
delete msg.transferData;
|
||||
}
|
||||
for (auto& msg : workerPort->workerMessages.messages) {
|
||||
free(msg.data);
|
||||
delete msg.transferData;
|
||||
}
|
||||
kinc_mutex_destroy(&workerPort->ownerMessages.messageMutex);
|
||||
kinc_mutex_destroy(&workerPort->workerMessages.messageMutex);
|
||||
kinc_event_destroy(&workerPort->ownerMessages.event);
|
||||
kinc_event_destroy(&workerPort->workerMessages.event);
|
||||
delete workerPort;
|
||||
}
|
||||
|
||||
@ -321,6 +563,9 @@ namespace {
|
||||
delete worker_data;
|
||||
delete context_data;
|
||||
isolate->TerminateExecution();
|
||||
}
|
||||
isolate->Dispose();
|
||||
delete array_buffer_allocator;
|
||||
}
|
||||
|
||||
void owner_onmessage_get(Local<String> property,const PropertyCallbackInfo<Value>& info) {
|
||||
@ -373,9 +618,6 @@ namespace {
|
||||
if (args.Length() < 1) {
|
||||
return;
|
||||
}
|
||||
if (args.Length() > 1) {
|
||||
kinc_log(KINC_LOG_LEVEL_WARNING, "RunT workers only support 1 argument for postMessage");
|
||||
}
|
||||
|
||||
Local<External> external = Local<External>::Cast(args.This()->GetInternalField(0));
|
||||
WorkerMessagePort* messagePort = (WorkerMessagePort*)external->Value();
|
||||
@ -390,7 +632,8 @@ namespace {
|
||||
Local<External> external = Local<External>::Cast(args.This()->GetInternalField(0));
|
||||
WorkerMessagePort* messagePort = (WorkerMessagePort*)external->Value();
|
||||
|
||||
messagePort->isTerminated = true;
|
||||
messagePort->isTerminated.store(true);
|
||||
kinc_event_signal(&messagePort->ownerMessages.event);
|
||||
|
||||
ContextData* context_data = (ContextData*)isolate->GetData(worker_data_slot);
|
||||
for (std::vector<OwnedWorker>::iterator it = context_data->workers.begin(); it != context_data->workers.end(); ++it) {
|
||||
@ -401,8 +644,18 @@ namespace {
|
||||
}
|
||||
}
|
||||
|
||||
for (auto& msg : messagePort->ownerMessages.messages) {
|
||||
free(msg.data);
|
||||
delete msg.transferData;
|
||||
}
|
||||
for (auto& msg : messagePort->workerMessages.messages) {
|
||||
free(msg.data);
|
||||
delete msg.transferData;
|
||||
}
|
||||
kinc_mutex_destroy(&messagePort->ownerMessages.messageMutex);
|
||||
kinc_mutex_destroy(&messagePort->workerMessages.messageMutex);
|
||||
kinc_event_destroy(&messagePort->ownerMessages.event);
|
||||
kinc_event_destroy(&messagePort->workerMessages.event);
|
||||
delete messagePort;
|
||||
|
||||
args.This()->SetInternalField(0, External::New(isolate, nullptr));
|
||||
@ -421,13 +674,13 @@ namespace {
|
||||
return;
|
||||
}
|
||||
|
||||
if (args.Length() > 1) {
|
||||
kinc_log(KINC_LOG_LEVEL_WARNING, "RunT only supports one argument for worker constructor, ignoring extra arguments");
|
||||
}
|
||||
|
||||
WorkerMessagePort* messagePort = new WorkerMessagePort;
|
||||
messagePort->ownerMessages.label = "ownerMessages(main->worker)";
|
||||
messagePort->workerMessages.label = "workerMessages(worker->main)";
|
||||
kinc_mutex_init(&messagePort->ownerMessages.messageMutex);
|
||||
kinc_mutex_init(&messagePort->workerMessages.messageMutex);
|
||||
kinc_event_init(&messagePort->ownerMessages.event, true);
|
||||
kinc_event_init(&messagePort->workerMessages.event, true);
|
||||
|
||||
Local<Context> context = isolate->GetCurrentContext();
|
||||
|
||||
@ -452,14 +705,22 @@ namespace {
|
||||
worker_data->fileName[length] = 0;
|
||||
|
||||
// Check if file exists before we create thread
|
||||
kinc_file_reader fileCheck;
|
||||
kinc_file_reader_t fileCheck;
|
||||
if (!kinc_file_reader_open(&fileCheck, worker_data->fileName, KINC_FILE_TYPE_ASSET)) {
|
||||
const char* errorFmt = "Worker constructor: file %s does not exist";
|
||||
int length = snprintf(nullptr, 0, errorFmt, worker_data->fileName);
|
||||
char* errorMessage = (char*)malloc(length + 1);
|
||||
snprintf(errorMessage, length, errorFmt, worker_data->fileName);
|
||||
snprintf(errorMessage, length + 1, errorFmt, worker_data->fileName);
|
||||
isolate->ThrowException(String::NewFromUtf8(isolate, errorMessage).ToLocalChecked());
|
||||
free(errorMessage);
|
||||
delete owned_worker.workerThread;
|
||||
delete worker_data;
|
||||
kinc_mutex_destroy(&messagePort->ownerMessages.messageMutex);
|
||||
kinc_mutex_destroy(&messagePort->workerMessages.messageMutex);
|
||||
kinc_event_destroy(&messagePort->ownerMessages.event);
|
||||
kinc_event_destroy(&messagePort->workerMessages.event);
|
||||
delete messagePort;
|
||||
args.This()->SetInternalField(0, External::New(isolate, nullptr));
|
||||
return;
|
||||
}
|
||||
kinc_file_reader_close(&fileCheck);
|
||||
|
||||
Reference in New Issue
Block a user