ext/pgsql: adding postgresql 17 new libpq wrapper call.

pg_set_chunked_rows_mode to allow to fetch results in chunk of max N rows.

Author: devnexen

ext/pgsql/config.m4

@@ -70,6 +70,7 @@ if test "$PHP_PGSQL" != "no"; then
   AC_CHECK_LIB(pq, PQresultMemorySize, AC_DEFINE(HAVE_PG_RESULT_MEMORY_SIZE,1,[PostgreSQL 12 or later]))
   AC_CHECK_LIB(pq, PQchangePassword, AC_DEFINE(HAVE_PG_CHANGE_PASSWORD,1,[PostgreSQL 17 or later]))
   AC_CHECK_LIB(pq, PQsocketPoll, AC_DEFINE(HAVE_PG_SOCKET_POLL,1,[PostgreSQL 17 or later]))
+  AC_CHECK_LIB(pq, PQsetChunkedRowsMode, AC_DEFINE(HAVE_PG_SET_CHUNKED_ROWS_MODE,1,[PostgreSQL 17 or later]))
 
   dnl Available since PostgreSQL 12.
   AC_CACHE_CHECK([if PGVerbosity enum has PQERRORS_SQLSTATE],

ext/pgsql/pgsql.c

@@ -40,9 +40,7 @@
 #include "php_globals.h"
 #include "zend_exceptions.h"
 #include "zend_attributes.h"
-#if !defined(HAVE_PG_SOCKET_POLL)
 #include "php_network.h"
-#endif
 
 #ifdef HAVE_PGSQL
 
@@ -935,16 +933,16 @@ static void php_pgsql_get_link_info(INTERNAL_FUNCTION_PARAMETERS, int entry_type
 			array_init(return_value);
 			res = PQexec(pgsql, "SHOW jit_provider");
 			if (PQresultStatus(res) != PGRES_TUPLES_OK) {
-                            add_assoc_null(return_value, "jit_provider");
+				add_assoc_null(return_value, "jit_provider");
 			} else {
-                            add_assoc_string(return_value, "jit_provider", PQgetvalue(res, 0, 0));
+				add_assoc_string(return_value, "jit_provider", PQgetvalue(res, 0, 0));
 			}
 			PQclear(res);
 			res = PQexec(pgsql, "SHOW jit");
 			if (PQresultStatus(res) != PGRES_TUPLES_OK) {
-                            add_assoc_null(return_value, "jit");
+				add_assoc_null(return_value, "jit");
 			} else {
-                            add_assoc_string(return_value, "jit", PQgetvalue(res, 0, 0));
+				add_assoc_string(return_value, "jit", PQgetvalue(res, 0, 0));
 			}
 			PQclear(res);
 			return;
@@ -4346,7 +4344,7 @@ static int php_pgsql_fd_cast(php_stream *stream, int cast_as, void **ret) /* {{{
 				}
 
 				if (ret) {
-				*(php_socket_t *)ret = fd_number;
+					*(php_socket_t *)ret = fd_number;
 				}
 			}
 				return SUCCESS;
@@ -6248,3 +6246,29 @@ PHP_FUNCTION(pg_socket_poll)
 
 	RETURN_LONG((zend_long)PQsocketPoll(socket, (int)read, (int)write, (int)ts));
 }
+
+#if defined(HAVE_PG_SET_CHUNKED_ROWS_MODE)
+PHP_FUNCTION(pg_set_chunked_rows_mode)
+{
+	zval *pgsql_link;
+	pgsql_link_handle *link;
+	zend_long size;
+
+	ZEND_PARSE_PARAMETERS_START(2, 2)
+		Z_PARAM_OBJECT_OF_CLASS(pgsql_link, pgsql_link_ce)
+		Z_PARAM_LONG(size)
+	ZEND_PARSE_PARAMETERS_END();
+
+	if (size < 1 || size > INT_MAX) {
+		zend_argument_value_error(2, "must be between 1 and %d", INT_MAX);
+		RETURN_THROWS();
+	}
+
+	link = Z_PGSQL_LINK_P(pgsql_link);
+	CHECK_PGSQL_LINK(link);
+
+	/** can still fail if it is not allowed e.g. already fetched results **/
+
+	RETURN_LONG((zend_long)PQsetChunkedRowsMode(link->conn, (int)size));
+}
+#endif

ext/pgsql/pgsql.stub.php

@@ -963,6 +963,10 @@ function pg_put_copy_end(PgSql\Connection $connection, ?string $error = null): i
      * @param resource $socket
      */
     function pg_socket_poll($socket, int $read, int $write, int $timeout = -1): int {}
+
+#ifdef HAVE_PG_SET_CHUNKED_ROWS_MODE
+    function pg_set_chunked_rows_mode(Pgsql\Connection $connection, int $size): int {}
+#endif
 }
 
 namespace PgSql {