Commit 9599f734 authored by Marshall Pierce's avatar Marshall Pierce
Browse files

Merge pull request #13 from Seagate/features/p2p 

Created p2p copydrive example
parents 776c2ab6 91cb33f3

CMakeLists.txt

+2 −1
Original line number Diff line number Diff line
@@ -18,7 +18,7 @@ else(USE_LOCAL_KINETIC_CLIENT) 
        kinetic_cpp_client

        PREFIX "vendor"

        GIT_REPOSITORY "git@github.com:Seagate/Kinetic-C-Client.git"

        GIT_TAG "183fbf055169b36e48f9fb4695cd6f81ba9aef42"

        GIT_TAG "510bc7dea3ab162abc87ac87c2e933ea0833a7bc"

        BUILD_IN_SOURCE 1

        INSTALL_COMMAND ""

    )
@@ -71,3 +71,4 @@ add_example_target(write_file_nonblocking) 
add_example_target(firmware_update)

add_example_target(write_file_blocking_threads)

add_example_target(dump_keyspace)

add_example_target(copydrive)

src/copydrive.cc

0 → 100644
+94 −0
Original line number Diff line number Diff line 
#include <stdio.h>



#include "kinetic/kinetic.h"



static const int kP2PBatchSize = 300;

using kinetic::KineticConnectionFactory;

using kinetic::Status;

using kinetic::KineticRecord;



using std::shared_ptr;

using std::string;

using std::unique_ptr;

using std::vector;



void dispatch_request(kinetic::BlockingKineticConnection& connection, const kinetic::P2PPushRequest& request) {

    unique_ptr<vector<kinetic::KineticStatus>> statuses(new vector<kinetic::KineticStatus>());

    if (!connection.P2PPush(request, statuses).ok()) {

        printf("Error pushing\n");

        exit(1);

    }



    for (auto it = statuses->begin(); it != statuses->end(); ++it) {

        if (it->ok()) {

            printf(".");

        } else {

            printf("X");

        }

    }

    fflush(stdout);

}



int main(int argc, char* argv[]) {

    if (argc != 5) {

        printf("%s: <source host> <source port> <destination host> <destination port>\n", argv[0]);

        return 1;

    }



    const char* source_host = argv[1];

    int source_port = atoi(argv[2]);

    const char* dest_host = argv[3];

    int dest_port = atoi(argv[4]);



    printf("Copying from %s:%d -> %s:%d\n", source_host, source_port, dest_host, dest_port);





    kinetic::ConnectionOptions options;

    options.host = source_host;

    options.port = source_port;

    options.user_id = 1;

    options.hmac_key = "asdfasdf";



    kinetic::KineticConnectionFactory kinetic_connection_factory = kinetic::NewKineticConnectionFactory();



    unique_ptr<kinetic::ConnectionHandle> connection;

    if (!kinetic_connection_factory.NewConnection(options, 5, connection).ok()) {

        printf("Unable to connect\n");

        return 1;

    }



    kinetic::P2PPushRequest request;

    request.host = dest_host;

    request.port = dest_port;





    // Build a key consisting of "FFFFFF...". In almost all cases this will come after the last

    // key in the drive

    string last_key;

    for (int i = 0; i < 4*1024; i++) {

        last_key += "\xFF";

    }



    // Iterate over all the keys and print them out

    for (kinetic::KeyRangeIterator it = connection->blocking().IterateKeyRange("", true, last_key, true, 100); it != kinetic::KeyRangeEnd(); ++it) {

        kinetic::P2PPushOperation op;

        op.key = *it;

        op.force = true;

        op.newKey = *it;

        request.operations.push_back(op);



        if (request.operations.size() > kP2PBatchSize) {

            dispatch_request(connection->blocking(), request);

            request.operations.clear();

        }

    }



    dispatch_request(connection->blocking(), request);



    printf("\n");







    return 0;

}