From 441abf1e6a15a39f1ef7bf5e4ee2a4a43bff327b Mon Sep 17 00:00:00 2001 From: Joel Grunbaum <joelgrun@gmail.com> Date: Thu, 20 Jan 2022 05:18:00 +0000 Subject: [PATCH] Added names and ready notice --- protocol.cpp | 134 ++++++++++++++++++++++++++++++-------------- 1 files changed, 90 insertions(+), 44 deletions(-) diff --git a/protocol.cpp b/protocol.cpp index 0396038..f336597 100644 --- a/protocol.cpp +++ b/protocol.cpp @@ -22,6 +22,7 @@ mapAnnounce; std::string server = std::string(HOST) + ":" + std::string(PORT); httplib::Client cli("http://" + server); +std::unique_ptr<easywsclient::WebSocket> ws; double lastime = 0; @@ -57,32 +58,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 +122,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 +175,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 +242,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