2 +--------------------------------------------------------------------+
4 +--------------------------------------------------------------------+
5 | Redistribution and use in source and binary forms, with or without |
6 | modification, are permitted provided that the conditions mentioned |
7 | in the accompanying LICENSE file are met. |
8 +--------------------------------------------------------------------+
9 | Copyright (c) 2013, Michael Wallner <mike@php.net> |
10 +--------------------------------------------------------------------+
19 #include <ext/spl/spl_iterators.h>
20 #if PHP_PQ_HAVE_PHP_JSON_H
21 #include <php_json.h> /* we've added the include directory to INCLUDES */
24 #include <libpq-events.h>
27 #include "php_pq_misc.h"
28 #include "php_pq_object.h"
29 #include "php_pqexc.h"
30 #include "php_pqres.h"
32 #include "php_pq_type.h"
34 zend_class_entry
*php_pqres_class_entry
;
35 static zend_object_handlers php_pqres_object_handlers
;
36 static HashTable php_pqres_object_prophandlers
;
37 static zend_object_iterator_funcs php_pqres_iterator_funcs
;
39 static zend_object_iterator
*php_pqres_iterator_init(zend_class_entry
*ce
, zval
*object
, int by_ref
)
41 php_pqres_iterator_t
*iter
;
42 zval tmp
, *zfetch_type
;
44 iter
= ecalloc(1, sizeof(*iter
));
45 iter
->zi
.funcs
= &php_pqres_iterator_funcs
;
46 iter
->zi
.data
= PHP_PQ_OBJ(object
, NULL
);
47 php_pq_object_addref(iter
->zi
.data
);
49 zfetch_type
= zend_read_property(ce
, object
, ZEND_STRL("fetchType"), 0, &tmp
);
50 iter
->fetch_type
= zval_get_long(zfetch_type
);
52 return (zend_object_iterator
*) iter
;
55 static void php_pqres_iterator_dtor(zend_object_iterator
*i
)
57 php_pqres_iterator_t
*iter
= (php_pqres_iterator_t
*) i
;
58 php_pqres_object_t
*obj
= i
->data
;
60 if (!Z_IS_UNDEF(&iter
->current_val
)) {
61 zval_ptr_dtor(&iter
->current_val
);
63 php_pq_object_delref(obj
);
67 static ZEND_RESULT_CODE
php_pqres_iterator_valid(zend_object_iterator
*i
)
69 php_pqres_iterator_t
*iter
= (php_pqres_iterator_t
*) i
;
70 php_pqres_object_t
*obj
= i
->data
;
72 switch (PQresultStatus(obj
->intern
->res
)) {
74 #ifdef HAVE_PGRES_SINGLE_TUPLE
75 case PGRES_SINGLE_TUPLE
:
77 if (PQntuples(obj
->intern
->res
) <= iter
->index
) {
88 #define PHP_PQRES_JSON_OPTIONS(res) \
89 (php_pqres_fetch_type(res) != PHP_PQRES_FETCH_OBJECT ? PHP_JSON_OBJECT_AS_ARRAY:0)
91 zval
*php_pqres_typed_zval(php_pqres_t
*res
, Oid typ
, zval
*zv
)
97 if ((zconv
= zend_hash_index_find(&res
->converters
, typ
))) {
101 ZVAL_LONG(&ztype
, typ
);
102 zend_call_method_with_2_params(zconv
, NULL
, NULL
, "convertfromstring", &rv
, zv
, &ztype
);
104 ZVAL_ZVAL(&rv
, zv
, 1, 1);
109 str
= zval_get_string(zv
);
113 case PHP_PQ_OID_BOOL
:
114 if (!(res
->auto_convert
& PHP_PQRES_CONV_BOOL
)) {
117 ZVAL_BOOL(zv
, *str
->val
== 't');
120 case PHP_PQ_OID_INT8
:
122 case PHP_PQ_OID_INT4
:
123 case PHP_PQ_OID_INT2
:
126 if (!(res
->auto_convert
& PHP_PQRES_CONV_INT
)) {
133 switch (is_numeric_str_function(str
, &lval
, &dval
)) {
138 ZVAL_DOUBLE(zv
, dval
);
146 case PHP_PQ_OID_FLOAT4
:
147 case PHP_PQ_OID_FLOAT8
:
148 if (!(res
->auto_convert
& PHP_PQRES_CONV_FLOAT
)) {
151 ZVAL_DOUBLE(zv
, zend_strtod(str
->val
, NULL
));
154 case PHP_PQ_OID_DATE
:
155 if (!(res
->auto_convert
& PHP_PQRES_CONV_DATETIME
)) {
158 php_pqdt_from_string(zv
, NULL
, str
->val
, str
->len
, "Y-m-d", NULL
);
161 case PHP_PQ_OID_ABSTIME
:
162 if (!(res
->auto_convert
& PHP_PQRES_CONV_DATETIME
)) {
165 php_pqdt_from_string(zv
, NULL
, str
->val
, str
->len
, "Y-m-d H:i:s", NULL
);
168 case PHP_PQ_OID_TIMESTAMP
:
169 if (!(res
->auto_convert
& PHP_PQRES_CONV_DATETIME
)) {
172 php_pqdt_from_string(zv
, NULL
, str
->val
, str
->len
, "Y-m-d H:i:s.u", NULL
);
175 case PHP_PQ_OID_TIMESTAMPTZ
:
176 if (!(res
->auto_convert
& PHP_PQRES_CONV_DATETIME
)) {
179 php_pqdt_from_string(zv
, NULL
, str
->val
, str
->len
, "Y-m-d H:i:s.uO", NULL
);
182 #if PHP_PQ_HAVE_PHP_JSON_H && defined(PHP_PQ_OID_JSON)
183 # ifdef PHP_PQ_OID_JSONB
184 case PHP_PQ_OID_JSONB
:
186 case PHP_PQ_OID_JSON
:
187 if (!(res
->auto_convert
& PHP_PQRES_CONV_JSON
)) {
190 php_json_decode_ex(zv
, str
->val
, str
->len
, PHP_PQRES_JSON_OPTIONS(res
), 512 /* PHP_JSON_DEFAULT_DEPTH */);
195 if (!(res
->auto_convert
& PHP_PQRES_CONV_ARRAY
)) {
198 if (PHP_PQ_TYPE_IS_ARRAY(typ
) && (ht
= php_pq_parse_array(res
, str
->val
, str
->len
, PHP_PQ_TYPE_OF_ARRAY(typ
)))) {
206 zend_sring_release(str
);
214 static inline zval
*php_pqres_get_col(php_pqres_t
*r
, unsigned row
, unsigned col
, zval
*zv
)
216 if (PQgetisnull(r
->res
, row
, col
)) {
219 ZVAL_STRINGL(zv
, PQgetvalue(r
->res
, row
, col
), PQgetlength(r
->res
, row
, col
));
220 zv
= php_pqres_typed_zval(r
, PQftype(r
->res
, col
), zv
);
226 static inline void php_pqres_add_col_to_zval(php_pqres_t
*r
, unsigned row
, unsigned col
, php_pqres_fetch_t fetch_type
, zval
*data
)
228 if (PQgetisnull(r
->res
, row
, col
)) {
229 switch (fetch_type
) {
230 case PHP_PQRES_FETCH_OBJECT
:
231 add_property_null(data
, PQfname(r
->res
, col
));
234 case PHP_PQRES_FETCH_ASSOC
:
235 add_assoc_null(data
, PQfname(r
->res
, col
));
238 case PHP_PQRES_FETCH_ARRAY
:
239 add_index_null(data
, col
);
245 ZVAL_STRINGL(&zv
, PQgetvalue(r
->res
, row
, col
), PQgetlength(r
->res
, row
, col
));
246 php_pqres_typed_zval(r
, PQftype(r
->res
, col
), &zv
);
248 switch (fetch_type
) {
249 case PHP_PQRES_FETCH_OBJECT
:
250 add_property_zval(data
, PQfname(r
->res
, col
), &zv
);
254 case PHP_PQRES_FETCH_ASSOC
:
255 add_assoc_zval(data
, PQfname(r
->res
, col
), &zv
);
258 case PHP_PQRES_FETCH_ARRAY
:
259 add_index_zval(data
, col
, &zv
);
265 zval
*php_pqres_row_to_zval(PGresult
*res
, unsigned row
, php_pqres_fetch_t fetch_type
, zval
*data
)
268 php_pqres_object_t
*res_obj
= PQresultInstanceData(res
, php_pqconn_event
);
270 if (Z_TYPE_P(data
) != IS_OBJECT
&& Z_TYPE_P(data
) != IS_ARRAY
) {
271 if (PHP_PQRES_FETCH_OBJECT
== fetch_type
) {
278 if (PQntuples(res
) > row
) {
279 for (c
= 0, cols
= PQnfields(res
); c
< cols
; ++c
) {
280 php_pqres_add_col_to_zval(res_obj
->intern
, row
, c
, fetch_type
, data
);
287 static void php_pqres_iterator_current(zend_object_iterator
*i
, zval
***data_ptr TSRMLS_DC
)
289 php_pqres_iterator_t
*iter
= (php_pqres_iterator_t
*) i
;
290 php_pqres_object_t
*obj
= i
->data
;
292 if (!iter
->current_val
) {
293 iter
->current_val
= php_pqres_row_to_zval(obj
->intern
->res
, iter
->index
, iter
->fetch_type
, NULL TSRMLS_CC
);
295 *data_ptr
= &iter
->current_val
;
298 #if PHP_VERSION_ID >= 50500
299 static void php_pqres_iterator_key(zend_object_iterator
*i
, zval
*key TSRMLS_DC
)
301 php_pqres_iterator_t
*iter
= (php_pqres_iterator_t
*) i
;
303 ZVAL_LONG(key
, iter
->index
);
306 static int php_pqres_iterator_key(zend_object_iterator
*i
, char **key_str
, uint
*key_len
, ulong
*key_num TSRMLS_DC
)
308 php_pqres_iterator_t
*iter
= (php_pqres_iterator_t
*) i
;
310 *key_num
= (ulong
) iter
->index
;
312 return HASH_KEY_IS_LONG
;
316 static void php_pqres_iterator_invalidate(zend_object_iterator
*i TSRMLS_DC
)
318 php_pqres_iterator_t
*iter
= (php_pqres_iterator_t
*) i
;
320 if (iter
->current_val
) {
321 zval_ptr_dtor(&iter
->current_val
);
322 iter
->current_val
= NULL
;
326 static void php_pqres_iterator_next(zend_object_iterator
*i TSRMLS_DC
)
328 php_pqres_iterator_t
*iter
= (php_pqres_iterator_t
*) i
;
330 php_pqres_iterator_invalidate(i TSRMLS_CC
);
334 static void php_pqres_iterator_rewind(zend_object_iterator
*i TSRMLS_DC
)
336 php_pqres_iterator_t
*iter
= (php_pqres_iterator_t
*) i
;
338 php_pqres_iterator_invalidate(i TSRMLS_CC
);
342 static zend_object_iterator_funcs php_pqres_iterator_funcs
= {
343 php_pqres_iterator_dtor
,
344 /* check for end of iteration (FAILURE or SUCCESS if data is valid) */
345 php_pqres_iterator_valid
,
346 /* fetch the item data for the current element */
347 php_pqres_iterator_current
,
348 /* fetch the key for the current element (return HASH_KEY_IS_STRING or HASH_KEY_IS_LONG) (optional, may be NULL) */
349 php_pqres_iterator_key
,
350 /* step forwards to next element */
351 php_pqres_iterator_next
,
352 /* rewind to start of data (optional, may be NULL) */
353 php_pqres_iterator_rewind
,
354 /* invalidate current value/key (optional, may be NULL) */
355 php_pqres_iterator_invalidate
358 static int php_pqres_count_elements(zval
*object
, long *count TSRMLS_DC
)
360 php_pqres_object_t
*obj
= zend_object_store_get_object(object TSRMLS_CC
);
365 *count
= (long) PQntuples(obj
->intern
->res
);
370 ZEND_RESULT_CODE
php_pqres_success(PGresult
*res TSRMLS_DC
)
374 switch (PQresultStatus(res
)) {
375 case PGRES_BAD_RESPONSE
:
376 case PGRES_NONFATAL_ERROR
:
377 case PGRES_FATAL_ERROR
:
378 zexc
= throw_exce(EX_SQL TSRMLS_CC
, "%s", PHP_PQresultErrorMessage(res
));
379 zend_update_property_string(Z_OBJCE_P(zexc
), zexc
, ZEND_STRL("sqlstate"), PQresultErrorField(res
, PG_DIAG_SQLSTATE
) TSRMLS_CC
);
386 void php_pqres_init_instance_data(PGresult
*res
, php_pqconn_object_t
*conn_obj
, php_pqres_object_t
**ptr TSRMLS_DC
)
388 php_pqres_object_t
*obj
;
389 php_pqres_t
*r
= ecalloc(1, sizeof(*r
));
392 zend_hash_init(&r
->bound
, 0, 0, ZVAL_PTR_DTOR
, 0);
393 zend_hash_init(&r
->converters
, 0, 0, ZVAL_PTR_DTOR
, 0);
394 zend_hash_copy(&r
->converters
, &conn_obj
->intern
->converters
, (copy_ctor_func_t
) zval_add_ref
, NULL
, sizeof(zval
*));
396 r
->auto_convert
= conn_obj
->intern
->default_auto_convert
;
397 r
->default_fetch_type
= conn_obj
->intern
->default_fetch_type
;
399 php_pqres_create_object_ex(php_pqres_class_entry
, r
, &obj TSRMLS_CC
);
400 PQresultSetInstanceData(res
, php_pqconn_event
, obj
);
407 php_pqres_fetch_t
php_pqres_fetch_type(php_pqres_t
*res
)
409 return res
->iter
? res
->iter
->fetch_type
: res
->default_fetch_type
;
412 static void php_pqres_object_free(void *o TSRMLS_DC
)
414 php_pqres_object_t
*obj
= o
;
416 fprintf(stderr
, "FREE res(#%d) %p\n", obj
->zv
.handle
, obj
);
419 if (obj
->intern
->res
) {
420 PQresultSetInstanceData(obj
->intern
->res
, php_pqconn_event
, NULL
);
421 PQclear(obj
->intern
->res
);
422 obj
->intern
->res
= NULL
;
425 if (obj
->intern
->iter
) {
426 php_pqres_iterator_dtor((zend_object_iterator
*) obj
->intern
->iter TSRMLS_CC
);
427 obj
->intern
->iter
= NULL
;
430 zend_hash_destroy(&obj
->intern
->bound
);
431 zend_hash_destroy(&obj
->intern
->converters
);
436 zend_object_std_dtor((zend_object
*) o TSRMLS_CC
);
440 zend_object_value
php_pqres_create_object_ex(zend_class_entry
*ce
, php_pqres_t
*intern
, php_pqres_object_t
**ptr TSRMLS_DC
)
442 php_pqres_object_t
*o
;
444 o
= ecalloc(1, sizeof(*o
));
445 zend_object_std_init((zend_object
*) o
, ce TSRMLS_CC
);
446 object_properties_init((zend_object
*) o
, ce
);
447 o
->prophandler
= &php_pqres_object_prophandlers
;
457 o
->zv
.handle
= zend_objects_store_put((zend_object
*) o
, NULL
, php_pqres_object_free
, NULL TSRMLS_CC
);
458 o
->zv
.handlers
= &php_pqres_object_handlers
;
463 static zend_object_value
php_pqres_create_object(zend_class_entry
*class_type TSRMLS_DC
)
465 return php_pqres_create_object_ex(class_type
, NULL
, NULL TSRMLS_CC
);
468 static void php_pqres_object_read_status(zval
*object
, void *o
, zval
*return_value TSRMLS_DC
)
470 php_pqres_object_t
*obj
= o
;
472 RETVAL_LONG(PQresultStatus(obj
->intern
->res
));
475 static void php_pqres_object_read_status_message(zval
*object
, void *o
, zval
*return_value TSRMLS_DC
)
477 php_pqres_object_t
*obj
= o
;
479 RETVAL_STRING(PQresStatus(PQresultStatus(obj
->intern
->res
))+sizeof("PGRES"), 1);
482 static void php_pqres_object_read_error_message(zval
*object
, void *o
, zval
*return_value TSRMLS_DC
)
484 php_pqres_object_t
*obj
= o
;
485 char *error
= PHP_PQresultErrorMessage(obj
->intern
->res
);
488 RETVAL_STRING(error
, 1);
494 static void php_pqres_object_read_num_rows(zval
*object
, void *o
, zval
*return_value TSRMLS_DC
)
496 php_pqres_object_t
*obj
= o
;
498 RETVAL_LONG(PQntuples(obj
->intern
->res
));
501 static void php_pqres_object_read_num_cols(zval
*object
, void *o
, zval
*return_value TSRMLS_DC
)
503 php_pqres_object_t
*obj
= o
;
505 RETVAL_LONG(PQnfields(obj
->intern
->res
));
508 static void php_pqres_object_read_affected_rows(zval
*object
, void *o
, zval
*return_value TSRMLS_DC
)
510 php_pqres_object_t
*obj
= o
;
512 RETVAL_LONG(atoi(PQcmdTuples(obj
->intern
->res
)));
515 static void php_pqres_object_read_fetch_type(zval
*object
, void *o
, zval
*return_value TSRMLS_DC
)
517 php_pqres_object_t
*obj
= o
;
519 RETVAL_LONG(php_pqres_fetch_type(obj
->intern
));
522 static void php_pqres_object_write_fetch_type(zval
*object
, void *o
, zval
*value TSRMLS_DC
)
524 php_pqres_object_t
*obj
= o
;
525 zval
*zfetch_type
= value
;
527 if (Z_TYPE_P(value
) != IS_LONG
) {
528 if (Z_REFCOUNT_P(value
) > 1) {
531 ZVAL_ZVAL(tmp
, zfetch_type
, 1, 0);
532 convert_to_long(tmp
);
535 convert_to_long_ex(&zfetch_type
);
539 if (!obj
->intern
->iter
) {
540 obj
->intern
->iter
= (php_pqres_iterator_t
*) php_pqres_iterator_init(Z_OBJCE_P(object
), object
, 0 TSRMLS_CC
);
541 obj
->intern
->iter
->zi
.funcs
->rewind((zend_object_iterator
*) obj
->intern
->iter TSRMLS_CC
);
543 obj
->intern
->iter
->fetch_type
= Z_LVAL_P(zfetch_type
);
545 if (zfetch_type
!= value
) {
546 zval_ptr_dtor(&zfetch_type
);
550 static void php_pqres_object_read_auto_conv(zval
*object
, void *o
, zval
*return_value TSRMLS_DC
)
552 php_pqres_object_t
*obj
= o
;
554 RETVAL_LONG(obj
->intern
->auto_convert
);
557 static void php_pqres_object_write_auto_conv(zval
*object
, void *o
, zval
*value TSRMLS_DC
)
559 php_pqres_object_t
*obj
= o
;
560 zval
*zauto_conv
= value
;
562 if (Z_TYPE_P(value
) != IS_LONG
) {
563 if (Z_REFCOUNT_P(value
) > 1) {
566 ZVAL_ZVAL(tmp
, zauto_conv
, 1, 0);
567 convert_to_long(tmp
);
570 convert_to_long_ex(&zauto_conv
);
574 obj
->intern
->auto_convert
= Z_LVAL_P(zauto_conv
);
576 if (zauto_conv
!= value
) {
577 zval_ptr_dtor(&zauto_conv
);
581 static ZEND_RESULT_CODE
php_pqres_iteration(zval
*this_ptr
, php_pqres_object_t
*obj
, php_pqres_fetch_t fetch_type
, zval
***row TSRMLS_DC
)
584 php_pqres_fetch_t orig_fetch
;
587 obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
590 if (obj
->intern
->iter
) {
591 obj
->intern
->iter
->zi
.funcs
->move_forward((zend_object_iterator
*) obj
->intern
->iter TSRMLS_CC
);
593 obj
->intern
->iter
= (php_pqres_iterator_t
*) php_pqres_iterator_init(Z_OBJCE_P(getThis()), getThis(), 0 TSRMLS_CC
);
594 obj
->intern
->iter
->zi
.funcs
->rewind((zend_object_iterator
*) obj
->intern
->iter TSRMLS_CC
);
596 orig_fetch
= obj
->intern
->iter
->fetch_type
;
597 obj
->intern
->iter
->fetch_type
= fetch_type
;
598 if (SUCCESS
== (rv
= obj
->intern
->iter
->zi
.funcs
->valid((zend_object_iterator
*) obj
->intern
->iter TSRMLS_CC
))) {
599 obj
->intern
->iter
->zi
.funcs
->get_current_data((zend_object_iterator
*) obj
->intern
->iter
, row TSRMLS_CC
);
601 obj
->intern
->iter
->fetch_type
= orig_fetch
;
606 typedef struct php_pqres_col
{
611 static ZEND_RESULT_CODE
column_nn(php_pqres_object_t
*obj
, zval
*zcol
, php_pqres_col_t
*col TSRMLS_DC
)
619 switch (Z_TYPE_P(zcol
)) {
625 index
= Z_LVAL_P(zcol
);
629 convert_to_string(zcol
);
633 if (!is_numeric_string(Z_STRVAL_P(zcol
), Z_STRLEN_P(zcol
), &index
, NULL
, 0)) {
634 name
= Z_STRVAL_P(zcol
);
642 col
->num
= PQfnumber(obj
->intern
->res
, name
);
644 col
->name
= PQfname(obj
->intern
->res
, index
);
649 php_error_docref(NULL TSRMLS_CC
, E_WARNING
, "Failed to find column at index %ld", index
);
652 if (col
->num
== -1) {
653 php_error_docref(NULL TSRMLS_CC
, E_WARNING
, "Failed to find column with name '%s'", name
);
659 ZEND_BEGIN_ARG_INFO_EX(ai_pqres_bind
, 0, 0, 2)
660 ZEND_ARG_INFO(0, col
)
661 ZEND_ARG_INFO(1, ref
)
663 static PHP_METHOD(pqres
, bind
) {
665 zend_error_handling zeh
;
668 zend_replace_error_handling(EH_THROW
, exce(EX_INVALID_ARGUMENT
), &zeh TSRMLS_CC
);
669 rv
= zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC
, "z/z", &zcol
, &zref
);
670 zend_restore_error_handling(&zeh TSRMLS_CC
);
673 php_pqres_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
676 throw_exce(EX_UNINITIALIZED TSRMLS_CC
, "pq\\Result not initialized");
680 if (SUCCESS
!= column_nn(obj
, zcol
, &col TSRMLS_CC
)) {
685 if (SUCCESS
!= zend_hash_index_update(&obj
->intern
->bound
, col
.num
, (void *) &zref
, sizeof(zval
*), NULL
)) {
686 php_error_docref(NULL TSRMLS_CC
, E_WARNING
, "Failed to bind column %s@%d", col
.name
, col
.num
);
689 zend_hash_sort(&obj
->intern
->bound
, zend_qsort
, php_pq_compare_index
, 0 TSRMLS_CC
);
697 static int apply_bound(void *p TSRMLS_DC
, int argc
, va_list argv
, zend_hash_key
*key
)
699 zval
**zvalue
, **zbound
= p
;
700 zval
**zrow
= va_arg(argv
, zval
**);
701 ZEND_RESULT_CODE
*rv
= va_arg(argv
, ZEND_RESULT_CODE
*);
703 if (SUCCESS
!= zend_hash_index_find(Z_ARRVAL_PP(zrow
), key
->h
, (void *) &zvalue
)) {
704 php_error_docref(NULL TSRMLS_CC
, E_WARNING
, "Failed to find column ad index %lu", key
->h
);
706 return ZEND_HASH_APPLY_STOP
;
709 ZVAL_COPY_VALUE(*zbound
, *zvalue
);
711 zval_ptr_dtor(zvalue
);
715 return ZEND_HASH_APPLY_KEEP
;
719 ZEND_BEGIN_ARG_INFO_EX(ai_pqres_fetch_bound
, 0, 0, 0)
721 static PHP_METHOD(pqres
, fetchBound
) {
722 zend_error_handling zeh
;
725 zend_replace_error_handling(EH_THROW
, exce(EX_INVALID_ARGUMENT
), &zeh TSRMLS_CC
);
726 rv
= zend_parse_parameters_none();
727 zend_restore_error_handling(&zeh TSRMLS_CC
);
730 php_pqres_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
733 throw_exce(EX_UNINITIALIZED TSRMLS_CC
, "pq\\Result not initialized");
737 if (SUCCESS
== php_pqres_iteration(getThis(), obj
, PHP_PQRES_FETCH_ARRAY
, &row TSRMLS_CC
) && row
) {
738 zend_replace_error_handling(EH_THROW
, exce(EX_RUNTIME
), &zeh TSRMLS_CC
);
739 zend_hash_apply_with_arguments(&obj
->intern
->bound TSRMLS_CC
, apply_bound
, 2, row
, &rv
);
740 zend_restore_error_handling(&zeh TSRMLS_CC
);
745 RETVAL_ZVAL(*row
, 1, 0);
752 ZEND_BEGIN_ARG_INFO_EX(ai_pqres_fetch_row
, 0, 0, 0)
753 ZEND_ARG_INFO(0, fetch_type
)
755 static PHP_METHOD(pqres
, fetchRow
) {
756 zend_error_handling zeh
;
757 php_pqres_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
758 long fetch_type
= -1;
761 zend_replace_error_handling(EH_THROW
, exce(EX_INVALID_ARGUMENT
), &zeh TSRMLS_CC
);
762 rv
= zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC
, "|l", &fetch_type
);
763 zend_restore_error_handling(&zeh TSRMLS_CC
);
767 throw_exce(EX_UNINITIALIZED TSRMLS_CC
, "pq\\Result not initialized");
771 if (fetch_type
== -1) {
772 fetch_type
= php_pqres_fetch_type(obj
->intern
);
775 zend_replace_error_handling(EH_THROW
, exce(EX_RUNTIME
), &zeh TSRMLS_CC
);
776 php_pqres_iteration(getThis(), obj
, fetch_type
, &row TSRMLS_CC
);
777 zend_restore_error_handling(&zeh TSRMLS_CC
);
780 RETVAL_ZVAL(*row
, 1, 0);
786 static zval
**column_at(zval
*row
, int col TSRMLS_DC
)
789 HashTable
*ht
= HASH_OF(row
);
790 int count
= zend_hash_num_elements(ht
);
793 php_error_docref(NULL TSRMLS_CC
, E_WARNING
, "Column index %d exceeds column count %d", col
, count
);
795 zend_hash_internal_pointer_reset(ht
);
797 zend_hash_move_forward(ht
);
799 zend_hash_get_current_data(ht
, (void *) &data
);
804 ZEND_BEGIN_ARG_INFO_EX(ai_pqres_fetch_col
, 0, 0, 1)
805 ZEND_ARG_INFO(1, ref
)
806 ZEND_ARG_INFO(0, col
)
808 static PHP_METHOD(pqres
, fetchCol
) {
809 zend_error_handling zeh
;
810 zval
*zcol
= NULL
, *zref
;
813 zend_replace_error_handling(EH_THROW
, exce(EX_INVALID_ARGUMENT
), &zeh TSRMLS_CC
);
814 rv
= zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC
, "z|z/!", &zref
, &zcol
);
815 zend_restore_error_handling(&zeh TSRMLS_CC
);
818 php_pqres_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
821 throw_exce(EX_UNINITIALIZED TSRMLS_CC
, "pq\\Result not initialized");
825 zend_replace_error_handling(EH_THROW
, exce(EX_RUNTIME
), &zeh TSRMLS_CC
);
826 php_pqres_iteration(getThis(), obj
, php_pqres_fetch_type(obj
->intern
), &row TSRMLS_CC
);
830 if (SUCCESS
!= column_nn(obj
, zcol
, &col TSRMLS_CC
)) {
833 zval
**zres
= column_at(*row
, col
.num TSRMLS_CC
);
839 ZVAL_ZVAL(zref
, *zres
, 1, 0);
844 zend_restore_error_handling(&zeh TSRMLS_CC
);
849 ZEND_BEGIN_ARG_INFO_EX(ai_pqres_fetch_all_cols
, 0, 0, 0)
850 ZEND_ARG_INFO(0, col
)
852 static PHP_METHOD(pqres
, fetchAllCols
) {
853 zend_error_handling zeh
;
857 zend_replace_error_handling(EH_THROW
, exce(EX_INVALID_ARGUMENT
), &zeh TSRMLS_CC
);
858 rv
= zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC
, "|z!", &zcol
);
859 zend_restore_error_handling(&zeh TSRMLS_CC
);
862 php_pqres_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
865 throw_exce(EX_UNINITIALIZED TSRMLS_CC
, "pq\\Result not initialized");
869 zend_replace_error_handling(EH_THROW
, exce(EX_RUNTIME
), &zeh TSRMLS_CC
);
870 if (SUCCESS
== column_nn(obj
, zcol
, &col TSRMLS_CC
)) {
871 int r
, rows
= PQntuples(obj
->intern
->res
);
873 array_init(return_value
);
874 for (r
= 0; r
< rows
; ++r
) {
875 add_next_index_zval(return_value
, php_pqres_get_col(obj
->intern
, r
, col
.num TSRMLS_CC
));
878 zend_restore_error_handling(&zeh TSRMLS_CC
);
883 struct apply_to_col_arg
{
884 php_pqres_object_t
*obj
;
885 php_pqres_col_t
*cols
;
886 ZEND_RESULT_CODE status
;
889 static int apply_to_col(void *p
, void *a TSRMLS_DC
)
892 struct apply_to_col_arg
*arg
= a
;
894 if (SUCCESS
!= column_nn(arg
->obj
, *c
, arg
->cols TSRMLS_CC
)) {
895 arg
->status
= FAILURE
;
896 return ZEND_HASH_APPLY_STOP
;
898 arg
->status
= SUCCESS
;
900 return ZEND_HASH_APPLY_KEEP
;
904 static php_pqres_col_t
*php_pqres_convert_to_cols(php_pqres_object_t
*obj
, HashTable
*ht TSRMLS_DC
)
906 struct apply_to_col_arg arg
= {NULL
};
907 php_pqres_col_t
*tmp
;
910 arg
.cols
= ecalloc(zend_hash_num_elements(ht
), sizeof(*tmp
));
912 zend_hash_apply_with_argument(ht
, apply_to_col
, &arg TSRMLS_CC
);
914 if (SUCCESS
== arg
.status
) {
922 ZEND_BEGIN_ARG_INFO_EX(ai_pqres_map
, 0, 0, 0)
923 ZEND_ARG_INFO(0, keys
)
924 ZEND_ARG_INFO(0, vals
)
925 ZEND_ARG_INFO(0, fetch_type
)
927 static PHP_METHOD(pqres
, map
) {
928 zend_error_handling zeh
;
929 zval
*zkeys
= 0, *zvals
= 0;
930 long fetch_type
= -1;
933 zend_replace_error_handling(EH_THROW
, exce(EX_INVALID_ARGUMENT
), &zeh TSRMLS_CC
);
934 rv
= zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC
, "|z/!z/!l", &zkeys
, &zvals
, &fetch_type
);
935 zend_restore_error_handling(&zeh TSRMLS_CC
);
938 php_pqres_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
941 throw_exce(EX_UNINITIALIZED TSRMLS_CC
, "pq\\Result not initialized");
944 php_pqres_col_t def
= {PQfname(obj
->intern
->res
, 0), 0}, *keys
= NULL
, *vals
= NULL
;
947 convert_to_array(zkeys
);
949 if ((ks
= zend_hash_num_elements(Z_ARRVAL_P(zkeys
)))) {
950 keys
= php_pqres_convert_to_cols(obj
, Z_ARRVAL_P(zkeys
) TSRMLS_CC
);
960 convert_to_array(zvals
);
962 if ((vs
= zend_hash_num_elements(Z_ARRVAL_P(zvals
)))) {
963 vals
= php_pqres_convert_to_cols(obj
, Z_ARRVAL_P(zvals
) TSRMLS_CC
);
967 if (fetch_type
== -1) {
968 fetch_type
= php_pqres_fetch_type(obj
->intern
);
975 switch (fetch_type
) {
976 case PHP_PQRES_FETCH_ARRAY
:
977 case PHP_PQRES_FETCH_ASSOC
:
978 array_init(return_value
);
980 case PHP_PQRES_FETCH_OBJECT
:
981 object_init(return_value
);
984 for (r
= 0, rows
= PQntuples(obj
->intern
->res
); r
< rows
; ++r
) {
988 for (k
= 0; k
< ks
; ++k
) {
989 char *key
= PQgetvalue(obj
->intern
->res
, r
, keys
[k
].num
);
990 int len
= PQgetlength(obj
->intern
->res
, r
, keys
[k
].num
);
992 if (SUCCESS
!= zend_symtable_find(HASH_OF(*cur
), key
, len
+ 1, (void *) &cur
)) {
996 switch (fetch_type
) {
997 case PHP_PQRES_FETCH_ARRAY
:
998 case PHP_PQRES_FETCH_ASSOC
:
1001 case PHP_PQRES_FETCH_OBJECT
:
1005 if (SUCCESS
!= zend_symtable_update(HASH_OF(*cur
), key
, len
+ 1, (void *) &tmp
, sizeof(zval
*), (void *) &cur
)) {
1006 throw_exce(EX_RUNTIME TSRMLS_CC
, "Failed to create map");
1012 for (v
= 0; v
< vs
; ++v
) {
1013 char *val
= PQgetvalue(obj
->intern
->res
, r
, vals
[v
].num
);
1014 int len
= PQgetlength(obj
->intern
->res
, r
, vals
[v
].num
);
1016 switch (fetch_type
) {
1017 case PHP_PQRES_FETCH_ARRAY
:
1018 add_index_stringl(*cur
, vals
[v
].num
, val
, len
, 1);
1020 case PHP_PQRES_FETCH_ASSOC
:
1021 add_assoc_stringl(*cur
, vals
[v
].name
, val
, len
, 1);
1023 case PHP_PQRES_FETCH_OBJECT
:
1024 add_property_stringl(*cur
, vals
[v
].name
, val
, len
, 1);
1029 php_pqres_row_to_zval(obj
->intern
->res
, r
, fetch_type
, cur TSRMLS_CC
);
1035 if (keys
&& keys
!= &def
) {
1045 ZEND_BEGIN_ARG_INFO_EX(ai_pqres_fetch_all
, 0, 0, 0)
1046 ZEND_ARG_INFO(0, fetch_type
)
1047 ZEND_END_ARG_INFO();
1048 static PHP_METHOD(pqres
, fetchAll
) {
1049 zend_error_handling zeh
;
1050 long fetch_type
= -1;
1051 ZEND_RESULT_CODE rv
;
1053 zend_replace_error_handling(EH_THROW
, exce(EX_INVALID_ARGUMENT
), &zeh TSRMLS_CC
);
1054 rv
= zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC
, "|l", &fetch_type
);
1055 zend_restore_error_handling(&zeh TSRMLS_CC
);
1057 if (SUCCESS
== rv
) {
1058 php_pqres_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
1060 throw_exce(EX_UNINITIALIZED TSRMLS_CC
, "pq\\Result not initialized");
1062 int r
, rows
= PQntuples(obj
->intern
->res
);
1064 if (fetch_type
== -1) {
1065 fetch_type
= php_pqres_fetch_type(obj
->intern
);
1068 array_init(return_value
);
1069 for (r
= 0; r
< rows
; ++r
) {
1070 add_next_index_zval(return_value
, php_pqres_row_to_zval(obj
->intern
->res
, r
, fetch_type
, NULL TSRMLS_CC
));
1076 ZEND_BEGIN_ARG_INFO_EX(ai_pqres_count
, 0, 0, 0)
1077 ZEND_END_ARG_INFO();
1078 static PHP_METHOD(pqres
, count
) {
1079 zend_error_handling zeh
;
1080 ZEND_RESULT_CODE rv
;
1082 zend_replace_error_handling(EH_THROW
, exce(EX_INVALID_ARGUMENT
), &zeh TSRMLS_CC
);
1083 rv
= zend_parse_parameters_none();
1084 zend_restore_error_handling(&zeh TSRMLS_CC
);
1086 if (SUCCESS
== rv
) {
1089 if (SUCCESS
!= php_pqres_count_elements(getThis(), &count TSRMLS_CC
)) {
1090 throw_exce(EX_UNINITIALIZED TSRMLS_CC
, "pq\\Result not initialized");
1097 ZEND_BEGIN_ARG_INFO_EX(ai_pqres_desc
, 0, 0, 0)
1098 ZEND_END_ARG_INFO();
1099 static PHP_METHOD(pqres
, desc
) {
1100 zend_error_handling zeh
;
1101 ZEND_RESULT_CODE rv
;
1103 zend_replace_error_handling(EH_THROW
, exce(EX_INVALID_ARGUMENT
), &zeh TSRMLS_CC
);
1104 rv
= zend_parse_parameters_none();
1105 zend_restore_error_handling(&zeh TSRMLS_CC
);
1107 if (SUCCESS
== rv
) {
1108 php_pqres_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
1111 throw_exce(EX_UNINITIALIZED TSRMLS_CC
, "pq\\Result not initialized");
1115 array_init(return_value
);
1116 for (p
= 0, params
= PQnparams(obj
->intern
->res
); p
< params
; ++p
) {
1117 add_next_index_long(return_value
, PQparamtype(obj
->intern
->res
, p
));
1123 static zend_function_entry php_pqres_methods
[] = {
1124 PHP_ME(pqres
, bind
, ai_pqres_bind
, ZEND_ACC_PUBLIC
)
1125 PHP_ME(pqres
, fetchBound
, ai_pqres_fetch_bound
, ZEND_ACC_PUBLIC
)
1126 PHP_ME(pqres
, fetchRow
, ai_pqres_fetch_row
, ZEND_ACC_PUBLIC
)
1127 PHP_ME(pqres
, fetchCol
, ai_pqres_fetch_col
, ZEND_ACC_PUBLIC
)
1128 PHP_ME(pqres
, fetchAll
, ai_pqres_fetch_all
, ZEND_ACC_PUBLIC
)
1129 PHP_ME(pqres
, fetchAllCols
, ai_pqres_fetch_all_cols
, ZEND_ACC_PUBLIC
)
1130 PHP_ME(pqres
, count
, ai_pqres_count
, ZEND_ACC_PUBLIC
)
1131 PHP_ME(pqres
, map
, ai_pqres_map
, ZEND_ACC_PUBLIC
)
1132 PHP_ME(pqres
, desc
, ai_pqres_desc
, ZEND_ACC_PUBLIC
)
1136 PHP_MSHUTDOWN_FUNCTION(pqres
)
1138 zend_hash_destroy(&php_pqres_object_prophandlers
);
1142 PHP_MINIT_FUNCTION(pqres
)
1144 zend_class_entry ce
= {0};
1145 php_pq_object_prophandler_t ph
= {0};
1147 INIT_NS_CLASS_ENTRY(ce
, "pq", "Result", php_pqres_methods
);
1148 php_pqres_class_entry
= zend_register_internal_class_ex(&ce
, NULL
, NULL TSRMLS_CC
);
1149 php_pqres_class_entry
->create_object
= php_pqres_create_object
;
1150 php_pqres_class_entry
->iterator_funcs
.funcs
= &php_pqres_iterator_funcs
;
1151 php_pqres_class_entry
->get_iterator
= php_pqres_iterator_init
;
1152 zend_class_implements(php_pqres_class_entry TSRMLS_CC
, 2, zend_ce_traversable
, spl_ce_Countable
);
1154 memcpy(&php_pqres_object_handlers
, zend_get_std_object_handlers(), sizeof(zend_object_handlers
));
1155 php_pqres_object_handlers
.read_property
= php_pq_object_read_prop
;
1156 php_pqres_object_handlers
.write_property
= php_pq_object_write_prop
;
1157 php_pqres_object_handlers
.clone_obj
= NULL
;
1158 php_pqres_object_handlers
.get_property_ptr_ptr
= NULL
;
1159 php_pqres_object_handlers
.get_gc
= NULL
;
1160 php_pqres_object_handlers
.get_debug_info
= php_pq_object_debug_info
;
1161 php_pqres_object_handlers
.get_properties
= php_pq_object_properties
;
1162 php_pqres_object_handlers
.count_elements
= php_pqres_count_elements
;
1164 zend_hash_init(&php_pqres_object_prophandlers
, 8, NULL
, NULL
, 1);
1166 zend_declare_property_null(php_pqres_class_entry
, ZEND_STRL("status"), ZEND_ACC_PUBLIC TSRMLS_CC
);
1167 ph
.read
= php_pqres_object_read_status
;
1168 zend_hash_add(&php_pqres_object_prophandlers
, "status", sizeof("status"), (void *) &ph
, sizeof(ph
), NULL
);
1170 zend_declare_property_null(php_pqres_class_entry
, ZEND_STRL("statusMessage"), ZEND_ACC_PUBLIC TSRMLS_CC
);
1171 ph
.read
= php_pqres_object_read_status_message
;
1172 zend_hash_add(&php_pqres_object_prophandlers
, "statusMessage", sizeof("statusMessage"), (void *) &ph
, sizeof(ph
), NULL
);
1174 zend_declare_property_null(php_pqres_class_entry
, ZEND_STRL("errorMessage"), ZEND_ACC_PUBLIC TSRMLS_CC
);
1175 ph
.read
= php_pqres_object_read_error_message
;
1176 zend_hash_add(&php_pqres_object_prophandlers
, "errorMessage", sizeof("errorMessage"), (void *) &ph
, sizeof(ph
), NULL
);
1178 zend_declare_property_long(php_pqres_class_entry
, ZEND_STRL("numRows"), 0, ZEND_ACC_PUBLIC TSRMLS_CC
);
1179 ph
.read
= php_pqres_object_read_num_rows
;
1180 zend_hash_add(&php_pqres_object_prophandlers
, "numRows", sizeof("numRows"), (void *) &ph
, sizeof(ph
), NULL
);
1182 zend_declare_property_long(php_pqres_class_entry
, ZEND_STRL("numCols"), 0, ZEND_ACC_PUBLIC TSRMLS_CC
);
1183 ph
.read
= php_pqres_object_read_num_cols
;
1184 zend_hash_add(&php_pqres_object_prophandlers
, "numCols", sizeof("numCols"), (void *) &ph
, sizeof(ph
), NULL
);
1186 zend_declare_property_long(php_pqres_class_entry
, ZEND_STRL("affectedRows"), 0, ZEND_ACC_PUBLIC TSRMLS_CC
);
1187 ph
.read
= php_pqres_object_read_affected_rows
;
1188 zend_hash_add(&php_pqres_object_prophandlers
, "affectedRows", sizeof("affectedRows"), (void *) &ph
, sizeof(ph
), NULL
);
1190 zend_declare_property_long(php_pqres_class_entry
, ZEND_STRL("fetchType"), PHP_PQRES_FETCH_ARRAY
, ZEND_ACC_PUBLIC TSRMLS_CC
);
1191 ph
.read
= php_pqres_object_read_fetch_type
;
1192 ph
.write
= php_pqres_object_write_fetch_type
;
1193 zend_hash_add(&php_pqres_object_prophandlers
, "fetchType", sizeof("fetchType"), (void *) &ph
, sizeof(ph
), NULL
);
1196 zend_declare_property_long(php_pqres_class_entry
, ZEND_STRL("autoConvert"), PHP_PQRES_CONV_ALL
, ZEND_ACC_PUBLIC TSRMLS_CC
);
1197 ph
.read
= php_pqres_object_read_auto_conv
;
1198 ph
.write
= php_pqres_object_write_auto_conv
;
1199 zend_hash_add(&php_pqres_object_prophandlers
, "autoConvert", sizeof("autoConvert"), (void *) &ph
, sizeof(ph
), NULL
);
1202 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("EMPTY_QUERY"), PGRES_EMPTY_QUERY TSRMLS_CC
);
1203 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("COMMAND_OK"), PGRES_COMMAND_OK TSRMLS_CC
);
1204 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("TUPLES_OK"), PGRES_TUPLES_OK TSRMLS_CC
);
1205 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("COPY_OUT"), PGRES_COPY_OUT TSRMLS_CC
);
1206 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("COPY_IN"), PGRES_COPY_IN TSRMLS_CC
);
1207 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("BAD_RESPONSE"), PGRES_BAD_RESPONSE TSRMLS_CC
);
1208 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("NONFATAL_ERROR"), PGRES_NONFATAL_ERROR TSRMLS_CC
);
1209 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("FATAL_ERROR"), PGRES_FATAL_ERROR TSRMLS_CC
);
1210 #ifdef HAVE_PGRES_COPY_BOTH
1211 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("COPY_BOTH"), PGRES_COPY_BOTH TSRMLS_CC
);
1213 #ifdef HAVE_PGRES_SINGLE_TUPLE
1214 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("SINGLE_TUPLE"), PGRES_SINGLE_TUPLE TSRMLS_CC
);
1217 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("FETCH_ARRAY"), PHP_PQRES_FETCH_ARRAY TSRMLS_CC
);
1218 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("FETCH_ASSOC"), PHP_PQRES_FETCH_ASSOC TSRMLS_CC
);
1219 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("FETCH_OBJECT"), PHP_PQRES_FETCH_OBJECT TSRMLS_CC
);
1221 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("CONV_BOOL"), PHP_PQRES_CONV_BOOL TSRMLS_CC
);
1222 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("CONV_INT"), PHP_PQRES_CONV_INT TSRMLS_CC
);
1223 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("CONV_FLOAT"), PHP_PQRES_CONV_FLOAT TSRMLS_CC
);
1224 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("CONV_SCALAR"), PHP_PQRES_CONV_SCALAR TSRMLS_CC
);
1225 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("CONV_ARRAY"), PHP_PQRES_CONV_ARRAY TSRMLS_CC
);
1226 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("CONV_DATETIME"), PHP_PQRES_CONV_DATETIME TSRMLS_CC
);
1227 #if PHP_PQ_HAVE_PHP_JSON_H
1228 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("CONV_JSON"), PHP_PQRES_CONV_JSON TSRMLS_CC
);
1230 zend_declare_class_constant_long(php_pqres_class_entry
, ZEND_STRL("CONV_ALL"), PHP_PQRES_CONV_ALL TSRMLS_CC
);
1240 * vim600: noet sw=4 ts=4 fdm=marker
1241 * vim<600: noet sw=4 ts=4