From 9c837f55239b8ae09eb4702db7c06d6d38acbe8b Mon Sep 17 00:00:00 2001
From: Vignesh C <vignesh21@gmail.com>
Date: Fri, 18 Jul 2025 21:05:50 +0530
Subject: [PATCH v6 2/3] Use libpqsrv_notice_receiver for remote server
 connections in dblink

This patch integrates the previously introduced libpqsrv_notice_receiver
as the custom PQsetNoticeReceiver handler for dblink's remote server
connections. It ensures that server notices are logged via ereport(LOG).
---
 contrib/dblink/dblink.c | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/contrib/dblink/dblink.c b/contrib/dblink/dblink.c
index 8a0b112a7ff..de5bed282f3 100644
--- a/contrib/dblink/dblink.c
+++ b/contrib/dblink/dblink.c
@@ -240,6 +240,10 @@ dblink_get_conn(char *conname_or_str,
 					 errmsg("could not establish connection"),
 					 errdetail_internal("%s", msg)));
 		}
+
+		PQsetNoticeReceiver(conn, libpqsrv_notice_receiver,
+							"received message via remote connection");
+
 		dblink_security_check(conn, NULL, connstr);
 		if (PQclientEncoding(conn) != GetDatabaseEncoding())
 			PQsetClientEncoding(conn, GetDatabaseEncodingName());
@@ -338,6 +342,9 @@ dblink_connect(PG_FUNCTION_ARGS)
 				 errdetail_internal("%s", msg)));
 	}
 
+	PQsetNoticeReceiver(conn, libpqsrv_notice_receiver,
+						"received message via remote connection");
+
 	/* check password actually used if not superuser */
 	dblink_security_check(conn, connname, connstr);
 
-- 
2.43.0

