#include <cstdio>
#include <iostream>
#include "HEpoll.h"
#include "AliasRef.h"

/**
 * 测试Client端功能
 */

#define CHKRET(r) if(r){printf(#r "=%d\n", r); return r;}
#define ELSEPRINT(tip, ...) else {printf("F: " tip "\n", ##__VA_ARGS__);}

atomic_int g_send_count(0);
atomic_int g_recv_count(0);
atomic_ullong g_send_bytes(0);
atomic_ullong g_recv_bytes(0);
chrono::steady_clock::time_point g_send_tbeg1;
chrono::steady_clock::time_point g_send_tbeg2;
chrono::steady_clock::time_point g_recv_tend;

int toTimeT(chrono::steady_clock::time_point tp)
{
    return tp.time_since_epoch().count();
}

void OnRespond(const string& msg, weak_ptr<ConnectMeta> wconnMeta)
{
    g_recv_bytes += msg.size();
    g_recv_count += 1;
    g_recv_tend = chrono::steady_clock::now();

    printf("recv=%llu bytes %d rows, time=%f(%d->%d->%d)\n", g_recv_bytes.load(), g_recv_count.load(),
        chrono::duration_cast<chrono::milliseconds>(g_recv_tend-g_send_tbeg1).count()/1000.0, 
        toTimeT(g_send_tbeg1), toTimeT(g_send_tbeg2), toTimeT(g_recv_tend));
}

int main(int argc, char** argv)
{
    if (argc < 3)
    {
        CHKRET(argc);
    }

    string svrHost(argv[1]);
    int svrPort = stoi(argv[2]);

    HEpoll hep;
    int ret = hep.init();
    CHKRET(ret);
    hep.setAliasNameInstance(make_shared<AliasRef<ConnectMetaSPtr>>());

    hep.run(true);

    string name0("cli-");
    string input;
    static int cIdx = 0;
    while (cin >> input, input != "q")
    {
        if ('c' == input[0])
        {
            int cCnt = input.size() > 1 ? stoi(input.c_str() +1) : 1;
            for (int i = 0; i < cCnt; i++)
            {
                auto conn = hep.addTcpConnect(svrHost, svrPort, 1, false);
                if (conn)
                {
                    hep.addAliasName(name0 + to_string(++cIdx), conn);
                    //int rSndBuf = conn->setSndBuffSize(10);
                    //int rRcvBuf = 0; //conn->setRcvBuffSize(10);
                    //printf("connect sndBuffSize %d rcvBuffSize=%d\n", rSndBuf, rRcvBuf);
                    conn->onMessageReachFunc = OnRespond;
                    //conn->sendMsg("begin test\n");
                    conn->start();
                }
                ELSEPRINT("addTcpConnect fail");
            }
            
        }
        else if ('s' == input[0])
        {
            int sCnt = stoi(input.c_str() + 1);
            if (sCnt > 0)
            {
                string datBuf(sCnt-1, '1');
                datBuf.append("2\n");
                g_send_tbeg1 = chrono::steady_clock::now();
                for (int i = 1; i <= cIdx; i++)
                {
                    string aliName = name0 + to_string(i);
                    auto conn = hep.find(aliName);
                    if (conn)
                    {
                        int muti = 3;
                        for (int i = 0; i < muti; i++)
                        {
                            if (0 == conn->sendMsg(datBuf))
                            {
                                g_send_count += 1;
                                g_send_bytes += datBuf.size() - 1;
                            }
                            ELSEPRINT("XXX sendMsg fail");
                        }
                        
                    }
                    ELSEPRINT("hep.find(%s) fail", aliName.c_str());
                }
            }

            g_send_tbeg2 = chrono::steady_clock::now();
            printf("sending=%llu bytes %d rows\n", g_send_bytes.load(), g_send_count.load());
        }
        else if ("r" == input)
        {
            g_send_count = 0;
            g_recv_count = 0;
            g_send_bytes = 0;
            g_recv_bytes = 0;
        }
        else if ('p' == input[0])
        {
            printf("hep status=%s\n", hep.toString().c_str());
            printf("send=%llu@%d  recv=%llu@%d\n", g_send_bytes.load(), g_send_count.load(),
                g_recv_bytes.load(), g_recv_count.load());
            printf("send_t1=%d send_t2=%d recv=%d %fsec\n", toTimeT(g_send_tbeg1), toTimeT(g_send_tbeg2), 
                toTimeT(g_recv_tend), chrono::duration_cast<chrono::milliseconds>(g_recv_tend-g_send_tbeg1).count()/1000.0);
            
            if (input[1] > 0)
            {
                string aliName = name0 + input.substr(1);
                auto connMeta = hep.find(aliName);
                if (connMeta)
                {
                    printf("%s : %s\n", aliName.c_str(), connMeta->toString().c_str());
                }
            }
        }
        

        //
        input.clear();
    }

    printf("notify Exit");
    hep.notifyExit();
    hep.unInit();
    return 0;
}