From 2204ad9824a87b43e29b8bf7c3c7791387b1896c Mon Sep 17 00:00:00 2001
From: tikikun <daogiatuank54@gmail.com>
Date: Tue, 16 Jan 2024 20:32:23 +0700
Subject: [PATCH] feat: add uploads folder path

---
 main.cc | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/main.cc b/main.cc
index 050b7b844..595186072 100644
--- a/main.cc
+++ b/main.cc
@@ -21,6 +21,7 @@ int main(int argc, char *argv[]) {
   int thread_num = 1;
   std::string host = "127.0.0.1";
   int port = 3928;
+  std::string uploads_folder_path;
 
   // Number of nitro threads
   if (argc > 1) {
@@ -37,6 +38,11 @@ int main(int argc, char *argv[]) {
     port = std::atoi(argv[3]); // Convert string argument to int
   }
 
+  // Uploads folder path
+  if (argc > 4) {
+    uploads_folder_path = argv[4];
+  }
+
   int logical_cores = std::thread::hardware_concurrency();
   int drogon_thread_num = std::max(thread_num, logical_cores);
   nitro_utils::nitro_logo();
@@ -49,6 +55,10 @@ int main(int argc, char *argv[]) {
   LOG_INFO << "Please load your model";
   drogon::app().addListener(host, port);
   drogon::app().setThreadNum(drogon_thread_num);
+  if (!uploads_folder_path.empty()) {
+    LOG_INFO << "Drogon uploads folder is at: " << uploads_folder_path;
+    drogon::app().setUploadPath(uploads_folder_path);
+  }
   LOG_INFO << "Number of thread is:" << drogon::app().getThreadNum();
 
   drogon::app().run();