swt23w23/src/main/java/catering/order/CustomOrderRepository.java

package catering.order;

import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.Set;

/**
 * This class is only used for the prototype to avoid using the bloated Salespoint API
 */
@Component
public class CustomOrderRepository {
	private Set<CustomOrder> orders;

	public CustomOrderRepository() {
		this.orders = new HashSet<>();
	}

	public boolean addOrder(CustomOrder order) {
		return this.orders.add(order);
	}

	public boolean removeOrder(int orderID) {
		for (CustomOrder order : orders) {
			if (order.getId() == orderID) {
				return this.orders.remove(order);
			}
		}
		return false;
	}

	public Collection<CustomOrder> getOrders() {
		return new ArrayList<>(this.orders);
	}
}