From 128c6d51ec8c70e230dc86b100cb887ba3f0378d Mon Sep 17 00:00:00 2001 From: Joel Grunbaum <joelgrun@gmail.com> Date: Thu, 20 Jan 2022 07:00:50 +0000 Subject: [PATCH] Secrets to macros and c string concat at compile time --- protocol.cpp | 141 +++++++++++++++++++++++++++++++--------------- 1 files changed, 94 insertions(+), 47 deletions(-) diff --git a/protocol.cpp b/protocol.cpp index 0396038..f3c637a 100644 --- a/protocol.cpp +++ b/protocol.cpp @@ -16,12 +16,15 @@ #include <string> #include <unordered_map> +using namespace std::literals; + namespace protocol { static std::unordered_map<json::MessageTypes, book::ProductTypeEnum> mapAnnounce; -std::string server = std::string(HOST) + ":" + std::string(PORT); -httplib::Client cli("http://" + server); +#define SERVER HOST ":" PORT +httplib::Client cli("http://" SERVER); +std::unique_ptr<easywsclient::WebSocket> ws; double lastime = 0; @@ -39,7 +42,6 @@ // std::ifstream sampleFile("../data.test"); // std::stringstream ss; // ss << sampleFile.rdbuf(); - httplib::Client cli("http://" + server); auto res = cli.Get("/recover"); std::string l; // l = ss.str(); @@ -57,32 +59,44 @@ return bs; } -void catchUp(std::unordered_map<std::string, book::Book>& bs) +void createWebSocket() { - static std::unique_ptr<easywsclient::WebSocket> ws( - easywsclient::WebSocket::from_url("ws://" + server + "/information")); - std::string feed; - bool gotMessage = false; + ws = std::unique_ptr<easywsclient::WebSocket>( + easywsclient::WebSocket::pointer( + easywsclient::WebSocket::from_url("ws://" SERVER "/information"))); ws->poll(); - ws->dispatch([gotMessageOut = &gotMessage, messageOut = &feed, - ws = ws.get()](const std::string& message) { - *gotMessageOut = true; - *messageOut = message; - }); - if (gotMessage) { - std::queue<json::Message*> a(json::parse(feed)); - while (!a.empty()) { - if (static_cast<json::FromExchange*>(a.front()) != nullptr || - static_cast<json::FromExchange*>(a.front())->timestamp > - lastime) { - lastime = - static_cast<json::FromExchange*>(a.front())->timestamp; +} + +std::deque<json::Message*> +catchUp(std::unordered_map<std::string, book::Book>& bs) +{ + std::string feed; + bool gotMessage; + std::deque<json::Message*> out; + do { + gotMessage = false; + ws->poll(); + ws->dispatch([gotMessageOut = &gotMessage, messageOut = &feed, + ws = ws.get()](const std::string& message) { + *gotMessageOut = true; + *messageOut = message; + }); + if (gotMessage) { + std::queue<json::Message*> a(json::parse(feed)); + while (!a.empty()) { + if (static_cast<json::FromExchange*>(a.front()) != nullptr || + static_cast<json::FromExchange*>(a.front())->timestamp > + lastime) { + lastime = + static_cast<json::FromExchange*>(a.front())->timestamp; + } + protocol::handleMessage(bs, a.front()); + out.push_back(a.front()); + a.pop(); } - protocol::handleMessage(bs, a.front()); - delete a.front(); - a.pop(); } - } + } while (gotMessage); + return out; } json::Message* addOrder(json::AddMessage& order) @@ -109,24 +123,24 @@ case json::SPREAD_TYPE: case json::CALL_TYPE: case json::PUT_TYPE: - announce(bs, dynamic_cast<json::AnnounceMessage*>(message)); + announce(bs, static_cast<json::AnnounceMessage*>(message)); break; case json::SETTLEMENT: - settle(bs, dynamic_cast<json::SettleMessage*>(message)); + settle(bs, static_cast<json::SettleMessage*>(message)); break; case json::ADDED: - addedOrder(bs, dynamic_cast<json::AddedMessage*>(message)); + addedOrder(bs, static_cast<json::AddedMessage*>(message)); break; case json::DELETED: - deletedOrder(bs, dynamic_cast<json::DeletedMessage*>(message)); + deletedOrder(bs, static_cast<json::DeletedMessage*>(message)); break; case json::TRADE: - tradeOrder(bs, dynamic_cast<json::TradeMessage*>(message)); + tradeOrder(bs, static_cast<json::TradeMessage*>(message)); break; case json::BROKER_REQUEST: case json::BROKER_ACK: case json::BROKER_CONFIRM: - broker(bs, dynamic_cast<json::Broker*>(message)); + broker(bs, static_cast<json::Broker*>(message)); break; default:; } @@ -162,28 +176,58 @@ json::DeletedMessage* message) { if (message->side == book::Buy) { - bs[message->product].bidSide.erase(message->id); + for (auto i = bs[message->product].bidSide.begin(); + i != bs[message->product].bidSide.end(); i++) { + if (i->id == message->id) { + bs[message->product].bidSide.erase(i); + break; + } + } } else { - bs[message->product].askSide.erase(message->id); + for (auto i = bs[message->product].askSide.begin(); + i != bs[message->product].askSide.end(); i++) { + if (i->id == message->id) { + bs[message->product].askSide.erase(i); + break; + } + } } } void tradeOrder(std::unordered_map<std::string, book::Book>& bs, json::TradeMessage* message) { - if (bs.find(message->passiveOrder) != bs.end()) { - if (message->tradeType == json::BUY_AGGRESSOR) { - if (message->passiveOrderRemaining > 0) { - bs[message->product].askSide.at(message->passiveOrder).volume = - message->passiveOrderRemaining; - } else { - bs[message->product].askSide.erase(message->passiveOrder); + if (message->tradeType == json::BUY_AGGRESSOR) { + if (message->passiveOrderRemaining > 0) { + for (auto& i : bs[message->product].askSide) { + if (i.id == message->passiveOrder) { + i.volume = message->passiveOrderRemaining; + break; + } } - } else if (message->tradeType == json::SELL_AGGRESSOR) { - if (message->passiveOrderRemaining > 0) { - bs[message->product].bidSide.at(message->passiveOrder).volume = - message->passiveOrderRemaining; - } else { - bs[message->product].bidSide.erase(message->passiveOrder); + } else { + for (auto i = bs[message->product].askSide.begin(); + i != bs[message->product].askSide.end(); i++) { + if (i->id == message->passiveOrder) { + bs[message->product].askSide.erase(i); + break; + } + } + } + } else if (message->tradeType == json::SELL_AGGRESSOR) { + if (message->passiveOrderRemaining > 0) { + for (auto& i : bs[message->product].bidSide) { + if (i.id == message->passiveOrder) { + i.volume = message->passiveOrderRemaining; + break; + } + } + } else { + for (auto i = bs[message->product].bidSide.begin(); + i != bs[message->product].bidSide.end(); i++) { + if (i->id == message->passiveOrder) { + bs[message->product].bidSide.erase(i); + break; + } } } } @@ -199,7 +243,10 @@ httplib::MultipartFormDataItems a = {{"message", message, "", ""}, {"username", USER, "", ""}, {"password", PASS, "", ""}}; - auto res = cli.Post("/execution", a); + auto res = + cli.Post("/execution", + "message=" + message + "&username=" USER "&password=" PASS, + "application/x-www-form-urlencoded"); std::string b = res->body; std::queue<json::Message*> c = json::parse(b); return c.front(); -- Gitblit v1.9.3