mirror of https://github.com/citusdata/citus.git
Add node_conninfo GUC and related logic
To support more flexible (i.e. not at compile-time) specification of libpq connection parameters, this change adds a new GUC, node_conninfo, which must be a space-separated string of key-value pairs suitable for parsing by libpq's connection establishment methods. To avoid rebuilding and parsing these values at connection time, this change also adds a cache in front of the configuration params to permit immediate use of any previously-calculated parameters.pull/2191/head
parent
a0651df574
commit
57b3f253c5
1
Makefile
1
Makefile
|
@ -9,6 +9,7 @@ OBJS = src/backend/distributed/shared_library_init.o \
|
||||||
src/backend/distributed/commands/drop_distributed_table.o \
|
src/backend/distributed/commands/drop_distributed_table.o \
|
||||||
src/backend/distributed/commands/multi_copy.o \
|
src/backend/distributed/commands/multi_copy.o \
|
||||||
src/backend/distributed/commands/transmit.o \
|
src/backend/distributed/commands/transmit.o \
|
||||||
|
src/backend/distributed/connection/connection_configuration.o \
|
||||||
src/backend/distributed/connection/connection_management.o \
|
src/backend/distributed/connection/connection_management.o \
|
||||||
src/backend/distributed/connection/placement_connection.o \
|
src/backend/distributed/connection/placement_connection.o \
|
||||||
src/backend/distributed/connection/remote_commands.o \
|
src/backend/distributed/connection/remote_commands.o \
|
||||||
|
|
|
@ -0,0 +1,336 @@
|
||||||
|
/*-------------------------------------------------------------------------
|
||||||
|
*
|
||||||
|
* connection_configuration.c
|
||||||
|
* Functions for controlling configuration of Citus connections
|
||||||
|
*
|
||||||
|
* Copyright (c) 2018, Citus Data, Inc.
|
||||||
|
*
|
||||||
|
*-------------------------------------------------------------------------
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "postgres.h"
|
||||||
|
|
||||||
|
#include "distributed/connection_management.h"
|
||||||
|
#include "distributed/metadata_cache.h"
|
||||||
|
#include "distributed/worker_manager.h"
|
||||||
|
#include "distributed/task_tracker.h"
|
||||||
|
#include "postmaster/postmaster.h"
|
||||||
|
#include "mb/pg_wchar.h"
|
||||||
|
#include "utils/builtins.h"
|
||||||
|
|
||||||
|
/* stores the string representation of our node connection GUC */
|
||||||
|
char *NodeConninfo = "";
|
||||||
|
|
||||||
|
/* represents a list of libpq parameter settings */
|
||||||
|
typedef struct ConnParamsInfo
|
||||||
|
{
|
||||||
|
char **keywords; /* libpq keywords */
|
||||||
|
char **values; /* desired values for above */
|
||||||
|
Size size; /* current used size of arrays */
|
||||||
|
Size maxSize; /* maximum allocated size of arrays (similar to e.g. StringInfo) */
|
||||||
|
} ConnParamsInfo;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Stores parsed global libpq parameter settings. static because all access
|
||||||
|
* is encapsulated in the other public functions in this file.
|
||||||
|
*/
|
||||||
|
static ConnParamsInfo ConnParams;
|
||||||
|
|
||||||
|
/* helper functions for processing connection info */
|
||||||
|
static Size CalculateMaxSize(void);
|
||||||
|
static int uri_prefix_length(const char *connstr);
|
||||||
|
|
||||||
|
/*
|
||||||
|
* InitConnParms initializes the ConnParams field to point to enough memory to
|
||||||
|
* store settings for every valid libpq value, though these regions are set to
|
||||||
|
* zeros from the outset and the size appropriately also set to zero.
|
||||||
|
*
|
||||||
|
* This function must be called before others in this file, though calling it
|
||||||
|
* after use of the initialized ConnParams structure will result in any
|
||||||
|
* populated parameter settings being lost.
|
||||||
|
*/
|
||||||
|
void
|
||||||
|
InitConnParams()
|
||||||
|
{
|
||||||
|
Size maxSize = CalculateMaxSize();
|
||||||
|
ConnParamsInfo connParams = {
|
||||||
|
.keywords = malloc(maxSize * sizeof(char *)),
|
||||||
|
.values = malloc(maxSize * sizeof(char *)),
|
||||||
|
.size = 0,
|
||||||
|
.maxSize = maxSize
|
||||||
|
};
|
||||||
|
|
||||||
|
memset(connParams.keywords, 0, maxSize * sizeof(char *));
|
||||||
|
memset(connParams.values, 0, maxSize * sizeof(char *));
|
||||||
|
|
||||||
|
ConnParams = connParams;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
* ResetConnParams frees all strings in the keywords and parameters arrays,
|
||||||
|
* sets their elements to null, and resets the ConnParamsSize to zero before
|
||||||
|
* adding back any hardcoded global connection settings (at present, only the
|
||||||
|
* fallback_application_name of 'citus').
|
||||||
|
*/
|
||||||
|
void
|
||||||
|
ResetConnParams()
|
||||||
|
{
|
||||||
|
Index paramIdx = 0;
|
||||||
|
|
||||||
|
for (paramIdx = 0; paramIdx < ConnParams.size; paramIdx++)
|
||||||
|
{
|
||||||
|
free((void *) ConnParams.keywords[paramIdx]);
|
||||||
|
free((void *) ConnParams.values[paramIdx]);
|
||||||
|
|
||||||
|
ConnParams.keywords[paramIdx] = ConnParams.values[paramIdx] = NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
ConnParams.size = 0;
|
||||||
|
|
||||||
|
AddConnParam("fallback_application_name", "citus");
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
* AddConnParam adds a parameter setting to the global libpq settings according
|
||||||
|
* to the provided keyword and value. Under assert-enabled builds, array bounds
|
||||||
|
* checking is performed.
|
||||||
|
*/
|
||||||
|
void
|
||||||
|
AddConnParam(const char *keyword, const char *value)
|
||||||
|
{
|
||||||
|
Assert((ConnParams.size + 1) < ConnParams.maxSize);
|
||||||
|
|
||||||
|
ConnParams.keywords[ConnParams.size] = strdup(keyword);
|
||||||
|
ConnParams.values[ConnParams.size] = strdup(value);
|
||||||
|
ConnParams.size++;
|
||||||
|
|
||||||
|
ConnParams.keywords[ConnParams.size] = ConnParams.values[ConnParams.size] = NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
* CheckConninfo is a building block to help implement check constraints and
|
||||||
|
* other check hooks against libpq-like conninfo strings. In particular, the
|
||||||
|
* provided conninfo must:
|
||||||
|
*
|
||||||
|
* - Not use a uri-prefix such as postgres:// (it must be only keys and values)
|
||||||
|
* - Parse using PQconninfoParse
|
||||||
|
* - Only set keywords contained in the provided whitelist
|
||||||
|
*
|
||||||
|
* This function returns true if all of the above are satisfied, otherwise it
|
||||||
|
* returns false. If the provided errmsg pointer is not NULL, it will be set
|
||||||
|
* to an appropriate message if the check fails.
|
||||||
|
*
|
||||||
|
* The provided whitelist must be sorted in a manner usable by bsearch, though
|
||||||
|
* this is only validated during assert-enabled builds.
|
||||||
|
*/
|
||||||
|
bool
|
||||||
|
CheckConninfo(const char *conninfo, const char **whitelist,
|
||||||
|
Size whitelistLength, char **errorMsg)
|
||||||
|
{
|
||||||
|
PQconninfoOption *optionArray = NULL;
|
||||||
|
PQconninfoOption *option = NULL;
|
||||||
|
Index whitelistIdx PG_USED_FOR_ASSERTS_ONLY = 0;
|
||||||
|
char *errorMsgString = NULL;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* If the user doesn't need a message, just overwrite errmsg with a stack
|
||||||
|
* variable so we can always safely write to it.
|
||||||
|
*/
|
||||||
|
if (errorMsg == NULL)
|
||||||
|
{
|
||||||
|
errorMsg = &errorMsgString;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* sure, it can be null */
|
||||||
|
if (conninfo == NULL)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* the libpq prefix form is more complex than we need; ban it */
|
||||||
|
if (uri_prefix_length(conninfo) != 0)
|
||||||
|
{
|
||||||
|
*errorMsg = "Citus connection info strings must be in "
|
||||||
|
"'k1=v1 k2=v2 [...] kn=vn' format";
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* this should at least parse */
|
||||||
|
optionArray = PQconninfoParse(conninfo, NULL);
|
||||||
|
if (optionArray == NULL)
|
||||||
|
{
|
||||||
|
*errorMsg = "Provided string is not a valid libpq connection info string";
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef USE_ASSERT_CHECKING
|
||||||
|
|
||||||
|
/* verify that the whitelist is in ascending order */
|
||||||
|
for (whitelistIdx = 1; whitelistIdx < whitelistLength; whitelistIdx++)
|
||||||
|
{
|
||||||
|
const char *prev = whitelist[whitelistIdx - 1];
|
||||||
|
const char *curr = whitelist[whitelistIdx];
|
||||||
|
|
||||||
|
AssertArg(strcmp(prev, curr) < 0);
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
for (option = optionArray; option->keyword != NULL; option++)
|
||||||
|
{
|
||||||
|
void *matchingKeyword = NULL;
|
||||||
|
|
||||||
|
if (option->val == NULL || option->val[0] == '\0')
|
||||||
|
{
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
matchingKeyword = bsearch(&option->keyword, whitelist, whitelistLength,
|
||||||
|
sizeof(char *), pg_qsort_strcmp);
|
||||||
|
if (matchingKeyword == NULL)
|
||||||
|
{
|
||||||
|
/* the whitelist lacks this keyword; error out! */
|
||||||
|
StringInfoData msgString;
|
||||||
|
initStringInfo(&msgString);
|
||||||
|
|
||||||
|
appendStringInfo(&msgString, "Prohibited conninfo keyword detected: %s",
|
||||||
|
option->keyword);
|
||||||
|
|
||||||
|
*errorMsg = msgString.data;
|
||||||
|
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
PQconninfoFree(optionArray);
|
||||||
|
|
||||||
|
/* if error message is set we found an invalid keyword */
|
||||||
|
return (*errorMsg == NULL);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
* GetConnParams uses the provided key to determine libpq parameters needed to
|
||||||
|
* establish a connection using that key. The keywords and values are placed in
|
||||||
|
* the like-named out parameters. All parameter strings are allocated in the
|
||||||
|
* context provided by the caller, to save the caller needing to copy strings
|
||||||
|
* into an appropriate context later.
|
||||||
|
*/
|
||||||
|
void
|
||||||
|
GetConnParams(ConnectionHashKey *key, char ***keywords, char ***values,
|
||||||
|
MemoryContext context)
|
||||||
|
{
|
||||||
|
/* make space for the port as a string: sign, 10 digits, NUL */
|
||||||
|
char *nodePortString = MemoryContextAlloc(context, 12 * sizeof(char *));
|
||||||
|
|
||||||
|
/*
|
||||||
|
* This function has three sections:
|
||||||
|
* - Initialize the keywords and values with copies of global parameters
|
||||||
|
* - Append user/host-specific parameters calculated from the given key
|
||||||
|
* - (Enterprise-only) append user/host-specific authentication params
|
||||||
|
*
|
||||||
|
* The global parameters have already been assigned from a GUC, so begin by
|
||||||
|
* calculating the key-specific parameters (basically just the fields of
|
||||||
|
* the key and the active database encoding).
|
||||||
|
*/
|
||||||
|
const char *runtimeKeywords[] = {
|
||||||
|
"host", "port", "dbname", "user", "client_encoding"
|
||||||
|
};
|
||||||
|
const char *runtimeValues[] = {
|
||||||
|
MemoryContextStrdup(context, key->hostname),
|
||||||
|
nodePortString,
|
||||||
|
MemoryContextStrdup(context, key->database),
|
||||||
|
MemoryContextStrdup(context, key->user),
|
||||||
|
GetDatabaseEncodingName()
|
||||||
|
};
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Declare local params for readability; we'll assign to outparams at end.
|
||||||
|
* Need to zero enough space for all possible libpq parameters.
|
||||||
|
*/
|
||||||
|
char **connKeywords = MemoryContextAllocZero(context, ConnParams.maxSize *
|
||||||
|
sizeof(char *));
|
||||||
|
char **connValues = MemoryContextAllocZero(context, ConnParams.maxSize *
|
||||||
|
sizeof(char *));
|
||||||
|
|
||||||
|
/* auth keywords will begin after global and runtime ones are appended */
|
||||||
|
Index authParamsIdx = ConnParams.size + lengthof(runtimeKeywords);
|
||||||
|
|
||||||
|
pg_ltoa(key->port, nodePortString); /* populate node port string with port */
|
||||||
|
|
||||||
|
/* first step: copy global parameters to beginning of array */
|
||||||
|
memcpy(connKeywords, ConnParams.keywords, ConnParams.size * sizeof(char *));
|
||||||
|
memcpy(connValues, ConnParams.values, ConnParams.size * sizeof(char *));
|
||||||
|
|
||||||
|
/* second step: begin at end of global params and copy runtime ones */
|
||||||
|
memcpy(&connKeywords[ConnParams.size], runtimeKeywords, sizeof(runtimeKeywords));
|
||||||
|
memcpy(&connValues[ConnParams.size], runtimeValues, sizeof(runtimeValues));
|
||||||
|
|
||||||
|
/* final step: add terminal NULL, required by libpq */
|
||||||
|
connKeywords[authParamsIdx] = connValues[authParamsIdx] = NULL;
|
||||||
|
|
||||||
|
*keywords = connKeywords;
|
||||||
|
*values = connValues;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
* CalculateMaxSize simply counts the number of elements returned by
|
||||||
|
* PQconnDefaults, including the final NULL. This helps us know how space would
|
||||||
|
* be used if a connection utilizes every known libpq parameter.
|
||||||
|
*/
|
||||||
|
static Size
|
||||||
|
CalculateMaxSize()
|
||||||
|
{
|
||||||
|
PQconninfoOption *defaults = PQconndefaults();
|
||||||
|
PQconninfoOption *option = NULL;
|
||||||
|
Size maxSize = 0;
|
||||||
|
|
||||||
|
for (option = defaults;
|
||||||
|
option->keyword != NULL;
|
||||||
|
option++, maxSize++)
|
||||||
|
{
|
||||||
|
/* do nothing, we're just counting the elements */
|
||||||
|
}
|
||||||
|
|
||||||
|
PQconninfoFree(defaults);
|
||||||
|
|
||||||
|
/* we've counted elements but libpq needs a final NULL, so add one */
|
||||||
|
maxSize++;
|
||||||
|
|
||||||
|
return maxSize;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/* *INDENT-OFF* */
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Checks if connection string starts with either of the valid URI prefix
|
||||||
|
* designators.
|
||||||
|
*
|
||||||
|
* Returns the URI prefix length, 0 if the string doesn't contain a URI prefix.
|
||||||
|
*
|
||||||
|
* This implementation (mostly) taken from libpq/fe-connect.c.
|
||||||
|
*/
|
||||||
|
static int
|
||||||
|
uri_prefix_length(const char *connstr)
|
||||||
|
{
|
||||||
|
const char uri_designator[] = "postgresql://";
|
||||||
|
const char short_uri_designator[] = "postgres://";
|
||||||
|
|
||||||
|
if (strncmp(connstr, uri_designator,
|
||||||
|
sizeof(uri_designator) - 1) == 0)
|
||||||
|
return sizeof(uri_designator) - 1;
|
||||||
|
|
||||||
|
if (strncmp(connstr, short_uri_designator,
|
||||||
|
sizeof(short_uri_designator) - 1) == 0)
|
||||||
|
return sizeof(short_uri_designator) - 1;
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* *INDENT-ON* */
|
|
@ -33,11 +33,10 @@
|
||||||
|
|
||||||
|
|
||||||
int NodeConnectionTimeout = 5000;
|
int NodeConnectionTimeout = 5000;
|
||||||
int CitusSSLMode = CITUS_SSL_MODE_PREFER;
|
|
||||||
HTAB *ConnectionHash = NULL;
|
HTAB *ConnectionHash = NULL;
|
||||||
|
HTAB *ConnParamsHash = NULL;
|
||||||
MemoryContext ConnectionContext = NULL;
|
MemoryContext ConnectionContext = NULL;
|
||||||
|
|
||||||
|
|
||||||
static uint32 ConnectionHashHash(const void *key, Size keysize);
|
static uint32 ConnectionHashHash(const void *key, Size keysize);
|
||||||
static int ConnectionHashCompare(const void *a, const void *b, Size keysize);
|
static int ConnectionHashCompare(const void *a, const void *b, Size keysize);
|
||||||
static MultiConnection * StartConnectionEstablishment(ConnectionHashKey *key);
|
static MultiConnection * StartConnectionEstablishment(ConnectionHashKey *key);
|
||||||
|
@ -55,10 +54,9 @@ static int CitusNoticeLogLevel = DEFAULT_CITUS_NOTICE_LEVEL;
|
||||||
void
|
void
|
||||||
InitializeConnectionManagement(void)
|
InitializeConnectionManagement(void)
|
||||||
{
|
{
|
||||||
HASHCTL info;
|
HASHCTL info, connParamsInfo;
|
||||||
uint32 hashFlags = 0;
|
uint32 hashFlags = 0;
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Create a single context for connection and transaction related memory
|
* Create a single context for connection and transaction related memory
|
||||||
* management. Doing so, instead of allocating in TopMemoryContext, makes
|
* management. Doing so, instead of allocating in TopMemoryContext, makes
|
||||||
|
@ -79,8 +77,14 @@ InitializeConnectionManagement(void)
|
||||||
info.hcxt = ConnectionContext;
|
info.hcxt = ConnectionContext;
|
||||||
hashFlags = (HASH_ELEM | HASH_FUNCTION | HASH_CONTEXT | HASH_COMPARE);
|
hashFlags = (HASH_ELEM | HASH_FUNCTION | HASH_CONTEXT | HASH_COMPARE);
|
||||||
|
|
||||||
|
memcpy(&connParamsInfo, &info, sizeof(HASHCTL));
|
||||||
|
connParamsInfo.entrysize = sizeof(ConnParamsHashEntry);
|
||||||
|
|
||||||
ConnectionHash = hash_create("citus connection cache (host,port,user,database)",
|
ConnectionHash = hash_create("citus connection cache (host,port,user,database)",
|
||||||
64, &info, hashFlags);
|
64, &info, hashFlags);
|
||||||
|
|
||||||
|
ConnParamsHash = hash_create("citus connparams cache (host,port,user,database)",
|
||||||
|
64, &connParamsInfo, hashFlags);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@ -635,30 +639,29 @@ ConnectionHashCompare(const void *a, const void *b, Size keysize)
|
||||||
static MultiConnection *
|
static MultiConnection *
|
||||||
StartConnectionEstablishment(ConnectionHashKey *key)
|
StartConnectionEstablishment(ConnectionHashKey *key)
|
||||||
{
|
{
|
||||||
char nodePortString[12];
|
bool found = false;
|
||||||
const char *clientEncoding = GetDatabaseEncodingName();
|
|
||||||
MultiConnection *connection = NULL;
|
MultiConnection *connection = NULL;
|
||||||
const char *sslmode = CitusSSLModeString();
|
ConnParamsHashEntry *entry = NULL;
|
||||||
|
|
||||||
const char *keywords[] = {
|
/* search our cache for precomputed connection settings */
|
||||||
"host", "port", "dbname", "user", "sslmode",
|
entry = hash_search(ConnParamsHash, key, HASH_ENTER, &found);
|
||||||
"client_encoding", "fallback_application_name",
|
if (!found)
|
||||||
NULL
|
{
|
||||||
};
|
/* if they're not found, compute them from GUC, runtime, etc. */
|
||||||
const char *values[] = {
|
GetConnParams(key, &entry->keywords, &entry->values, ConnectionContext);
|
||||||
key->hostname, nodePortString, key->database, key->user, sslmode,
|
}
|
||||||
clientEncoding, "citus", NULL
|
|
||||||
};
|
|
||||||
|
|
||||||
connection = MemoryContextAllocZero(ConnectionContext, sizeof(MultiConnection));
|
connection = MemoryContextAllocZero(ConnectionContext, sizeof(MultiConnection));
|
||||||
sprintf(nodePortString, "%d", key->port);
|
|
||||||
|
|
||||||
strlcpy(connection->hostname, key->hostname, MAX_NODE_LENGTH);
|
strlcpy(connection->hostname, key->hostname, MAX_NODE_LENGTH);
|
||||||
connection->port = key->port;
|
connection->port = key->port;
|
||||||
strlcpy(connection->database, key->database, NAMEDATALEN);
|
strlcpy(connection->database, key->database, NAMEDATALEN);
|
||||||
strlcpy(connection->user, key->user, NAMEDATALEN);
|
strlcpy(connection->user, key->user, NAMEDATALEN);
|
||||||
|
|
||||||
connection->pgConn = PQconnectStartParams(keywords, values, false);
|
|
||||||
|
connection->pgConn = PQconnectStartParams((const char **) entry->keywords,
|
||||||
|
(const char **) entry->values,
|
||||||
|
false);
|
||||||
connection->connectionStart = GetCurrentTimestamp();
|
connection->connectionStart = GetCurrentTimestamp();
|
||||||
|
|
||||||
/*
|
/*
|
||||||
|
@ -674,52 +677,6 @@ StartConnectionEstablishment(ConnectionHashKey *key)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/*
|
|
||||||
* CitusSSLModeString returns the current value of citus.sslmode.
|
|
||||||
*/
|
|
||||||
char *
|
|
||||||
CitusSSLModeString(void)
|
|
||||||
{
|
|
||||||
switch (CitusSSLMode)
|
|
||||||
{
|
|
||||||
case CITUS_SSL_MODE_DISABLE:
|
|
||||||
{
|
|
||||||
return "disable";
|
|
||||||
}
|
|
||||||
|
|
||||||
case CITUS_SSL_MODE_ALLOW:
|
|
||||||
{
|
|
||||||
return "allow";
|
|
||||||
}
|
|
||||||
|
|
||||||
case CITUS_SSL_MODE_PREFER:
|
|
||||||
{
|
|
||||||
return "prefer";
|
|
||||||
}
|
|
||||||
|
|
||||||
case CITUS_SSL_MODE_REQUIRE:
|
|
||||||
{
|
|
||||||
return "require";
|
|
||||||
}
|
|
||||||
|
|
||||||
case CITUS_SSL_MODE_VERIFY_CA:
|
|
||||||
{
|
|
||||||
return "verify-ca";
|
|
||||||
}
|
|
||||||
|
|
||||||
case CITUS_SSL_MODE_VERIFY_FULL:
|
|
||||||
{
|
|
||||||
return "verify-full";
|
|
||||||
}
|
|
||||||
|
|
||||||
default:
|
|
||||||
{
|
|
||||||
ereport(ERROR, (errmsg("unrecognized value for citus.sslmode")));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Close all remote connections if necessary anymore (i.e. not session
|
* Close all remote connections if necessary anymore (i.e. not session
|
||||||
* lifetime), or if in a failed state.
|
* lifetime), or if in a failed state.
|
||||||
|
|
|
@ -67,6 +67,8 @@ static void RegisterCitusConfigVariables(void);
|
||||||
static bool ErrorIfNotASuitableDeadlockFactor(double *newval, void **extra,
|
static bool ErrorIfNotASuitableDeadlockFactor(double *newval, void **extra,
|
||||||
GucSource source);
|
GucSource source);
|
||||||
static void NormalizeWorkerListPath(void);
|
static void NormalizeWorkerListPath(void);
|
||||||
|
static bool NodeConninfoGucCheckHook(char **newval, void **extra, GucSource source);
|
||||||
|
static void NodeConninfoGucAssignHook(const char *newval, void *extra);
|
||||||
static bool StatisticsCollectionGucCheckHook(bool *newval, void **extra, GucSource
|
static bool StatisticsCollectionGucCheckHook(bool *newval, void **extra, GucSource
|
||||||
source);
|
source);
|
||||||
|
|
||||||
|
@ -74,6 +76,19 @@ static bool StatisticsCollectionGucCheckHook(bool *newval, void **extra, GucSour
|
||||||
static bool ExpireCachedShards = false;
|
static bool ExpireCachedShards = false;
|
||||||
static int LargeTableShardCount = 0;
|
static int LargeTableShardCount = 0;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* SSL modes available for connecting to worker nodes.
|
||||||
|
*/
|
||||||
|
enum CitusSSLMode
|
||||||
|
{
|
||||||
|
CITUS_SSL_MODE_DISABLE = 1 << 0,
|
||||||
|
CITUS_SSL_MODE_ALLOW = 1 << 1,
|
||||||
|
CITUS_SSL_MODE_PREFER = 1 << 2,
|
||||||
|
CITUS_SSL_MODE_REQUIRE = 1 << 3,
|
||||||
|
CITUS_SSL_MODE_VERIFY_CA = 1 << 4,
|
||||||
|
CITUS_SSL_MODE_VERIFY_FULL = 1 << 5
|
||||||
|
};
|
||||||
|
static int CitusSSLMode = CITUS_SSL_MODE_PREFER;
|
||||||
|
|
||||||
/* *INDENT-OFF* */
|
/* *INDENT-OFF* */
|
||||||
/* GUC enum definitions */
|
/* GUC enum definitions */
|
||||||
|
@ -184,6 +199,8 @@ _PG_init(void)
|
||||||
CreateRequiredDirectories();
|
CreateRequiredDirectories();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
InitConnParams();
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Register Citus configuration variables. Do so before intercepting
|
* Register Citus configuration variables. Do so before intercepting
|
||||||
* hooks or calling initialization functions, in case we want to do the
|
* hooks or calling initialization functions, in case we want to do the
|
||||||
|
@ -329,15 +346,13 @@ RegisterCitusConfigVariables(void)
|
||||||
|
|
||||||
DefineCustomEnumVariable(
|
DefineCustomEnumVariable(
|
||||||
"citus.sslmode",
|
"citus.sslmode",
|
||||||
gettext_noop("SSL mode to use for connections to worker nodes."),
|
gettext_noop("This GUC variable has been deprecated."),
|
||||||
gettext_noop("When connecting to a worker node, specify whether the SSL mode"
|
NULL,
|
||||||
"mode for the connection is 'disable', 'allow', 'prefer' "
|
|
||||||
"(the default), 'require', 'verify-ca' or 'verify-full'."),
|
|
||||||
&CitusSSLMode,
|
&CitusSSLMode,
|
||||||
CITUS_SSL_MODE_PREFER,
|
CITUS_SSL_MODE_PREFER,
|
||||||
citus_ssl_mode_options,
|
citus_ssl_mode_options,
|
||||||
PGC_POSTMASTER,
|
PGC_POSTMASTER,
|
||||||
GUC_SUPERUSER_ONLY,
|
GUC_SUPERUSER_ONLY | GUC_NO_SHOW_ALL,
|
||||||
NULL, NULL, NULL);
|
NULL, NULL, NULL);
|
||||||
|
|
||||||
DefineCustomBoolVariable(
|
DefineCustomBoolVariable(
|
||||||
|
@ -926,6 +941,20 @@ RegisterCitusConfigVariables(void)
|
||||||
&StatisticsCollectionGucCheckHook,
|
&StatisticsCollectionGucCheckHook,
|
||||||
NULL, NULL);
|
NULL, NULL);
|
||||||
|
|
||||||
|
DefineCustomStringVariable(
|
||||||
|
"citus.node_conninfo",
|
||||||
|
gettext_noop("Sets parameters used for outbound connections."),
|
||||||
|
NULL,
|
||||||
|
&NodeConninfo,
|
||||||
|
"sslmode=prefer",
|
||||||
|
PGC_POSTMASTER,
|
||||||
|
GUC_SUPERUSER_ONLY,
|
||||||
|
NodeConninfoGucCheckHook,
|
||||||
|
NodeConninfoGucAssignHook,
|
||||||
|
NULL);
|
||||||
|
NormalizeWorkerListPath();
|
||||||
|
|
||||||
|
|
||||||
/* warn about config items in the citus namespace that are not registered above */
|
/* warn about config items in the citus namespace that are not registered above */
|
||||||
EmitWarningsOnPlaceholders("citus");
|
EmitWarningsOnPlaceholders("citus");
|
||||||
}
|
}
|
||||||
|
@ -997,6 +1026,87 @@ NormalizeWorkerListPath(void)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
* NodeConninfoGucCheckHook ensures conninfo settings are in the expected form
|
||||||
|
* and that the keywords of all non-null settings are on a whitelist devised to
|
||||||
|
* keep users from setting options that may result in confusion.
|
||||||
|
*/
|
||||||
|
static bool
|
||||||
|
NodeConninfoGucCheckHook(char **newval, void **extra, GucSource source)
|
||||||
|
{
|
||||||
|
/* this array _must_ be kept in an order usable by bsearch */
|
||||||
|
const char *whitelist[] = {
|
||||||
|
"application_name",
|
||||||
|
"connect_timeout",
|
||||||
|
#if defined(ENABLE_GSS) && defined(ENABLE_SSPI)
|
||||||
|
"gsslib",
|
||||||
|
#endif
|
||||||
|
"keepalives",
|
||||||
|
"keepalives_count",
|
||||||
|
"keepalives_idle",
|
||||||
|
"keepalives_interval",
|
||||||
|
#if defined(ENABLE_GSS) || defined(ENABLE_SSPI)
|
||||||
|
"krbsrvname",
|
||||||
|
#endif
|
||||||
|
"sslcompression",
|
||||||
|
"sslcrl",
|
||||||
|
"sslmode",
|
||||||
|
"sslrootcert"
|
||||||
|
};
|
||||||
|
char *errorMsg = NULL;
|
||||||
|
bool conninfoValid = CheckConninfo(*newval, whitelist, lengthof(whitelist),
|
||||||
|
&errorMsg);
|
||||||
|
|
||||||
|
if (!conninfoValid)
|
||||||
|
{
|
||||||
|
GUC_check_errdetail("%s", errorMsg);
|
||||||
|
}
|
||||||
|
|
||||||
|
return conninfoValid;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
* NodeConninfoGucAssignHook is the assignment hook for the node_conninfo GUC
|
||||||
|
* variable. Though this GUC is a "string", we actually parse it as a non-URI
|
||||||
|
* PQconninfo key/value setting, storing the resultant PQconninfoOption values
|
||||||
|
* using the public functions in connection_configuration.c.
|
||||||
|
*/
|
||||||
|
static void
|
||||||
|
NodeConninfoGucAssignHook(const char *newval, void *extra)
|
||||||
|
{
|
||||||
|
PQconninfoOption *optionArray = NULL;
|
||||||
|
PQconninfoOption *option = NULL;
|
||||||
|
|
||||||
|
if (newval == NULL)
|
||||||
|
{
|
||||||
|
newval = "";
|
||||||
|
}
|
||||||
|
|
||||||
|
optionArray = PQconninfoParse(newval, NULL);
|
||||||
|
if (optionArray == NULL)
|
||||||
|
{
|
||||||
|
ereport(FATAL, (errmsg("cannot parse node_conninfo value"),
|
||||||
|
errdetail("The GUC check hook should prevent "
|
||||||
|
"all malformed values.")));
|
||||||
|
}
|
||||||
|
|
||||||
|
ResetConnParams();
|
||||||
|
|
||||||
|
for (option = optionArray; option->keyword != NULL; option++)
|
||||||
|
{
|
||||||
|
if (option->val == NULL || option->val[0] == '\0')
|
||||||
|
{
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
AddConnParam(option->keyword, option->val);
|
||||||
|
}
|
||||||
|
|
||||||
|
PQconninfoFree(optionArray);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
static bool
|
static bool
|
||||||
StatisticsCollectionGucCheckHook(bool *newval, void **extra, GucSource source)
|
StatisticsCollectionGucCheckHook(bool *newval, void **extra, GucSource source)
|
||||||
{
|
{
|
||||||
|
|
|
@ -9,7 +9,6 @@
|
||||||
|
|
||||||
#include "stdint.h"
|
#include "stdint.h"
|
||||||
#include "postgres.h"
|
#include "postgres.h"
|
||||||
|
|
||||||
#include "miscadmin.h"
|
#include "miscadmin.h"
|
||||||
|
|
||||||
#include "access/genam.h"
|
#include "access/genam.h"
|
||||||
|
|
|
@ -113,26 +113,21 @@ typedef struct ConnectionHashEntry
|
||||||
dlist_head *connections;
|
dlist_head *connections;
|
||||||
} ConnectionHashEntry;
|
} ConnectionHashEntry;
|
||||||
|
|
||||||
/*
|
/* hash entry for cached connection parameters */
|
||||||
* SSL modes available for connecting to worker nodes.
|
typedef struct ConnParamsHashEntry
|
||||||
*/
|
|
||||||
enum CitusSSLMode
|
|
||||||
{
|
{
|
||||||
CITUS_SSL_MODE_DISABLE = 1 << 0,
|
ConnectionHashKey key;
|
||||||
CITUS_SSL_MODE_ALLOW = 1 << 1,
|
char **keywords;
|
||||||
CITUS_SSL_MODE_PREFER = 1 << 2,
|
char **values;
|
||||||
CITUS_SSL_MODE_REQUIRE = 1 << 3,
|
} ConnParamsHashEntry;
|
||||||
CITUS_SSL_MODE_VERIFY_CA = 1 << 4,
|
|
||||||
CITUS_SSL_MODE_VERIFY_FULL = 1 << 5
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
/* SSL mode to use when connecting to worker nodes */
|
|
||||||
extern int CitusSSLMode;
|
|
||||||
|
|
||||||
/* maximum duration to wait for connection */
|
/* maximum duration to wait for connection */
|
||||||
extern int NodeConnectionTimeout;
|
extern int NodeConnectionTimeout;
|
||||||
|
|
||||||
|
/* parameters used for outbound connections */
|
||||||
|
extern char *NodeConninfo;
|
||||||
|
|
||||||
/* the hash table */
|
/* the hash table */
|
||||||
extern HTAB *ConnectionHash;
|
extern HTAB *ConnectionHash;
|
||||||
|
|
||||||
|
@ -143,6 +138,14 @@ extern struct MemoryContextData *ConnectionContext;
|
||||||
extern void AfterXactConnectionHandling(bool isCommit);
|
extern void AfterXactConnectionHandling(bool isCommit);
|
||||||
extern void InitializeConnectionManagement(void);
|
extern void InitializeConnectionManagement(void);
|
||||||
|
|
||||||
|
extern void InitConnParams(void);
|
||||||
|
extern void ResetConnParams(void);
|
||||||
|
extern void AddConnParam(const char *keyword, const char *value);
|
||||||
|
extern void GetConnParams(ConnectionHashKey *key, char ***keywords, char ***values,
|
||||||
|
MemoryContext context);
|
||||||
|
extern bool CheckConninfo(const char *conninfo, const char **whitelist,
|
||||||
|
Size whitelistLength, char **errmsg);
|
||||||
|
|
||||||
|
|
||||||
/* Low-level connection establishment APIs */
|
/* Low-level connection establishment APIs */
|
||||||
extern MultiConnection * GetNodeConnection(uint32 flags, const char *hostname,
|
extern MultiConnection * GetNodeConnection(uint32 flags, const char *hostname,
|
||||||
|
@ -157,7 +160,6 @@ extern MultiConnection * StartNodeUserDatabaseConnection(uint32 flags,
|
||||||
int32 port,
|
int32 port,
|
||||||
const char *user,
|
const char *user,
|
||||||
const char *database);
|
const char *database);
|
||||||
extern char * CitusSSLModeString(void);
|
|
||||||
extern void CloseNodeConnectionsAfterTransaction(char *nodeName, int nodePort);
|
extern void CloseNodeConnectionsAfterTransaction(char *nodeName, int nodePort);
|
||||||
extern void CloseConnection(MultiConnection *connection);
|
extern void CloseConnection(MultiConnection *connection);
|
||||||
extern void ShutdownConnection(MultiConnection *connection);
|
extern void ShutdownConnection(MultiConnection *connection);
|
||||||
|
|
Loading…
Reference in New Issue