ZEND_METHOD(Vedis, hgetall)
{
char *key;
int key_len;
VEDIS_ARGS_PARAM(HGETALL, 7, 1);
if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "s",
&key, &key_len) == FAILURE) {
return;
}
VEDIS_SELF(intern);
VEDIS_ARGS_STRING(key, key_len);
VEDIS_ARGS_EXEC(RETURN_FALSE);
array_init(return_value);
if (vedis_value_is_array(result)) {
int key_len = 0;
const char *key;
vedis_value *entry;
while ((entry = vedis_array_next_elem(result)) != 0) {
key = vedis_value_to_string(entry, &key_len);
if ((entry = vedis_array_next_elem(result)) == 0) {
break;
}
if (vedis_value_is_null(entry)) {
add_assoc_null_ex(return_value, key, key_len + 1);
} else {
int len = 0;
const char *str = vedis_value_to_string(entry, &len);
add_assoc_stringl_ex(return_value,
key, key_len + 1, (char *)str, len, 1);
}
}
}
}
ZEND_METHOD(Vedis, hmget)
{
char *key;
int key_len;
zval **val, *fields;
HashPosition pos;
size_t n;
VEDIS_PARAM(HMGET, 5);
if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "sa",
&key, &key_len, &fields) == FAILURE) {
return;
}
VEDIS_SELF(intern);
n = zend_hash_num_elements(HASH_OF(fields));
if (n == 0) {
RETURN_FALSE;
}
VEDIS_ARGS_INIT(n + 1);
VEDIS_ARGS_STRING(key, key_len);
zend_hash_internal_pointer_reset_ex(HASH_OF(fields), &pos);
while (zend_hash_get_current_data_ex(HASH_OF(fields),
(void **)&val, &pos) == SUCCESS) {
if (Z_TYPE_PP(val) != IS_STRING) {
convert_to_string(*val);
}
VEDIS_ARGS_STRING(Z_STRVAL_PP(val), Z_STRLEN_PP(val));
zend_hash_move_forward_ex(HASH_OF(fields), &pos);
}
VEDIS_ARGS_EXEC(RETURN_FALSE);
array_init(return_value);
if (vedis_value_is_array(result)) {
vedis_value *entry;
zend_hash_internal_pointer_reset_ex(HASH_OF(fields), &pos);
while (1) {
if (zend_hash_get_current_data_ex(HASH_OF(fields),
(void **)&val, &pos) != SUCCESS) {
break;
}
if ((entry = vedis_array_next_elem(result)) == 0) {
break;
}
if (Z_TYPE_PP(val) != IS_STRING) {
convert_to_string(*val);
}
if (Z_STRLEN_PP(val) > 0) {
if (vedis_value_is_null(entry)) {
add_assoc_null_ex(return_value,
Z_STRVAL_PP(val), Z_STRLEN_PP(val) + 1);
} else {
int len = 0;
const char *str = vedis_value_to_string(entry, &len);
add_assoc_stringl_ex(return_value,
Z_STRVAL_PP(val), Z_STRLEN_PP(val) + 1,
(char *)str, len, 1);
}
}
zend_hash_move_forward_ex(HASH_OF(fields), &pos);
}
}
}
/**
* Lists table references
*
* @param string $table
* @param string $schema
* @return Phalcon\Db\Reference[]
*/
PHP_METHOD(Phalcon_Db_Adapter_Pdo_Sqlite, describeReferences){
zval *table, *schema = NULL, *dialect, *sql, *fetch_assoc;
zval *describe, *reference_objects, *reference_describe = NULL;
zval *number = NULL, *constraint_name = NULL, *referenced_table = NULL;
zval *from = NULL, *to = NULL, *columns = NULL, *referenced_columns = NULL;
zval *reference_array = NULL, *reference = NULL;
HashTable *ah0;
HashPosition hp0;
zval **hd;
char *hash_index;
uint hash_index_len;
ulong hash_num;
int hash_type;
PHALCON_MM_GROW();
if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "z|z", &table, &schema) == FAILURE) {
PHALCON_MM_RESTORE();
RETURN_NULL();
}
if (!schema) {
PHALCON_INIT_NVAR(schema);
}
PHALCON_INIT_VAR(dialect);
phalcon_read_property(&dialect, this_ptr, SL("_dialect"), PH_NOISY_CC);
PHALCON_INIT_VAR(sql);
PHALCON_CALL_METHOD_PARAMS_2(sql, dialect, "describereferences", table, schema, PH_NO_CHECK);
PHALCON_INIT_VAR(fetch_assoc);
phalcon_get_class_constant(fetch_assoc, phalcon_db_ce, SS("FETCH_ASSOC") TSRMLS_CC);
PHALCON_INIT_VAR(describe);
PHALCON_CALL_METHOD_PARAMS_2(describe, this_ptr, "fetchall", sql, fetch_assoc, PH_NO_CHECK);
PHALCON_INIT_VAR(reference_objects);
array_init(reference_objects);
if (!phalcon_valid_foreach(describe TSRMLS_CC)) {
return;
}
ah0 = Z_ARRVAL_P(describe);
zend_hash_internal_pointer_reset_ex(ah0, &hp0);
ph_cycle_start_0:
if (zend_hash_get_current_data_ex(ah0, (void**) &hd, &hp0) != SUCCESS) {
goto ph_cycle_end_0;
}
PHALCON_GET_FOREACH_KEY(number, ah0, hp0);
PHALCON_GET_FOREACH_VALUE(reference_describe);
PHALCON_INIT_NVAR(constraint_name);
PHALCON_CONCAT_SV(constraint_name, "foreign_key_", number);
PHALCON_INIT_NVAR(referenced_table);
phalcon_array_fetch_string(&referenced_table, reference_describe, SL("table"), PH_NOISY_CC);
PHALCON_INIT_NVAR(from);
phalcon_array_fetch_string(&from, reference_describe, SL("from"), PH_NOISY_CC);
PHALCON_INIT_NVAR(to);
phalcon_array_fetch_string(&to, reference_describe, SL("to"), PH_NOISY_CC);
PHALCON_INIT_NVAR(columns);
array_init(columns);
phalcon_array_append(&columns, from, PH_SEPARATE TSRMLS_CC);
PHALCON_INIT_NVAR(referenced_columns);
array_init(referenced_columns);
phalcon_array_append(&referenced_columns, to, PH_SEPARATE TSRMLS_CC);
PHALCON_INIT_NVAR(reference_array);
array_init(reference_array);
add_assoc_null_ex(reference_array, SS("referencedSchema"));
phalcon_array_update_string(&reference_array, SL("referencedTable"), &referenced_table, PH_COPY | PH_SEPARATE TSRMLS_CC);
phalcon_array_update_string(&reference_array, SL("columns"), &columns, PH_COPY | PH_SEPARATE TSRMLS_CC);
phalcon_array_update_string(&reference_array, SL("referencedColumns"), &referenced_columns, PH_COPY | PH_SEPARATE TSRMLS_CC);
PHALCON_INIT_NVAR(reference);
object_init_ex(reference, phalcon_db_reference_ce);
PHALCON_CALL_METHOD_PARAMS_2_NORETURN(reference, "__construct", constraint_name, reference_array, PH_CHECK);
phalcon_array_update_zval(&reference_objects, constraint_name, &reference, PH_COPY | PH_SEPARATE TSRMLS_CC);
zend_hash_move_forward_ex(ah0, &hp0);
goto ph_cycle_start_0;
ph_cycle_end_0:
RETURN_CTOR(reference_objects);
}
PHP_COUCHBASE_LOCAL
void php_couchbase_get_impl(INTERNAL_FUNCTION_PARAMETERS,
int multi,
int oo,
int lock,
int touch)
{
char *key, **keys;
long *klens, klen = 0;
int nkey = 0;
long flag = 0;
lcb_time_t exp = {0};
long expiry = 0;
zval *res, *cas_token = NULL;
int argflags;
lcb_error_t retval;
php_couchbase_res *couchbase_res;
php_couchbase_ctx *ctx;
argflags = oo ? PHP_COUCHBASE_ARG_F_OO : PHP_COUCHBASE_ARG_F_FUNCTIONAL;
if (multi) {
zval *akeys;
zval **ppzval;
zend_bool preserve_order;
int i;
if (lock) {
PHP_COUCHBASE_GET_PARAMS_WITH_ZV(res, couchbase_res, argflags,
"az|ll",
&akeys, &cas_token,
&flag, &expiry);
} else if (touch) {
PHP_COUCHBASE_GET_PARAMS_WITH_ZV(res, couchbase_res, argflags,
"al|z",
&akeys, &expiry, &cas_token);
} else {
PHP_COUCHBASE_GET_PARAMS_WITH_ZV(res, couchbase_res, argflags,
"a|zl",
&akeys, &cas_token, &flag);
}
nkey = zend_hash_num_elements(Z_ARRVAL_P(akeys));
keys = ecalloc(nkey, sizeof(char *));
klens = ecalloc(nkey, sizeof(long));
preserve_order = (flag & COUCHBASE_GET_PRESERVE_ORDER);
array_init(return_value);
for (i = 0, zend_hash_internal_pointer_reset(Z_ARRVAL_P(akeys));
zend_hash_has_more_elements(Z_ARRVAL_P(akeys)) == SUCCESS;
zend_hash_move_forward(Z_ARRVAL_P(akeys)), i++) {
if (zend_hash_get_current_data(Z_ARRVAL_P(akeys), (void **)&ppzval) == FAILURE) {
nkey--;
continue;
}
if (IS_ARRAY != Z_TYPE_PP(ppzval)) {
convert_to_string_ex(ppzval);
}
if (!Z_STRLEN_PP(ppzval)) {
nkey--;
continue;
}
if (couchbase_res->prefix_key_len) {
klens[i] = spprintf(&(keys[i]), 0, "%s_%s", couchbase_res->prefix_key, Z_STRVAL_PP(ppzval));
} else {
keys[i] = Z_STRVAL_PP(ppzval);
klens[i] = Z_STRLEN_PP(ppzval);
}
if (preserve_order) {
add_assoc_null_ex(return_value, keys[i], klens[i] + 1);
}
}
if (!nkey) {
efree(keys);
efree(klens);
return;
}
if (cas_token && IS_ARRAY != Z_TYPE_P(cas_token)) {
zval_dtor(cas_token);
array_init(cas_token);
}
} else {
if (lock) {
PHP_COUCHBASE_GET_PARAMS_WITH_ZV(res, couchbase_res, argflags,
"sz|l", &key, &klen,
&cas_token, &expiry);
} else if (touch) {
PHP_COUCHBASE_GET_PARAMS_WITH_ZV(res, couchbase_res, argflags,
"sl|z", &key, &klen, &expiry,
&cas_token);
} else {
PHP_COUCHBASE_GET_PARAMS_WITH_ZV(res, couchbase_res, argflags,
"s|z", &key, &klen,
&cas_token);
}
if (!klen) {
return;
}
nkey = 1;
if (couchbase_res->prefix_key_len) {
klen = spprintf(&key, 0, "%s_%s", couchbase_res->prefix_key, key);
}
keys = &key;
klens = &klen;
if (cas_token) {
zval_dtor(cas_token);
ZVAL_NULL(cas_token);
}
}
{
lcb_get_cmd_t **commands = ecalloc(nkey, sizeof(lcb_get_cmd_t *));
int ii;
if (expiry) {
exp = pcbc_check_expiry(expiry);
}
for (ii = 0; ii < nkey; ++ii) {
lcb_get_cmd_t *cmd = ecalloc(1, sizeof(lcb_get_cmd_t));
commands[ii] = cmd;
cmd->v.v0.key = keys[ii];
cmd->v.v0.nkey = klens[ii];
cmd->v.v0.lock = (int)lock;
cmd->v.v0.exptime = exp; /* NB: this assumes sizeof(lcb_time_t) == sizeof(long) */
}
ctx = ecalloc(1, sizeof(php_couchbase_ctx));
ctx->res = couchbase_res;
ctx->rv = return_value;
ctx->cas = cas_token;
retval = lcb_get(couchbase_res->handle, ctx,
nkey, (const lcb_get_cmd_t * const *)commands);
for (ii = 0; ii < nkey; ++ii) {
efree(commands[ii]);
}
efree(commands);
if (LCB_SUCCESS != retval) {
if (couchbase_res->prefix_key_len) {
int i;
for (i = 0; i < nkey; i++) {
efree(keys[i]);
}
}
if (multi) {
efree(keys);
efree(klens);
zval_dtor(return_value);
}
efree(ctx);
couchbase_report_error(INTERNAL_FUNCTION_PARAM_PASSTHRU, oo,
cb_lcb_exception,
"Failed to schedule get request: %s",
lcb_strerror(couchbase_res->handle, retval));
return;
}
couchbase_res->seqno += nkey;
pcbc_start_loop(couchbase_res);
if (LCB_SUCCESS != ctx->res->rc) {
if (LCB_KEY_ENOENT != ctx->res->rc) {
couchbase_report_error(INTERNAL_FUNCTION_PARAM_PASSTHRU, oo,
cb_lcb_exception,
"Failed to get a value from server: %s",
lcb_strerror(couchbase_res->handle,
ctx->res->rc));
}
}
efree(ctx);
if (couchbase_res->prefix_key_len) {
int i;
for (i = 0; i < nkey; i++) {
efree(keys[i]);
}
}
if (multi) {
efree(keys);
efree(klens);
}
}
}
