Kaynağa Gözat

add application_name to pgbouncer connectio string

Grant Limberg 6 yıl önce
ebeveyn
işleme
0da2efa633
3 değiştirilmiş dosya ile 10 ekleme ve 2 silme
  1. 4 1
      controller/PostgreSQL.cpp
  2. 1 0
      cycle_controllers.sh
  3. 5 1
      docker/main.sh

+ 4 - 1
controller/PostgreSQL.cpp

@@ -1349,7 +1349,10 @@ PGconn *PostgreSQL::getPgConn(OverrideMode m) {
 		char *connStr = getenv("PGBOUNCER_CONNSTR");
 		if (connStr != NULL) {
 			fprintf(stderr, "PGBouncer Override\n");
-			return PQconnectdb(connStr);
+			std::string conn(connStr);
+			conn += " application_name=controller-"; 
+			conn += _myAddressStr.c_str();
+			return PQconnectdb(conn.c_str());
 		}
 	}
 

+ 1 - 0
cycle_controllers.sh

@@ -5,4 +5,5 @@ CONTROLLERS=`kubectl get pods -o=name | grep controller | sed "s/^.\{4\}//"`
 for c in ${CONTROLLERS[@]}
 do
     kubectl delete pod ${c}
+    sleep 30
 done

+ 5 - 1
docker/main.sh

@@ -12,6 +12,10 @@ if [ -z "$ZT_DB_PORT" ]; then
     echo '*** FAILED: ZT_DB_PORT environment variable not defined'
     exit 1
 fi
+if [ -z "$ZT_DB_NAME" ]; then
+    echo '*** FAILED: ZT_DB_NAME environment variable not defined'
+    exit 1
+fi
 if [ -z "$ZT_DB_USER" ]; then
     echo '*** FAILED: ZT_DB_USER environment variable not defined'
     exit 1
@@ -38,7 +42,7 @@ echo "{
             \"inot\",
             \"nat64\"
         ],
-        \"controllerDbPath\": \"postgres:host=${ZT_DB_HOST} port=${ZT_DB_PORT} dbname=ztc user=${ZT_DB_USER} password=${ZT_DB_PASSWORD} sslmode=require sslcert=${DB_CLIENT_CERT} sslkey=${DB_CLIENT_KEY} sslrootcert=${DB_SERVER_CA}\"
+        \"controllerDbPath\": \"postgres:host=${ZT_DB_HOST} port=${ZT_DB_PORT} dbname=${ZT_DB_NAME} user=${ZT_DB_USER} password=${ZT_DB_PASSWORD} sslmode=prefer sslcert=${DB_CLIENT_CERT} sslkey=${DB_CLIENT_KEY} sslrootcert=${DB_SERVER_CA}\"
     }
 }    
 " > /var/lib/zerotier-one/local.conf