Update atm.java

aditya0608 · web-flow · commit 84b66af9ad1d · 2023-06-17T20:46:57.000+05:30
Added validation checks for balance less than or equal to zero
diff --git a/ATM interface/atm.java b/ATM interface/atm.java
@@ -4,6 +4,10 @@
 //create ATMExample class to implement the ATM functionality  
 public class atm 
 {  
+  public static boolean validateInputAmount(Integer amount)
+  {
+    return amount>0;
+  }
     //main method starts   
     public static void main(String args[] )  
     {  
@@ -30,7 +34,12 @@ public static void main(String args[] )
         System.out.print("Enter money to be withdrawn:");  
                       
         //get the withdrawl money from user  
-        withdraw = sc.nextInt();  
+        withdraw = sc.nextInt(); 
+                if(!validateInputAmount(withdraw))
+                {
+                  System.out.println("Enter amount greater than zero");
+                  break;
+                }
                       
         //check whether the balance is greater than or equal to the withdrawal amount  
         if(balance >= withdraw)  
@@ -53,6 +62,11 @@ public static void main(String args[] )
                       
         //get deposite amount from te user  
         deposit = sc.nextInt();  
+                 if(!validateInputAmount(deposit))
+                {
+                  System.out.println("Enter amount greater than zero");
+                  break;
+                }
                       
         //add the deposit amount to the total balanace  
         balance = balance + deposit;  
