Wrap adoptSocket

uNetworking · Aug 28, 2024 · 6fb88f7 · 6fb88f7
1 parent c067109
commit 6fb88f7
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/src/AppWrapper.h b/src/AppWrapper.h
@@ -713,6 +713,19 @@ std::pair<uWS::SocketContextOptions, bool> readOptionsObject(const FunctionCallb
     return {options, true};
 }
 
+template <typename APP>
+void uWS_App_adoptSocket(const FunctionCallbackInfo<Value> &args) {
+    APP *app = (APP *) args.Holder()->GetAlignedPointerFromInternalField(0);
+
+    Isolate *isolate = args.GetIsolate();
+
+    int32_t fd = args[0]->Int32Value(isolate->GetCurrentContext()).ToChecked();
+
+    app->adoptSocket(fd);
+
+    args.GetReturnValue().Set(args.Holder());
+}
+
 template <typename APP>
 void uWS_App_addChildApp(const FunctionCallbackInfo<Value> &args) {
     APP *app = (APP *) args.Holder()->GetAlignedPointerFromInternalField(0);
@@ -971,7 +984,7 @@ void uWS_App(const FunctionCallbackInfo<Value> &args) {
         /* load balancing */
         appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "addChildAppDescriptor", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, uWS_App_addChildApp<APP>, args.Data()));
         appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "getDescriptor", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, uWS_App_getDescriptor<APP>, args.Data()));
-
+        appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "adoptSocket", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, uWS_App_adoptSocket<APP>, args.Data()));
 
         /* ws, listen */
         appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "ws", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, uWS_App_ws<APP>, args.Data()));