/*
* *************************************************************************************
* Copyright (C) 2008 EsperTech, Inc. All rights reserved. *
* http://esper.codehaus.org *
* http://www.espertech.com *
* ---------------------------------------------------------------------------------- *
* The software in this package is published under the terms of the GPL license *
* a copy of which has been included with this distribution in the license.txt file. *
* *************************************************************************************
*/
package com.espertech.esper.collection;
import com.espertech.esper.client.EventBean;
import java.util.Collection;
import java.util.Iterator;
import java.util.NoSuchElementException;
public abstract class MixedEventBeanAndCollectionIteratorBase implements Iterator<EventBean> {
private final Iterator keyIterator;
private Iterator<EventBean> currentListIterator;
private EventBean currentItem;
protected abstract Object getValue(Object iteratorKeyValue);
protected MixedEventBeanAndCollectionIteratorBase(Iterator keyIterator) {
this.keyIterator = keyIterator;
}
protected void init() {
if (keyIterator.hasNext()) {
goToNext();
}
}
public final EventBean next()
{
if (currentListIterator == null && currentItem == null) {
throw new NoSuchElementException();
}
EventBean eventBean;
if (currentListIterator != null) {
eventBean = currentListIterator.next();
if (!currentListIterator.hasNext()) {
currentListIterator = null;
currentItem = null;
if (keyIterator.hasNext())
{
goToNext();
}
}
}
else {
eventBean = currentItem;
currentItem = null;
if (keyIterator.hasNext()) {
goToNext();
}
}
return eventBean;
}
public final boolean hasNext()
{
if (currentListIterator == null && currentItem == null) {
return false;
}
if (currentItem != null) {
return true;
}
if (currentListIterator.hasNext()) {
return true;
}
currentListIterator = null;
currentItem = null;
return keyIterator.hasNext();
}
public final void remove()
{
throw new UnsupportedOperationException();
}
private void goToNext() {
Object nextKey = keyIterator.next();
Object entry = getValue(nextKey);
while(true) {
if (entry instanceof Collection) {
currentListIterator = ((Collection<EventBean>) entry).iterator();
if (currentListIterator.hasNext()) {
break;
}
else {
currentListIterator = null;
}
}
else if (entry instanceof EventBean) {
currentItem = (EventBean) entry;
break;
}
// next key
if (keyIterator.hasNext()) {
nextKey = keyIterator.next();
entry = getValue(nextKey);
}
else {
break;
}
}
}
}