mirror of
https://github.com/tinygo-org/tinygo.git
synced 2026-08-05 11:37:46 +00:00
runtime: map every goroutine to a new OS thread
This is not a scheduler in the runtime, instead every goroutine is mapped to a single OS thread - meaning 1:1 scheduling. While this may not perform well (or at all) for large numbers of threads, it greatly simplifies many things in the runtime. For example, blocking syscalls can be called directly instead of having to use epoll or similar. Also, we don't need to do anything special to call C code - the default stack is all we need.
This commit is contained in:
committed by
Ron Evans
parent
193f91b870
commit
120d17c124
@@ -0,0 +1,104 @@
|
||||
//go:build none
|
||||
|
||||
#define _GNU_SOURCE
|
||||
#include <pthread.h>
|
||||
#include <semaphore.h>
|
||||
#include <signal.h>
|
||||
#include <stdint.h>
|
||||
#include <stdio.h>
|
||||
|
||||
// BDWGC also uses SIGRTMIN+6 on Linux, which seems like a reasonable choice.
|
||||
#ifdef __linux__
|
||||
#define taskPauseSignal (SIGRTMIN + 6)
|
||||
#endif
|
||||
|
||||
// Pointer to the current task.Task structure.
|
||||
// Ideally the entire task.Task structure would be a thread-local variable but
|
||||
// this also works.
|
||||
static __thread void *current_task;
|
||||
|
||||
struct state_pass {
|
||||
void *(*start)(void*);
|
||||
void *args;
|
||||
void *task;
|
||||
uintptr_t *stackTop;
|
||||
sem_t startlock;
|
||||
};
|
||||
|
||||
// Handle the GC pause in Go.
|
||||
void tinygo_task_gc_pause(int sig);
|
||||
|
||||
// Initialize the main thread.
|
||||
void tinygo_task_init(void *mainTask, pthread_t *thread, void *context) {
|
||||
// Make sure the current task pointer is set correctly for the main
|
||||
// goroutine as well.
|
||||
current_task = mainTask;
|
||||
|
||||
// Store the thread ID of the main thread.
|
||||
*thread = pthread_self();
|
||||
|
||||
// Register the "GC pause" signal for the entire process.
|
||||
// Using pthread_kill, we can still send the signal to a specific thread.
|
||||
struct sigaction act = { 0 };
|
||||
act.sa_flags = SA_SIGINFO;
|
||||
act.sa_handler = &tinygo_task_gc_pause;
|
||||
sigaction(taskPauseSignal, &act, NULL);
|
||||
}
|
||||
|
||||
void tinygo_task_exited(void*);
|
||||
|
||||
// Helper to start a goroutine while also storing the 'task' structure.
|
||||
static void* start_wrapper(void *arg) {
|
||||
struct state_pass *state = arg;
|
||||
void *(*start)(void*) = state->start;
|
||||
void *args = state->args;
|
||||
current_task = state->task;
|
||||
|
||||
// Save the current stack pointer in the goroutine state, for the GC.
|
||||
int stackAddr;
|
||||
*(state->stackTop) = (uintptr_t)(&stackAddr);
|
||||
|
||||
// Notify the caller that the thread has successfully started and
|
||||
// initialized.
|
||||
sem_post(&state->startlock);
|
||||
|
||||
// Run the goroutine function.
|
||||
start(args);
|
||||
|
||||
// Notify the Go side this thread will exit.
|
||||
tinygo_task_exited(current_task);
|
||||
|
||||
return NULL;
|
||||
};
|
||||
|
||||
// Start a new goroutine in an OS thread.
|
||||
int tinygo_task_start(uintptr_t fn, void *args, void *task, pthread_t *thread, uintptr_t *stackTop, void *context) {
|
||||
// Sanity check. Should get optimized away.
|
||||
if (sizeof(pthread_t) != sizeof(void*)) {
|
||||
__builtin_trap();
|
||||
}
|
||||
|
||||
struct state_pass state = {
|
||||
.start = (void*)fn,
|
||||
.args = args,
|
||||
.task = task,
|
||||
.stackTop = stackTop,
|
||||
};
|
||||
sem_init(&state.startlock, 0, 0);
|
||||
int result = pthread_create(thread, NULL, &start_wrapper, &state);
|
||||
|
||||
// Wait until the thread has been created and read all state_pass variables.
|
||||
sem_wait(&state.startlock);
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
// Return the current task (for task.Current()).
|
||||
void* tinygo_task_current(void) {
|
||||
return current_task;
|
||||
}
|
||||
|
||||
// Send a signal to cause the task to pause for the GC mark phase.
|
||||
void tinygo_task_send_gc_signal(pthread_t thread) {
|
||||
pthread_kill(thread, taskPauseSignal);
|
||||
}
|
||||
Reference in New Issue
Block a user