[#13] Support Unix Domain Socket as a PostgreSQL connection

pgexporter · Oct 27, 2021 · cd898e1 · cd898e1
1 parent c4718d4
commit cd898e1
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/doc/CONFIGURATION.md b/doc/CONFIGURATION.md
@@ -56,6 +56,8 @@ See a [sample](./etc/pgexporter.conf) configuration for running `pgexporter` on
 
 Note, that PostgreSQL 10+ is required.
 
+Note, that if `host` starts with a `/` it represents a path and `pgexporter` will connect using a Unix Domain Socket.
+
 # pgexporter_users configuration
 
 The `pgexporter_users` configuration defines the users known to the system. This file is created and managed through

diff --git a/src/libpgexporter/security.c b/src/libpgexporter/security.c
@@ -1031,7 +1031,19 @@ pgexporter_server_authenticate(int server, char* database, char* username, char*
       memset(&security_messages[i], 0, SECURITY_BUFFER_SIZE);
    }
 
-   ret = pgexporter_connect(config->servers[server].host, config->servers[server].port, &server_fd);
+   if (config->servers[server].host[0] == '/')
+   {
+      char pgsql[MISC_LENGTH];
+
+      memset(&pgsql, 0, sizeof(pgsql));
+      snprintf(&pgsql[0], sizeof(pgsql), ".s.PGSQL.%d", config->servers[server].port);
+      ret = pgexporter_connect_unix_socket(config->servers[server].host, &pgsql[0], &server_fd);
+   }
+   else
+   {
+      ret = pgexporter_connect(config->servers[server].host, config->servers[server].port, &server_fd);
+   }
+
    if (ret != 0)
    {
       goto error;