ZEND_DECLARE_MODULE_GLOBALS(pq)
*/
-
/* {{{ PHP_INI
*/
/* Remove comments and fill if you need to have entries in php.ini
zend_objects_store_del_ref_by_handle_ex(obj->zv.handle, obj->zv.handlers TSRMLS_CC);
}
-static int apply_unlisten(void *p TSRMLS_DC, int argc, va_list argv, zend_hash_key *key)
-{
- php_pqconn_object_t *obj = va_arg(argv, php_pqconn_object_t *);
- char *quoted_channel = PQescapeIdentifier(obj->intern->conn, key->arKey, key->nKeyLength - 1);
-
- if (quoted_channel) {
- PGresult *res;
- char *cmd;
-
- spprintf(&cmd, 0, "UNLISTEN %s", quoted_channel);
- if ((res = PQexec(obj->intern->conn, cmd))) {
- PHP_PQclear(res);
- }
-
- efree(cmd);
- PQfreemem(quoted_channel);
- }
-
- return ZEND_HASH_APPLY_REMOVE;
-}
-
-static void php_pqconn_notice_ignore(void *p, const PGresult *res)
-{
-}
-
static void php_pqconn_object_free(void *o TSRMLS_DC)
{
php_pqconn_object_t *obj = o;
if (obj->intern) {
- php_pqconn_event_data_t *evdata;
- PGresult *res;
-
- if ((evdata = PQinstanceData(obj->intern->conn, php_pqconn_event))) {
- memset(evdata, 0, sizeof(*evdata));
- efree(evdata);
- PQsetInstanceData(obj->intern->conn, php_pqconn_event, NULL);
- }
- PQsetNoticeReceiver(obj->intern->conn, php_pqconn_notice_ignore, NULL);
-
- /* clean up async results */
- while ((res = PQgetResult(obj->intern->conn))) {
- PHP_PQclear(res);
- }
-
- zend_hash_apply_with_arguments(&obj->intern->listeners TSRMLS_CC, apply_unlisten, 1, obj);
-
php_resource_factory_handle_dtor(&obj->intern->factory, obj->intern->conn TSRMLS_CC);
php_resource_factory_dtor(&obj->intern->factory);
php_pq_callback_dtor(&obj->intern->onevent);
static void php_pqconn_notice_recv(void *p, const PGresult *res)
{
php_pqconn_event_data_t *data = p;
- zval **evhs;
- TSRMLS_DF(data);
- if (SUCCESS == zend_hash_find(&data->obj->intern->eventhandlers, ZEND_STRS("notice"), (void *) &evhs)) {
- zval *args, *connection = NULL;
+ if (data) {
+ zval **evhs;
+ TSRMLS_DF(data);
+
+ if (SUCCESS == zend_hash_find(&data->obj->intern->eventhandlers, ZEND_STRS("notice"), (void *) &evhs)) {
+ zval *args, *connection = NULL;
- MAKE_STD_ZVAL(args);
- array_init(args);
- php_pq_object_to_zval(data->obj, &connection TSRMLS_CC);
- add_next_index_zval(args, connection);
- add_next_index_string(args, PHP_PQresultErrorMessage(res), 1);
- zend_hash_apply_with_argument(Z_ARRVAL_PP(evhs), apply_event, args TSRMLS_CC);
- zval_ptr_dtor(&args);
+ MAKE_STD_ZVAL(args);
+ array_init(args);
+ php_pq_object_to_zval(data->obj, &connection TSRMLS_CC);
+ add_next_index_zval(args, connection);
+ add_next_index_string(args, PHP_PQresultErrorMessage(res), 1);
+ zend_hash_apply_with_argument(Z_ARRVAL_PP(evhs), apply_event, args TSRMLS_CC);
+ zval_ptr_dtor(&args);
+ }
}
}
static void *php_pqconn_resource_factory_ctor(void *data, void *init_arg TSRMLS_DC)
{
php_pqconn_resource_factory_data_t *o = init_arg;
+ PGconn *conn = NULL;;
if (o->flags & PHP_PQCONN_ASYNC) {
- return PQconnectStart(o->dsn);
+ conn = PQconnectStart(o->dsn);
} else {
- return PQconnectdb(o->dsn);
+ conn = PQconnectdb(o->dsn);
+ }
+
+ if (conn) {
+ PQregisterEventProc(conn, php_pqconn_event, "ext-pq", NULL);
}
+
+ return conn;
}
static void php_pqconn_resource_factory_dtor(void *opaque, void *handle TSRMLS_DC)
{
+ php_pqconn_event_data_t *evdata = PQinstanceData(handle, php_pqconn_event);
+
+ /* we don't care for anthing, except free'ing evdata */
+ if (evdata) {
+ PQsetInstanceData(handle, php_pqconn_event, NULL);
+ memset(evdata, 0, sizeof(*evdata));
+ efree(evdata);
+ }
+
PQfinish(handle);
}
php_pqconn_resource_factory_dtor
};
-static void php_pqconn_persistent_resource_factory_dtor(void *opaque, void *handle TSRMLS_DC)
+static void php_pqconn_wakeup(php_persistent_handle_factory_t *f, void **handle TSRMLS_DC)
{
+}
+
+static int apply_unlisten(void *p TSRMLS_DC, int argc, va_list argv, zend_hash_key *key)
+{
+ php_pqconn_object_t *obj = va_arg(argv, php_pqconn_object_t *);
+ char *quoted_channel = PQescapeIdentifier(obj->intern->conn, key->arKey, key->nKeyLength - 1);
+
+ if (quoted_channel) {
+ PGresult *res;
+ char *cmd;
+
+ spprintf(&cmd, 0, "UNLISTEN %s", quoted_channel);
+ if ((res = PQexec(obj->intern->conn, cmd))) {
+ PHP_PQclear(res);
+ }
+
+ efree(cmd);
+ PQfreemem(quoted_channel);
+ }
+
+ return ZEND_HASH_APPLY_REMOVE;
+}
+
+static void php_pqconn_notice_ignore(void *p, const PGresult *res)
+{
+}
+
+static void php_pqconn_retire(php_persistent_handle_factory_t *f, void **handle TSRMLS_DC)
+{
+ php_pqconn_event_data_t *evdata = PQinstanceData(*handle, php_pqconn_event);
PGresult *res;
+ /* go away */
+ PQsetInstanceData(*handle, php_pqconn_event, NULL);
+
+ /* ignore notices */
+ PQsetNoticeReceiver(*handle, php_pqconn_notice_ignore, NULL);
- /* clean up */
- if ((res = PQexec(handle, PQtransactionStatus(handle) == PQTRANS_IDLE ? "RESET ALL" : "ROLLBACK; RESET ALL"))) {
+ /* clean up async results */
+ while ((res = PQgetResult(*handle))) {
PHP_PQclear(res);
}
- /* release to the pool, if the connection is alive */
- if (CONNECTION_OK == PQstatus(handle)) {
- php_persistent_handle_release(opaque, handle TSRMLS_CC);
- } else {
- PQfinish(handle);
+ /* clean up transaction & session */
+ switch (PQtransactionStatus(*handle)) {
+ case PQTRANS_IDLE:
+ res = PQexec(*handle, "RESET ALL");
+ break;
+ default:
+ res = PQexec(*handle, "ROLLBACK; RESET ALL");
+ break;
}
-}
-static php_resource_factory_ops_t php_pqconn_persistent_resource_factory_ops = {
- (php_resource_factory_handle_ctor_t) php_persistent_handle_acquire,
- NULL,
- php_pqconn_persistent_resource_factory_dtor
-};
+ if (res) {
+ PHP_PQclear(res);
+ }
+
+ if (evdata) {
+ /* clean up notify listeners */
+ zend_hash_apply_with_arguments(&evdata->obj->intern->listeners TSRMLS_CC, apply_unlisten, 1, evdata->obj);
+
+ /* release instance data */
+ memset(evdata, 0, sizeof(*evdata));
+ efree(evdata);
+ }
+}
ZEND_BEGIN_ARG_INFO_EX(ai_pqconn_construct, 0, 0, 1)
ZEND_ARG_INFO(0, dsn)
zend_hash_init(&obj->intern->eventhandlers, 0, NULL, ZVAL_PTR_DTOR, 0);
if (flags & PHP_PQCONN_PERSISTENT) {
- php_persistent_handle_factory_t *phf = php_persistent_handle_concede(NULL, ZEND_STRL("pq\\Connection"), dsn_str, dsn_len TSRMLS_CC);
- php_resource_factory_init(&obj->intern->factory, &php_pqconn_persistent_resource_factory_ops, phf, (void (*)(void*)) php_persistent_handle_abandon);
- } else{
+ php_persistent_handle_factory_t *phf = php_persistent_handle_concede(NULL, ZEND_STRL("pq\\Connection"), dsn_str, dsn_len, php_pqconn_wakeup, php_pqconn_retire TSRMLS_CC);
+ php_resource_factory_init(&obj->intern->factory, php_persistent_handle_get_resource_factory_ops(), phf, (void (*)(void*)) php_persistent_handle_abandon);
+ } else {
php_resource_factory_init(&obj->intern->factory, &php_pqconn_resource_factory_ops, NULL, NULL);
}
+
if (flags & PHP_PQCONN_ASYNC) {
obj->intern->poller = (int (*)(PGconn*)) PQconnectPoll;
}
obj->intern->conn = php_resource_factory_handle_ctor(&obj->intern->factory, &rfdata TSRMLS_CC);
- PQregisterEventProc(obj->intern->conn, php_pqconn_event, "ext-pq", NULL);
- /* the connection might be persistent, so reset event_proc instance data */
PQsetInstanceData(obj->intern->conn, php_pqconn_event, evdata);
PQsetNoticeReceiver(obj->intern->conn, php_pqconn_notice_recv, evdata);
zend_declare_class_constant_long(php_pqcopy_class_entry, ZEND_STRL("FROM_STDIN"), PHP_PQCOPY_FROM_STDIN TSRMLS_CC);
zend_declare_class_constant_long(php_pqcopy_class_entry, ZEND_STRL("TO_STDOUT"), PHP_PQCOPY_TO_STDOUT TSRMLS_CC);
- php_persistent_handle_provide(ZEND_STRL("pq\\Connection"), &php_pqconn_resource_factory_ops, NULL, NULL);
+ php_persistent_handle_provide(ZEND_STRL("pq\\Connection"), &php_pqconn_resource_factory_ops, NULL, NULL TSRMLS_CC);
/*
REGISTER_INI_ENTRIES();
static zend_module_dep pq_module_deps[] = {
ZEND_MOD_REQUIRED("raphf")
+ ZEND_MOD_REQUIRED("spl")
ZEND_MOD_END
};
projects / m6w6 / ext-pq / commitdiff
