k18-ntcs-web-ntc/src/main/java/com/nis/web/service/basics/AsnIpCfgService.java

package com.nis.web.service.basics;

import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.BlockingQueue;

import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import com.beust.jcommander.internal.Lists;
import com.nis.domain.Page;
import com.nis.domain.basics.AsnIpCfg;
import com.nis.domain.basics.PolicyGroupInfo;
import com.nis.domain.basics.Varibles;
import com.nis.domain.configuration.CfgIndexInfo;
import com.nis.domain.maat.GroupReuseAddBean;
import com.nis.domain.maat.GroupReuseCfg;
import com.nis.domain.maat.MaatCfg.IpCfg;
import com.nis.domain.maat.MaatCfg.NumBoundaryCfg;
import com.nis.domain.maat.MaatCfg.StringCfg;
import com.nis.domain.maat.ToMaatResult;
import com.nis.domain.specific.ConfigGroupInfo;
import com.nis.exceptions.MaatConvertException;
import com.nis.util.AsnCacheUtils;
import com.nis.util.ConfigServiceUtil;
import com.nis.util.Constants;
import com.nis.web.dao.CrudDao;
import com.nis.web.dao.basics.AsnIpCfgDao;
import com.nis.web.dao.basics.PolicyGroupInfoDao;
import com.nis.web.dao.specific.ConfigGroupInfoDao;
import com.nis.web.security.UserUtils;
import com.nis.web.service.CrudService;
@Service
public class AsnIpCfgService extends CrudService<CrudDao<AsnIpCfg>, AsnIpCfg> {
	@Autowired
	private AsnIpCfgDao asnIpCfgDao;
//	@Autowired
//	private SpecificServiceCfgDao specificServiceCfgDao;
	@Autowired
	private ConfigGroupInfoDao configGroupInfoDao;
	@Autowired
	private PolicyGroupInfoDao policyGroupInfoDao;
	/**
	 * @param page
	 * @param entity
	 * @return
	 */
	public Page<AsnIpCfg> findPage(Page<AsnIpCfg> page, AsnIpCfg entity) {
		entity.getSqlMap().put("dsf", configScopeFilter(entity.getCurrentUser(),"r"));
		entity.setPage(page);
		List<AsnIpCfg> list=asnIpCfgDao.findPage(entity);
		page.setList(list);
		return page;
	}
	public AsnIpCfg get(Long id ) {
		return asnIpCfgDao.get(id);
	}
	
//	public void save(List<AsnIpCfg> entitys){
//		this.saveBatch(entitys, AsnIpCfgDao.class);
//	}
	@Transactional(readOnly=false,rollbackFor=RuntimeException.class)
	public void save(List<AsnIpCfg> entitys) {
		logger.warn("Start to save IP,size:"+entitys.size());
		long start=System.currentTimeMillis();
		int len=0;
		List<AsnIpCfg> tempList=Lists.newArrayList();
//		Varibles maxPacket=asnIpCfgDao.getVaribles("max_allowed_packet");
		for(AsnIpCfg asnIpCfg:entitys) {
			int tempLen=asnIpCfg.toString().getBytes(Charset.forName("UTF-8")).length;
			if((len+tempLen)<Constants.MAX_ALLOWED_PACKET) {
				tempList.add(asnIpCfg);
				len+=tempLen;
			}else {
//				logger.warn("save ip size:"+tempList.size());
				asnIpCfgDao.insertBatch(tempList);
				tempList.clear();
				tempList.add(asnIpCfg);
				len=tempLen;
			}
		}
		if(tempList.size()>0) {
//			logger.warn("save ip size:"+tempList.size());
			asnIpCfgDao.insertBatch(tempList);
			tempList.clear();
		}
//		entitys.clear();
		long end=System.currentTimeMillis();
		logger.warn("Save IP finish,cost:"+(end-start));
	}
	@Transactional(readOnly=false,rollbackFor=RuntimeException.class)
	public void update(AsnIpCfg entity){
		Date editTime=new Date();
		entity.setEditTime(editTime);
		entity.setEditorId(entity.getCurrentUser().getId());
		if(entity.getAsnIpGroup()==null) {
			throw new RuntimeException("asn group name not found!");
		}
		ConfigGroupInfo groupInfo=configGroupInfoDao.getConfigGroupInfoByGroupId(entity.getAsnIpGroup());
		if(groupInfo.getIsIssued()==0) {//未下发,可修改
			asnIpCfgDao.update(entity);
		}else {
			throw new RuntimeException("asn group is issued, cannot update asn ip!");
		}
		
	}
	@Transactional(readOnly=false,rollbackFor=RuntimeException.class)
	public void processGroup(Map<Long,Integer> asnNoMap){
		logger.warn("Start to save group");
		long start=System.currentTimeMillis();
		List<PolicyGroupInfo> policyGroupInfos=Lists.newArrayList();
		List<ConfigGroupInfo> configGroupInfoS=Lists.newArrayList();
		Date createTime=new Date();
		int len=0,len1=0;
//		Varibles maxPacket=asnIpCfgDao.getVaribles("max_allowed_packet");
		for(Entry<Long, Integer> e:asnNoMap.entrySet()) {
			PolicyGroupInfo info=new PolicyGroupInfo();
			info.setAsnNo(e.getKey());
			info.setDescription("Create by import excel");
			info.setGroupName("Import"+e.getKey());
			info.setGroupType(4);
			info.setServiceGroupId(asnNoMap.get(e.getKey()));
			info.setIsValid(1);
			info.setCreatorId(UserUtils.getUser().getId());
			info.setCreateTime(createTime);
			int tempLen=info.toString().getBytes(Charset.forName("UTF-8")).length;
			if((len+tempLen)<Constants.MAX_ALLOWED_PACKET) {
				policyGroupInfos.add(info);
				len+=tempLen;
			}else {
				policyGroupInfoDao.insertBatch(policyGroupInfos);
				policyGroupInfos.clear();
				policyGroupInfos.add(info);
				len=tempLen;
			}
			ConfigGroupInfo group=new ConfigGroupInfo();
			group.setGroupId(info.getServiceGroupId());
			group.setGroupName(info.getGroupName());
			group.setIsIssued(0);
			group.setGroupType(info.getGroupType());
			group.setAsnId(info.getAsnNo());
			int tempLen1=group.toString().getBytes(Charset.forName("UTF-8")).length;
			if((len1+tempLen1)<Constants.MAX_ALLOWED_PACKET) {
				configGroupInfoS.add(group);
				len1+=tempLen1;
			}else {
				configGroupInfoDao.insertBatch(configGroupInfoS);
				configGroupInfoS.clear();
				configGroupInfoS.add(group);
				len1=tempLen1;
			}
		}
		if(policyGroupInfos.size()>0) {
			policyGroupInfoDao.insertBatch(policyGroupInfos);
			policyGroupInfos.clear();
		}
		if(configGroupInfoS.size()>0) {
			configGroupInfoDao.insertBatch(configGroupInfoS);
			configGroupInfoS.clear();
		}
		//刷新缓存
		AsnCacheUtils.init(true);
		long end=System.currentTimeMillis();
		logger.warn("Save group finish,cost:"+(end-start));
	}
	public void deleteIps(Map<Long,Integer> asnNoMap){
		List<AsnIpCfg> toDelAndSendAsnIpCfgs=Lists.newArrayList();
		List<Long> asnIds=Lists.newArrayList(asnNoMap.size());
		for(Entry<Long, Integer> e:asnNoMap.entrySet()) {
			ConfigGroupInfo configGroupInfo=AsnCacheUtils.get(e.getKey());
			if(configGroupInfo==null) {
				configGroupInfo=this.getConfigGroupInfoByAsnNo(e.getKey());
			}
			if(configGroupInfo.getIsIssued()==1) {//已下发
				List<AsnIpCfg> _toDelAsnIpCfgs=this.getByAsnNo(configGroupInfo.getAsnId());
				toDelAndSendAsnIpCfgs.addAll(_toDelAsnIpCfgs);
			}else {
				asnIds.add(e.getKey());
			}
		}
		if(asnIds.size()>0) {
			this.deleteByAsnNo(asnIds);
		}
		if(toDelAndSendAsnIpCfgs.size()>0) {
			int pointsDataLimit = Constants.MAAT_JSON_SEND_SIZE;//限制条数
			Integer size = toDelAndSendAsnIpCfgs.size();
			//判断是否有必要分批
			if(pointsDataLimit<size){
				int part = size/pointsDataLimit;//分批数
				for (int i = 0; i < part; i++) {
					//pointsDataLimit条
					List<AsnIpCfg> listPage = toDelAndSendAsnIpCfgs.subList(0, pointsDataLimit);
					delAndSend(listPage);
					//剔除
					toDelAndSendAsnIpCfgs.subList(0, pointsDataLimit).clear();
				}
				//最后剩下的
				if(!toDelAndSendAsnIpCfgs.isEmpty()){
					delAndSend(toDelAndSendAsnIpCfgs);
					
				}
			}else {
				delAndSend(toDelAndSendAsnIpCfgs);
			}
		}
	}
	/**
	 * 根据Id 批量删除
	 * @param entities
	 */
	public void deleteByIdsBatch(List<AsnIpCfg> entities) {
		StringBuilder sb=new StringBuilder();
		for(AsnIpCfg ip:entities) {
			sb.append(ip.getCfgId());
			sb.append(",");
		}
		sb.deleteCharAt(sb.toString().lastIndexOf(","));
		asnIpCfgDao.delete(sb.toString());
	}
	@Transactional(readOnly=false,rollbackFor=RuntimeException.class)
	public void delAndSend(List<AsnIpCfg> entities) {
		deleteByIdsBatch(entities);
		asnIPRegionSendToMaat(entities,Constants.VALID_NO);
	}
	/**
	 * 分割IP region分次下发
	 * 由于在Controller中分割，此方法不需要调用，直接使用asnIPRegionSendToMaat方法即可
	 * @param asnIpCfgs
	 * @param valid
	 */
	@Deprecated
	public void splitAndSend(List<AsnIpCfg> asnIpCfgs,Integer valid){
		if(asnIpCfgs!=null&&asnIpCfgs.size()>0) {
//			List<AsnIpCfg> asnIpCfgs=Lists.newArrayList(list.size()); 
//			asnIpCfgs.addAll(list);
			int pointsDataLimit = Constants.MAAT_JSON_SEND_SIZE;//限制条数
			Integer size = asnIpCfgs.size();
			//判断是否有必要分批
			if(pointsDataLimit<size){
				int part = size/pointsDataLimit;//分批数
				for (int i = 0; i < part; i++) {
					//pointsDataLimit条
					List<AsnIpCfg> listPage = asnIpCfgs.subList(0, pointsDataLimit);
					asnIPRegionSendToMaat(listPage,valid);
					//剔除
					asnIpCfgs.subList(0, pointsDataLimit).clear();
				}
				//最后剩下的
				if(!asnIpCfgs.isEmpty()){
					asnIPRegionSendToMaat(asnIpCfgs,valid);
				}
			}else {
				asnIPRegionSendToMaat(asnIpCfgs,valid);
			}
			
		}
	}
	
	/**
	 * 批量保存asn PolicyGroupInfo
	 * @param entities
	 * @return
	 * @throws MaatConvertException
	 */
	@Transactional(readOnly=false,rollbackFor=RuntimeException.class)
	public Map<Long,ConfigGroupInfo> saveAsnGroupBatch(List<PolicyGroupInfo> entities) throws MaatConvertException{
		Map<Long,ConfigGroupInfo> configGroupInfoMap=new HashMap<>(entities.size());
		List<Integer> groupIds= ConfigServiceUtil.getId(2,entities.size());
		if(groupIds.size()!=entities.size()) {
			throw new MaatConvertException("Get asn group id failed");
		}
		int ind=0;
		for(PolicyGroupInfo entity:entities) {
			entity.setIsValid(1);
			ConfigGroupInfo group=new ConfigGroupInfo();
			//新增
			Date createTime=new Date();
			entity.setCreatorId(UserUtils.getUser().getId());
			entity.setCreateTime(createTime);
			//asn类型从综合服务获取groupId
			if(4!=entity.getGroupType().intValue()) {
				throw new RuntimeException("ConfigGroupInfo is not asn type 4!");
			}
			entity.setServiceGroupId(groupIds.get(ind).intValue());
			//新增协议分组
			group.setGroupId(groupIds.get(ind).intValue());
			group.setGroupName(entity.getGroupName());
			group.setIsIssued(0);
			group.setGroupType(entity.getGroupType());
			group.setAsnId(entity.getAsnNo());
			configGroupInfoDao.insertConfigGroupInfo(group);
			if(group.getGroupType().intValue()==4) {
				AsnCacheUtils.put(entity.getAsnNo(), group);
			}
			policyGroupInfoDao.insert(entity);
			configGroupInfoMap.put(group.getAsnId(), group);
			ind++;
		}
		return configGroupInfoMap;
	}
	@Transactional(readOnly=false,rollbackFor=RuntimeException.class)
	public void saveAsnIpCfg(CfgIndexInfo entity) {
		Date createTime=new Date();
		ConfigGroupInfo groupInfo=configGroupInfoDao.getConfigGroupInfoByGroupId(entity.getAsnIpGroup().intValue());
		if(groupInfo==null) {
			throw new RuntimeException("ConfigGroupInfo is null!");
		}
		List<Integer> idList=ConfigServiceUtil.getId(3, entity.getAsnIpCfgs().size());
		int index=0;
		if(groupInfo.getIsIssued()==1) {//如果已经下发，则需要下到综合服务中
			if(entity.getAsnIpCfgs()!=null) {
				for(AsnIpCfg cfg:entity.getAsnIpCfgs()) {
					if(StringUtils.isNotBlank(entity.getCfgDesc())) {
						BeanUtils.copyProperties(entity, cfg, new String[]{"cfgId"});
					}else {
						BeanUtils.copyProperties(entity, cfg, new String[]{"cfgId","cfgDesc"});
					}
					
					cfg.setCreateTime(createTime);
					cfg.setCreatorId(entity.getCurrentUser().getId());
					cfg.setAsnIpGroup(entity.getAsnIpGroup());
					cfg.setIsValid(Constants.VALID_YES);
					cfg.setRegionId(idList.get(index));
					index++;
				}
			}
			this.save(entity.getAsnIpCfgs());
			asnIPRegionSendToMaat(entity.getAsnIpCfgs(),Constants.VALID_YES);
//			splitAndSend(entity.getAsnIpCfgs(),Constants.VALID_YES);
		}else {
			if(entity.getAsnIpCfgs()!=null) {
				for(AsnIpCfg cfg:entity.getAsnIpCfgs()) {
					BeanUtils.copyProperties(entity, cfg, new String[]{"cfgId"});
					cfg.setCreateTime(createTime);
					cfg.setCreatorId(entity.getCurrentUser().getId());
					cfg.setAsnIpGroup(entity.getAsnIpGroup().intValue());
					cfg.setIsValid(Constants.VALID_NO);
					cfg.setRegionId(idList.get(index));
					index++;
				}
			}
			this.save(entity.getAsnIpCfgs());
		}
	}
	@Transactional(readOnly=false,rollbackFor=RuntimeException.class)
	public void delete(Integer isValid,String ids) {
		/*for(String id:ids.split(",")) {
			Long.parseLong(id);
		}*/
		List<AsnIpCfg> issuedList=new ArrayList<>();
		List<AsnIpCfg> asnIpCfgList= asnIpCfgDao.getByIds(ids);
		
		for(AsnIpCfg asnIpCfg:asnIpCfgList) {
			if(asnIpCfg.getIsValid()==Constants.VALID_YES) {//代表下发过了
				asnIpCfg.setIsValid(isValid);
				issuedList.add(asnIpCfg);
				Integer groupId=asnIpCfg.getAsnIpGroup();
				List result=asnIpCfgDao.findOtherIps(groupId, asnIpCfg.getCfgId().intValue());
				if(result==null||result.size()==0) {
					throw new RuntimeException("Cant not delete ip from asn group, there is only one ip left in the group!");
				}
			}
		}
		asnIpCfgDao.delete(ids);
		//已经下发过的,调用分组复用配置删除接口
//		splitAndSend(issuedList,Constants.VALID_NO);
		if(issuedList.size() > 0){
			asnIPRegionSendToMaat(issuedList,Constants.VALID_NO);
		}
		
	}
//	public List<ConfigGroupInfo> findPolicyGroupInfosByType(Integer groupId) {
//		// TODO Auto-generated method stub
//		return asnIpCfgDao.findPolicyGroupInfosByType(groupId);
//	}
	/**
	 * 选中组中是否含有只剩一个未删除IP的组
	 * @param serviceGroupIds
	 * @return
	 */
	public boolean hasLastIp(String serviceGroupIds,String ids) {
		// TODO Auto-generated method stub
		for(String groupId:serviceGroupIds.split(",")) {
			Long.parseLong(groupId);
		}
		List<Integer> countList=asnIpCfgDao.countValidIPs(serviceGroupIds,ids);
		if(countList.size()==0) {
			return true;
		}
		for(Integer count:countList) {
			if(count==0) {
				return true;
			}
		}
		return false;
	}
	public ConfigGroupInfo getConfigGroupInfoByAsnNo(Long asnNo) {
		ConfigGroupInfo configGroupInfo=configGroupInfoDao.getInfoByAsnNo(asnNo);
		AsnCacheUtils.put(asnNo, configGroupInfo);
		return configGroupInfo;
	}
	public List<AsnIpCfg> getByAsnNo(Long asnNo) {
		if(asnNo==null) {
			throw new RuntimeException("groupId or asnNo is null!");
		}
		return asnIpCfgDao.getByAsnNo(asnNo.longValue());
	}
//	public void deleteByAsnGroup(Long asnNo) {
//		if(asnNo==null) {
//			throw new RuntimeException("asnNo is null!");
//		}
//		asnIpCfgDao.deleteByAsnGroup(asnNo.longValue());
//	}
	@Transactional(readOnly=false,rollbackFor=RuntimeException.class)
	public void deleteByAsnNo(List<Long> asnNoList) {
		int pointsDataLimit = 1000;//限制条数
		Integer size = asnNoList.size();
		if(pointsDataLimit<size){
			int part = size/pointsDataLimit;//分批数
			for (int i = 0; i < part; i++) {
				List<Long> listPage = asnNoList.subList(0, pointsDataLimit);
				StringBuilder asnNoStr=new StringBuilder();
				for(Long asnNo:asnNoList) {
					asnNoStr.append(asnNo);
					asnNoStr.append(",");
				}
				int result=0;
				do {
					result=asnIpCfgDao.deleteByAsnGroup(asnNoStr.toString().substring(0,asnNoStr.toString().length()-1));
				}while(result>0);
				//剔除
				asnNoList.subList(0, pointsDataLimit).clear();
			}
			if(!asnNoList.isEmpty()){
				StringBuilder asnNoStr=new StringBuilder();
				for(Long asnNo:asnNoList) {
					asnNoStr.append(asnNo);
					asnNoStr.append(",");
				}
				int result=0;
				do {
					result=asnIpCfgDao.deleteByAsnGroup(asnNoStr.toString().substring(0,asnNoStr.toString().length()-1));
				}while(result>0);
				asnNoList.clear();
			}
		}else {
			StringBuilder asnNoStr=new StringBuilder();
			for(Long asnNo:asnNoList) {
				asnNoStr.append(asnNo);
				asnNoStr.append(",");
			}
			int result=0;
			do {
				result=asnIpCfgDao.deleteByAsnGroup(asnNoStr.toString().substring(0,asnNoStr.toString().length()-1));
			}while(result>0);
			asnNoList.clear();
		}
	}
	public void ajaxDeleteAsnIp(String ids) {
		asnIpCfgDao.ajaxDeleteAsnIp(ids);
	}
}