1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
#pragma once
#include "coro.h"
#include "simple.h"
#include "task.h"
namespace NRainCheck {
namespace NPrivate {
void SpawnTaskImpl(TTaskRunnerBase* task);
template <typename TTask, typename ITask, typename TRunner, typename TEnv, typename TParam>
TIntrusivePtr<TRunner> SpawnTaskWithRunner(TEnv* env, TParam param1, ISubtaskListener* subtaskListener) {
static_assert((std::is_base_of<ITask, TTask>::value), "expect (std::is_base_of<ITask, TTask>::value)");
TIntrusivePtr<TRunner> task(new TRunner(env, subtaskListener, new TTask(env, param1)));
NPrivate::SpawnTaskImpl(task.Get());
return task;
}
template <typename TTask, typename ITask, typename TRunner, typename TEnv>
void SpawnSubtaskWithRunner(TEnv* env, TSubtaskCompletion* completion) {
static_assert((std::is_base_of<ITask, TTask>::value), "expect (std::is_base_of<ITask, TTask>::value)");
TTaskRunnerBase* current = TTaskRunnerBase::CurrentTask();
completion->SetRunning(current);
NPrivate::SpawnTaskImpl(new TRunner(env, completion, new TTask(env)));
}
template <typename TTask, typename ITask, typename TRunner, typename TEnv, typename TParam>
void SpawnSubtaskWithRunner(TEnv* env, TSubtaskCompletion* completion, TParam param) {
static_assert((std::is_base_of<ITask, TTask>::value), "expect (std::is_base_of<ITask, TTask>::value)");
TTaskRunnerBase* current = TTaskRunnerBase::CurrentTask();
completion->SetRunning(current);
NPrivate::SpawnTaskImpl(new TRunner(env, completion, new TTask(env, param)));
}
}
// Instantiate and start a task with given parameter.
template <typename TTask, typename TEnv, typename TParam>
TIntrusivePtr<typename TTask::TTaskRunner> SpawnTask(TEnv* env, TParam param1, ISubtaskListener* subtaskListener = &TNopSubtaskListener::Instance) {
return NPrivate::SpawnTaskWithRunner<
TTask, typename TTask::ITask, typename TTask::TTaskRunner, TEnv, TParam>(env, param1, subtaskListener);
}
// Instantiate and start subtask of given task.
template <typename TTask, typename TEnv, typename TParam>
void SpawnSubtask(TEnv* env, TSubtaskCompletion* completion, TParam param) {
return NPrivate::SpawnSubtaskWithRunner<TTask, typename TTask::ITask, typename TTask::TTaskRunner>(env, completion, param);
}
}
|
