From 91e486362258f6223cc351e5f5c7a0c987e1766e Mon Sep 17 00:00:00 2001
From: JayLiu <38887641+luky116@users.noreply.github.com>
Date: Sat, 6 Jul 2024 19:32:18 +0800
Subject: [PATCH] feat: add InitLimit (#346)

* add InitLimit
---
 src/pikiwidb.cc | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/src/pikiwidb.cc b/src/pikiwidb.cc
index 8a17d4b76..5108e8d9d 100644
--- a/src/pikiwidb.cc
+++ b/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();