package com.nis.web.service.configuration; import java.util.ArrayList; import java.util.Date; import java.util.List; import java.util.Map; 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.nis.domain.configuration.AreaIpCfg; import com.nis.domain.configuration.BaseIpCfg; import com.nis.domain.configuration.HttpUrlCfg; import com.nis.domain.maat.MaatCfg; import com.nis.domain.maat.MaatCfg.DigestCfg; import com.nis.domain.maat.MaatCfg.GroupCfg; 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.ToMaatBean; import com.nis.domain.maat.ToMaatResult; import com.nis.exceptions.MaatConvertException; import com.nis.util.ConfigServiceUtil; import com.nis.util.Constants; import com.nis.util.StringUtil; import com.nis.web.dao.configuration.AreaIpCfgDao; import com.nis.web.dao.configuration.DomainDao; import com.nis.web.dao.configuration.StringCfgDao; import com.nis.web.service.CrudService; /** * String相关配置事务类 * @author dell * */ @Service public class DomainService extends CrudService { @Autowired DomainDao domainDao; @Autowired protected StringCfgDao stringCfgDao; @Autowired protected AreaIpCfgDao areaIpCfgDao; @Transactional(readOnly=false,rollbackFor=RuntimeException.class) public void addStringCfg(HttpUrlCfg cfg){ //调用服务接口获取compileId Integer compileId = 0; try { List compileIds = ConfigServiceUtil.getId(1,1); if(!StringUtil.isEmpty(compileIds)){ compileId = compileIds.get(0); } } catch (MaatConvertException e) { e.printStackTrace(); logger.info("获取编译ID出错"); throw new MaatConvertException(":"+e.getMessage()); } if(compileId!=0){ cfg.setCompileId(compileId); setAreaEffectiveIds(cfg); if(cfg.getAreaCfg()!=null&&cfg.getAreaCfg().size()>0){ for(AreaIpCfg c:cfg.getAreaCfg()){ c.initDefaultValue(); BeanUtils.copyProperties(cfg, c,new String[]{"cfgId","ipType","direction", "protocol","protocolId","areaEffectiveIds","cfgRegionCode", "cfgType","ipPattern","srcIpAddress","portPattern","srcPort","destIpAddress","destPort"}); c.setTableName(AreaIpCfg.getTablename()); } this.saveIpBatch(cfg.getAreaCfg()); } if(cfg instanceof HttpUrlCfg){ domainDao.insert((HttpUrlCfg)cfg); }else{ stringCfgDao.insert(cfg); } }else{ throw new MaatConvertException(": compileId is 0"); } } @Transactional(readOnly=false,rollbackFor=RuntimeException.class) public void updateStringCfg(HttpUrlCfg cfg){ AreaIpCfg area=new AreaIpCfg(); area.setCompileId(cfg.getCompileId()); area.setFunctionId(cfg.getFunctionId()); areaIpCfgDao.deleteAreaIpCfg(area); //区域IPsetAreaEffectiveIds设置 setAreaEffectiveIds(cfg); Date date=new Date(); if(cfg.getAreaCfg()!=null&&cfg.getAreaCfg().size()>0){ for(AreaIpCfg c:cfg.getAreaCfg()){ c.initDefaultValue(); BeanUtils.copyProperties(cfg, c,new String[]{"cfgId","ipType","direction", "protocol","protocolId","areaEffectiveIds","cfgRegionCode", "cfgType","ipPattern","srcIpAddress","portPattern","srcPort","destIpAddress","destPort"}); c.setTableName(AreaIpCfg.getTablename()); c.setCreatorId(cfg.getCurrentUser().getId()); c.setCreateTime(date); } this.saveIpBatch(cfg.getAreaCfg()); } if(cfg instanceof HttpUrlCfg && cfg.getCfgId() == null){ domainDao.insert((HttpUrlCfg)cfg); }else{ stringCfgDao.update(cfg); } } public void auditWhiteDomain(HttpUrlCfg cfg) throws Exception{ List beans=new ArrayList<>(); beans.add(cfg); domainDao.audit(cfg); List configCompileList = new ArrayList<>(); List groupRelationList = new ArrayList<>(); List ipRegionList = new ArrayList<>(); List strRegionList = new ArrayList<>(); List numRegionList = new ArrayList<>(); List digestRegionList = new ArrayList<>(); List areaIpRegionList = new ArrayList<>(); ToMaatBean maatBean = new ToMaatBean(); MaatCfg maatCfg = new MaatCfg(); maatCfg.initDefaultValue(); BeanUtils.copyProperties(cfg, maatCfg); if(cfg.getIsAudit()==Constants.AUDIT_YES){ maatBean.setOpAction(Constants.INSERT_ACTION); Map map = cfgConvert(strRegionList,beans,2,cfg,groupRelationList); groupRelationList=map.get("groupList"); strRegionList=map.get("dstList"); maatCfg.setAreaEffectiveIds("0"); maatCfg.setAction(cfg.getAction()); maatCfg.setAuditTime(cfg.getAuditTime()); maatCfg.setIpRegionList(ipRegionList); maatCfg.setStrRegionList(strRegionList); maatCfg.setNumRegionList(numRegionList); maatCfg.setDigestRegionList(digestRegionList); maatCfg.setGroupRelationList(groupRelationList); maatCfg.setGroupNum(groupRelationList.size()); maatCfg.setAreaIpRegionList(areaIpRegionList); configCompileList.add(maatCfg); maatBean.setConfigCompileList(configCompileList); maatBean.setAuditTime(cfg.getAuditTime()); maatBean.setCreatorName(cfg.getCurrentUser().getName()); maatBean.setVersion(Constants.MAAT_VERSION); //调用服务接口下发配置数据 String json=gsonToJson(maatBean); logger.info("域名白名单配置:"+json); //调用服务接口下发配置 try { ToMaatResult result = ConfigServiceUtil.postMaatCfg(json); logger.info("域名白名单配置下发响应信息:"+result.getMsg()); } catch (Exception e) { e.printStackTrace(); logger.info("域名白名单配置下发失败"); throw e; } }else if(cfg.getIsAudit()==Constants.AUDIT_NOT_YES){ maatCfg.setCompileId(cfg.getCompileId()); maatCfg.setServiceId(cfg.getServiceId()); maatCfg.setIsValid(Constants.VALID_NO);//无效 configCompileList.add(maatCfg); maatBean.setConfigCompileList(configCompileList); maatBean.setAuditTime(cfg.getAuditTime()); maatBean.setCreatorName(cfg.getCurrentUser().getName()); maatBean.setVersion(Constants.MAAT_VERSION); maatBean.setOpAction(Constants.UPDATE_ACTION); //调用服务接口取消配置 String json=gsonToJson(maatBean); logger.info("域名白名单配置参数:"+json); //调用服务接口取消配置 try { ToMaatResult result = ConfigServiceUtil.put(json, 1); logger.info("域名白名单取消配置响应信息:"+result.getMsg()); } catch (Exception e) { e.printStackTrace(); logger.info("域名白名单取消配置失败"); throw e; } } } public void auditDomain(HttpUrlCfg cfg) throws Exception{ List beans=new ArrayList<>(); beans.add(cfg); domainDao.audit(cfg); List beans1=new ArrayList<>(); List areaIpCfgList=areaIpCfgDao.getByCompileId(cfg.getCompileId()); for(AreaIpCfg area:areaIpCfgList){ BeanUtils.copyProperties(cfg,area ,new String[]{"cfgRegionCode" ,"cfgType" ,"cfgId" ,"ipType" ,"ipPattern" ,"srcIpAddress" ,"portPattern" ,"srcPort" ,"destPort" ,"protocol" ,"direction" ,"protocolId" }); beans1.add(area); } this.auditIpBatch(beans1); List configCompileList = new ArrayList<>(); List groupRelationList = new ArrayList<>(); List ipRegionList = new ArrayList<>(); List strRegionList = new ArrayList<>(); List numRegionList = new ArrayList<>(); List digestRegionList = new ArrayList<>(); List areaIpRegionList = new ArrayList<>(); ToMaatBean maatBean = new ToMaatBean(); MaatCfg maatCfg = new MaatCfg(); maatCfg.initDefaultValue(); BeanUtils.copyProperties(cfg, maatCfg); if(cfg.getIsAudit()==Constants.AUDIT_YES){ maatBean.setOpAction(Constants.INSERT_ACTION); Map map = cfgConvert(strRegionList,beans,2,cfg,groupRelationList); groupRelationList=map.get("groupList"); strRegionList=map.get("dstList"); numRegionList=map.get("numRegionList")==null?new ArrayList<>():map.get("numRegionList"); Map areaMap = cfgConvert(areaIpRegionList,beans1,1,cfg,groupRelationList); groupRelationList=areaMap.get("groupList"); areaIpRegionList=areaMap.get("dstList"); maatCfg.setAreaEffectiveIds(StringUtils.isBlank(cfg.getAreaEffectiveIds())?"0":cfg.getAreaEffectiveIds()); if(cfg.getServiceId().intValue()==Constants.SERVICE_PXY_DOMAIN_INTERCEPT){ String region=Constants.USERREGION_DOMAIN_ID+"="+cfg.getCompileId().intValue()+ Constants.USER_REGION_SPLIT+Constants.USERREGION_DOMAIN_STR+"="+this.keywordsEscape(cfg.getCfgKeywords()); maatCfg.setUserRegion(region); }else if(Constants.SERVICE_DOMAIN_RATELIMIT==cfg.getServiceId().intValue()){ String region=Constants.USERREGION_RATE_LIMIT+"="+cfg.getRatelimit()+ Constants.USER_REGION_SPLIT+Constants.USERREGION_DOMAIN_ID+"="+cfg.getCompileId().intValue()+ Constants.USER_REGION_SPLIT+Constants.USERREGION_DOMAIN_STR+"="+this.keywordsEscape(cfg.getCfgKeywords()); maatCfg.setUserRegion(region); } maatCfg.setAction(cfg.getAction()); maatCfg.setAuditTime(cfg.getAuditTime()); maatCfg.setIpRegionList(ipRegionList); maatCfg.setStrRegionList(strRegionList); maatCfg.setNumRegionList(numRegionList); maatCfg.setAreaIpRegionList(areaIpRegionList); maatCfg.setDigestRegionList(digestRegionList); maatCfg.setGroupRelationList(groupRelationList); maatCfg.setGroupNum(groupRelationList.size()); maatCfg.setAreaIpRegionList(areaIpRegionList); configCompileList.add(maatCfg); maatBean.setOpAction(Constants.INSERT_ACTION); maatBean.setConfigCompileList(configCompileList); maatBean.setAuditTime(cfg.getAuditTime()); maatBean.setCreatorName(cfg.getCurrentUser().getName()); maatBean.setVersion(Constants.MAAT_VERSION); //调用服务接口下发配置数据 String json=gsonToJson(maatBean); logger.info("域名配置:"+json); //调用服务接口下发配置 try { ToMaatResult result = ConfigServiceUtil.postMaatCfg(json); logger.info("域名配置下发响应信息:"+result.getMsg()); } catch (Exception e) { e.printStackTrace(); logger.info("域名配置下发失败"); throw e; } }else if(cfg.getIsAudit()==Constants.AUDIT_NOT_YES){ maatCfg.setCompileId(cfg.getCompileId()); maatCfg.setServiceId(cfg.getServiceId()); maatCfg.setIsValid(Constants.VALID_NO);//无效 configCompileList.add(maatCfg); maatBean.setConfigCompileList(configCompileList); maatBean.setAuditTime(cfg.getAuditTime()); maatBean.setCreatorName(cfg.getCurrentUser().getName()); maatBean.setVersion(Constants.MAAT_VERSION); maatBean.setOpAction(Constants.UPDATE_ACTION); //调用服务接口取消配置 String json=gsonToJson(maatBean); logger.info("域名配置参数:"+json); //调用服务接口取消配置 try { ToMaatResult result = ConfigServiceUtil.put(json, 1); logger.info("域名取消配置响应信息:"+result.getMsg()); } catch (Exception e) { e.printStackTrace(); logger.info("域名取消配置失败"); throw e; } } } @Transactional(readOnly=false,rollbackFor=RuntimeException.class) public void deleteWhiteDomain(String ids){ List cfgs=new ArrayList(); Date date =new Date(); if(StringUtils.isNotBlank(ids)){ for(String idStr:ids.split(",")){ if(StringUtils.isNotBlank(idStr)){ HttpUrlCfg cfg=new HttpUrlCfg(); cfg.setCfgId(Long.parseLong(idStr)); cfg.setTableName(HttpUrlCfg.getTablename()); cfg.setEditorId(cfg.getCurrentUser().getId()); cfg.setEditTime(date); cfg.setIsValid(Constants.VALID_DEL); cfgs.add(cfg); } } } this.deleteBatch(cfgs, DomainDao.class); } @Transactional(readOnly=false,rollbackFor=RuntimeException.class) public void deleteDomain(String ids,String compileIds,int functionId){ if(StringUtils.isNotBlank(compileIds)){ for(String compileId:compileIds.split(",")){//强转数字,防止注入 Integer.parseInt(compileId); } stringCfgDao.deleteByCompileIds(new HttpUrlCfg().getCurrentUser().getId(),AreaIpCfg.getTablename(),compileIds); } List stringCfgs=new ArrayList(); Date date =new Date(); if(StringUtils.isNotBlank(ids)){ for(String idStr:ids.split(",")){ if(StringUtils.isNotBlank(idStr)){ HttpUrlCfg cfg=new HttpUrlCfg(); cfg.setCfgId(Long.parseLong(idStr)); cfg.setTableName(HttpUrlCfg.getTablename()); cfg.setEditorId(cfg.getCurrentUser().getId()); cfg.setEditTime(date); cfg.setIsValid(Constants.VALID_DEL); stringCfgs.add(cfg); } } } this.deleteBatch(stringCfgs, StringCfgDao.class); } /** * * deleteStringCfg(删除IP类配置) * (继承BaseStringCfg这个类方可使用) * @param baseStringCfg * @return *int * @exception * @since 1.0.0 */ @Transactional(readOnly=false,rollbackFor=RuntimeException.class) public void deleteStringCfg(List stringCfg,List areaCfg){ if(areaCfg!=null&&areaCfg.size()>0){ this.deleteIpBatch(areaCfg); } this.deleteBatch(stringCfg, StringCfgDao.class); } /** * * getStringCfg(根据IP与类名获取IP配置) * (继承BaseStringCfg这个类方可使用) * @param clazz * @param id * @return *BaseStringCfg * @exception * @since 1.0.0 */ public HttpUrlCfg getStringCfgById(HttpUrlCfg baseStringCfg){ return domainDao.getById(baseStringCfg.getCfgId()); } public HttpUrlCfg getStringCfgById(long id){ return domainDao.getById(id); } public Integer getIsValid(HttpUrlCfg baseStringCfg){ return domainDao.getIsValid(baseStringCfg); } public Integer getIsValid(long id){ return domainDao.getIsValid(id); } public Integer getIsAudit(HttpUrlCfg baseStringCfg){ return domainDao.getIsAudit(baseStringCfg); } public Integer getIsAudit(long id){ return domainDao.getIsAudit(id); } /** * getList(这里用一句话描述这个方法的作用) * (这里描述这个方法适用条件 – 可选) * @param tableName * @param ids * @return *List * @exception * @since 1.0.0 */ public List getListByCompileId(String ids) { // TODO Auto-generated method stub return domainDao.getListByCompileId(ids); } /** * getListByCfgId(这里用一句话描述这个方法的作用) * (这里描述这个方法适用条件 – 可选) * @param tablename * @param ids * @return *List * @exception * @since 1.0.0 */ public List getListByCfgId(String ids) { // TODO Auto-generated method stub return domainDao.getListByCfgId(ids); } }