main.cpp

/************************************************************************
  > File Name: main.cpp
  > Author: leaker
  > Mail: xiao13149920@foxmail.com
  > Created Time: Sat 21 Nov 2020 04:06:40 PM CST
 ************************************************************************/

#include "annhoo_md_sdk/ah_md_spi.h"
#include "annhoo_md_sdk/ah_md_api.h"
#include "cmdline.h"
#include "ah_md_spi_inl.h"

int main(int argc, char* argv[])
{
    cmdline::parser a;
    a.add<std::string>("config", 'c', "config file", false, "../conf/config_annhoo_md_sdk.yaml");
    a.add<std::string>("path", 'p', "data path", false, "./data");
    a.parse_check(argc, argv);

    // const char* config_file = "../conf/config_quotes.yaml";
    std::string config_file = a.get<std::string>("config");
    std::string data_path = a.get<std::string>("path");
    auto        end = data_path.rend();
    for (auto it = data_path.rbegin(); it != end; ++it) {
        if (*it == '/') {
            data_path.pop_back();
        } else {
            break;
        }
    }

    time_t tt;
    time(&tt);
    // tm* t = gmtime(&tt, );
    struct tm* info = localtime(&tt);
    char       day[11];
    // sprintf(day, "%04d-%02d-%02d", t->tm_year + 1900, t->tm_mon + 1, t->tm_mday);
    strftime(day, sizeof day, "%Y-%m-%d", info);
    std::string today = day;
    std::string index_data = data_path + "/index_" + day + ".csv";
    // std::string entrust   = data_path + "/entrust_" + day + ".csv";
    // std::string trade     = data_path + "/trade_" + day + ".csv";
    std::string       entrust_trade = data_path + "/tick_" + day + ".csv";
    std::string       snap_shot = data_path + "/snapshot_" + day + ".csv";
    annhoo::IAHMdApi* ahMdApi = NewMdApi(config_file.c_str());
    // annhoo::IAHMdSpi* ahMdSpi = new annhoo::AHMdSpiInl("mkt_data.csv", "entrust.csv", "trade.csv", "snap_shot.csv",true);
    //  annhoo::IAHMdSpi* ahMdSpi = new annhoo::AHMdSpiInl(mkt_data, entrust, trade, snap_shot);
    annhoo::IAHMdSpi* ahMdSpi = new annhoo::AHMdSpiInl(index_data, entrust_trade, snap_shot);
    ahMdApi->RegisterSpi(ahMdSpi);
    ahMdApi->SubscribeAll();
    // ahMdApi->Login();
    ahMdApi->Connect(true);
    ahMdApi->Start();
    ahMdApi->Join();

    ahMdApi->UnRegisterSpi(ahMdSpi);
    delete ahMdSpi;
    delete ahMdApi;

    return 0;
}