/*
 * blsm_client.cpp
 *
 * Copyright 2009-2012 Yahoo! Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */
#include "MapKeeper.h"
#include <transport/TSocket.h>
#include <transport/TBufferTransports.h>
#include <protocol/TBinaryProtocol.h>
#include <iostream>
using namespace std;

using namespace apache::thrift;
using namespace apache::thrift::protocol;
using namespace apache::thrift::transport;

int main(int argc, char **argv) {
    boost::shared_ptr<TSocket> socket(new TSocket("localhost", 9090));
    boost::shared_ptr<TTransport> transport(new TFramedTransport(socket));
    boost::shared_ptr<TProtocol> protocol(new TBinaryProtocol(transport));

    mapkeeper::MapKeeperClient client(protocol);
    transport->open();
    socket->setNoDelay(true);
    mapkeeper::BinaryResponse getResponse;
    mapkeeper::RecordListResponse scanResponse;
    std::string db = "db";
    std::string db1 = "db1";
    cout << client.addMap(db) << endl;;
    cout << client.insert(db, "kkkkkkkkkkkkk1", "v1") << endl;
    cout << client.insert(db, "kkkkkkkkkkkkk2", "v2") << endl;
    cout << client.insert(db, "kkkkkkkkkkkkk3", "v3") << endl;
    client.get(getResponse, db, "kkkkkkkkkkkkk1");
    cout << getResponse.responseCode << endl;
    cout << getResponse.value << endl;
    client.get(getResponse, db, "kkkkkkkkkkkkk2");
    cout << getResponse.responseCode << endl;
    cout << getResponse.value << endl;
    client.get(getResponse, db, "kkkkkkkkkkkkk3");
    cout << getResponse.responseCode << endl;
    cout << getResponse.value << endl;

    cout << client.update(db, "k0", "v11") << endl;
    cout << client.update(db, "kkkkkkkkkkkkk1", "v11") << endl;
    cout << client.update(db, "kkkkkkkkkkkkk2", "v12") << endl;
    cout << client.update(db, "kkkkkkkkkkkkk3", "v13") << endl;
    client.get(getResponse, db, "kkkkkkkkkkkkk1");
    cout << getResponse.responseCode << endl;
    cout << getResponse.value << endl;
    client.get(getResponse, db, "kkkkkkkkkkkkk2");
    cout << getResponse.responseCode << endl;
    cout << getResponse.value << endl;
    client.get(getResponse, db, "kkkkkkkkkkkkk3");
    cout << getResponse.responseCode << endl;
    cout << getResponse.value << endl;

/*
    client.get(getResponse, "fdsafdasfdasfdsaf", "kkkkkkkkkkkkk3");
    cout << getResponse.responseCode << endl;

    client.get(getResponse, db, "k4");
    cout << getResponse.responseCode << endl;
    */

    cout << "adding db one more time" << endl;
    cout << client.addMap(db) << endl;;

    cout << client.addMap(db1) << endl;;
    cout << client.insert(db1, "new key", "new value") << endl;
    client.get(getResponse, db1, "new key");
    cout << getResponse.responseCode << endl;
    cout << getResponse.value << endl;

    client.get(getResponse, db, "new key");
    cout << getResponse.responseCode << endl;
    cout << getResponse.value << endl;

    cout << client.remove(db1, "new key") << endl;
    client.get(getResponse, db1, "new key");
    cout << getResponse.responseCode << endl;
    cout << client.remove(db1, "new key") << endl;

    client.scan(scanResponse, db, mapkeeper::ScanOrder::Ascending, "", true, "", true, 100, 100);
    std::vector<mapkeeper::Record>::iterator itr;
    for (itr = scanResponse.records.begin(); itr != scanResponse.records.end(); itr++) {
        cout << itr->key << " " << itr->value << endl;
    }
    std::cout << std::endl;


    client.scan(scanResponse, db, mapkeeper::ScanOrder::Ascending, "kkkkkkkkkkkkk1", false, "kkkkkkkkkkkkk3", false, 100, 100);
    for (itr = scanResponse.records.begin(); itr != scanResponse.records.end(); itr++) {
        cout << itr->key << " " << itr->value << endl;
    }
    std::cout << std::endl;

    client.scan(scanResponse, db, mapkeeper::ScanOrder::Ascending, "kkkkkkkkkkkkk1", true, "kkkkkkkkkkkkk3", true, 100, 100);
    for (itr = scanResponse.records.begin(); itr != scanResponse.records.end(); itr++) {
        cout << itr->key << " " << itr->value << endl;
    }
    std::cout << std::endl;

    client.scan(scanResponse, db, mapkeeper::ScanOrder::Ascending, "kkkkkkkkkkkkk2", true, "kkkkkkkkkkkkk2", true, 100, 100);
    for (itr = scanResponse.records.begin(); itr != scanResponse.records.end(); itr++) {
        cout << itr->key << " " << itr->value << endl;
    }
    std::cout << std::endl;

    client.scan(scanResponse, db, mapkeeper::ScanOrder::Ascending, "k", true, "kkkkkkkkkkkkk4", true, 100, 100);
    for (itr = scanResponse.records.begin(); itr != scanResponse.records.end(); itr++) {
        cout << itr->key << " " << itr->value << endl;
    }
    return 0;
}