feat: add InitLimit

src/pikiwidb.cc

@@ -11,6 +11,7 @@
 #include "pikiwidb.h"
 
 #include <sys/fcntl.h>
+#include <sys/resource.h>
 #include <sys/wait.h>
 #include <unistd.h>
 #include <iostream>
@@ -216,6 +217,29 @@ static void InitLogs() {
 #endif
 }
 
+static int InitLimit() {
+  rlimit limit;
+  rlim_t maxfiles = g_config.max_clients;
+  if (getrlimit(RLIMIT_NOFILE, &limit) == -1) {
+    WARN("getrlimit error: {}", strerror(errno));
+  } else if (limit.rlim_cur < maxfiles) {
+    rlim_t old_limit = limit.rlim_cur;
+    limit.rlim_cur = maxfiles;
+    limit.rlim_max = maxfiles;
+    if (setrlimit(RLIMIT_NOFILE, &limit) != -1) {
+      WARN("your 'limit -n ' of {} is not enough for PikiwiDB to start. PikiwiDB have successfully reconfig it to ",
+           old_limit, limit.rlim_cur);
+    } else {
+      ERROR(
+          "your 'limit -n ' of {} is not enough for PikiwiDB to start."
+          " PikiwiDB can not reconfig it({}), do it by yourself",
+          old_limit, strerror(errno));
+      return -1;
+    }
+  }
+  return 0;
+}
+
 static void daemonize() {
   if (fork()) {
     exit(0); /* parent exits */
@@ -258,6 +282,7 @@ int main(int ac, char* av[]) {
     daemonize();
   }
 
+  InitLimit();
   pstd::InitRandom();
   SignalSetup();
   InitLogs();