46 lines
1.3 KiB
C++
46 lines
1.3 KiB
C++
#include <iostream>
|
|
#include <fstream>
|
|
|
|
#include <argparse.hpp>
|
|
|
|
#include <daggy/Server.hpp>
|
|
|
|
// Add executors here
|
|
#include <daggy/executors/task/ForkingTaskExecutor.hpp>
|
|
|
|
// Add loggers here
|
|
#include <daggy/loggers/dag_run/OStreamLogger.hpp>
|
|
#include <daggy/loggers/dag_run/FileSystemLogger.hpp>
|
|
|
|
struct Options {
|
|
std::string listenIP = "localhost";
|
|
fs::path logFile = "daggy.log";
|
|
uint16_t listenPort = 2503;
|
|
size_t webThreads = 50;
|
|
size_t dagThreads = 20;
|
|
size_t taskThreads = 20;
|
|
};
|
|
|
|
int main(int argc, char **argv) {
|
|
Options options;
|
|
|
|
argparse::ArgumentParser args("Daggy");
|
|
|
|
args.add_argument("-v", "--verbose");
|
|
args.add_argument("-c", "--config")
|
|
.help("Config file");
|
|
args.add_argument("--ip")
|
|
.help("IP address to listen to");
|
|
args.add_argument("--port")
|
|
.help("Port to listen to")
|
|
.action([](const std::string &value) { return std::stoi(value); });
|
|
|
|
std::ofstream logFile(options.logFile, std::ios::app);
|
|
daggy::loggers::dag_run::OStreamLogger logger(logFile);
|
|
daggy::executors::task::ForkingTaskExecutor executor(options.taskThreads);
|
|
Pistache::Address listenSpec(options.listenIP, options.listenPort);
|
|
|
|
daggy::Server server(listenSpec, logger, executor, options.dagThreads);
|
|
server.init(options.webThreads);
|
|
server.start();
|
|
} |