Skip to content

Commit c5cfdd4

Browse files
authored
src: rewrite task runner in c++
PR-URL: #52609 Reviewed-By: Matteo Collina <matteo.collina@gmail.com> Reviewed-By: Daniel Lemire <daniel@lemire.me> Reviewed-By: James M Snell <jasnell@gmail.com> Reviewed-By: Stephen Belanger <admin@stephenbelanger.com>
1 parent 905ab3e commit c5cfdd4

15 files changed

+404
-189
lines changed

lib/internal/main/run.js

-74
This file was deleted.

lib/internal/shell.js

-37
This file was deleted.

node.gyp

+2
Original file line numberDiff line numberDiff line change
@@ -136,6 +136,7 @@
136136
'src/node_stat_watcher.cc',
137137
'src/node_symbols.cc',
138138
'src/node_task_queue.cc',
139+
'src/node_task_runner.cc',
139140
'src/node_trace_events.cc',
140141
'src/node_types.cc',
141142
'src/node_url.cc',
@@ -397,6 +398,7 @@
397398
'test/cctest/test_base_object_ptr.cc',
398399
'test/cctest/test_cppgc.cc',
399400
'test/cctest/test_node_postmortem_metadata.cc',
401+
'test/cctest/test_node_task_runner.cc',
400402
'test/cctest/test_environment.cc',
401403
'test/cctest/test_linked_binding.cc',
402404
'test/cctest/test_node_api.cc',

src/node.cc

+21-8
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121

2222
#include "node.h"
2323
#include "node_dotenv.h"
24+
#include "node_task_runner.h"
2425

2526
// ========== local headers ==========
2627

@@ -360,10 +361,6 @@ MaybeLocal<Value> StartExecution(Environment* env, StartExecutionCallback cb) {
360361
return StartExecution(env, "internal/main/watch_mode");
361362
}
362363

363-
if (!env->options()->run.empty()) {
364-
return StartExecution(env, "internal/main/run");
365-
}
366-
367364
if (!first_argv.empty() && first_argv != "-") {
368365
return StartExecution(env, "internal/main/run_main_module");
369366
}
@@ -976,11 +973,11 @@ int InitializeNodeWithArgs(std::vector<std::string>* argv,
976973
InitializeNodeWithArgsInternal(argv, exec_argv, errors, flags));
977974
}
978975

979-
static std::unique_ptr<InitializationResultImpl>
976+
static std::shared_ptr<InitializationResultImpl>
980977
InitializeOncePerProcessInternal(const std::vector<std::string>& args,
981978
ProcessInitializationFlags::Flags flags =
982979
ProcessInitializationFlags::kNoFlags) {
983-
auto result = std::make_unique<InitializationResultImpl>();
980+
auto result = std::make_shared<InitializationResultImpl>();
984981
result->args_ = args;
985982

986983
if (!(flags & ProcessInitializationFlags::kNoParseGlobalDebugVariables)) {
@@ -1010,6 +1007,22 @@ InitializeOncePerProcessInternal(const std::vector<std::string>& args,
10101007
}
10111008
}
10121009

1010+
if (!per_process::cli_options->run.empty()) {
1011+
// TODO(@anonrig): Handle NODE_NO_WARNINGS, NODE_REDIRECT_WARNINGS,
1012+
// --disable-warning and --redirect-warnings.
1013+
if (per_process::cli_options->per_isolate->per_env->warnings) {
1014+
fprintf(stderr,
1015+
"ExperimentalWarning: Task runner is an experimental feature and "
1016+
"might change at any time\n\n");
1017+
}
1018+
1019+
auto positional_args = task_runner::GetPositionalArgs(args);
1020+
result->early_return_ = true;
1021+
task_runner::RunTask(
1022+
result, per_process::cli_options->run, positional_args);
1023+
return result;
1024+
}
1025+
10131026
if (!(flags & ProcessInitializationFlags::kNoPrintHelpOrVersionOutput)) {
10141027
if (per_process::cli_options->print_version) {
10151028
printf("%s\n", NODE_VERSION);
@@ -1169,7 +1182,7 @@ InitializeOncePerProcessInternal(const std::vector<std::string>& args,
11691182
return result;
11701183
}
11711184

1172-
std::unique_ptr<InitializationResult> InitializeOncePerProcess(
1185+
std::shared_ptr<InitializationResult> InitializeOncePerProcess(
11731186
const std::vector<std::string>& args,
11741187
ProcessInitializationFlags::Flags flags) {
11751188
return InitializeOncePerProcessInternal(args, flags);
@@ -1375,7 +1388,7 @@ static ExitCode StartInternal(int argc, char** argv) {
13751388
// Hack around with the argv pointer. Used for process.title = "blah".
13761389
argv = uv_setup_args(argc, argv);
13771390

1378-
std::unique_ptr<InitializationResultImpl> result =
1391+
std::shared_ptr<InitializationResultImpl> result =
13791392
InitializeOncePerProcessInternal(
13801393
std::vector<std::string>(argv, argv + argc));
13811394
for (const std::string& error : result->errors()) {

src/node.h

+2-2
Original file line numberDiff line numberDiff line change
@@ -349,7 +349,7 @@ NODE_DEPRECATED("Use InitializeOncePerProcess() instead",
349349
// including the arguments split into argv/exec_argv, a list of potential
350350
// errors encountered during initialization, and a potential suggested
351351
// exit code.
352-
NODE_EXTERN std::unique_ptr<InitializationResult> InitializeOncePerProcess(
352+
NODE_EXTERN std::shared_ptr<InitializationResult> InitializeOncePerProcess(
353353
const std::vector<std::string>& args,
354354
ProcessInitializationFlags::Flags flags =
355355
ProcessInitializationFlags::kNoFlags);
@@ -358,7 +358,7 @@ NODE_EXTERN std::unique_ptr<InitializationResult> InitializeOncePerProcess(
358358
NODE_EXTERN void TearDownOncePerProcess();
359359
// Convenience overload for specifying multiple flags without having
360360
// to worry about casts.
361-
inline std::unique_ptr<InitializationResult> InitializeOncePerProcess(
361+
inline std::shared_ptr<InitializationResult> InitializeOncePerProcess(
362362
const std::vector<std::string>& args,
363363
std::initializer_list<ProcessInitializationFlags::Flags> list) {
364364
uint64_t flags_accum = ProcessInitializationFlags::kNoFlags;

src/node_modules.cc

-24
Original file line numberDiff line numberDiff line change
@@ -367,28 +367,6 @@ void BindingData::GetNearestParentPackageJSONType(
367367
args.GetReturnValue().Set(Array::New(realm->isolate(), values, 3));
368368
}
369369

370-
void BindingData::GetPackageJSONScripts(
371-
const FunctionCallbackInfo<Value>& args) {
372-
Realm* realm = Realm::GetCurrent(args);
373-
std::string_view path = "package.json";
374-
375-
THROW_IF_INSUFFICIENT_PERMISSIONS(
376-
realm->env(), permission::PermissionScope::kFileSystemRead, path);
377-
378-
auto package_json = GetPackageJSON(realm, path);
379-
if (package_json == nullptr) {
380-
printf("Can't read package.json\n");
381-
return;
382-
} else if (!package_json->scripts.has_value()) {
383-
printf("Can't read package.json \"scripts\" object\n");
384-
return;
385-
}
386-
387-
args.GetReturnValue().Set(
388-
ToV8Value(realm->context(), package_json->scripts.value())
389-
.ToLocalChecked());
390-
}
391-
392370
void BindingData::GetPackageScopeConfig(
393371
const FunctionCallbackInfo<Value>& args) {
394372
CHECK_GE(args.Length(), 1);
@@ -469,7 +447,6 @@ void BindingData::CreatePerIsolateProperties(IsolateData* isolate_data,
469447
"getNearestParentPackageJSON",
470448
GetNearestParentPackageJSON);
471449
SetMethod(isolate, target, "getPackageScopeConfig", GetPackageScopeConfig);
472-
SetMethod(isolate, target, "getPackageJSONScripts", GetPackageJSONScripts);
473450
}
474451

475452
void BindingData::CreatePerContextProperties(Local<Object> target,
@@ -486,7 +463,6 @@ void BindingData::RegisterExternalReferences(
486463
registry->Register(GetNearestParentPackageJSONType);
487464
registry->Register(GetNearestParentPackageJSON);
488465
registry->Register(GetPackageScopeConfig);
489-
registry->Register(GetPackageJSONScripts);
490466
}
491467

492468
} // namespace modules

src/node_options.cc

+5-4
Original file line numberDiff line numberDiff line change
@@ -566,10 +566,7 @@ EnvironmentOptionsParser::EnvironmentOptionsParser() {
566566
"process V8 profiler output generated using --prof",
567567
&EnvironmentOptions::prof_process);
568568
// Options after --prof-process are passed through to the prof processor.
569-
AddAlias("--prof-process", { "--prof-process", "--" });
570-
AddOption("--run",
571-
"Run a script specified in package.json",
572-
&EnvironmentOptions::run);
569+
AddAlias("--prof-process", {"--prof-process", "--"});
573570
#if HAVE_INSPECTOR
574571
AddOption("--cpu-prof",
575572
"Start the V8 CPU profiler on start up, and write the CPU profile "
@@ -1069,6 +1066,10 @@ PerProcessOptionsParser::PerProcessOptionsParser(
10691066
"Generate a blob that can be embedded into the single executable "
10701067
"application",
10711068
&PerProcessOptions::experimental_sea_config);
1069+
1070+
AddOption("--run",
1071+
"Run a script specified in package.json",
1072+
&PerProcessOptions::run);
10721073
}
10731074

10741075
inline std::string RemoveBrackets(const std::string& host) {

src/node_options.h

+1-1
Original file line numberDiff line numberDiff line change
@@ -161,7 +161,6 @@ class EnvironmentOptions : public Options {
161161
bool heap_prof = false;
162162
#endif // HAVE_INSPECTOR
163163
std::string redirect_warnings;
164-
std::string run;
165164
std::string diagnostic_dir;
166165
std::string env_file;
167166
bool has_env_file_string = false;
@@ -280,6 +279,7 @@ class PerProcessOptions : public Options {
280279
bool print_v8_help = false;
281280
bool print_version = false;
282281
std::string experimental_sea_config;
282+
std::string run;
283283

284284
#ifdef NODE_HAVE_I18N_SUPPORT
285285
std::string icu_data_dir;

0 commit comments

Comments
 (0)