database/openGauss-server
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add auth for subscription.

Browse Source
This commit is contained in:
TotaJ
2022-01-20 17:57:37 +08:00
parent 23ddd1ca38
commit 5e11ed9133
4 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/common/backend/utils/init/postinit.cpp
View File

@ -1812,8 +1812,8 @@ void PostgresInitializer::InitWAL()
CheckAuthentication();
/* Don't set superuser when connection is from gs_basebackup */
if (u_sess->proc_cxt.clientIsGsBasebackup) {
/* Don't set superuser when connection is from gs_basebackup or subscription */
if (u_sess->proc_cxt.clientIsGsBasebackup || u_sess->proc_cxt.clientIsSubscription) {
InitUser();
} else {
SetSuperUserStandalone();

3
src/gausskernel/process/postmaster/postmaster.cpp
View File

@ -3750,6 +3750,9 @@ int ProcessStartupPacket(Port* port, bool SSLdone)
} else if (strcmp(valptr, "autonomoustransaction") == 0) {
clientIsAutonomousTransaction = true;
ereport(DEBUG5, (errmsg("autonomoustransaction connected")));
} else if (strcmp(valptr, "subscription") == 0) {
u_sess->proc_cxt.clientIsSubscription = true;
ereport(DEBUG5, (errmsg("subscription connected")));
} else {
ereport(DEBUG5, (errmsg("application %s connected", valptr)));
}

4
src/gausskernel/storage/replication/subscription_walreceiver.cpp
View File

@ -28,7 +28,7 @@ bool sub_connect(char *conninfo, XLogRecPtr *startpoint, char *appname, int chan
/*
* We use the expand_dbname parameter to process the connection string (or
* URI), and pass some extra options. The deliberately undocumented
* parameter "replication=true" makes it a replication connection. The
* parameter "replication=database" makes it a replication connection. The
* database name is ignored by the server in replication mode.
*/
keys[i] = "dbname";
@ -36,7 +36,7 @@ bool sub_connect(char *conninfo, XLogRecPtr *startpoint, char *appname, int chan
keys[++i] = "replication";
vals[i] = "database";
keys[++i] = "fallback_application_name";
vals[i] = appname;
vals[i] = "subscription";
keys[++i] = "client_encoding";
vals[i] = GetDatabaseEncodingName();
keys[++i] = NULL;

1
src/include/knl/knl_session.h
View File

@ -1134,6 +1134,7 @@ typedef struct knl_u_proc_context {
bool clientIsGsBasebackup; /* gs_basebackup tool check flag */
bool clientIsGsroach; /* gs_roach tool check flag */
bool clientIsGsRestore; /* gs_restore tool check flag */
bool clientIsSubscription; /* subscription client check flag */
bool IsBinaryUpgrade;
bool IsWLMWhiteList; /* this proc will not be controled by WLM */
bool gsRewindAddCount;