k18-ntcs-web-ntc/src/main/java/com/nis/web/task/StatisticASNIpNumer.java

package com.nis.web.task;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import org.apache.ibatis.session.ExecutorType;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.log4j.Logger;
import org.springframework.beans.factory.annotation.Autowired;

import com.nis.domain.basics.AsnGroupInfo;
import com.nis.domain.basics.AsnIpCfg;
import com.nis.util.IPUtil;
import com.nis.util.StringUtil;
import com.nis.web.dao.basics.AsnGroupInfoDao;
import com.nis.web.dao.basics.AsnIpCfgDao;
import com.nis.web.service.SpringContextHolder;

import net.sf.json.JSONObject;
/**
 * 定时统计ASN 下IPv4和IPV6的个数
 * @author intraUser
 *
 */
public class StatisticASNIpNumer {
	
	private Logger logger = Logger.getLogger(this.getClass());
	
	@Autowired
	protected AsnIpCfgDao asnIpCfgDao;
	public void calculateASNIp() {
		logger.info("定时统计ASN IP个数开始。。。。。。。。。。。。。。");
		long start=System.currentTimeMillis();
		Map<Integer,Map<String,Long>> asnNumerMap=new HashMap<>();
		List<Object[]> asnIpList=new ArrayList<>();
		asnIpList=asnIpCfgDao.findAllAsnIpCfgList();
		
		getAllASNIPNumber(asnIpList, asnNumerMap);
		
		updateAllASNIPNumber(asnNumerMap);
		
		long end=System.currentTimeMillis();
		logger.info("定时统计ASN IP个数结束:"+(end-start)+"。。。。。。。。。。。。。。");
	}
	public void getAllASNIPNumber(List<Object[]> asnIpList,Map<Integer,Map<String,Long>> asnNumerMap){
		logger.info("计算ASN IP开始：asn ip size:"+asnIpList.size()+"。。。。。。。。。。。。。。");
		long start=System.currentTimeMillis();
		if(!StringUtil.isEmpty(asnIpList)){
			try {
				for (Iterator iterator = asnIpList.iterator(); iterator.hasNext();) {
					Map asnIpMap=(Map) iterator.next();
					//groupId
					Integer asnGroupId = (Integer)asnIpMap.get("asn_ip_group");
					//ipType(4:v4,6:v6)
					Integer ipType= (Integer)asnIpMap.get("ip_type");
					//(1:mask,2:range,3:ip)
					//Integer ipPattern= (Integer)asnIpMap.get("ip_pattern");
					//dest_ip_address
					String ipAddress= (String)asnIpMap.get("dest_ip_address");
					long IPNumber=0;
					if(ipType.equals(4)){
						if(ipAddress.indexOf("/") > -1){
							Integer mask=Integer.parseInt(ipAddress.split("/")[1]);
							ipAddress=ipAddress.split("/")[0];
							IPNumber=IPUtil.getIpNum(ipAddress, mask);
						}else{
							IPNumber=1;
						}
						//判断组是否已经存在
						if(asnNumerMap.keySet().contains(asnGroupId)){
							asnNumerMap.get(asnGroupId).put("v4", asnNumerMap.get(asnGroupId).get("v4")+IPNumber);
							asnNumerMap.put(asnGroupId, asnNumerMap.get(asnGroupId));
						}else{
							Map<String, Long> map=new HashMap<>();
							map.put("v4", IPNumber);
							map.put("v6", 0l);
							asnNumerMap.put(asnGroupId, map);
						}
					}else{
						IPNumber=1;
						//判断组是否已经存在
						if(asnNumerMap.keySet().contains(asnGroupId)){
							asnNumerMap.get(asnGroupId).put("v6", asnNumerMap.get(asnGroupId).get("v6")+IPNumber);
							asnNumerMap.put(asnGroupId, asnNumerMap.get(asnGroupId));
						}else{
							Map<String, Long> map=new HashMap<>();
							map.put("v4", 0l);
							map.put("v6", IPNumber);
							asnNumerMap.put(asnGroupId, map);
						}
					}
				}
			} catch (Exception e) {
				logger.error("计算ASN IP个数失败",e);
			}
		}
		
		long end=System.currentTimeMillis();
		logger.info("计算ASN IP个数结束:"+(end-start)+"。。。。。。。。。。。。。。");
	}
	public void updateAllASNIPNumber(Map<Integer,Map<String,Long>> asnNumerMap) {
		logger.info("修改ASN IP个数开始：asn size:"+asnNumerMap.size()+"。。。。。。。。。。。。。。");
		long start=System.currentTimeMillis();
		if(!StringUtil.isEmpty(asnNumerMap)){
			int index=0;
			SqlSessionFactory sqlSessionFactory=SpringContextHolder.getBean(SqlSessionFactory.class);
			SqlSession batchSqlSession = null;
			try {
				for(Entry<Integer,Map<String,Long>> e: asnNumerMap.entrySet()) {
					batchSqlSession = sqlSessionFactory.openSession(ExecutorType.BATCH, false);
					((AsnGroupInfoDao) batchSqlSession.getMapper(AsnGroupInfoDao.class)).updateIpNum(e.getValue().get("v4"),e.getValue().get("v6"),e.getKey());
					if(index!= 0 && index%20000==0){
						batchSqlSession.commit();
					}
					index++;
				}
				batchSqlSession.commit();
			}catch (Exception e) {
				logger.error("修改ASN IP个数失败", e);
			} finally {
				if(batchSqlSession != null){
					batchSqlSession.close();
				}
			}
		}
		long end=System.currentTimeMillis();
		logger.info("修改ASN IP个数结束:"+(end-start)+"。。。。。。。。。。。。。。");
	}
}