implementations.terrastoreDB.java Source code

Introduction

Here is the source code for implementations.terrastoreDB.java

Source

package implementations;
/**
 * Copyright 2011 Thibault Dory
 * Licensed under the GPL Version 3 license
 */

import java.nio.ByteBuffer;
import java.nio.CharBuffer;
import java.nio.charset.Charset;
import java.nio.charset.CharsetDecoder;
import java.nio.charset.CodingErrorAction;
import org.json.JSONException;
import org.json.XML;

import core.BenchDB;
import terrastore.client.TerrastoreClient;
import terrastore.client.connection.resteasy.HTTPConnectionFactory;

/**
 * 
 * @author Thibault Dory
 * This class will not work with the current update function as it add a "1" outside of the JSON document
 */

public class terrastoreDB extends BenchDB {
    TerrastoreClient client;
    final String UTF8 = "UTF8";

    @Override
    public int connectNode(String nodeAddress) {
        int ret;
        try {
            client = new TerrastoreClient("http://" + nodeAddress + ":8000", new HTTPConnectionFactory());
            ret = 1;
        } catch (Exception e) {
            ret = -1;
            e.printStackTrace();
        }
        return ret;
    }

    @Override
    public String readDB(String ID) {
        String xml;
        try {
            xml = client.bucket("test").key(ID).get(String.class);
        } catch (Exception e) {
            xml = null;
            e.printStackTrace();
        }
        return xml;
    }

    @Override
    public int updateDB(String ID, String newValue) {
        int ret;
        org.json.JSONObject myjson = null;
        try {
            myjson = XML.toJSONObject(newValue);
        } catch (JSONException e1) {
            e1.printStackTrace();
        }
        try {
            newValue = myjson.toString();
            CharsetDecoder utf8Decoder = Charset.forName("UTF-8").newDecoder();
            utf8Decoder.onMalformedInput(CodingErrorAction.REPLACE);
            utf8Decoder.onUnmappableCharacter(CodingErrorAction.REPLACE);
            ByteBuffer bytes = ByteBuffer.wrap(newValue.getBytes("UTF8"));
            CharBuffer parsed = utf8Decoder.decode(bytes);
            client.bucket("test").key(ID).put(parsed.toString());
            ret = 1;
        } catch (Exception e) {
            ret = -1;
            e.printStackTrace();
        }
        return ret;
    }

    @Override
    public int writeDB(String ID, String Value) {
        return updateDB(ID, Value);
    }

    @Override
    public void searchDB(String keyword) {
        // TODO Auto-generated method stub
    }

    @Override
    public void close() {

    }

}