feat: update httplib.h, handle new env vars

Signed-off-by: Leonardo Cecchi <[email protected]>

README.md

@@ -15,6 +15,16 @@ Environment variables are your friends:
 * `SQL_QUERY` is the SQL query that will be executed when accessing the `/tx`
   endpoint.
 
+* `KEEP_ALIVE_MAX_COUNT` TCP keepalive configuration
+
+* `KEEP_ALIVE_TIMEOUT` TCP keepalive configuration
+
+* `READ_TIMEOUT` HTTP request handling read timeout
+
+* `WRITE_TIMEOUT` HTTP request handling write timeout
+
+* `IDLE_INTERVAL` HTTP request timeout idle interval
+
 The endpoint `/.well-known/check` will always return "ok".
 
 ## Other considerations

http_test.cpp

@@ -25,12 +25,15 @@ void ok(const httplib::Request &req, httplib::Response &res);
 void logger(const httplib::Request &req, const httplib::Response &res);
 std::string get_configuration(const std::string &env_name,
                               const std::string &default_value = "");
+void set_server_option_from_env(const std::string& env_name, std::function<void(int)>setter);
 
 /**
  * The database URL to be used as a libpq connection string
  */
 const std::string database_url = get_configuration("DATABASE_URL");
 const std::string sql_query = get_configuration("SQL_QUERY", "SELECT 1");
+const std::string keep_alive_max_count = get_configuration("KEEP_ALIVE_MAX_COUNT");
+const std::string keep_alive_timeout = get_configuration("KEEP_ALIVE_TIMEOUT");
 
 /**
  * Everything starts from here
@@ -45,11 +48,35 @@ int main() {
   httplib::Server svr;
   svr.Get("/tx", tx);
   svr.Get("/.well-known/check", ok);
+
+  // Keepalive settings
+  set_server_option_from_env("KEEP_ALIVE_MAX_COUNT", [&svr](int value){ svr.set_keep_alive_max_count(value); });
+  set_server_option_from_env("KEEP_ALIVE_TIMEOUT", [&svr](int value){ svr.set_keep_alive_timeout(value); });
+
+  // Timeout settings
+  set_server_option_from_env("READ_TIMEOUT", [&svr](int value){ svr.set_read_timeout(value); });
+  set_server_option_from_env("WRITE_TIMEOUT", [&svr](int value){ svr.set_write_timeout(value); });
+  set_server_option_from_env("IDLE_INTERVAL", [&svr](int value){ svr.set_idle_interval(value); });
+
   svr.listen("0.0.0.0", 8080);
 
   return 0;
 }
 
+void set_server_option_from_env(const std::string& env_name, std::function<void(int)>setter) {
+  const std::string env_value = get_configuration(env_name);
+
+  if (!env_value.empty()) {
+    try {
+      std::cout << env_name << ": " << std::quoted(env_value) << std::endl;
+      setter(std::stoi(env_value));
+    } catch(std::exception &e) {
+      std::cout << "Error parsing " << env_name << " as an integer" << std::endl;
+    }
+  }
+
+}
+
 /**
  * This is a context class that to be used inside a request handler. It will
  * log the request handling time