catalog.java

SSD3 Exercise Four Ex 4

import java.util.*;

public class Catalog implements Iterable<Product>{

	private ArrayList<Product> products;
	
	public Catalog()
	{
		products = new ArrayList<Product>();
	}
	
	public void addProduct(Product product)
	{
		products.add(product);
	}
	
	public Iterator<Product> iterator()
	{
		return products.iterator();
	}
	
	public Product getProduct(String code)
	{
		for(Iterator<Product> itr = products.iterator(); itr.hasNext();)
		{   
			
			Product product = itr.next();   
			
			if(product.getCode().equals(code))   
				return product;   
		}  
		
		return null;
	}
	
}