orderBook app

Author: pran9v

OrderBook/bin/orderbook/Order.class

@@ -34,15 +34,23 @@ public boolean removeOrders(String orderId) {
         return true;
     }
 
-    public void viewOrder() {
+    public void viewOrders() {
         System.out.println("Buy Orders are: ");
-        for(Map.Entry<Double, List<Order>> entry : buyOrders.entrySet()) {
-            System.out.println("Price: "+entry.getKey()+" -> "+entry.getValue());
+        if(!buyOrders.isEmpty()) {
+            for(Map.Entry<Double, List<Order>> entry : buyOrders.entrySet()) {
+                System.out.println("Price: "+entry.getKey()+" -> "+entry.getValue());
+            }
+        } else {
+            System.out.println("No buy orders.");
         }
 
-        System.out.println("Sell Order are: ");
-        for(Map.Entry<Double, List<Order>> entry : sellOrders.entrySet()) {
-            System.out.println("Price: "+entry.getKey()+" -> "+entry.getValue());
+        System.out.println("Sell Orders are: ");
+        if(!sellOrders.isEmpty()) {
+            for(Map.Entry<Double, List<Order>> entry : sellOrders.entrySet()) {
+                System.out.println("Price: "+entry.getKey()+" -> "+entry.getValue());
+            }
+        } else {
+            System.out.println("No sell orders.");
         }
     }
 

OrderBook/bin/orderbook/OrderBook.class

@@ -1,5 +1,63 @@
 package orderbook;
 
+import java.util.Scanner;
+
 public class OrderBookApp {
-    // TODO
+    
+    public static void main(String[] args) {
+
+        OrderBook orderBook = new OrderBook();
+        Scanner sc = new Scanner(System.in);
+
+        while(true) {
+
+            System.out.println("---ORDER BOOK MENU---");
+            System.out.println("1) Add Order");
+            System.out.println("2) View Orders");
+            System.out.println("3) Delete Order");
+            System.out.println("4) Exit");
+
+            int choice = sc.nextInt();
+            String id;
+
+            switch (choice) {
+                case 1:
+                    System.out.print("Enter OrderId: ");
+                    id = sc.next();
+                    System.out.print("Enter Price: ");
+                    double price = sc.nextDouble();
+                    System.out.print("Enter quantity: ");
+                    int quantity = sc.nextInt();
+                    System.out.print("Is it a Buy Order(true/false): ");
+                    boolean isBuyOrder = sc.nextBoolean();
+                    orderBook.addOrder(new Order(id, price, quantity, isBuyOrder));
+                    System.out.println("Order added successfully!");
+                    break;
+                case 2:
+                    orderBook.viewOrders();
+                    break;
+                case 3:
+                    if(orderBook.buyOrders.isEmpty() && orderBook.sellOrders.isEmpty()) {
+                        System.out.println("There are no orders to be removed.");
+                        break;
+                    }
+                    System.out.print("Enter the OrderId to be removed: ");
+                    id = sc.next();
+                    boolean isRemoved = orderBook.removeOrders(id);
+                    if(isRemoved) {
+                        System.out.println("Order Removed Successfully!");
+                    } else {
+                        System.out.println("No such order.");
+                    }
+                    break;
+                case 4: 
+                    System.out.println("exited.");
+                    sc.close();
+                    return;
+
+                default:
+                    System.out.println("Invalid choice, please try again.");
+            }
+        }
+    }
 }