Skip to content
Draft
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
112 changes: 110 additions & 2 deletions ext/pgsql/pgsql.c
Original file line number Diff line number Diff line change
Expand Up @@ -194,6 +194,11 @@ static void pgsql_link_free(pgsql_link_handle *link)
FREE_HASHTABLE(link->notices);
link->notices = NULL;
}
if (link->meta_cache) {
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is pretty repetitive. Could be a helper.

zend_hash_destroy(link->meta_cache);
FREE_HASHTABLE(link->meta_cache);
link->meta_cache = NULL;
}
}

static void pgsql_link_free_obj(zend_object *obj)
Expand Down Expand Up @@ -765,6 +770,7 @@ static void php_pgsql_do_connect(INTERNAL_FUNCTION_PARAMETERS, int persistent)
link->conn = pgsql;
link->hash = zend_string_copy(str.s);
link->notices = NULL;
link->meta_cache = NULL;
link->persistent = 1;
} else { /* Non persistent connection */
zval *index_ptr;
Expand Down Expand Up @@ -816,6 +822,7 @@ static void php_pgsql_do_connect(INTERNAL_FUNCTION_PARAMETERS, int persistent)
link->conn = pgsql;
link->hash = zend_string_copy(str.s);
link->notices = NULL;
link->meta_cache = NULL;
link->persistent = 0;

/* add it to the hash */
Expand Down Expand Up @@ -1182,6 +1189,13 @@ PHP_FUNCTION(pg_query)
php_error_docref(NULL, E_NOTICE,"Cannot set connection to blocking mode");
RETURN_FALSE;
}

if (link->meta_cache) {
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Invalidation is incomplete. Wiping on pg_query() only — pg_query_params, pg_execute, pg_prepare, and the pg_send_* family all leave stale entries

zend_hash_destroy(link->meta_cache);
FREE_HASHTABLE(link->meta_cache);
link->meta_cache = NULL;
}

while ((pgsql_result = PQgetResult(pgsql))) {
PQclear(pgsql_result);
leftover = true;
Expand Down Expand Up @@ -1310,6 +1324,13 @@ PHP_FUNCTION(pg_query_params)
php_error_docref(NULL, E_NOTICE,"Cannot set connection to blocking mode");
RETURN_FALSE;
}

if (link->meta_cache) {
zend_hash_destroy(link->meta_cache);
FREE_HASHTABLE(link->meta_cache);
link->meta_cache = NULL;
}

while ((pgsql_result = PQgetResult(pgsql))) {
PQclear(pgsql_result);
leftover = true;
Expand Down Expand Up @@ -1405,6 +1426,11 @@ PHP_FUNCTION(pg_prepare)
php_error_docref(NULL, E_NOTICE,"Cannot set connection to blocking mode");
RETURN_FALSE;
}
if (link->meta_cache) {
zend_hash_destroy(link->meta_cache);
FREE_HASHTABLE(link->meta_cache);
link->meta_cache = NULL;
}
while ((pgsql_result = PQgetResult(pgsql))) {
PQclear(pgsql_result);
leftover = true;
Expand Down Expand Up @@ -1493,6 +1519,11 @@ PHP_FUNCTION(pg_execute)
php_error_docref(NULL, E_NOTICE,"Cannot set connection to blocking mode");
RETURN_FALSE;
}
if (link->meta_cache) {
zend_hash_destroy(link->meta_cache);
FREE_HASHTABLE(link->meta_cache);
link->meta_cache = NULL;
}
while ((pgsql_result = PQgetResult(pgsql))) {
PQclear(pgsql_result);
leftover = true;
Expand Down Expand Up @@ -1699,6 +1730,25 @@ static zend_string *get_field_name(PGconn *pgsql, Oid oid)
return ret;
}

static pgsql_link_handle *pgsql_get_link_from_conn(PGconn *conn)
{
zval *pgsql_link;

if (!conn) {
return NULL;
}

ZEND_HASH_FOREACH_VAL(&PGG(connections), pgsql_link) {
pgsql_link_handle *link = Z_PGSQL_LINK_P(pgsql_link);

if (link && link->conn == conn) {
return link;
}
} ZEND_HASH_FOREACH_END();

return NULL;
}

/* Returns the name of the table field belongs to, or table's oid if oid_only is true */
PHP_FUNCTION(pg_field_table)
{
Expand Down Expand Up @@ -3979,6 +4029,12 @@ PHP_FUNCTION(pg_send_query)
RETURN_FALSE;
}

if (link->meta_cache) {
zend_hash_destroy(link->meta_cache);
FREE_HASHTABLE(link->meta_cache);
link->meta_cache = NULL;
}

if (_php_pgsql_link_has_results(pgsql)) {
php_error_docref(NULL, E_NOTICE,
"There are results on this connection. Call pg_get_result() until it returns FALSE");
Expand Down Expand Up @@ -4053,6 +4109,12 @@ PHP_FUNCTION(pg_send_query_params)
RETURN_FALSE;
}

if (link->meta_cache) {
zend_hash_destroy(link->meta_cache);
FREE_HASHTABLE(link->meta_cache);
link->meta_cache = NULL;
}

if (_php_pgsql_link_has_results(pgsql)) {
php_error_docref(NULL, E_NOTICE,
"There are results on this connection. Call pg_get_result() until it returns FALSE");
Expand Down Expand Up @@ -4133,6 +4195,12 @@ PHP_FUNCTION(pg_send_prepare)
RETURN_FALSE;
}

if (link->meta_cache) {
zend_hash_destroy(link->meta_cache);
FREE_HASHTABLE(link->meta_cache);
link->meta_cache = NULL;
}

if (_php_pgsql_link_has_results(pgsql)) {
php_error_docref(NULL, E_NOTICE,
"There are results on this connection. Call pg_get_result() until it returns FALSE");
Expand Down Expand Up @@ -4208,6 +4276,12 @@ PHP_FUNCTION(pg_send_execute)
RETURN_FALSE;
}

if (link->meta_cache) {
zend_hash_destroy(link->meta_cache);
FREE_HASHTABLE(link->meta_cache);
link->meta_cache = NULL;
}

if (_php_pgsql_link_has_results(pgsql)) {
php_error_docref(NULL, E_NOTICE,
"There are results on this connection. Call pg_get_result() until it returns FALSE");
Expand Down Expand Up @@ -4539,7 +4613,6 @@ PHP_FUNCTION(pg_flush)

/* {{{ php_pgsql_meta_data
* table_name must not be empty
* TODO: Add meta_data cache for better performance
*/
PHP_PGSQL_API zend_result php_pgsql_meta_data(PGconn *pg_link, const zend_string *table_name, zval *meta, bool extended)
{
Expand All @@ -4550,6 +4623,7 @@ PHP_PGSQL_API zend_result php_pgsql_meta_data(PGconn *pg_link, const zend_string
size_t new_len, len;
int i, num_rows, err;
zval elem;
pgsql_link_handle *link;

ZEND_ASSERT(ZSTR_LEN(table_name) != 0);

Expand Down Expand Up @@ -4620,14 +4694,28 @@ PHP_PGSQL_API zend_result php_pgsql_meta_data(PGconn *pg_link, const zend_string
smart_str_0(&querystr);
efree(src);

link = pgsql_get_link_from_conn(pg_link);

if (link && link->meta_cache) {
zval *meta_cache = zend_hash_find(link->meta_cache, querystr.s);

if (meta_cache) {
if (Z_TYPE_P(meta) != IS_UNDEF) {
zval_ptr_dtor(meta);
}
ZVAL_COPY(meta, meta_cache);
smart_str_free(&querystr);
return SUCCESS;
}
}

pg_result = PQexec(pg_link, ZSTR_VAL(querystr.s));
if (PQresultStatus(pg_result) != PGRES_TUPLES_OK || (num_rows = PQntuples(pg_result)) == 0) {
php_error_docref(NULL, E_WARNING, "Table '%s' doesn't exists", ZSTR_VAL(table_name));
smart_str_free(&querystr);
PQclear(pg_result);
return FAILURE;
}
smart_str_free(&querystr);

for (i = 0; i < num_rows; i++) {
char *name;
Expand Down Expand Up @@ -4658,6 +4746,21 @@ PHP_PGSQL_API zend_result php_pgsql_meta_data(PGconn *pg_link, const zend_string
name = PQgetvalue(pg_result,i,0);
add_assoc_zval(meta, name, &elem);
}

if(link) {
if (!link->meta_cache) {
ALLOC_HASHTABLE(link->meta_cache);
zend_hash_init(link->meta_cache, 8, NULL, ZVAL_PTR_DTOR, 0);
}

zval meta_copy;
ZVAL_COPY(&meta_copy, meta);
zend_string *key = zend_string_copy(querystr.s);

zend_hash_update( link->meta_cache, key, &meta_copy);
zend_string_release(key);
}
smart_str_free(&querystr);
PQclear(pg_result);

return SUCCESS;
Expand Down Expand Up @@ -6089,6 +6192,11 @@ PHP_FUNCTION(pg_delete)
if (php_pgsql_delete(pg_link, table, ids, option, &sql) == FAILURE) {
RETURN_FALSE;
}
if (link->meta_cache) {
zend_hash_destroy(link->meta_cache);
FREE_HASHTABLE(link->meta_cache);
link->meta_cache = NULL;
}
if (option & PGSQL_DML_STRING) {
RETURN_STR(sql);
}
Expand Down
1 change: 1 addition & 0 deletions ext/pgsql/php_pgsql.h
Original file line number Diff line number Diff line change
Expand Up @@ -139,6 +139,7 @@ typedef struct pgsql_link_handle {
PGconn *conn;
zend_string *hash;
HashTable *notices;
HashTable *meta_cache;
bool persistent;
zend_object std;
} pgsql_link_handle;
Expand Down
Loading