Muutettu HashMapit

Verkkokauppa.java

@@ -2,28 +2,28 @@ import java.util.ArrayList;
 import java.util.HashMap;
 
 public class Verkkokauppa {
-    private ArrayList<Asiakas> asiakkaat;
-    private ArrayList<Tuote> tuotteet;
-    private ArrayList<Myyja> myyjat;
+    private HashMap<String, Asiakas> asiakkaat;
+    private HashMap<String, Tuote> tuotteet;
+    private HashMap<String, Myyja> myyjat;
     private ArrayList<Ostotapahtuma> tapahtumat;
 
     public Verkkokauppa() {
-        asiakkaat = new ArrayList<>();
-        tuotteet = new ArrayList<>();
-        myyjat = new ArrayList<>();
+        asiakkaat = new HashMap<>();
+        tuotteet = new HashMap<>();
+        myyjat = new HashMap<>();
         tapahtumat = new ArrayList<>();
     }
 
     public void lisaaAsiakas(Asiakas asiakas) {
-        asiakkaat.add(asiakas);
+        asiakkaat.put(asiakas.getAsiakasNumero(), asiakas);
     }
 
     public void lisaaTuote(Tuote tuote) {
-        tuotteet.add(tuote);
+        tuotteet.put(tuote.getNimi(), tuote);
     }
 
     public void lisaaMyyja(Myyja myyja) {
-        myyjat.add(myyja);
+        myyjat.put(myyja.getTunniste(), myyja);
     }
 
     public void lisaaTapahtuma(Ostotapahtuma tapahtuma) {
@@ -37,8 +37,8 @@ public class Verkkokauppa {
      * @return true, jos asiakas löytyi ja poistettiin, muuten false
      */
     public boolean poistaAsiakas(Asiakas asiakas) {
-        if (asiakkaat.contains(asiakas)) {
-            asiakkaat.remove(asiakas);
+        if (asiakkaat.containsValue(asiakas)) {
+            asiakkaat.remove(asiakas.getAsiakasNumero(), asiakas);
             return true;
         }
         return false;
@@ -51,8 +51,8 @@ public class Verkkokauppa {
      * @return true, jos tuote löytyi ja poistettiin, muuten false
      */
     public boolean poistaTuote(Tuote tuote) {
-        if (tuotteet.contains(tuote)) {
-            tuotteet.remove(tuote);
+        if (tuotteet.containsValue(tuote)) {
+            tuotteet.remove(tuote.getNimi(), tuote);
             return true;
         }
         return false;
@@ -65,8 +65,8 @@ public class Verkkokauppa {
      * @return true, jos myyjä löytyi ja poistettiin, muuten false
      */
     public boolean poistaMyyja(Myyja myyja) {
-        if (myyjat.contains(myyja)) {
-            myyjat.remove(myyja);
+        if (myyjat.containsValue(myyja)) {
+            myyjat.remove(myyja.getTunniste(), myyja);
             return true;
         }
         return false;
@@ -94,7 +94,7 @@ public class Verkkokauppa {
      * @return pyydetyn asiakkaan tai null, jos asiakasta ei löydy
      */
     public Asiakas annaAsiakas(String asiakasnumero) {
-        for (Asiakas asiakas : asiakkaat) {
+        for (Asiakas asiakas : asiakkaat.values()) {
             if (asiakas.getAsiakasNumero().equals(asiakasnumero)) {
                 return asiakas;
             }
@@ -110,7 +110,7 @@ public class Verkkokauppa {
      * @return pyydetyn tuotteen tai null, jos tuotetta ei löydy
      */
     public Tuote annaTuote(String nimi) {
-        for (Tuote tuote : tuotteet) {
+        for (Tuote tuote : tuotteet.values()) {
             if (tuote.getNimi().equals(nimi)) {
                 return tuote;
             }
@@ -126,7 +126,7 @@ public class Verkkokauppa {
      * @return pyydetyn myyjän tai null, jos myyjää ei löydy
      */
     public Myyja annaMyyja(String tunniste) {
-        for (Myyja myyja : myyjat) {
+        for (Myyja myyja : myyjat.values()) {
             if (myyja.getTunniste().equals(tunniste)) {
                 return myyja;
             }
@@ -160,7 +160,7 @@ public class Verkkokauppa {
         // \t on tabulaattori, joka tasaa tulosteen
         // sarakkeisiin; \n tekee rivinvaihdon
         StringBuilder s = new StringBuilder("Asiakasnumero\tNimi\t\t\tOstoja\tKanta-asiakas\n");
-        for (Asiakas asiakas : asiakkaat) {
+        for (Asiakas asiakas : asiakkaat.values()) {
             s.append(asiakas.getAsiakasNumero() + "\t\t\t");
             s.append(asiakas.getNimi() + "\t");
             s.append(asiakas.getOstojaTehty() + "\t\t");
@@ -183,7 +183,7 @@ public class Verkkokauppa {
         // \t on tabulaattori, joka tasaa tulosteen
         // sarakkeisiin; \n tekee rivinvaihdon
         StringBuilder s = new StringBuilder("Tunniste\tNimi\t\t\tProvisiot\n");
-        for (Myyja myyja : myyjat) {
+        for (Myyja myyja : myyjat.values()) {
             s.append(myyja.getTunniste() + "\t\t");
             s.append(myyja.getNimi() + "\t\t");
             s.append(myyja.getProvisiot() + "\n");
@@ -201,7 +201,7 @@ public class Verkkokauppa {
         // \t on tabulaattori, joka tasaa tulosteen
         // sarakkeisiin; \n tekee rivinvaihdon
         StringBuilder s = new StringBuilder("Nimi\tSaldo\t\t\tHinta\tVirtuaalinen\n");
-        for (Tuote tuote : tuotteet) {
+        for (Tuote tuote : tuotteet.values()) {
             s.append(tuote.getNimi() + "\t");
             s.append(tuote.getSaldo() + "\t\t");
             s.append(tuote.getHinta() + "\t");