From 29bb01fc1d45d6430fcfe0f1bba41ad6976ff694 Mon Sep 17 00:00:00 2001 From: Parambir Singh Date: Mon, 29 May 2017 17:37:08 +1000 Subject: [PATCH] Use Iterator instead of Iterator Use Iterator instead of Iterator for better type safety. Also, remove risky cast that could fail at runtime. --- .../designpatterns/iterator/transition/Waitress.java | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/headfirst/designpatterns/iterator/transition/Waitress.java b/src/headfirst/designpatterns/iterator/transition/Waitress.java index 2d35ddeb..083d43bd 100755 --- a/src/headfirst/designpatterns/iterator/transition/Waitress.java +++ b/src/headfirst/designpatterns/iterator/transition/Waitress.java @@ -11,16 +11,16 @@ public Waitress(ArrayList menus) { } public void printMenu() { - Iterator menuIterator = menus.iterator(); + Iterator menuIterator = menus.iterator(); while(menuIterator.hasNext()) { - Menu menu = (Menu)menuIterator.next(); + Menu menu = menuIterator.next(); printMenu(menu.createIterator()); } } - void printMenu(Iterator iterator) { + void printMenu(Iterator iterator) { while (iterator.hasNext()) { - MenuItem menuItem = (MenuItem)iterator.next(); + MenuItem menuItem = iterator.next(); System.out.print(menuItem.getName() + ", "); System.out.print(menuItem.getPrice() + " -- "); System.out.println(menuItem.getDescription());