Aerospike.cpp

Go to the documentation of this file.

#ifdef _USE_AEROSPIKE_DB_


#include "Aerospike.hpp"

#include <string.h>

HCE::AerospikeDB::AerospikeDB() {
        //citrusleaf_init ();
        _clc = citrusleaf_cluster_create();
        if (!_clc) {
                throw Poco::Exception("clc is not created");
        }
}

HCE::AerospikeDB::~AerospikeDB() {
        if (_clc) {
            citrusleaf_cluster_destroy(_clc);
        }
        //citrusleaf_shutdown();
}

void HCE::AerospikeDB::setDriverUrl( const std::string & dbUrl) {
        Poco::URI url(dbUrl);
        std::string host = url.getHost();
        unsigned short port = url.getPort();
        citrusleaf_cluster_add_host(_clc, host.c_str(), port, TIMEOUT);
}

void HCE::AerospikeDB::put(const std::string& key, const std::string& _value) {
        HCE::BaseDB::_error = ACTION_SUCCESS;
        cl_object key_obj;
        citrusleaf_object_init_str(&key_obj, key.c_str());
        cl_bin value;
        strcpy( value.bin_name, docs[DEFAULT_DOCUMENT].c_str());
        citrusleaf_object_init_str(&value.object, _value.c_str());
        int return_value = citrusleaf_put(_clc, _dbName.c_str(), sets[DEFAULT_SET].c_str(), &key_obj, &value, 1, NULL);
        if( return_value != CITRUSLEAF_OK ) {
                std::cerr << "Failure setting values " << return_value << std::endl;
                _error = ACTION_GENERAL_FAILURE;
        }
}

std::string HCE::AerospikeDB::get(const std::string& key) {
    HCE::BaseDB::_error = DOCUMENT_NOT_PRESENT;
    cl_object key_obj;
    citrusleaf_object_init_str(&key_obj, key.c_str());
    cl_bin *cl_v = NULL;
    uint32_t generation;
    int cl_v_len;
    int rv = citrusleaf_get_all(_clc, _dbName.c_str(), sets[DEFAULT_SET].c_str(), &key_obj, &cl_v, &cl_v_len, TIMEOUT, &generation);
    if (rv != CITRUSLEAF_OK) {
        std::cerr << "get failed - " << rv << std::endl;
        if (cl_v) free(cl_v);
        return std::string ("");
    }
    std::string s;
    for (int i=0;i< cl_v_len;i++) {
        #ifdef _DEBUG_
                fprintf(stderr, "%d:  bin %s ",i,cl_v[i].bin_name);
        #endif
        switch (cl_v[i].object.type) {
        case CL_STR:
                        #ifdef _DEBUG_
                        fprintf(stderr, "type string: value %s\n", cl_v[i].object.u.str);
                #endif
                s = cl_v[i].object.u.str;
                _error = DOCUMENT_PRESENT;
                break;
        case CL_INT:
                #ifdef _DEBUG_
                        fprintf(stderr, "type int: value %ld64\n",cl_v[i].object.u.i64);
                #endif
                break;
        default:
                #ifdef _DEBUG_
                        fprintf(stderr, "type unknown! (%d)\n",(int)cl_v[i].object.type);
                #endif
                break;
        }
    }
    for (int i = 0; i < cl_v_len; i++) {
        citrusleaf_object_free(&cl_v[i].object);
    }
    if (cl_v) free(cl_v);
    return s;
}

std::vector<std::string> HCE::AerospikeDB::get_all() {
        std::vector<std::string> data;
        return data;
}

void HCE::AerospikeDB::del(const std::string& key) {
        BaseDB::_error = ACTION_SUCCESS;
        cl_object key_obj;
        citrusleaf_object_init_str(&key_obj, key.c_str());
        int return_value = citrusleaf_delete(_clc, _dbName.c_str(), sets[DEFAULT_SET].c_str(), &key_obj, NULL);
        if( return_value != CITRUSLEAF_OK ) {
                _error = ACTION_GENERAL_FAILURE;
        }
}

#endif