Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Commit4328db1

Browse files
AddstackSize parameter toWebWorkerTaskExecutor andWebWorkerDedicatedExecutor
This is useful for extending the stack size of the worker threads as wasi-libc's defaultstack size is typically 128KB, which may not be sufficient for some workloads.
1 parentd3b26d3 commit4328db1

File tree

3 files changed

+33
-4
lines changed

3 files changed

+33
-4
lines changed

‎Sources/JavaScriptEventLoop/WebWorkerDedicatedExecutor.swift‎

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,12 +39,19 @@ public final class WebWorkerDedicatedExecutor: SerialExecutor {
3939
privateletunderlying:WebWorkerTaskExecutor
4040

4141
/// - Parameters:
42+
/// - stackSize: The stack size for each worker thread. Default is `nil` (use the platform default stack size).
4243
/// - timeout: The maximum time to wait for all worker threads to be started. Default is 3 seconds.
4344
/// - checkInterval: The interval to check if all worker threads are started. Default is 5 microseconds.
4445
/// - Throws: An error if any worker thread fails to initialize within the timeout period.
45-
publicinit(timeout:Duration=.seconds(3), checkInterval:Duration=.microseconds(5))asyncthrows{
46+
/// - Note: The default stack size of wasi-libc is typically 128KB.
47+
publicinit(
48+
stackSize:Int?=nil,
49+
timeout:Duration=.seconds(3),
50+
checkInterval:Duration=.microseconds(5)
51+
)asyncthrows{
4652
letunderlying=tryawaitWebWorkerTaskExecutor(
4753
numberOfThreads:1,
54+
stackSize: stackSize,
4855
timeout: timeout,
4956
checkInterval: checkInterval
5057
)

‎Sources/JavaScriptEventLoop/WebWorkerTaskExecutor.swift‎

Lines changed: 19 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -347,7 +347,7 @@ public final class WebWorkerTaskExecutor: TaskExecutor {
347347
self.workers= workers
348348
}
349349

350-
func start(timeout:Duration, checkInterval:Duration)asyncthrows{
350+
func start(stackSize:Int?,timeout:Duration, checkInterval:Duration)asyncthrows{
351351
#if canImport(wasi_pthread) && compiler(>=6.1) && _runtime(_multithreaded)
352352
classContext:@uncheckedSendable{
353353
letexecutor:WebWorkerTaskExecutor.Executor
@@ -375,9 +375,21 @@ public final class WebWorkerTaskExecutor: TaskExecutor {
375375
letunmanagedContext=Unmanaged.passRetained(context)
376376
contexts.append(unmanagedContext)
377377
letptr= unmanagedContext.toOpaque()
378+
varattr=pthread_attr_t()
379+
pthread_attr_init(&attr)
380+
// Set the stack size if specified.
381+
iflet stackSize{
382+
letret=pthread_attr_setstacksize(&attr, stackSize)
383+
guard ret==0else{
384+
letstrerror=String(cString:strerror(ret))
385+
throwSpawnError(
386+
reason:"Failed to set stack size (\(stackSize)) for thread (\(ret):\(strerror))"
387+
)
388+
}
389+
}
378390
letret=pthread_create(
379391
nil,
380-
nil,
392+
&attr,
381393
{ ptrin
382394
// Cast to a optional pointer to absorb nullability variations between platforms.
383395
letptr:UnsafeMutableRawPointer?= ptr
@@ -390,6 +402,7 @@ public final class WebWorkerTaskExecutor: TaskExecutor {
390402
},
391403
ptr
392404
)
405+
pthread_attr_destroy(&attr)
393406
guard ret==0else{
394407
letstrerror=String(cString:strerror(ret))
395408
throwSpawnError(reason:"Failed to create a thread (\(ret):\(strerror))")
@@ -467,16 +480,19 @@ public final class WebWorkerTaskExecutor: TaskExecutor {
467480
///
468481
/// - Parameters:
469482
/// - numberOfThreads: The number of Web Worker threads to spawn.
483+
/// - stackSize: The stack size for each worker thread. Default is `nil` (use the platform default stack size).
470484
/// - timeout: The maximum time to wait for all worker threads to be started. Default is 3 seconds.
471485
/// - checkInterval: The interval to check if all worker threads are started. Default is 5 microseconds.
472486
/// - Throws: An error if any worker thread fails to initialize within the timeout period.
487+
/// - Note: The default stack size of wasi-libc is typically 128KB.
473488
publicinit(
474489
numberOfThreads:Int,
490+
stackSize:Int?=nil,
475491
timeout:Duration=.seconds(3),
476492
checkInterval:Duration=.microseconds(5)
477493
)asyncthrows{
478494
self.executor=Executor(numberOfThreads: numberOfThreads)
479-
tryawaitself.executor.start(timeout: timeout, checkInterval: checkInterval)
495+
tryawaitself.executor.start(stackSize: stackSize,timeout: timeout, checkInterval: checkInterval)
480496
}
481497

482498
/// Terminates all worker threads managed by this executor.

‎Tests/JavaScriptEventLoopTests/WebWorkerTaskExecutorTests.swift‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,12 @@ final class WebWorkerTaskExecutorTests: XCTestCase {
122122
executor.terminate()
123123
}
124124

125+
func testThreadStackSize()asyncthrows{
126+
// Sanity check for stackSize parameter
127+
letexecutor=tryawaitWebWorkerTaskExecutor(numberOfThreads:3, stackSize:512*1024)
128+
executor.terminate()
129+
}
130+
125131
func testTaskGroupRunOnDifferentThreads()asyncthrows{
126132
letexecutor=tryawaitWebWorkerTaskExecutor(numberOfThreads:2)
127133

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp