Utility class getSize method updated

.idea/.name

+OrderManagementForRetailStore
\ No newline at end of file

src/main/java/com/retail_store/order_management/service/RetailStoreService.java

@@ -213,11 +213,11 @@ public class RetailStoreService implements RetailStore, ProductFilter {
 
     @Override
     public synchronized boolean addProductToCart(Collection<OrderedProduct> cart, Product product, Connection connection) {
-        if (product.outOfStock()) {
-            return false;
-        }
         getLogger().info(product.showSizes());
         Size size = getSize(true, "Please Select Size");
+        if (product.outOfStock() || size == null) {
+            return false;
+        }
         getLogger().info("\n-----------------------------\nPlease enter quantity required\n-----------------------------\n");
         int quantity = getInt();
         int availableQuantity = product.getSize().get(size);

src/main/java/com/retail_store/order_management/utils/Utility.java

@@ -52,15 +52,18 @@ public final class Utility {
      * @return
      */
     public static Size getSize(boolean check, String message) {
+        Size sizeValue;
         String categoryOptions = Arrays.stream(Size.values()).filter(size -> !size.equals(Size.EXIT)).map(size -> "(" + size.getOption() + ")" + size).reduce((sizes, size) -> sizes + " " + size).get() + "\nPress 0 to go back.";
         if (check) {
             getLogger().info("\n------------------------\n" + message + "\n------------------------\n" + categoryOptions + "\n------------------------\n");
         }
         int option = getInt();
-        return Arrays.stream(Size.values()).filter(size -> size.getOption() == option).findFirst().orElseGet(() -> {
+        sizeValue = Arrays.stream(Size.values()).filter(size -> size.getOption() == option).findFirst().orElseGet(() -> {
             getLogger().info("\n------------------------\nPlease enter valid category\n------------------------\n" + categoryOptions + "\n------------------------\n");
             return getSize(false, message);
+
         });
+        return sizeValue.equals(Size.EXIT) ? null : sizeValue;
     }
 
     /**