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 +--------------------------------------------------------------------+
18 #include <ext/standard/php_smart_str.h>
21 #include "php_pq_misc.h"
22 #include "php_pq_object.h"
23 #include "php_pqexc.h"
24 #include "php_pqconn.h"
25 #include "php_pqres.h"
26 #include "php_pqcur.h"
28 zend_class_entry
*php_pqcur_class_entry
;
29 static zend_object_handlers php_pqcur_object_handlers
;
30 static HashTable php_pqcur_object_prophandlers
;
32 static void cur_close(php_pqcur_object_t
*obj TSRMLS_DC
)
34 if (obj
->intern
->open
) {
38 smart_str_appends(&cmd
, "CLOSE ");
39 smart_str_appends(&cmd
, obj
->intern
->name
);
42 if ((res
= PQexec(obj
->intern
->conn
->intern
->conn
, cmd
.c
))) {
47 obj
->intern
->open
= 0;
51 static void cur_fetch_or_move(INTERNAL_FUNCTION_PARAMETERS
, const char *action
, zend_bool async
)
56 php_pq_callback_t resolver
= {{0}};
57 zend_error_handling zeh
;
59 zend_replace_error_handling(EH_THROW
, exce(EX_INVALID_ARGUMENT
), &zeh TSRMLS_CC
);
60 rv
= zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC
, async
? "|sf" : "|s", &spec_str
, &spec_len
, &resolver
.fci
, &resolver
.fcc
);
61 zend_restore_error_handling(&zeh TSRMLS_CC
);
64 php_pqcur_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
67 throw_exce(EX_UNINITIALIZED TSRMLS_CC
, "pq\\Cursor not initialized");
71 smart_str_appends(&cmd
, *action
== 'f' ? "FETCH " : "MOVE ");
72 smart_str_appendl(&cmd
, spec_str
, spec_len
);
73 smart_str_appends(&cmd
, " FROM ");
74 smart_str_appends(&cmd
, obj
->intern
->name
);
78 int rc
= PQsendQuery(obj
->intern
->conn
->intern
->conn
, cmd
.c
);
81 throw_exce(EX_IO TSRMLS_CC
, "Failed to %s cursor (%s)", *action
== 'f' ? "fetch from" : "move in", PHP_PQerrorMessage(obj
->intern
->conn
->intern
->conn
));
82 } else if (obj
->intern
->conn
->intern
->unbuffered
&& !PQsetSingleRowMode(obj
->intern
->conn
->intern
->conn
)) {
83 throw_exce(EX_RUNTIME TSRMLS_CC
, "Failed to enable unbuffered mode (%s)", PHP_PQerrorMessage(obj
->intern
->conn
->intern
->conn
));
85 php_pq_callback_recurse(&obj
->intern
->conn
->intern
->onevent
, &resolver TSRMLS_CC
);
86 obj
->intern
->conn
->intern
->poller
= PQconsumeInput
;
89 PGresult
*res
= PQexec(obj
->intern
->conn
->intern
->conn
, cmd
.c
);
92 throw_exce(EX_RUNTIME TSRMLS_CC
, "Failed to %s cursor (%s)", *action
== 'f' ? "fetch from" : "move in", PHP_PQerrorMessage(obj
->intern
->conn
->intern
->conn
));
93 } else if (SUCCESS
== php_pqres_success(res TSRMLS_CC
)) {
94 php_pq_object_to_zval_no_addref(PQresultInstanceData(res
, php_pqconn_event
), &return_value TSRMLS_CC
);
99 php_pqconn_notify_listeners(obj
->intern
->conn TSRMLS_CC
);
104 static void php_pqcur_object_free(void *o TSRMLS_DC
)
106 php_pqcur_object_t
*obj
= o
;
108 fprintf(stderr
, "FREE cur(#%d) %p (conn: %p)\n", obj
->zv
.handle
, obj
, obj
->intern
->conn
);
111 //cur_close(obj TSRMLS_CC);
112 //php_pq_object_delref(obj->intern->conn TSRMLS_CC);
113 efree(obj
->intern
->decl
);
114 efree(obj
->intern
->name
);
118 zend_object_std_dtor((zend_object
*) o TSRMLS_CC
);
122 zend_object_value
php_pqcur_create_object_ex(zend_class_entry
*ce
, php_pqcur_t
*intern
, php_pqcur_object_t
**ptr TSRMLS_DC
)
124 php_pqcur_object_t
*o
;
126 o
= ecalloc(1, sizeof(*o
));
127 zend_object_std_init((zend_object
*) o
, ce TSRMLS_CC
);
128 object_properties_init((zend_object
*) o
, ce
);
129 o
->prophandler
= &php_pqcur_object_prophandlers
;
139 o
->zv
.handle
= zend_objects_store_put((zend_object
*) o
, NULL
, php_pqcur_object_free
, NULL TSRMLS_CC
);
140 o
->zv
.handlers
= &php_pqcur_object_handlers
;
145 static zend_object_value
php_pqcur_create_object(zend_class_entry
*class_type TSRMLS_DC
)
147 return php_pqcur_create_object_ex(class_type
, NULL
, NULL TSRMLS_CC
);
150 static void php_pqcur_object_read_name(zval
*object
, void *o
, zval
*return_value TSRMLS_DC
)
152 php_pqcur_object_t
*obj
= o
;
154 RETVAL_STRING(obj
->intern
->name
, 1);
157 static void php_pqcur_object_read_connection(zval
*object
, void *o
, zval
*return_value TSRMLS_DC
)
159 php_pqcur_object_t
*obj
= o
;
161 php_pq_object_to_zval(obj
->intern
->conn
, &return_value TSRMLS_CC
);
164 ZEND_BEGIN_ARG_INFO_EX(ai_pqcur_open
, 0, 0, 0)
166 static PHP_METHOD(pqcur
, open
)
168 zend_error_handling zeh
;
171 zend_replace_error_handling(EH_THROW
, exce(EX_INVALID_ARGUMENT
), &zeh TSRMLS_CC
);
172 rv
= zend_parse_parameters_none();
173 zend_restore_error_handling(&zeh TSRMLS_CC
);
176 php_pqcur_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
179 throw_exce(EX_UNINITIALIZED TSRMLS_CC
, "pq\\Cursor not initialized");
180 } else if (!obj
->intern
->open
) {
181 if (SUCCESS
== php_pqconn_declare(NULL
, obj
->intern
->conn
, obj
->intern
->decl TSRMLS_CC
)) {
182 obj
->intern
->open
= 1;
188 ZEND_BEGIN_ARG_INFO_EX(ai_pqcur_close
, 0, 0, 0)
190 static PHP_METHOD(pqcur
, close
)
192 zend_error_handling zeh
;
195 zend_replace_error_handling(EH_THROW
, exce(EX_INVALID_ARGUMENT
), &zeh TSRMLS_CC
);
196 rv
= zend_parse_parameters_none();
197 zend_restore_error_handling(&zeh TSRMLS_CC
);
200 php_pqcur_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
203 throw_exce(EX_UNINITIALIZED TSRMLS_CC
, "pq\\Cursor not initialized");
205 cur_close(obj TSRMLS_CC
);
210 ZEND_BEGIN_ARG_INFO_EX(ai_pqcur_fetch
, 0, 0, 1)
211 ZEND_ARG_INFO(0, spec
)
213 static PHP_METHOD(pqcur
, fetch
)
215 cur_fetch_or_move(INTERNAL_FUNCTION_PARAM_PASSTHRU
, "fetch", 0);
218 ZEND_BEGIN_ARG_INFO_EX(ai_pqcur_move
, 0, 0, 0)
219 ZEND_ARG_INFO(0, spec
)
221 static PHP_METHOD(pqcur
, move
)
223 cur_fetch_or_move(INTERNAL_FUNCTION_PARAM_PASSTHRU
, "move", 0);
226 ZEND_BEGIN_ARG_INFO_EX(ai_pqcur_fetchAsync
, 0, 0, 0)
227 ZEND_ARG_INFO(0, spec
)
228 ZEND_ARG_INFO(0, callback
)
230 static PHP_METHOD(pqcur
, fetchAsync
)
232 cur_fetch_or_move(INTERNAL_FUNCTION_PARAM_PASSTHRU
, "fetch", 1);
235 ZEND_BEGIN_ARG_INFO_EX(ai_pqcur_moveAsync
, 0, 0, 0)
236 ZEND_ARG_INFO(0, spec
)
237 ZEND_ARG_INFO(0, callback
)
239 static PHP_METHOD(pqcur
, moveAsync
)
241 cur_fetch_or_move(INTERNAL_FUNCTION_PARAM_PASSTHRU
, "move", 1);
244 static zend_function_entry php_pqcur_methods
[] = {
245 PHP_ME(pqcur
, open
, ai_pqcur_open
, ZEND_ACC_PUBLIC
)
246 PHP_ME(pqcur
, close
, ai_pqcur_close
, ZEND_ACC_PUBLIC
)
247 PHP_ME(pqcur
, fetch
, ai_pqcur_fetch
, ZEND_ACC_PUBLIC
)
248 PHP_ME(pqcur
, move
, ai_pqcur_move
, ZEND_ACC_PUBLIC
)
249 PHP_ME(pqcur
, fetchAsync
, ai_pqcur_fetchAsync
, ZEND_ACC_PUBLIC
)
250 PHP_ME(pqcur
, moveAsync
, ai_pqcur_moveAsync
, ZEND_ACC_PUBLIC
)
254 PHP_MSHUTDOWN_FUNCTION(pqcur
)
256 zend_hash_destroy(&php_pqcur_object_prophandlers
);
260 PHP_MINIT_FUNCTION(pqcur
)
262 zend_class_entry ce
= {0};
263 php_pq_object_prophandler_t ph
= {0};
265 INIT_NS_CLASS_ENTRY(ce
, "pq", "Cursor", php_pqcur_methods
);
266 php_pqcur_class_entry
= zend_register_internal_class_ex(&ce
, NULL
, NULL TSRMLS_CC
);
267 php_pqcur_class_entry
->create_object
= php_pqcur_create_object
;
269 memcpy(&php_pqcur_object_handlers
, zend_get_std_object_handlers(), sizeof(zend_object_handlers
));
270 php_pqcur_object_handlers
.read_property
= php_pq_object_read_prop
;
271 php_pqcur_object_handlers
.write_property
= php_pq_object_write_prop
;
272 php_pqcur_object_handlers
.clone_obj
= NULL
;
273 php_pqcur_object_handlers
.get_property_ptr_ptr
= NULL
;
274 php_pqcur_object_handlers
.get_gc
= NULL
;
275 php_pqcur_object_handlers
.get_properties
= php_pq_object_properties
;
276 php_pqcur_object_handlers
.get_debug_info
= php_pq_object_debug_info
;
278 zend_hash_init(&php_pqcur_object_prophandlers
, 2, NULL
, NULL
, 1);
280 zend_declare_class_constant_long(php_pqcur_class_entry
, ZEND_STRL("BINARY"), PHP_PQ_DECLARE_BINARY TSRMLS_CC
);
281 zend_declare_class_constant_long(php_pqcur_class_entry
, ZEND_STRL("INSENSITIVE"), PHP_PQ_DECLARE_INSENSITIVE TSRMLS_CC
);
282 zend_declare_class_constant_long(php_pqcur_class_entry
, ZEND_STRL("WITH_HOLD"), PHP_PQ_DECLARE_WITH_HOLD TSRMLS_CC
);
283 zend_declare_class_constant_long(php_pqcur_class_entry
, ZEND_STRL("SCROLL"), PHP_PQ_DECLARE_SCROLL TSRMLS_CC
);
284 zend_declare_class_constant_long(php_pqcur_class_entry
, ZEND_STRL("NO_SCROLL"), PHP_PQ_DECLARE_NO_SCROLL TSRMLS_CC
);
286 zend_declare_property_null(php_pqcur_class_entry
, ZEND_STRL("name"), ZEND_ACC_PUBLIC TSRMLS_CC
);
287 ph
.read
= php_pqcur_object_read_name
;
288 zend_hash_add(&php_pqcur_object_prophandlers
, "name", sizeof("name"), (void *) &ph
, sizeof(ph
), NULL
);
290 zend_declare_property_null(php_pqcur_class_entry
, ZEND_STRL("connection"), ZEND_ACC_PUBLIC TSRMLS_CC
);
291 ph
.read
= php_pqcur_object_read_connection
;
292 zend_hash_add(&php_pqcur_object_prophandlers
, "connection", sizeof("connection"), (void *) &ph
, sizeof(ph
), NULL
);
302 * vim600: noet sw=4 ts=4 fdm=marker
303 * vim<600: noet sw=4 ts=4