Initial import

Author: leonardoce

.gitignore

@@ -0,0 +1,2 @@
+*.swp
+http_test

COPYING

@@ -0,0 +1,11 @@
+FROM ubuntu:bionic AS builder
+RUN apt-get -y update && apt-get install -y postgresql-client build-essential clang libpq-dev
+COPY . /app
+WORKDIR /app
+RUN make
+
+FROM ubuntu:bionic
+RUN apt-get -y update && apt-get install -y postgresql-client && rm -rf /var/lib/apt/lists/*
+COPY --from=builder /app/http_test /app/http_test
+CMD ["/app/http_test"]
+

Dockerfile

@@ -0,0 +1,5 @@
+http_test: http_test.cpp
+	clang++ -o http_test -std=c++17 http_test.cpp -I$$(pg_config --includedir) $$(pg_config --ldflags) -lpq -lpthread
+
+fmt:
+	clang-format -i http_test.cpp

Makefile

@@ -0,0 +1,26 @@
+# Simple HTTP stress testing
+
+This small C++ software has only one final scope: to have a simple backend to
+be used to test the speed of PostgreSQL workloads when a new connection is
+established per every request.
+
+
+## Configuration
+
+Environment variables are your friends:
+
+* `DATABASE_URL` is used as a connection string (libpq is being used to
+  communicate with the PostgreSQL server);
+
+* `SQL_QUERY` is the SQL query that will be executed when accessing the `/tx`
+  endpoint.
+
+
+## Other considerations
+
+As previously said, a new connection is established per request, without any
+connection pool. Use pgbouncer if you need something like that.
+
+## Thanks
+
+https://github.com/yhirose/cpp-httplib is a great tool, really.

README.md

@@ -0,0 +1,109 @@
+/**
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ **/
+ 
+#include "httplib.h"
+#include <chrono>
+#include <ctime>
+#include <iostream>
+#include <libpq-fe.h>
+
+void tx(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 = "");
+
+/**
+ * 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");
+
+/**
+ * Everything starts from here
+ */
+int main() {
+  // Initial logging
+  std::cout << "Database URL: " << std::quoted(database_url) << std::endl;
+  std::cout << "SQL query: " << std::quoted(sql_query) << std::endl;
+  std::cout.flush();
+
+  // Starting HTTP server
+  httplib::Server svr;
+  svr.set_logger(logger);
+  svr.Get("/tx", tx);
+  svr.listen("0.0.0.0", 8080);
+
+  return 0;
+}
+
+/**
+ * Given a request/response pair, put a small log message in stdout
+ */
+void logger(const httplib::Request &req, const httplib::Response &res) {
+  using namespace std;
+
+  const auto now = std::chrono::system_clock::now();
+  const auto nowAsTimeT = std::chrono::system_clock::to_time_t(now);
+  const auto nowMs = std::chrono::duration_cast<std::chrono::milliseconds>(
+                         now.time_since_epoch()) %
+                     1000;
+
+  cout << std::put_time(std::localtime(&nowAsTimeT), "%a %b %d %Y %T") << '.'
+       << std::setfill('0') << std::setw(3) << nowMs.count();
+  cout << " " << req.remote_addr << " " << req.path << " - " << res.status
+       << endl;
+  cout.flush();
+}
+
+/**
+ * The test transaction on PostgreSQL
+ */
+void tx(const httplib::Request &req, httplib::Response &res) {
+  PGconn *conn = PQconnectdb(database_url.c_str());
+
+  if (PQstatus(conn) != CONNECTION_OK) {
+    res.set_content("Connection error", "text/plain");
+    res.status = 500;
+    std::cout << PQerrorMessage(conn) << std::endl;
+    std::cout.flush();
+  } else {
+    PGresult *pgres = PQexec(conn, sql_query.c_str());
+    if (PQresultStatus(pgres) != PGRES_COMMAND_OK &&
+        PQresultStatus(pgres) != PGRES_TUPLES_OK) {
+      res.set_content("Query error", "text/plain");
+      res.status = 500;
+      std::cout << PQresultErrorMessage(pgres) << std::endl;
+      std::cout.flush();
+    } else {
+      res.set_content("Ok!", "text/plain");
+    }
+    PQclear(pgres);
+  }
+
+  PQfinish(conn);
+}
+
+/**
+ * Get a configuration parameter or returns the default value
+ */
+std::string get_configuration(const std::string &env_name,
+                              const std::string &default_value) {
+  auto value = std::getenv(env_name.c_str());
+  if (value == NULL) {
+    return default_value;
+  } else {
+    return value;
+  }
+}