spymemcached是一个memcached的客户端

http://code.google.com/p/spymemcached/

测试代码如下：

1.SpyMemcachedConstants

import java.util.concurrent.TimeUnit; public interface SpyMemcachedConstants { public static int DEFAULT_TIMEOUT = 5; public static TimeUnit DEFAULT_TIMEUNIT = TimeUnit.SECONDS; }

2.SpyMemcachedServer

public class SpyMemcachedServer { private String ip; private int port; public void setIp(String ip) { this.ip = ip; } public String getIp() { return ip; } public void setPort(int port) { if (port < 0 || port > 65535) { throw new IllegalArgumentException("Port number must be between 0 to 65535"); } this.port = port; } public int getPort() { return port; } public String toString() { return getIp() + ":" + getPort(); } }

3.SpyMemcachedManager

import java.io.IOException; import java.io.OutputStream; import java.net.SocketAddress; import java.util.Collection; import java.util.Iterator; import java.util.List; import java.util.Map; import java.util.Set; import java.util.concurrent.ExecutionException; import java.util.concurrent.Future; import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeoutException; import net.spy.memcached.AddrUtil; import net.spy.memcached.BinaryConnectionFactory; import net.spy.memcached.CASMutation; import net.spy.memcached.CASMutator; import net.spy.memcached.CASValue; import net.spy.memcached.ConnectionObserver; import net.spy.memcached.MemcachedClient; import net.spy.memcached.transcoders.LongTranscoder; import net.spy.memcached.transcoders.SerializingTranscoder; import net.spy.memcached.transcoders.Transcoder; public class SpyMemcachedManager { private List<SpyMemcachedServer> servers; private MemcachedClient memClient; public SpyMemcachedManager(List<SpyMemcachedServer> servers) { this.servers = servers; } public void connect() throws IOException { if (memClient != null) { return; } StringBuffer buf = new StringBuffer(); for (int i = 0; i < servers.size(); i++) { SpyMemcachedServer server = servers.get(i); buf.append(server.toString()).append(" "); } memClient = new MemcachedClient( AddrUtil.getAddresses(buf.toString())); } public void disConnect() { if (memClient == null) { return; } memClient.shutdown(); } public void addObserver(ConnectionObserver obs) { memClient.addObserver(obs); } public void removeObserver(ConnectionObserver obs) { memClient.removeObserver(obs); } //---- Basic Operation Start ----// public boolean set(String key, Object value, int expire) { Future<Boolean> f = memClient.set(key, expire, value); return getBooleanValue(f); } public Object get(String key) { return memClient.get(key); } public Object asyncGet(String key) { Object obj = null; Future<Object> f = memClient.asyncGet(key); try { obj = f.get(SpyMemcachedConstants.DEFAULT_TIMEOUT, SpyMemcachedConstants.DEFAULT_TIMEUNIT); } catch (Exception e) { f.cancel(false); } return obj; } public boolean add(String key, Object value, int expire) { Future<Boolean> f = memClient.add(key, expire, value); return getBooleanValue(f); } public boolean replace(String key, Object value, int expire) { Future<Boolean> f = memClient.replace(key, expire, value); return getBooleanValue(f); } public boolean delete(String key) { Future<Boolean> f = memClient.delete(key); return getBooleanValue(f); } public boolean flush() { Future<Boolean> f = memClient.flush(); return getBooleanValue(f); } public Map<String, Object> getMulti(Collection<String> keys) { return memClient.getBulk(keys); } public Map<String, Object> getMulti(String[] keys) { return memClient.getBulk(keys); } public Map<String, Object> asyncGetMulti(Collection<String> keys) { Map map = null; Future<Map<String, Object>> f = memClient.asyncGetBulk(keys); try { map = f.get(SpyMemcachedConstants.DEFAULT_TIMEOUT, SpyMemcachedConstants.DEFAULT_TIMEUNIT); } catch (Exception e) { f.cancel(false); } return map; } public Map<String, Object> asyncGetMulti(String keys[]) { Map map = null; Future<Map<String, Object>> f = memClient.asyncGetBulk(keys); try { map = f.get(SpyMemcachedConstants.DEFAULT_TIMEOUT, SpyMemcachedConstants.DEFAULT_TIMEUNIT); } catch (Exception e) { f.cancel(false); } return map; } //---- Basic Operation End ----// //---- increment & decrement Start ----// public long increment(String key, int by, long defaultValue, int expire) { return memClient.incr(key, by, defaultValue, expire); } public long increment(String key, int by) { return memClient.incr(key, by); } public long decrement(String key, int by, long defaultValue, int expire) { return memClient.decr(key, by, defaultValue, expire); } public long decrement(String key, int by) { return memClient.decr(key, by); } public long asyncIncrement(String key, int by) { Future<Long> f = memClient.asyncIncr(key, by); return getLongValue(f); } public long asyncDecrement(String key, int by) { Future<Long> f = memClient.asyncDecr(key, by); return getLongValue(f); } // ---- increment & decrement End ----// public void printStats() throws IOException { printStats(null); } public void printStats(OutputStream stream) throws IOException { Map<SocketAddress, Map<String, String>> statMap = memClient.getStats(); if (stream == null) { stream = System.out; } StringBuffer buf = new StringBuffer(); Set<SocketAddress> addrSet = statMap.keySet(); Iterator<SocketAddress> iter = addrSet.iterator(); while (iter.hasNext()) { SocketAddress addr = iter.next(); buf.append(addr.toString() + "/n"); Map<String, String> stat = statMap.get(addr); Set<String> keys = stat.keySet(); Iterator<String> keyIter = keys.iterator(); while (keyIter.hasNext()) { String key = keyIter.next(); String value = stat.get(key); buf.append(" key=" + key + ";value=" + value + "/n"); } buf.append("/n"); } stream.write(buf.toString().getBytes()); stream.flush(); } public Transcoder getTranscoder() { return memClient.getTranscoder(); } private long getLongValue(Future<Long> f) { try { Long l = f.get(SpyMemcachedConstants.DEFAULT_TIMEOUT, SpyMemcachedConstants.DEFAULT_TIMEUNIT); return l.longValue(); } catch (Exception e) { f.cancel(false); } return -1; } private boolean getBooleanValue(Future<Boolean> f) { try { Boolean bool = f.get(SpyMemcachedConstants.DEFAULT_TIMEOUT, SpyMemcachedConstants.DEFAULT_TIMEUNIT); return bool.booleanValue(); } catch (Exception e) { f.cancel(false); return false; } } }

4.SpyMemcachedTest

import java.io.IOException; import java.net.SocketAddress; import java.util.ArrayList; import java.util.Iterator; import java.util.List; import java.util.Map; import java.util.Set; import net.spy.memcached.CASValue; import net.spy.memcached.ConnectionObserver; import net.spy.memcached.transcoders.Transcoder; import edu.tju.mslab.baseframe.util.DebugPrinter; import junit.framework.TestCase; public class SpyMemcachedTest extends TestCase { private SpyMemcachedManager manager; protected void setUp() throws Exception { super.setUp(); String[][] servs = new String[][] { {"localhost", "11211"}, //{"localhost", "11212"} }; List<SpyMemcachedServer> servers = new ArrayList<SpyMemcachedServer>(); for (int i = 0; i < servs.length; i++) { SpyMemcachedServer server = new SpyMemcachedServer(); server.setIp(servs[i][0]); server.setPort(Integer.parseInt(servs[i][1])); servers.add(server); } manager = new SpyMemcachedManager(servers); manager.connect(); addObserver(); } protected void tearDown() throws Exception { super.tearDown(); manager.disConnect(); } public void testSet() { System.out.println("====TestSet===="); for (int i = 0; i < 10; i++) { String key = "key" + i; String value = "value" + i; manager.set(key, value, 20); } } public void testGet() { System.out.println("====TestGet===="); for (int i = 0; i < 10; i++) { String key = "key" + i; Object value = manager.get(key); if (value != null) { System.out.println("From memcached"); System.out.println("key=" + key + ";value=" + value); } else { System.out.println("Not found"); System.out.println("key=" + key + ";value=" + value); } } } public void testAdd() { System.out.println("====TestAdd===="); boolean flag = manager.add("key1", "value1-added", 20); //exist assertEquals(false, flag); flag = manager.add("key100", "value100", 20); // don't exist assertEquals(true, flag); testGet(); } public void testReplace() { System.out.println("====TestReplace===="); boolean flag = manager.replace("key2", "value2-replaced", 20); assertEquals(true, flag); flag = manager.replace("key1000", "value1000", 20); assertEquals(false, flag); testGet(); } public void testDelete() { System.out.println("====TestDelete===="); boolean flag = manager.delete("key3"); assertEquals(true, flag); flag = manager.delete("key1000"); assertEquals(false, flag); testGet(); } public void testAsyncGet() { System.out.println("====TestAsyncGet===="); Object value = manager.asyncGet("key4"); if (value != null) { assertEquals("value4", (String)value); } System.out.println("value=" + value); } public void testGetMulti() { System.out.println("====TestGetMulti===="); List<String> keys = new ArrayList<String>(); String[] strKeys = new String[5]; for (int i = 0; i < 5; i++) { keys.add("key" + i); strKeys[i] = "key" + i; } Map<String, Object> cache = manager.getMulti(keys); printMap(cache); cache = manager.getMulti(keys); printMap(cache); } public void testAsyncGetMulti() { System.out.println("====TestAsyncGetMulti===="); List<String> keys = new ArrayList<String>(); String[] strKeys = new String[5]; for (int i = 0; i < 5; i++) { keys.add("key" + i); strKeys[i] = "key" + i; } Map<String, Object> cache = manager.asyncGetMulti(keys); printMap(cache); cache = manager.asyncGetMulti(keys); printMap(cache); } public void testIncrAndDecr() { System.out.println("====TestIncrAndDecr===="); long l = -1; l = manager.increment("incr", 2, 100, 60); assertEquals(100, l); l = manager.increment("incr", 4); assertEquals(104, l); l = manager.decrement("decr", 4, 100, 60); assertEquals(100, l); l = manager.decrement("decr", 3); assertEquals(97, l); System.out.println("incr=" + manager.get("incr").toString()); System.out.println("decr=" + manager.get("decr").toString()); } public void testAsyncIncrAndDecr() { System.out.println("====TestAsyncIncrAndDecr===="); long l = -1; l = manager.asyncIncrement("incr", 2); assertEquals(106, l); l = manager.asyncDecrement("decr", 4); assertEquals(93, l); System.out.println("aincr=" + manager.get("incr").toString()); System.out.println("adecr=" + manager.get("decr").toString()); } public void testPrintStat() throws IOException { System.out.println("====TestPrintStat===="); manager.printStats(); } private void addObserver() { //System.out.println("====AddObserver===="); ConnectionObserver obs = new ConnectionObserver() { public void connectionEstablished(SocketAddress sa, int reconnectCount) { System.out.println("Established " + sa.toString()); } public void connectionLost(SocketAddress sa) { System.out.println("Lost " + sa.toString()); } }; manager.addObserver(obs); } public void testGetTranscoder() { System.out.println("====TestGetTranscoder===="); Transcoder tran = manager.getTranscoder(); System.out.println(tran.getClass().toString()); } private void printMap(Map map) { StringBuffer temp = new StringBuffer(); Set set = map.keySet(); Iterator iter = set.iterator(); while(iter.hasNext()) { String key = (String)iter.next(); Object value = map.get(key); temp.append("key=" + key + ";value=" + value + "/n"); } System.out.println(temp.toString()); } }