#include "Capability.h"
#include "Stats.h"
#include "RtsFlags.h"
+#include "Storage.h"
#include "Schedule.h"
#include "Hash.h"
+#include "Trace.h"
#if HAVE_SIGNAL_H
#include <signal.h>
Task *all_tasks = NULL;
static Task *task_free_list = NULL; // singly-linked
static nat taskCount;
-#define DEFAULT_MAX_WORKERS 64
-static nat maxWorkers; // we won't create more workers than this
static nat tasksRunning;
static nat workerCount;
taskCount = 0;
workerCount = 0;
tasksRunning = 0;
- maxWorkers = DEFAULT_MAX_WORKERS;
initialized = 1;
#if defined(THREADED_RTS)
newThreadLocalKey(¤tTaskKey);
void
stopTaskManager (void)
{
- IF_DEBUG(scheduler, sched_belch("stopping task manager, %d tasks still running", tasksRunning));
+ debugTrace(DEBUG_sched,
+ "stopping task manager, %d tasks still running",
+ tasksRunning);
+ /* nothing to do */
+}
+
+
+void
+freeTaskManager (void)
+{
+ Task *task, *next;
+
+ debugTrace(DEBUG_sched, "freeing task manager");
+
+ ACQUIRE_LOCK(&sched_mutex);
+ for (task = all_tasks; task != NULL; task = next) {
+ next = task->all_link;
+ if (task->stopped) {
+ // We only free resources if the Task is not in use. A
+ // Task may still be in use if we have a Haskell thread in
+ // a foreign call while we are attempting to shut down the
+ // RTS (see conc059).
+#if defined(THREADED_RTS)
+ closeCondition(&task->cond);
+ closeMutex(&task->lock);
+#endif
+ stgFree(task);
+ }
+ }
+ all_tasks = NULL;
+ task_free_list = NULL;
+#if defined(THREADED_RTS)
+ freeThreadLocalKey(¤tTaskKey);
+#endif
+ RELEASE_LOCK(&sched_mutex);
}
#if defined(THREADED_RTS)
currentUserTime = getThreadCPUTime();
currentElapsedTime = getProcessElapsedTime();
- task->mut_time = 0.0;
- task->mut_etime = 0.0;
- task->gc_time = 0.0;
- task->gc_etime = 0.0;
+ task->mut_time = 0;
+ task->mut_etime = 0;
+ task->gc_time = 0;
+ task->gc_etime = 0;
task->muttimestart = currentUserTime;
task->elapsedtimestart = currentElapsedTime;
#endif
taskEnter(task);
- IF_DEBUG(scheduler,sched_belch("new task (taskCount: %d)", taskCount););
+ debugTrace(DEBUG_sched, "new task (taskCount: %d)", taskCount);
return task;
}
task_free_list = task;
RELEASE_LOCK(&sched_mutex);
- IF_DEBUG(scheduler,sched_belch("task exiting"));
+ debugTrace(DEBUG_sched, "task exiting");
}
#ifdef THREADED_RTS
{
ASSERT_LOCK_HELD(&sched_mutex);
if (!task->stopped) {
- IF_DEBUG(scheduler,sched_belch("discarding task %p", TASK_ID(task)));
+ debugTrace(DEBUG_sched, "discarding task %ld", (long)TASK_ID(task));
task->cap = NULL;
task->tso = NULL;
task->stopped = rtsTrue;
}
void
-taskStop (Task *task)
+taskTimeStamp (Task *task USED_IF_THREADS)
{
#if defined(THREADED_RTS)
- OSThreadId id;
Ticks currentElapsedTime, currentUserTime, elapsedGCTime;
- id = osThreadId();
- ASSERT(task->id == id);
- ASSERT(myTask() == task);
-
currentUserTime = getThreadCPUTime();
currentElapsedTime = getProcessElapsedTime();
task->mut_etime =
currentElapsedTime - task->elapsedtimestart - elapsedGCTime;
- if (task->mut_time < 0.0) { task->mut_time = 0.0; }
- if (task->mut_etime < 0.0) { task->mut_etime = 0.0; }
+ if (task->mut_time < 0) { task->mut_time = 0; }
+ if (task->mut_etime < 0) { task->mut_etime = 0; }
+#endif
+}
+
+void
+workerTaskStop (Task *task)
+{
+#if defined(THREADED_RTS)
+ OSThreadId id;
+ id = osThreadId();
+ ASSERT(task->id == id);
+ ASSERT(myTask() == task);
#endif
+ task->cap = NULL;
+ taskTimeStamp(task);
task->stopped = rtsTrue;
tasksRunning--;
+
+ ACQUIRE_LOCK(&sched_mutex);
+ task->next = task_free_list;
+ task_free_list = task;
+ RELEASE_LOCK(&sched_mutex);
}
void
resetTaskManagerAfterFork (void)
{
-#warning TODO!
+ // TODO!
taskCount = 0;
}
OSThreadId tid;
Task *task;
- if (workerCount >= maxWorkers) {
- barf("too many workers; runaway worker creation?");
- }
workerCount++;
// A worker always gets a fresh Task structure.
r = createOSThread(&tid, (OSThreadProc *)taskStart, task);
if (r != 0) {
- barf("startTask: Can't create new task");
+ sysErrorBelch("failed to create OS thread");
+ stg_exit(EXIT_FAILURE);
}
- IF_DEBUG(scheduler,sched_belch("new worker task (taskCount: %d)", taskCount););
+ debugTrace(DEBUG_sched, "new worker task (taskCount: %d)", taskCount);
task->id = tid;
debugBelch("on capability %d, ", task->cap->no);
}
if (task->tso) {
- debugBelch("bound to thread %d", task->tso->id);
+ debugBelch("bound to thread %lu", (unsigned long)task->tso->id);
} else {
debugBelch("worker");
}