threadsafeiterator.java

来自「这个是perst-269.zip下面的SOURCECODE,和大家分享了。」· Java 代码 · 共 59 行

package org.garret.perst;

import java.util.Iterator;

/**
 * This class is used to make it possible to iterate collection without locking it and 
 * when cokkection can be currently updated by other threads.
 * Please notice that if other threads are deleting elements from the tree, then
 * deleting current elemet of the iterator can still cause the problem.
 * Also this iterator should be used only inside one thread - sharing iterator between multiple
 * threads will not work correctly.
 */
public class ThreadSafeIterator<T> implements Iterator<T> { 
    private IResource   collection;
    private Iterator<T> iterator;
    private T           next;

    public boolean hasNext() { 
        boolean result;
        if (next == null) {
            collection.sharedLock();
            if (iterator.hasNext()) { 
                next = iterator.next();
                result = true;
            } else { 
                result = false; 
            }
            collection.unlock();
        } else { 
            result = true;
        }
        return result;
    }
    
    public T next() { 
        T obj = next;
        if (obj == null) { 
            collection.sharedLock();
            obj = iterator.next();
            collection.unlock();
        } else { 
            next = null;
        }
        return obj;
    }

    public ThreadSafeIterator(IResource collection, Iterator<T> iterator) { 
        this.collection = collection;
        this.iterator = iterator;
    }

    public void remove() {
        collection.exclusiveLock();
        iterator.remove();
        collection.unlock();
        next = null;
    }
}