Konto - obsługa błędów za pomocą wyjątków

src/p11_klasy/enkapsulacja/BrakSrodkow.java

+package p11_klasy.enkapsulacja;
+
+public class BrakSrodkow extends Exception {
+
+	public BrakSrodkow(String message) {
+		super(message);
+	}
+
+}

src/p11_klasy/enkapsulacja/Konto.java

@@ -39,14 +39,19 @@ public class Konto {
 	}
 	
 	public void wplata(int kwota) {
-		if(kwota > 0) {
-			this.saldo += kwota;
-		} else {
-			System.err.println("ujemna kwota wpłaty");
+		if(kwota <= 0) {
+			throw new IllegalArgumentException("ujemna kwota wpłaty");
 		}
+		this.saldo += kwota;
 	}
 	
-	public void wyplata(int kwota) {
+	public void wyplata(int kwota) throws BrakSrodkow {
+		if(kwota <= 0) {
+			throw new IllegalArgumentException("ujemna kwota wypłaty");
+		}
+		if(kwota > saldo) {
+			throw new BrakSrodkow("za mało kasy");
+		}
 		this.saldo -= kwota;
 	}
 }

src/p11_klasy/enkapsulacja/Program.java

@@ -35,10 +35,21 @@ public class Program {
 		System.out.println(konto1);
 		konto1.wplata(300);
 		System.out.println(konto1);
+		try {
 			konto1.wyplata(400);
+			System.out.println("nie było błędu wypłaty");
+		} catch (Exception e) {
+			// e.printStackTrace();
+			System.out.println(e);
+		}
 		System.out.println(konto1);
 		
+		try {
 			konto1.wplata(-2000);
+			System.out.println("nie było błędu wpłaty");
+		} catch (Exception e) {
+			System.out.println(e);
+		}
 		System.out.println(konto1);
 	}
 

src/p11_klasy/enkapsulacja/ProgramBankowy.java

@@ -51,8 +51,8 @@ public class ProgramBankowy {
             // to w programie możemy te sytuacje rozróżnić i obsłużyć w róznych catchach
             } catch(IllegalArgumentException e) {
                 System.out.println("Niepoprawny argument: " + e.getMessage());
-//            } catch(BrakSrodkow e) {
-//                System.out.println(e.getMessage());
+            } catch(BrakSrodkow e) {
+                System.out.println(e.getMessage());
             } catch(Exception e) {
                 System.out.println("Inny błąd: " + e);
             }