您的位置:首页 > 数据库 > Memcache

集群环境中利用Memcached CAS原子操作计数

2013-10-16 17:55 525 查看
利用Memcached CAS原子操作,确保集群/多线程环境汇总累加操作是有效的,而不会覆盖其他线程的累加结果。

package org.pile.memcached;

import java.io.IOException;
import java.util.Random;
import java.util.concurrent.atomic.AtomicInteger;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.pile.util.LogWriter;

import net.spy.memcached.CASValue;

public class CASLockServlet extends HttpServlet {
//private static final int[] LEVEL = {10,50,200,500,1000,5000,10000,20000};
private static final String CACHE_KEY = "CASLOCK";
private volatile static AtomicInteger visitTimes = new AtomicInteger();
//private Random		  random     = new Random();

public void doGet(HttpServletRequest request, HttpServletResponse response)
throws ServletException, IOException {
String responseBody = null;
if(null != request.getParameter("reset")){
visitTimes = new AtomicInteger();
MemCache.setCache(CACHE_KEY, 0, 0);
LogWriter.printLog("Reset servlet status.");
ResponseBuilder.buildResponse("Reset ok!", response);
return;
}
if(null != request.getParameter("print")){
responseBody = "Servlet Object value ::" + visitTimes.intValue();
responseBody += " Memcached Object value ::" + MemCache.getCache(CACHE_KEY).toString();
LogWriter.printLog("Print servlet status.");
ResponseBuilder.buildResponse(responseBody, response);
return;
}
int step  = 1;
visitTimes.addAndGet(step);
LogWriter.printLog("Update local value::" + visitTimes.intValue());
while(true){
LogWriter.printLog("Begin to get CASValue::" + visitTimes.intValue());
CASValue<Object> casValue = MemCache.getCasId(CACHE_KEY);
LogWriter.printLog("End to get CASValue::" + casValue.getValue().toString());
int cachTimes = Integer.parseInt(casValue.getValue().toString()) + step;
if(MemCache.casCache(CACHE_KEY, casValue.getCas(), String.valueOf(cachTimes))){
LogWriter.printLog("Success to CAS Cache::" + cachTimes);
break;
}
else{
LogWriter.printLog("Fail to CAS Cache::" + cachTimes);
try{
Thread.sleep(200);
}
catch(Exception exp){
exp.printStackTrace();
}
}
}
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: