From 6bdd28a09c589cf631fce948476d48e9375f72a0 Mon Sep 17 00:00:00 2001
From: Joel Grunbaum <joelgrun@gmail.com>
Date: Mon, 10 Jan 2022 04:50:34 +0000
Subject: [PATCH] Click trader working, with exchange communication

---
 click.cpp |  175 ++++++++++++++++++++++++++++++++++++++++-----------------
 1 files changed, 122 insertions(+), 53 deletions(-)

diff --git a/click.cpp b/click.cpp
index 4b43885..749d649 100644
--- a/click.cpp
+++ b/click.cpp
@@ -7,71 +7,140 @@
 #include <iostream>
 #include <unordered_map>
 
-enum clickType { Buy, Sell, Flash };
+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}};
+	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)" << std::endl
-	      << "\t-p price" << std::endl
-	      << "\t-v volume" << std::endl
-	      << "\t-i id" << std::endl
-	      << "Always need product, need side, price and volume for "
-		 "adding/flash, need id for deleting"
-	      << std::endl;
+	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
+			  << "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 {
+		std::cout << static_cast<json::RejectMessage*>(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 {
+        std::cout << static_cast<json::ErrorMessage*>(b)->as_string() << std::endl;
+    }
 }
 
 int main(int argc, char** argv)
 {
-    int c;
-    int index;
-    std::string product, id;
-    double price;
-    clickType click;
-    uint64_t volume;
-    initialise();
-    if (argc == 1) {
-	usage(), exit(0);
-    }
-    while ((c = getopt(argc, argv, "a::t::p::v::i::")) != -1) {
-	switch (c) {
-	case 'a':
-	    product = std::string(optarg);
-	    break;
-	case 's':
-	    click = 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 '?':
-	default:
-	    usage();
-	    exit(0);
+	int c;
+	std::string product, id;
+	double price;
+	clickType click, side;
+	uint64_t volume;
+	initialise();
+	if (argc == 1) {
+		usage(), exit(0);
 	}
-    }
-    switch (click) {
-    case Buy:
-    case Sell:
-    case Flash:;
-    }
+	while ((c = getopt(argc, argv, "a::t::s::p::v::i::")) != -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 '?':
+            std::cout << "*1" << std::endl;
+		default:
+            std::cout << "*2 " << (char) c << std::endl;
+			usage();
+			exit(0);
+		}
+	}
+	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;
+	}
 }

--
Gitblit v1.9.3