/************************************************************************
 *
 * Copyright (c) 2016 Alibaba.com, Inc. All Rights Reserved
 * $Id:  echo_purgelog.cc,v 1.0 Jan 17, 2017 3:08:17 PM jarry.zj(jarry.zj@alibaba-inc.com) $
 *
 ************************************************************************/

/**
 * @file echo_purgelog.cc
 * @author jarry.zj(jarry.zj@alibaba-inc.com)
 * @date Jan 17, 2017 3:08:17 PM
 * @version 1.0
 * @brief
 *
 **/


#include <cstdlib>
#include <iostream>
#include <thread>
#include <unistd.h>
#include "paxos.h"
#include "rd_paxos_log.h"
#include "paxos_log.h"
#include "paxos_server.h"
#include "paxos.pb.h"

using namespace alisql;

/*
 * Apply thread: once a log entry is committed, the apply thread will echo the value of the entry.
 * it also can be set to state machine or ack to the client in KV server.
 */
void applyThread(Paxos *paxos)
{
  std::shared_ptr<PaxosLog> log= paxos->getLog();
  uint64_t appliedIndex= 0;
  while (1)
  {
    /*uint64_t commitIndex= */paxos->waitCommitIndexUpdate(appliedIndex);
    uint64_t i= 0;
    for (i= appliedIndex + 1; i <= paxos->getCommitIndex(); ++i)
    {
      LogEntry entry;
      log->getEntry(i, entry);
      if (entry.optype() > 10)
        continue;
      //std::cout<< "LogIndex "<<i <<": key:"<< entry.ikey()<<", value:"<< entry.value()<< std::endl<< std::flush;
      std::cout<< "====> CommittedMsg:"<< entry.value() <<", LogIndex:"<< i<< std::endl<< std::flush;
    }
    appliedIndex= i - 1;
    paxos->updateAppliedIndex(appliedIndex);
  }

  std::cout<< "====> ApplyThread: exit."<<std::endl<< std::flush;
}

bool myPurgeLog(const LogEntry &le)
{
    std::cout << "Purge log "<< le.value() << std::endl;
    return true;
}

int main(int argc, char *argv[])
{
  if (argc < 5)
  {
    std::cerr<< "Usage: ./echo <ip1:port1> <ip2:port2> <ip3:port3> <currentIndex> [<witness_ip:witness_port>]" <<std::endl;
    std::cerr<< "Example: ./echo 127.0.0.1:10001 127.0.0.1:10002 127.0.0.1:10003 1 127.0.0.1:10004" <<std::endl;
    return 1;
  }
  int index= atol(argv[4]);
  std::cout<< "Current Instance IP:PORT " << argv[index] <<std::endl;

  setenv("easy_log_level", "3", 1);
  extern easy_log_level_t easy_log_level;
  easy_log_level= easy_log_level;
  //easy_log_level= EASY_LOG_ERROR;

  /* Server list. */
  std::vector<std::string> strConfig;
  strConfig.emplace_back(argv[1]);
  strConfig.emplace_back(argv[2]);
  strConfig.emplace_back(argv[3]);

  /* You can use the RDPaxosLog (based on RocksDB) by default, you can also implement a new log based on the interface PaxosLog by yourself. */
  std::shared_ptr<PaxosLog> rlog= std::make_shared<RDPaxosLog>(std::string("paxosLogTestDir")+strConfig[index-1], true, 4 * 1024 * 1024);

  /* Init paxos consensus here. */
  Paxos *paxos1= new Paxos(5000, rlog);
  paxos1->init(strConfig, index, new ClientService());

  /* init witness node */
  std::vector<std::string> witnesslist;
  for (int i = 5; i < argc; ++i)
  {
    witnesslist.emplace_back(argv[i]);
  }
  /* add learner after we have a leader */
  while (1) {
      sleep(1);
      if (paxos1->getState() == paxos1->LEADER) {
          paxos1->changeLearners(Paxos::CCAddNode, witnesslist);
      }
      if (paxos1->getCurrentLeader() != 0) {
          /* have leader */
          break;
      }
  }

  paxos1->initAutoPurgeLog(true, true, myPurgeLog);
  /* Start the apply thread. */
  std::thread th1(applyThread, paxos1);

  while (true)
  {
    std::string key= "keykey", value, line;
    std::cin>> value;
    if (paxos1->getState() != Paxos::LEADER)
    {
      std::cerr<< "====> Error: I'm not leader!! Leader is server:"<<paxos1->getCurrentLeader() << std::endl;
      continue;
    }

    //std::cin>> "set">> key>> "=">> value>> std::endl;
    std::cout<< "====> Input value:"<< value<< " ."<< std::endl;

    LogEntry le;
    le.set_index(0);
    le.set_optype(1);
    //le.set_ikey(key);
    le.set_value(value);

    /* Send the log entry by Paxos consensus. */
    paxos1->replicateLog(le);
  }

  th1.join();

  delete paxos1;

  return 0;
}                 //function main