|| || || || || || || ||

---

mainFileClient.cpp

Go to the documentation of this file.

#include "Util/Channel.h"
#include "Basis/CmdOptions.h"
#include "Util/ChannelPool.h"
#include "Basis/Timer.h"
#include "Util/IOBufferFile.h"

namespace Impala
{
namespace Application
{
namespace FileClient
{

using namespace Util;

int      gVerbose;
Channel* gChannel;

void
DoGet(String fileName)
{
    std::cout << "Getting " << fileName << " ... " << std::flush;
    Timer timer(1);
    char* buf = gChannel->Buffer();
    sprintf(buf, "get %s\0", fileName.c_str());
    int len = gChannel->SendRequest(strlen(buf)+1);
    buf[len] = 0;
    if ((len > 5) && (strncmp(buf, "ERROR", 5) == 0))
        return;
    IOBufferFile ioBuf(fileName, false, false);
    ioBuf.Write(buf, len);
    double b = len;
    while (len == Channel::DATA_BUFFER_SIZE)
    {
        sprintf(buf, "getmore\0");
        len = gChannel->SendRequest(strlen(buf)+1);
        buf[len] = 0;
        ioBuf.Write(buf, len);
        b += len;
        std::cout << "#" << std::flush;
    }
    double timeVal = timer.SplitTime();
    double bps = b / (1024.0*1024.0 * timeVal);
    std::cout << " done (in " << timeVal << " sec = " << bps << " Mb/sec)"
              << std::endl;
}

void
DoPut(String fileName)
{
    std::cout << "Putting " << fileName << " ... " << std::flush;
    Timer timer(1);
    char* buf = gChannel->Buffer();
    String dstFileName = FileNameTail(fileName);
    sprintf(buf, "put %s\0", dstFileName.c_str());
    int len = gChannel->SendRequest(strlen(buf)+1);
    buf[len] = 0;
    if ((len > 5) && (strncmp(buf, "ERROR", 5) == 0))
        return;
    IOBufferFile ioBuf(fileName, true, false);
    double b = ioBuf.Size();
    while (ioBuf.Available() > 0)
    {
        sprintf(buf, "putmore\0");
        int nrData = Min<Int64>(Channel::DATA_BUFFER_SIZE-8, ioBuf.Available());
        ioBuf.Read(buf+8, nrData);
        len = gChannel->SendRequest(nrData+8);
        if ((len > 5) && (strncmp(buf, "ERROR", 5) == 0))
            return;
        std::cout << "#" << std::flush;
    }
    double timeVal = timer.SplitTime();
    double bps = b / (1024.0*1024.0 * timeVal);
    std::cout << " done (in " << timeVal << " sec = " << bps << " Mb/sec)"
              << std::endl;
}

int
mainFileClient(int argc, char* argv[])
{
    CmdOptions& options = CmdOptions::GetInstance();
    options.Initialise(false);
    String usageStr = "{put|get} servername port filename(s)\n\n";
    if (! options.ParseArgs(argc, argv, usageStr, 4))
        return 1;

    String cmd = options.GetArg(0);
    String server = options.GetArg(1);
    int port = atol(options.GetArg(2));

    gVerbose = options.GetInt("verb");

    //gChannel = new Channel(server, port, gVerbose);
    gChannel = ChannelPool::Instance().Get(server, port, ".passwords"); 
    if (!gChannel || !gChannel->Valid())
        return 1;

    if (cmd == "get")
    {
        for (int i=3 ; i<options.GetNrArg() ; i++)
            DoGet(options.GetArg(i));
    }
    else if (cmd == "put")
    {
        for (int i=3 ; i<options.GetNrArg() ; i++)
            DoPut(options.GetArg(i));
    }

    // SK: Closing channels is handled by ChannelPool
    //gChannel->CloseChannel();

    return 0;
}

} // namespace FileClient
} // namespace Application
} // namespace Impala

int
main(int argc, char* argv[])
{
    return Impala::Application::FileClient::mainFileClient(argc, argv);
}

---