/*
******************************************************************************************************
Lists the UDF modules registered with the server.
*
* @param aerospike_obj_p The C client's aerospike object.
* @param error_p The C client's as_error to be set to the encountered error.
* @param array_of_modules_p An array of registered UDF modules
* against the Aerospike DB to be populated by
* this function.
* @param language Optionally filters a subset of modules
* matching the given type.
* @options_p The optional parameters to the
* Aerospike::listRegistered().
*
* @return AEROSPIKE_OK if success. Otherwise AEROSPIKE_x.
******************************************************************************************************
*/
extern as_status
aerospike_list_registered_udf_modules(Aerospike_object* aerospike_obj_p,
as_error* error_p, zval* array_of_modules_p, long language,
zval* options_p)
{
as_udf_files udf_files;
uint32_t init_udf_files = 0;
uint32_t i=0;
as_policy_info info_policy;
if ((language != -1) && ((language & AS_UDF_TYPE) != AS_UDF_TYPE)) {
PHP_EXT_SET_AS_ERR(error_p, AEROSPIKE_ERR_PARAM,
"Invalid Value for language");
DEBUG_PHP_EXT_ERROR("Invalid value for language");
goto exit;
}
set_policy(NULL, NULL, NULL, NULL, &info_policy, NULL, options_p, error_p);
if (AEROSPIKE_OK != (error_p->code)) {
DEBUG_PHP_EXT_DEBUG("Unable to set policy");
goto exit;
}
as_udf_files_init(&udf_files, 0);
init_udf_files = 1;
if (AEROSPIKE_OK != aerospike_udf_list(aerospike_obj_p->as_ref_p->as_p,
error_p, &info_policy, &udf_files)) {
DEBUG_PHP_EXT_ERROR(error_p->message);
goto exit;
}
for (i = 0; i < udf_files.size; i++) {
if ((language != -1) && ((udf_files.entries[i].type) !=
(language - AS_UDF_TYPE))) {
continue;
}
zval* module_p = NULL;
MAKE_STD_ZVAL(module_p);
array_init(module_p);
add_assoc_stringl(module_p, UDF_MODULE_NAME, udf_files.entries[i].name,
strlen(udf_files.entries[i].name), 1);
add_assoc_long(module_p, UDF_MODULE_TYPE,
(udf_files.entries[i].type + AS_UDF_TYPE));
add_next_index_zval(array_of_modules_p, module_p);
}
exit:
if (init_udf_files) {
as_udf_files_destroy(&udf_files);
}
return error_p->code;
}
// ----------------------------------------------------------------------------------
//
// list all udfs
//
// def list_udf(options = {})
//
// params:
// options - ?
//
// ------
// RETURN:
// 1. array of hashes representing each udf
//
// @TODO options policy
//
static VALUE list_udf(int argc, VALUE * argv, VALUE self) {
as_error err;
aerospike * as = rb_aero_CLIENT;
VALUE options;
rb_scan_args(argc, argv, "01", &options);
// default values for optional arguments
if ( NIL_P(options) ) options = rb_hash_new();
as_udf_files files;
as_udf_files_init(&files, 0);
if ( aerospike_udf_list(as, &err, NULL, &files) != AEROSPIKE_OK )
raise_as_error(err);
VALUE udfs = rb_ary_new();
for( int i = 0; i < files.size; i++ ) {
as_udf_file * file = &files.entries[i];
VALUE udf_file = rb_hash_new();
VALUE udf_type;
if ( file->type == 0 ) udf_type = lua_sym;
rb_hash_aset(udf_file, name_sym, rb_str_new2(file->name));
rb_hash_aset(udf_file, udf_type_sym, udf_type);
rb_hash_aset(udf_file, hash_sym, INT2FIX(file->hash));
rb_ary_push(udfs, udf_file);
}
as_udf_files_destroy(&files);
return udfs;
}
/**
*******************************************************************************************************
* Lists the UDF modules registered with the server
*
* @param self AerospikeClient object
* @param args The args is a tuple object containing an argument
* list passed from Python to a C function
* @param kwds Dictionary of keywords
*
* Returns list of modules that are registered with Aerospike DB.
*******************************************************************************************************
*/
PyObject * AerospikeClient_UDF_List(AerospikeClient * self, PyObject *args, PyObject * kwds)
{
// Initialize error
as_error err;
as_error_init(&err);
int init_udf_files = 0;
// Python Function Arguments
PyObject * py_policy = NULL;
as_policy_info info_policy;
as_policy_info *info_policy_p = NULL;
// Python Function Keyword Arguments
static char * kwlist[] = {"policy", NULL};
// Python Function Argument Parsing
if ( PyArg_ParseTupleAndKeywords(args, kwds, "|O:udf_list", kwlist, &py_policy) == false ) {
return NULL;
}
if (!self || !self->as) {
as_error_update(&err, AEROSPIKE_ERR_PARAM, "Invalid aerospike object");
goto CLEANUP;
}
if (!self->is_conn_16) {
as_error_update(&err, AEROSPIKE_ERR_CLUSTER, "No connection to aerospike cluster");
goto CLEANUP;
}
// Convert python object to policy_info
pyobject_to_policy_info( &err, py_policy, &info_policy, &info_policy_p,
&self->as->config.policies.info);
if ( err.code != AEROSPIKE_OK ) {
goto CLEANUP;
}
as_udf_files files;
as_udf_files_init(&files, 0);
init_udf_files = 1;
// Invoke operation
Py_BEGIN_ALLOW_THREADS
aerospike_udf_list(self->as, &err, info_policy_p, &files);
Py_END_ALLOW_THREADS
if ( err.code != AEROSPIKE_OK ) {
as_error_update(&err, err.code, NULL);
goto CLEANUP;
}
// Convert as_udf_files struct into python object
PyObject * py_files;
as_udf_files_to_pyobject(&err, &files, &py_files);
if ( err.code != AEROSPIKE_OK ) {
as_error_update(&err, err.code, NULL);
goto CLEANUP;
}
CLEANUP:
if (init_udf_files) {
as_udf_files_destroy(&files);
}
if ( err.code != AEROSPIKE_OK ) {
PyObject * py_err = NULL;
error_to_pyobject(&err, &py_err);
PyObject *exception_type = raise_exception(&err);
if(PyObject_HasAttrString(exception_type, "module")) {
PyObject_SetAttrString(exception_type, "module", Py_None);
}
if(PyObject_HasAttrString(exception_type, "func")) {
PyObject_SetAttrString(exception_type, "func", Py_None);
}
PyErr_SetObject(exception_type, py_err);
Py_DECREF(py_err);
return NULL;
}
return py_files;
}
