net.spy.memcached.protocol.ProxyCallback Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spymemcached Show documentation

A simple, asynchronous, single-threaded Memcached client written in java.

The newest version!

/**
 *
 */
package net.spy.memcached.protocol;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

import net.spy.memcached.ops.GetOperation;
import net.spy.memcached.ops.OperationStatus;

/**
 * Proxy callback used for dispatching callbacks over optimized gets.
 */
public class ProxyCallback implements GetOperation.Callback {

	private final Map> callbacks=
		new HashMap>();
	private final Collection allCallbacks=
		new ArrayList();

	public void addCallbacks(GetOperation o) {
		GetOperation.Callback c=new GetCallbackWrapper(o.getKeys().size(),
				(GetOperation.Callback)o.getCallback());
		allCallbacks.add(c);
		for(String s : o.getKeys()) {
			Collection cbs=callbacks.get(s);
			if(cbs == null) {
				cbs=new ArrayList();
				callbacks.put(s, cbs);
			}
			cbs.add(c);
		}
	}

	public void gotData(String key, int flags, byte[] data) {
		Collection cbs=callbacks.get(key);
		assert cbs != null : "No callbacks for key " + key;
		for(GetOperation.Callback c : cbs) {
			c.gotData(key, flags, data);
		}
	}

	public void receivedStatus(OperationStatus status) {
		for(GetOperation.Callback c : allCallbacks) {
			c.receivedStatus(status);
		}
	}

	public void complete() {
		for(GetOperation.Callback c : allCallbacks) {
			c.complete();
		}
	}

	public int numKeys() {
		return callbacks.size();
	}

	public int numCallbacks() {
		return allCallbacks.size();
	}
}