com.hybhub.util.concurrent.ConcurrentSetCollection Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-batch-redis Show documentation
Spring Batch reader and writer implementations for Redis
The newest version!
package com.hybhub.util.concurrent;


import java.util.Collection;
import java.util.Iterator;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.atomic.AtomicInteger;

public class ConcurrentSetCollection extends AbstractConcurrentSet implements Collection {

	ConcurrentSetCollection(final int capacity) {
		super(capacity);
	}

	@Override
	public int size() {
		return count.get();
	}

	@Override
	public void clear() {
		fullyLock();
		try {
			set.clear();
		}
		finally {
			fullyUnlock();
		}

	}

	@Override
	public boolean isEmpty() {
		return count.get() == 0;
	}

	public boolean offer(final E e) {
		if (e == null) {
			throw new NullPointerException();
		}
		final AtomicInteger count = this.count;
		if (count.get() == capacity) {
			return false;
		}
		int c = -1;
		fullyLock();
		try {
			if (count.get() < capacity && ! set.contains(e)) {
				set.add(e);
				c = count.getAndIncrement();
				if (c + 1 < capacity) {
					notFull.signal();
				}
			}
		} finally {
			fullyUnlock();
		}
		if (c == 0) {
			signalNotEmpty();
		}
		return c >= 0 || set.contains(e);
	}

	@Override
	public boolean add(final E e) {
		if (this.offer(e)) {
			return true;
		}
		else {
			throw new IllegalStateException("Queue full");
		}
	}

	@Override
	public boolean remove(final Object o) {
		if (o == null) {
			return false;
		}
		fullyLock();
		try {
			if(set.remove(o)){
				count.decrementAndGet();
				return true;
			}
			return false;
		} finally {
			fullyUnlock();
		}
	}

	@Override
	public boolean contains(final Object o) {
		if(o == null) {
			return false;
		}
		fullyLock();
		try {
			return set.contains(o);
		}
		finally {
			fullyUnlock();
		}
	}

	@Override
	public boolean containsAll(final Collection c) {
		for (Object o : c){
			if(!this.contains(o)){
				return false;
			}
		}
		return true;
	}

	@Override
	public Iterator iterator() {
		return new CopyOnWriteArrayList<>(set).iterator();
	}

	@Override
	public Object[] toArray() {
		fullyLock();
		try {
			return set.toArray();
		}
		finally {
			fullyUnlock();
		}
	}

	@Override
	public  T[] toArray(final T[] a) {
		fullyLock();
		try {
			return set.toArray(a);
		}
		finally {
			fullyUnlock();
		}
	}

	@Override
	public boolean addAll(final Collection c) {
		throw new UnsupportedOperationException();
	}

	@Override
	public boolean removeAll(final Collection c) {
		throw new UnsupportedOperationException();
	}

	@Override
	public boolean retainAll(final Collection c) {
		throw new UnsupportedOperationException();
	}
}