java后台使用计数器实现轮询

现在遇到一个业务场景：有四个区域的若干用户。每天通过APP上传照片。四个区域每个区域各分配一个评分员对用户上传的照片进行评分，评分是匿名的，但为了确保公平公正，每个区域评分员又不能只对自己本区域的人员照片进行评分。同时每个评分员的评分工作量又要求大致相同。

这个场景下需要在用户评分时就指定相应的评分员，但这个指定是随机的，而每个评分员被指定的概率又是相同的。那么就可以考虑轮询。我这个场景是广义上的轮询。就是说确保每个人按顺序被指定分配。

最后用java计数器实现了预期功能。首先这个场景是并发的，要求计数器必须是线程间可见的而且是线程安全的。我的实现是这样的：先创建计数器工具类，确保每个线程取到时加1，且不被覆盖，然后将评分员列表查询出来，加入hashMap。再在图片上传接口里取到这个list，利用计数器当前的值对list的长度取余作为下标去从list里取评分员对象。因为list有序集合，只要计数器正常就可以按顺序不断取到评分员对象，实现轮询。

第一步》创建计数器工具类：

package com.xcwlkj.util;

import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicReferenceFieldUpdater;

public class CounterUtil {

	    private static final AtomicReferenceFieldUpdater<CounterUtil, AtomicInteger> valueUpdater =
	            AtomicReferenceFieldUpdater.newUpdater(CounterUtil.class, AtomicInteger.class, "value");
	    //保证变量线程间可见
	    private volatile String key;
	    //保证计数器原子性
	    private volatile AtomicInteger value;
	    private static final String DATE_PATTERN = "yyyy-MM-dd";


	    public CounterUtil() {
	        /**
	         *key值设定为当前日期是因为计数器每天重置一次，否则数值一天天增大
	         */
	        this.key = getCurrentDateString() ;
	        this.value = new AtomicInteger(0);
	    }


	    /**
	     * 获取计数器加1以后的值
	     *
	     * @return
	     */
	    public Integer addAndGet() {

	        AtomicInteger oldValue = value;
	        AtomicInteger newInteger = new AtomicInteger(0);

	        int newVal = -1;
	        String newDateStr = getCurrentDateString();
	        //日期一致，计数器加1后返回
	        if (isDateEquals(newDateStr)) {
	            newVal = add(1);
	            return newVal;
	        }

	        //日期不一致，保证有一个线程重置技术器
	        reSet(oldValue, newInteger, newDateStr);
	        this.key = newDateStr;
	        //重置后加1返回
	        newVal = add(1);
	        return newVal;
	    }

	    /**
	     * 获取计数器的当前值
	     * @return
	     */
	    public Integer get() {
	        return value.get();
	    }

	    /**
	     * 判断当前日期与老的日期（也即key成员变量记录的值）是否一致
	     *
	     * @return
	     */
	    private boolean isDateEquals(String newDateStr) {
	        String oldDateStr = key;
	        if (!isBlank(oldDateStr) && oldDateStr.equals(newDateStr)) {
	            return true;
	        }

	        return false;
	    }


	    /**
	     * 如果日期发生变化，重置计数器，也即将key设置为当前日期，并将value重置为0，重置后才能接着累加，
	     */
	    private void reSet(AtomicInteger oldValue, AtomicInteger newValue, String newDateStr) {
	        if(valueUpdater.compareAndSet(this, oldValue, newValue)) {
	            System.out.println("线程" + Thread.currentThread().getName() + "发现日期发生变化");
	        }
	    }

	    /**
	     * 获取当前日期字符串
	     *
	     * @return
	     */
	    private String getCurrentDateString() {
	        Date date = new Date();
	        String newDateStr = new SimpleDateFormat(DATE_PATTERN).format(date);
	        return newDateStr;
	    }

	    /**
	     * 计数器的值加1
	     * @param increment
	     */
	    private int add(int increment) {
	       return value.addAndGet(increment);
	    }

	    public static boolean isBlank(CharSequence cs) {
	        int strLen;
	        if(cs != null && (strLen = cs.length()) != 0) {
	            for(int i = 0; i < strLen; ++i) {
	                if(!Character.isWhitespace(cs.charAt(i))) {
	                    return false;
	                }
	            }

	            return true;
	        } else {
	            return true;
	        }
	    }
}

           

第二步》将待轮询的评分员放入缓存（代码片段）

@Service("manageUserService")
public class ManageUserServiceImpl implements ManageUserService {
 private static  Map<String,List<ManageUser>> userMap = new HashMap<String,List<ManageUser>>();
@Override
public void initVillageVerifyRoleCache() {

         ManageUserExample ex=new ManageUserExample();
         ex.createCriteria().andRoleNbrEqualTo(systemCfg.getVillageVerifyRoleNbr());
         //查询所有评分员列表
       	 List<ManageUser> userList=manageUserMapper.selectByExample(ex);
       	 if (userList.size()>0) {
       	    //清除map所有key-value
       		userMap.clear();
       		//将查询到的最新值加入map
       		userMap.put(systemCfg.getVerifyKey(), userList);          
        }
  }
  //在修改评分员时刷新缓存
@Override
public void modifyManageUser(ManageUser model) {
    if (StringUtil.equals(systemCfg.getVillageVerifyRoleNbr(), model.getRoleNbr())) {
         initVillageVerifyRoleCache()；
    }
  }
  //从缓存获取评分员列表
@Override
public List<ManageUser> queryManageUsersFromCache(String veriftKey) {
	
	return userMap.get(veriftKey);
}
}
           

第三步》在图片上传的接口里使用计数器（代码片段）

@Service("garbageScoreManager")
public class GarbageScoreManagerImpl implements GarbageScoreManager {
    private CounterUtil count=new CounterUtil();
    @Override
	public ReviewScoreResModel reviewScore(ReviewScoreReqModel req) {
		//从缓存取出评分员列表
		List<ManageUser> userList=manageUserService.queryManageUsersFromCache(systemCfg.getVerifyKey());
		//获取计数器当前值
		int currCount=count.addAndGet();
		if (userList.size()<=0) {
			throw new BusinessParamException("请先配置评分员", "请先配置评分员");
		}
		//取余算法确保每次都能取到不同评分员对象
		int newCount=currCount%userList.size();
		//获取评分员对象编号
		String manageUserNbr=userList.get(newCount).getNbr();
	}
}


    
           

第四步》按评分员编号查询照片记录（实现略）

第五步》将hashmap在启动时放置在spring上下文里

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationListener;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.stereotype.Service;
//初始化启动类
@Service("systemInitService")
public class SystemInitService implements ApplicationListener<ContextRefreshedEvent> {
	@Autowired
	private OrganService organService;
	@Autowired
	private ManageUserService manageUserService;
	private Logger log = LoggerFactory.getLogger(SystemInitService.class);
	/** 
	 * @see org.springframework.context.ApplicationListener#onApplicationEvent(org.springframework.context.ApplicationEvent)
	 */
	@Override
	public void onApplicationEvent(ContextRefreshedEvent event) {
		
		if (event.getApplicationContext().getParent() == null) {

			log.info("-----------初始化评分员到缓存开始--------------");
			manageUserService.initVillageVerifyRoleCache();
			log.info("-----------初始化评分员到缓存结束--------------");
	    }
		
	}

}
`