From 0bef191acd5a77544852e6a3daae2df05bd34a31 Mon Sep 17 00:00:00 2001
From: Joel Grunbaum <joelgrun@gmail.com>
Date: Wed, 19 Jan 2022 11:20:45 +0000
Subject: [PATCH] Reformat and change send to x-www-form-urlencoded

---
 click.cpp |  100 +++++++++++++++++++++++++++++--------------------
 1 files changed, 59 insertions(+), 41 deletions(-)

diff --git a/click.cpp b/click.cpp
index 749d649..ba85093 100644
--- a/click.cpp
+++ b/click.cpp
@@ -2,9 +2,11 @@
 #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 };
@@ -13,10 +15,10 @@
 
 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}};
+	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()
@@ -30,6 +32,7 @@
 			  << "\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;
@@ -38,34 +41,42 @@
 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));
+	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));
+	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)
+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);
+	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 {
+	} 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;
@@ -74,17 +85,18 @@
 
 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) {
+	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;
-    }
+	} else if (b->type == json::ERROR) {
+		std::cout << static_cast<json::ErrorMessage*>(b)->as_string()
+				  << std::endl;
+	}
 }
 
 int main(int argc, char** argv)
@@ -94,11 +106,12 @@
 	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::")) != -1) {
+	while ((c = getopt(argc, argv, "a::t::s::p::v::i::l::")) != -1) {
 		switch (c) {
 		case 'a':
 			product = std::string(optarg);
@@ -106,9 +119,9 @@
 		case 't':
 			click = mapClick[optarg];
 			break;
-        case 's':
-            side = mapClick[optarg];
-            break;
+		case 's':
+			side = mapClick[optarg];
+			break;
 		case 'p':
 			price = std::stod(optarg);
 			break;
@@ -118,29 +131,34 @@
 		case 'i':
 			id = std::string(optarg);
 			break;
+		case 'l':
+			times = std::stoll(optarg);
+			break;
 		case '?':
-            std::cout << "*1" << std::endl;
+			std::cout << "*1" << std::endl;
 		default:
-            std::cout << "*2 " << (char) c << std::endl;
+			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;
+	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;
+		}
 	}
 }

--
Gitblit v1.9.3