#include "book.hpp"
|
#include "json.hpp"
|
#include "protocol.hpp"
|
|
#include <cstdint>
|
#include <cstdlib>
|
#include <getopt.h>
|
#include <iostream>
|
#include <string>
|
#include <unordered_map>
|
|
enum clickType { Buy, Sell, Flash, Delete };
|
|
static std::unordered_map<std::string, clickType> mapClick;
|
|
void initialise()
|
{
|
mapClick = {{"BUY", Buy}, {"SELL", Sell}, {"FLASH", Flash},
|
{"b", Buy}, {"s", Sell}, {"f", Flash},
|
{"B", Buy}, {"S", Sell}, {"F", Flash},
|
{"DELETE", Delete}, {"D", Delete}, {"d", Delete}};
|
}
|
|
void usage()
|
{
|
std::cout << "DESCRIPTION" << std::endl
|
<< "Click trader using same algs" << std::endl
|
<< std::endl
|
<< "USAGE" << std::endl
|
<< "\t-a product" << std::endl
|
<< "\t-t click type (buy, sell, flash, delete)" << std::endl
|
<< "\t-p price" << std::endl
|
<< "\t-v volume" << std::endl
|
<< "\t-i id" << std::endl
|
<< "\t-l loop count (default to 1)" << std::endl
|
<< "Always need product, need side, price and volume for "
|
"adding/flash, need id for deleting"
|
<< std::endl;
|
}
|
|
void buy(std::string& product, double price, uint64_t volume)
|
{
|
json::AddMessage a(json::ADD, product, price, book::Buy, volume);
|
json::AddedMessage* b =
|
static_cast<json::AddedMessage*>(protocol::addOrder(a));
|
std::cout << b->as_string() << std::endl;
|
}
|
|
void sell(std::string& product, double price, uint64_t volume)
|
{
|
json::AddMessage a(json::ADD, product, price, book::Sell, volume);
|
json::AddedMessage* b =
|
static_cast<json::AddedMessage*>(protocol::addOrder(a));
|
std::cout << b->as_string() << std::endl;
|
delete b;
|
}
|
|
void flash(std::string& product, double price, uint64_t volume,
|
book::OrderSideEnum side)
|
{
|
json::AddMessage a(json::ADD, product, price, side, volume);
|
json::Message* b = static_cast<json::Message*>(protocol::addOrder(a));
|
if (b->type == json::ERROR) {
|
std::cout << static_cast<json::ErrorMessage*>(b)->as_string()
|
<< std::endl;
|
return;
|
}
|
json::DeleteMessage c(json::DELETE, product,
|
static_cast<json::AddedMessage*>(b)->id);
|
json::Message* d = protocol::deleteOrder(c);
|
std::cout << static_cast<json::AddedMessage*>(b)->as_string() << std::endl;
|
if (d->type == json::DELETED) {
|
std::cout << static_cast<json::DeletedMessage*>(d)->as_string()
|
<< std::endl;
|
} else if (d->type == json::REJECT) {
|
std::cout << static_cast<json::RejectMessage*>(d)->as_string()
|
<< std::endl;
|
} else if (d->type == json::ERROR) {
|
std::cout << static_cast<json::ErrorMessage*>(d)->as_string() << std::endl;
|
}
|
delete b;
|
delete d;
|
}
|
|
void deleteOrder(std::string& product, std::string& id)
|
{
|
json::DeleteMessage a(json::DELETE, product, id);
|
json::Message* b = protocol::deleteOrder(a);
|
if (b->type == json::DELETED) {
|
std::cout << static_cast<json::DeletedMessage*>(b)->as_string()
|
<< std::endl;
|
} else if (b->type == json::REJECT) {
|
std::cout << static_cast<json::RejectMessage*>(b)->as_string()
|
<< std::endl;
|
} else if (b->type == json::ERROR) {
|
std::cout << static_cast<json::ErrorMessage*>(b)->as_string()
|
<< std::endl;
|
}
|
}
|
|
int main(int argc, char** argv)
|
{
|
int c;
|
std::string product, id;
|
double price;
|
clickType click, side;
|
uint64_t volume;
|
uint64_t times = 1;
|
initialise();
|
if (argc == 1) {
|
usage(), exit(0);
|
}
|
while ((c = getopt(argc, argv, "a::t::s::p::v::i::l::")) != -1) {
|
switch (c) {
|
case 'a':
|
product = std::string(optarg);
|
break;
|
case 't':
|
click = mapClick[optarg];
|
break;
|
case 's':
|
side = mapClick[optarg];
|
break;
|
case 'p':
|
price = std::stod(optarg);
|
break;
|
case 'v':
|
volume = std::stoll(optarg);
|
break;
|
case 'i':
|
id = std::string(optarg);
|
break;
|
case 'l':
|
times = std::stoll(optarg);
|
break;
|
case '?':
|
std::cout << "*1" << std::endl;
|
default:
|
std::cout << "*2 " << (char)c << std::endl;
|
usage();
|
exit(0);
|
}
|
}
|
for (std::uint64_t i = 0; i < times; i++)
|
{
|
switch (click) {
|
case Buy:
|
buy(product, price, volume);
|
break;
|
case Sell:
|
sell(product, price, volume);
|
break;
|
case Flash:
|
if (side == clickType::Buy)
|
flash(product, price, volume, book::Buy);
|
else
|
flash(product, price, volume, book::Sell);
|
break;
|
case Delete:
|
deleteOrder(product, id);
|
break;
|
}
|
}
|
}
|
