sha512sum/cserver
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add cli args support

Browse source
This commit is contained in:
sha512sum 2024-07-02 11:11:32 +00:00
parent 4d917d6375
commit b51cdd0d8a
1 changed files with 36 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
include/cserver/engine/components.hpp
View file

@ -160,22 +160,47 @@ struct ServiceContextForComponent {
}; };
}; };
template <typename Component, typename T>
struct ServiceContextForComponentWithCliArgs : ServiceContextForComponent<Component, T> {
int argc;
const char** argv;
constexpr ServiceContextForComponentWithCliArgs(T& context, int argc, const char** argv) :
ServiceContextForComponent<Component, T>(context),
argc(argc),
argv(argv) {};
};
template <utempl::Tuple DependencyGraph, typename T> template <utempl::Tuple DependencyGraph, typename T>
inline constexpr auto InitComponents(T& ccontext) -> void { inline constexpr auto InitComponents(T& ccontext, int argc, const char** argv) -> void {
static auto& context = ccontext; static auto& context = ccontext;
static auto& taskProcessor = context.taskProcessor; static auto& taskProcessor = context.taskProcessor;
static auto& ioContext = taskProcessor.ioContext; static auto& ioContext = taskProcessor.ioContext;
static constexpr utempl::ConstexprString cliArgs = [] -> decltype(auto) {
if constexpr(!std::is_same_v<decltype(T::kConfig.template Get<"cliArgs">()), void>) {
return T::kConfig.template Get<"cliArgs">();
} else {
return "";
};
}();
static utempl::Tuple inited = TransformDependencyGraphToTupleWithDependenciesToInitedFlagChanges<AsyncConditionVariable, DependencyGraph>(ioContext); static utempl::Tuple inited = TransformDependencyGraphToTupleWithDependenciesToInitedFlagChanges<AsyncConditionVariable, DependencyGraph>(ioContext);
auto work = make_work_guard(ioContext); auto work = make_work_guard(ioContext);
[]<std::size_t... Is>(std::index_sequence<Is...>){ [argc, argv]<std::size_t... Is>(std::index_sequence<Is...>){
(boost::asio::co_spawn(ioContext, []() -> cserver::Task<> { (boost::asio::co_spawn(ioContext, [argc, argv]() -> cserver::Task<> {
auto& dependencies = Get<Is>(inited); auto& dependencies = Get<Is>(inited);
for(auto* flag : dependencies) { for(auto* flag : dependencies) {
co_await flag->AsyncWait(); co_await flag->AsyncWait();
}; };
ServiceContextForComponent<decltype(Get<Is>(decltype(T::kUtils)::kComponentConfigs)), T> currentContext{context}; using Current = decltype(Get<Is>(decltype(T::kUtils)::kComponentConfigs));
Get<Is>(context.storage).emplace(currentContext); if constexpr(Current::kName == cliArgs) {
ServiceContextForComponentWithCliArgs<Current, T> currentContext{context, argc, argv};
Get<Is>(context.storage).emplace(currentContext);
} else {
ServiceContextForComponent<Current, T> currentContext{context};
Get<Is>(context.storage).emplace(currentContext);
};
auto& componentInitFlag = GetInitFlagFor<AsyncConditionVariable, Is>(ioContext); auto& componentInitFlag = GetInitFlagFor<AsyncConditionVariable, Is>(ioContext);
componentInitFlag.NotifyAll(); componentInitFlag.NotifyAll();
if constexpr(requires{Get<Is>(context.storage)->Run();}) { if constexpr(requires{Get<Is>(context.storage)->Run();}) {
@ -251,8 +276,8 @@ struct ServiceContext {
taskProcessor{utempl::Wrapper<utempl::ConstexprString{kBasicTaskProcessorName}>{}, *this} taskProcessor{utempl::Wrapper<utempl::ConstexprString{kBasicTaskProcessorName}>{}, *this}
,storage{} { ,storage{} {
}; };
constexpr auto Run() { constexpr auto Run(int argc, const char** argv) {
impl::InitComponents<DependencyGraph>(*this); impl::InitComponents<DependencyGraph>(*this, argc, argv);
this->taskProcessor.Run(); this->taskProcessor.Run();
}; };
@ -309,6 +334,8 @@ consteval auto Ignore() {};
template <typename Current, ConstexprConfig Config, typename... Ts> template <typename Current, ConstexprConfig Config, typename... Ts>
struct DependencyInfoInjector { struct DependencyInfoInjector {
int argc{};
const char** argv{};
static constexpr ConstexprConfig kConfig = Config; static constexpr ConstexprConfig kConfig = Config;
static constexpr DependenciesUtils<Ts...> kUtils; static constexpr DependenciesUtils<Ts...> kUtils;
static constexpr utempl::ConstexprString kName = Current::kName; static constexpr utempl::ConstexprString kName = Current::kName;
@ -545,9 +572,9 @@ struct ServiceContextBuilder {
return ServiceContext<config, GetIndexDependencyGraph(), ComponentConfigs...>{}; return ServiceContext<config, GetIndexDependencyGraph(), ComponentConfigs...>{};
}; };
static constexpr auto Run() -> void { static constexpr auto Run(int argc = 0, const char** argv = nullptr) -> void {
static auto context = GetServiceContext(); static auto context = GetServiceContext();
context.Run(); context.Run(argc, argv);
context.taskProcessor.ioContext.run(); context.taskProcessor.ioContext.run();
}; };
}; };