/*
Post a message to this worker. Note: the worker is the destination worker which may be the parent.
function postMessage(data: Object, ports: Array = null): Void
*/
static EjsObj *workerPostMessage(Ejs *ejs, EjsWorker *worker, int argc, EjsObj **argv)
{
EjsString *data;
EjsWorker *target;
MprDispatcher *dispatcher;
Message *msg;
if (worker->state >= EJS_WORKER_CLOSED) {
ejsThrowStateError(ejs, "Worker has completed");
return 0;
}
/*
Create the event with serialized data in the originating interpreter. It owns the data.
*/
ejsBlockGC(ejs);
if ((data = ejsToJSON(ejs, argv[0], NULL)) == 0) {
ejsThrowArgError(ejs, "Cannot serialize message data");
return 0;
}
if ((msg = createMessage()) == 0) {
ejsThrowMemoryError(ejs);
return 0;
}
target = worker->pair;
msg->data = ejsToMulti(ejs, data);
msg->worker = target;
msg->callback = "onmessage";
msg->callbackSlot = ES_Worker_onmessage;
dispatcher = target->ejs->dispatcher;
mprCreateEvent(dispatcher, "postMessage", 0, doMessage, msg, 0);
return 0;
}
/*
Worker thread main procedure. Worker is the inside worker.
*/
static int workerMain(EjsWorker *insideWorker, MprEvent *event)
{
Ejs *outside, *inside;
EjsWorker *outsideWorker;
MprDispatcher *dispatcher;
Message *msg;
assert(insideWorker->inside);
outsideWorker = insideWorker->pair;
assert(!outsideWorker->inside);
assert(insideWorker->state == EJS_WORKER_BEGIN);
outside = outsideWorker->ejs;
inside = insideWorker->ejs;
insideWorker->state = EJS_WORKER_STARTED;
/*
Run the script or file
*/
if (outsideWorker->scriptFile) {
loadFile(insideWorker, outsideWorker->scriptFile);
} else if (outsideWorker->scriptLiteral) {
if (outside->service->loadScriptLiteral == 0) {
ejsThrowIOError(outside, "worker: Compiling is not enabled");
return 0;
}
(outside->service->loadScriptLiteral)(inside, outsideWorker->scriptLiteral, NULL);
}
/*
Check for exceptions
*/
if (inside->exception) {
handleError(outside, outsideWorker, inside->exception, 0);
inside->exception = 0;
}
ejsBlockGC(inside);
if ((msg = createMessage()) == 0) {
ejsThrowMemoryError(outside);
return 0;
}
/*
Post "onclose" finalization message
*/
msg->worker = outsideWorker;
msg->callback = "onclose";
msg->callbackSlot = ES_Worker_onclose;
insideWorker->state = EJS_WORKER_CLOSED;
outsideWorker->state = EJS_WORKER_CLOSED;
dispatcher = outside->dispatcher;
mprCreateEvent(dispatcher, "doMessage", 0, (MprEventProc) doMessage, msg, 0);
return 0;
}
/*
function emit(level: Number, ...data): Number
*/
static EjsNumber *lf_emit(Ejs *ejs, EjsObj *unused, int argc, EjsObj **argv)
{
EjsArray *args;
EjsByteArray *ap;
EjsObj *vp;
EjsString *str;
char *msg, *arg;
ssize len, written;
int i, level, paused;
assure(argc >= 2 && ejsIs(ejs, argv[1], Array));
level = ejsGetInt(ejs, argv[0]);
args = (EjsArray*) argv[1];
written = 0;
msg = 0;
paused = ejsBlockGC(ejs);
for (i = 0; i < args->length; i++) {
vp = ejsGetProperty(ejs, args, i);
assure(vp);
switch (TYPE(vp)->sid) {
case S_ByteArray:
ap = (EjsByteArray*) vp;
// TODO ENCODING
arg = (char*) &ap->value[ap->readPosition];
len = ap->writePosition - ap->readPosition;
break;
case S_String:
// MOB - use NULL instead of &len
arg = awtom(((EjsString*) vp)->value, &len);
break;
default:
str = ejsToString(ejs, vp);
// MOB - use NULL instead of &len
arg = awtom(((EjsString*) str)->value, &len);
break;
}
msg = srejoin(msg, arg, NULL);
}
if (msg) {
mprRawLog(level, "%s", msg);
written += slen(msg);
}
ejsUnblockGC(ejs, paused);
return ejsCreateNumber(ejs, (MprNumber) slen(msg));
}
int ecCompile(EcCompiler *cp, int argc, char **argv)
{
Ejs *ejs;
int rc, saveCompiling, paused;
ejs = cp->ejs;
saveCompiling = ejs->compiling;
ejs->compiling = 1;
paused = ejsBlockGC(ejs);
rc = compileInner(cp, argc, argv);
ejsUnblockGC(ejs, paused);
ejs->compiling = saveCompiling;
return rc;
}
/*
WARNING: the inside interpreter owns the exception object. Must fully extract all fields
*/
static void handleError(Ejs *ejs, EjsWorker *worker, EjsObj *exception, int throwOutside)
{
Ejs *inside;
EjsObj *e;
MprDispatcher *dispatcher;
Message *msg;
assert(!worker->inside);
assert(exception);
assert(ejs == worker->ejs);
ejsBlockGC(ejs);
if ((msg = createMessage()) == 0) {
ejsThrowMemoryError(ejs);
return;
}
msg->worker = worker;
msg->callback = "onerror";
msg->callbackSlot = ES_Worker_onerror;
inside = worker->pair->ejs;
inside->exception = 0;
e = ejsDeserialize(ejs, ejsSerialize(inside, exception, 0));
inside->exception = exception;
/*
Inside interpreter owns the exception object, so must fully extract all exception.
Allocate into the outside worker's interpreter.
*/
if (ejsIsError(inside, exception)) {
msg->message = ejsGetPropertyByName(ejs, e, EN("message"));
msg->stack = ejsGetPropertyByName(ejs, e, EN("stack"));
} else {
msg->message = e;
msg->stack = 0;
}
if (throwOutside) {
if (msg->stack) {
ejsThrowStateError(ejs, "%@\n%@", ejsToString(ejs, msg->message), ejsToString(ejs, msg->stack));
} else {
ejsThrowStateError(ejs, "%@", ejsToString(ejs, msg->message));
}
}
dispatcher = ejs->dispatcher;
mprCreateEvent(dispatcher, "doMessage-error", 0, (MprEventProc) doMessage, msg, 0);
}
static EjsWorker *workerConstructor(Ejs *ejs, EjsWorker *worker, int argc, EjsObj **argv)
{
EjsArray *search;
EjsObj *options, *value;
cchar *name, *scriptFile;
ejsBlockGC(ejs);
scriptFile = (argc >= 1 && argv[0] != ESV(null)) ? ((EjsPath*) argv[0])->value : 0;
options = (argc == 2 && argv[1] != ESV(null)) ? (EjsObj*) argv[1]: NULL;
name = 0;
search = 0;
if (options) {
search = ejsGetPropertyByName(ejs, options, EN("search"));
value = ejsGetPropertyByName(ejs, options, EN("name"));
if (ejsIs(ejs, value, String)) {
name = ejsToMulti(ejs, value);
}
}
worker->ejs = ejs;
worker->state = EJS_WORKER_BEGIN;
return initWorker(ejs, worker, 0, name, search, scriptFile);
}
/*
Process a message sent from postMessage. This may run inside the worker or outside in the parent depending on the
direction of the message. But it ALWAYS runs in the appropriate thread for the interpreter.
*/
static int doMessage(Message *msg, MprEvent *mprEvent)
{
Ejs *ejs;
EjsObj *event, *frame;
EjsWorker *worker;
EjsFunction *callback;
EjsObj *argv[1];
worker = msg->worker;
worker->gotMessage = 1;
ejs = worker->ejs;
assert(!ejs->exception);
event = 0;
ejsBlockGC(ejs);
callback = ejsGetProperty(ejs, worker, msg->callbackSlot);
switch (msg->callbackSlot) {
case ES_Worker_onerror:
event = ejsCreateObj(ejs, ESV(ErrorEvent), 0);
break;
case ES_Worker_onclose:
case ES_Worker_onmessage:
event = ejsCreateObj(ejs, ESV(Event), 0);
break;
default:
assert(msg->callbackSlot == 0);
return 0;
}
worker->event = event;
if (msg->data) {
ejsSetProperty(ejs, event, ES_Event_data, ejsCreateStringFromAsc(ejs, msg->data));
}
if (msg->message) {
ejsSetProperty(ejs, event, ES_ErrorEvent_message, msg->message);
}
if (msg->stack) {
ejsSetProperty(ejs, event, ES_ErrorEvent_stack, msg->stack);
if ((frame = ejsGetProperty(ejs, msg->stack, 0)) != 0 && !ejsIs(ejs, frame, Void)) {
ejsSetProperty(ejs, event, ES_ErrorEvent_filename, ejsGetPropertyByName(ejs, frame, EN("filename")));
ejsSetProperty(ejs, event, ES_ErrorEvent_lineno, ejsGetPropertyByName(ejs, frame, EN("lineno")));
}
}
assert(!ejs->exception);
if (callback == 0 || ejsIs(ejs, callback, Null)) {
if (msg->callbackSlot == ES_Worker_onmessage) {
mprTrace(6, "Discard message as no onmessage handler defined for worker");
} else if (msg->callbackSlot == ES_Worker_onerror) {
if (ejsIs(ejs, msg->message, String)) {
ejsThrowError(ejs, "Exception in Worker: %@", ejsToString(ejs, msg->message));
} else {
ejsThrowError(ejs, "Exception in Worker: %s", ejsGetErrorMsg(worker->pair->ejs, 1));
}
} else {
/* Ignore onclose message */
}
} else if (!ejsIsFunction(ejs, callback)) {
ejsThrowTypeError(ejs, "Worker callback %s is not a function", msg->callback);
} else {
assert(!ejs->exception);
argv[0] = event;
ejsRunFunction(ejs, callback, worker, 1, argv);
}
if (msg->callbackSlot == ES_Worker_onclose) {
assert(!worker->inside);
worker->state = EJS_WORKER_COMPLETE;
mprTrace(5, "Worker.doMessage: complete");
/* Worker and insider interpreter are now eligible for garbage collection */
removeWorker(worker);
}
mprSignalDispatcher(ejs->dispatcher);
worker->event = 0;
return 0;
}
static EjsWorker *initWorker(Ejs *ejs, EjsWorker *worker, Ejs *baseVM, cchar *name, EjsArray *search, cchar *scriptFile)
{
Ejs *wejs;
EjsWorker *self;
EjsName sname;
static int workerSeqno = 0;
ejsBlockGC(ejs);
if (worker == 0) {
worker = ejsCreateWorker(ejs);
}
worker->ejs = ejs;
worker->state = EJS_WORKER_BEGIN;
if (name) {
worker->name = sclone(name);
} else {
lock(ejs);
worker->name = sfmt("worker-%d", workerSeqno++);
unlock(ejs);
}
/*
Create a new interpreter and an "inside" worker object and pair it with the current "outside" worker.
The worker interpreter gets a new dispatcher
*/
if (baseVM) {
if ((wejs = ejsCloneVM(baseVM)) == 0) {
ejsThrowMemoryError(ejs);
return 0;
}
} else {
if ((wejs = ejsCreateVM(0, 0, ejs->flags)) == 0) {
ejsThrowMemoryError(ejs);
return 0;
}
if (ejsLoadModules(wejs, 0, 0) < 0) {
return 0;
}
}
worker->pair = self = ejsCreateWorker(wejs);
self->state = EJS_WORKER_BEGIN;
self->ejs = wejs;
self->inside = 1;
self->pair = worker;
self->name = sjoin("inside-", worker->name, NULL);
#if MOB
mprEnableDispatcher(wejs->dispatcher);
#endif
if (search) {
ejsSetSearchPath(ejs, (EjsArray*) search);
}
// TODO - these should be don't delete
ejsSetProperty(ejs, worker, ES_Worker_name, ejsCreateStringFromAsc(ejs, self->name));
ejsSetProperty(wejs, self, ES_Worker_name, ejsCreateStringFromAsc(wejs, self->name));
sname = ejsName(wejs, EJS_WORKER_NAMESPACE, "self");
ejsSetPropertyByName(wejs, wejs->global, sname, self);
/*
Workers have a dedicated namespace to enable viewing of the worker globals (self, onmessage, postMessage...)
*/
ejsDefineReservedNamespace(wejs, wejs->global, NULL, EJS_WORKER_NAMESPACE);
addWorker(ejs, worker);
if (scriptFile) {
worker->scriptFile = sclone(scriptFile);
worker->state = EJS_WORKER_STARTED;
if (mprCreateEvent(wejs->dispatcher, "workerMain", 0, (MprEventProc) workerMain, self, 0) < 0) {
mprRemoveItem(ejs->workers, worker);
ejsThrowStateError(ejs, "Cannot create worker event");
return 0;
}
}
return worker;
}
static int compileInner(EcCompiler *cp, int argc, char **argv)
{
Ejs *ejs;
EjsModule *mp;
MprList *nodes;
EjsBlock *block;
EcLocation loc;
cchar *ext;
char *msg;
int next, i, j, nextModule, lflags, rc, paused;
ejs = cp->ejs;
if ((nodes = mprCreateList(-1, 0)) == 0) {
return EJS_ERR;
}
cp->nodes = nodes;
/*
Warn about source files mentioned multiple times.
TODO OPT. This is slow.
*/
for (i = 0; i < argc; i++) {
for (j = 0; j < argc; j++) {
if (i == j) {
continue;
}
if (mprSamePath(argv[i], argv[j])) {
compileError(cp, "Loading source %s multiple times. Ignoring extra copies.", argv[i]);
return EJS_ERR;
}
}
if (cp->outputFile && mprSamePath(cp->outputFile, argv[i])) {
compileError(cp, "Output file is the same as input file: %s", argv[i]);
return EJS_ERR;
}
}
/*
Compile source files and load any module files
*/
for (i = 0; i < argc && !cp->fatalError; i++) {
ext = mprGetPathExt(argv[i]);
if (scasecmp(ext, "mod") == 0 || scasecmp(ext, BIT_SHOBJ) == 0) {
nextModule = mprGetListLength(ejs->modules);
lflags = cp->strict ? EJS_LOADER_STRICT : 0;
if ((rc = ejsLoadModule(cp->ejs, ejsCreateStringFromAsc(ejs, argv[i]), -1, -1, lflags)) < 0) {
msg = sfmt("Error initializing module %s\n%s", argv[i], ejsGetErrorMsg(cp->ejs, 1));
memset(&loc, 0, sizeof(EcLocation));
loc.filename = sclone(argv[i]);
if (rc == MPR_ERR_CANT_INITIALIZE) {
ecError(cp, "Error", &loc, msg);
} else {
ecError(cp, "Error", &loc, msg);
}
cp->nodes = NULL;
return EJS_ERR;
}
if (cp->merge) {
/*
If merging, we must emit the loaded module into the output. So add to the compiled modules list.
*/
for (next = nextModule; (mp = mprGetNextItem(ejs->modules, &next)) != 0; ) {
if (mprLookupItem(cp->modules, mp) < 0 && mprAddItem(cp->modules, mp) < 0) {
compileError(cp, "Can't add module %s", mp->name);
}
}
}
mprAddItem(nodes, 0);
} else {
mprAssert(!MPR->marking);
paused = ejsBlockGC(ejs);
mprAddItem(nodes, ecParseFile(cp, argv[i]));
ejsUnblockGC(ejs, paused);
}
mprAssert(!MPR->marking);
}
mprAssert(ejs->result == 0 || (MPR_GET_GEN(MPR_GET_MEM(ejs->result)) != MPR->heap->dead));
/*
Allocate the eval frame stack. This is used for property lookups. We have one dummy block at the top always.
*/
block = ejsCreateBlock(ejs, 0);
mprSetName(block, "Compiler");
ejsPushBlock(ejs, block);
/*
Process the internal representation and generate code
*/
paused = ejsBlockGC(ejs);
if (!cp->parseOnly && cp->errorCount == 0) {
ecResetParser(cp);
if (ecAstProcess(cp) < 0) {
ejsPopBlock(ejs);
cp->nodes = NULL;
ejsUnblockGC(ejs, paused);
return EJS_ERR;
}
if (cp->errorCount == 0) {
ecResetParser(cp);
if (ecCodeGen(cp) < 0) {
ejsPopBlock(ejs);
cp->nodes = NULL;
ejsUnblockGC(ejs, paused);
return EJS_ERR;
}
}
}
ejsPopBlock(ejs);
mprAssert(ejs->result == 0 || (MPR_GET_GEN(MPR_GET_MEM(ejs->result)) != MPR->heap->dead));
/*
Add compiled modules to the interpreter
*/
for (next = 0; ((mp = (EjsModule*) mprGetNextItem(cp->modules, &next)) != 0); ) {
ejsAddModule(cp->ejs, mp);
}
cp->nodes = NULL;
ejsUnblockGC(ejs, paused);
if (!paused) {
mprYield(0);
}
mprAssert(ejs->result == 0 || (MPR_GET_GEN(MPR_GET_MEM(ejs->result)) != MPR->heap->dead));
return (cp->errorCount > 0) ? EJS_ERR: 0;
}
