e2m/uWebSockets.js
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Make websockets benchmarkable

Browse Source
This commit is contained in:
Alex Hultman 2018-12-26 15:23:14 +01:00
parent 7fdba0bdbd
commit fb10ac9b61
3 changed files with 40 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/HelloWorld.js
View File

@ -11,10 +11,10 @@ const app = uWS./*SSL*/App({
res.end('Hejdå!'); res.end('Hejdå!');
}).ws('/*', { }).ws('/*', {
open: (ws, req) => { open: (ws, req) => {
console.log('hello websocket!'); console.log(ws);
}, },
message: () => { message: (ws, message) => {
console.log('message from websocket yo'); ws.send(message);
} }
}).listen(port, (token) => { }).listen(port, (token) => {
if (token) { if (token) {

44
src/addon.cpp
View File

@ -21,6 +21,7 @@ using namespace v8;
Isolate *isolate; Isolate *isolate;
Persistent<Object> resTemplate; Persistent<Object> resTemplate;
Persistent<Object> reqTemplate; Persistent<Object> reqTemplate;
Persistent<Object> wsTemplate;
#include <iostream> #include <iostream>
#include <vector> #include <vector>
@ -101,8 +102,15 @@ void req_getHeader(const FunctionCallbackInfo<Value> &args) {
} }
/* WebSocket send */ /* WebSocket send */
void uWS_WebSocket_send() { // not properly templated, just like the httpsockets are not!
void uWS_WebSocket_send(const FunctionCallbackInfo<Value> &args) {
uWS::WebSocket<false, true> *ws = (uWS::WebSocket<false, true> *) args.Holder()->GetAlignedPointerFromInternalField(0);
NativeString nativeString(args.GetIsolate(), args[0]);
//std::cout << std::string_view(nativeString.getData(), nativeString.getLength()) << std::endl;
ws->send(std::string_view(nativeString.getData(), nativeString.getLength()), uWS::OpCode::TEXT);
} }
/* uWS.App.ws('/pattern', options) */ /* uWS.App.ws('/pattern', options) */
@ -115,6 +123,10 @@ void uWS_App_ws(const FunctionCallbackInfo<Value> &args) {
Persistent<Function> *openPf = new Persistent<Function>(); Persistent<Function> *openPf = new Persistent<Function>();
Persistent<Function> *messagePf = new Persistent<Function>(); Persistent<Function> *messagePf = new Persistent<Function>();
struct PerSocketData {
Persistent<Object> *socketPf;
};
/* Get the behavior object */ /* Get the behavior object */
if (args.Length() == 2) { if (args.Length() == 2) {
Local<Object> behaviorObject = Local<Object>::Cast(args[1]); Local<Object> behaviorObject = Local<Object>::Cast(args[1]);
@ -125,24 +137,31 @@ void uWS_App_ws(const FunctionCallbackInfo<Value> &args) {
messagePf->Reset(args.GetIsolate(), Local<Function>::Cast(behaviorObject->Get(String::NewFromUtf8(isolate, "message")))); messagePf->Reset(args.GetIsolate(), Local<Function>::Cast(behaviorObject->Get(String::NewFromUtf8(isolate, "message"))));
} }
app->template ws<void>(std::string(nativeString.getData(), nativeString.getLength()), { app->template ws<PerSocketData>(std::string(nativeString.getData(), nativeString.getLength()), {
/*.compression = uWS::SHARED_COMPRESSOR, /*.compression = uWS::SHARED_COMPRESSOR,
.maxPayloadLength = 16 * 1024,*/ .maxPayloadLength = 16 * 1024,*/
.open = [openPf](auto *ws, auto *req) { .open = [openPf](auto *ws, auto *req) {
HandleScope hs(isolate); HandleScope hs(isolate);
/* Create a new websocket object */
Local<Object> wsObject = Local<Object>::New(isolate, wsTemplate)->Clone();
wsObject->SetAlignedPointerInInternalField(0, ws);
/* Attach a new V8 object with pointer to us, to us */ /* Attach a new V8 object with pointer to us, to us */
PerSocketData *perSocketData = (PerSocketData *) ws->getUserData();
perSocketData->socketPf = new Persistent<Object>;
perSocketData->socketPf->Reset(isolate, wsObject);
//Local<Value> argv[] = {resObject, reqObject}; Local<Value> argv[] = {wsObject};
Local<Function>::New(isolate, *openPf)->Call(isolate->GetCurrentContext()->Global(), 0, nullptr); Local<Function>::New(isolate, *openPf)->Call(isolate->GetCurrentContext()->Global(), 1, argv);
}, },
.message = [messagePf](auto *ws, std::string_view message, uWS::OpCode opCode) { .message = [messagePf](auto *ws, std::string_view message, uWS::OpCode opCode) {
HandleScope hs(isolate); HandleScope hs(isolate);
/* ws's user data holds pointer to persistent object which is the V8 representation we pass here */
Local<Value> argv[] = {}; // ws, message, opCode PerSocketData *perSocketData = (PerSocketData *) ws->getUserData();
Local<Function>::New(isolate, *messagePf)->Call(isolate->GetCurrentContext()->Global(), 1, argv); Local<Value> argv[2] = {Local<Object>::New(isolate, *(perSocketData->socketPf)), ArrayBuffer::New(isolate, (void *) message.data(), message.length())}; // ws, message, opCode
Local<Function>::New(isolate, *messagePf)->Call(isolate->GetCurrentContext()->Global(), 2, argv);
}/* }/*
.drain = []() {}, .drain = []() {},
.ping = []() {}, .ping = []() {},
@ -301,6 +320,15 @@ void Main(Local<Object> exports) {
exports->Set(String::NewFromUtf8(isolate, "SSLApp"), FunctionTemplate::New(isolate, uWS_App<uWS::SSLApp>)->GetFunction()); exports->Set(String::NewFromUtf8(isolate, "SSLApp"), FunctionTemplate::New(isolate, uWS_App<uWS::SSLApp>)->GetFunction());
exports->Set(String::NewFromUtf8(isolate, "nextTick"), FunctionTemplate::New(isolate, nextTick)->GetFunction()); exports->Set(String::NewFromUtf8(isolate, "nextTick"), FunctionTemplate::New(isolate, nextTick)->GetFunction());
/* The template for websockets */
Local<FunctionTemplate> wsTemplateLocal = FunctionTemplate::New(isolate);
wsTemplateLocal->SetClassName(String::NewFromUtf8(isolate, "uWS.WebSocket"));
wsTemplateLocal->InstanceTemplate()->SetInternalFieldCount(1);
wsTemplateLocal->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "send"), FunctionTemplate::New(isolate, uWS_WebSocket_send));
Local<Object> wsObjectLocal = wsTemplateLocal->GetFunction()->NewInstance(isolate->GetCurrentContext()).ToLocalChecked();
wsTemplate.Reset(isolate, wsObjectLocal);
// HttpResponse template (not templated) // HttpResponse template (not templated)
Local<FunctionTemplate> resTemplateLocal = FunctionTemplate::New(isolate); Local<FunctionTemplate> resTemplateLocal = FunctionTemplate::New(isolate);
resTemplateLocal->SetClassName(String::NewFromUtf8(isolate, "uWS.HttpResponse")); resTemplateLocal->SetClassName(String::NewFromUtf8(isolate, "uWS.HttpResponse"));

2
uWebSockets

@ -1 +1 @@
Subproject commit fb95abad7180a22f8f2a07c627f9bd3daf97c7a5 Subproject commit c0af883bf9517670d7b729b23baeb5346766ed45