k18-ntcs-web-argus-service/src/main/java/com/nis/web/service/restful/DashboardService.java

package com.nis.web.service.restful;

import com.beust.jcommander.internal.Maps;
import com.nis.domain.restful.NtcEntranceReport;
import com.nis.domain.restful.dashboard.*;
import com.nis.util.StringUtils;
import com.nis.web.dao.dashboard.*;
import com.nis.web.service.BaseService;
import com.zdjizhi.utils.StringUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.*;

@Service
public class DashboardService extends BaseService{

	@Autowired
	public TrafficIpActiveStatisticDao trafficIpActiveStatisticDao;
	@Autowired
	public TrafficProtocolStatisticDao trafficProtocolStatisticDao;
	@Autowired
	public TrafficAppStatisticDao trafficAppStatisticDao;
	@Autowired
	public TrafficUaStatisticDao trafficUaStatisticDao;
	@Autowired
	public TrafficHttpStatisticDao trafficHttpStatisticDao;
	@Autowired
	public NtcTotalReportDao ntcTotalReportDao;
	@Autowired
	public TrafficPortActiveStatisticDao trafficPortActiveStatisticDao;
	
	/**
	 * 流量统计 数据显示
	 * info：先查询最近时间，根据时间条件查询数据 效率提高
	 */
	public List<Map> getTotalReportList(){
		List<Map> totalReportList =new ArrayList<Map>();
		List<Map> newData = new ArrayList<Map>();
		// 流量统计阻断丢弃回流等数据显示最近时间
		NtcTotalReport maxReportTime = ntcTotalReportDao.getMaxReportTime();
		if(maxReportTime!=null&&maxReportTime.getReportTime()!=null) {
			Date reportTime = maxReportTime.getReportTime();
			totalReportList = ntcTotalReportDao.getTotalReportList(reportTime);// 返回阻断，监测等結果
		}
		Map maxRecvtTime = ntcTotalReportDao.getMaxStatTime();
		if(maxRecvtTime!=null && maxRecvtTime.get("statTime")!=null) {
			Date recvTime = (Date) maxRecvtTime.get("statTime");
			newData = ntcTotalReportDao.getNetFlowPortInfoNew(recvTime);
		}
		// 统计带宽的流入流出 单位 五分钟 的 byte
		if(StringUtil.isNotEmpty(newData)){
			Double inoctets = Double.parseDouble(newData.get(0).get("inoctets").toString()) ;
			Double outoctets = Double.parseDouble(newData.get(0).get("outoctets").toString()) ;

			if (StringUtil.isEmpty(totalReportList)) {
				Map valueMap = Maps.newHashMap();
				valueMap.put("inoctets", inoctets);
				valueMap.put("outoctets", outoctets);
				totalReportList.add(valueMap);
			} else {
				totalReportList.get(0).put("inoctets", inoctets);
				totalReportList.get(0).put("outoctets", outoctets);
			}
		}
		return totalReportList;
	}
	/**
	 * 根据ip46,协议tcp，udp查询带宽
	 * @param addrType
	 * @param transType
	 * @return
	 */
	public List<HashMap> getBandwidthTrans(String addrType,Integer transType) {
		ArrayList<HashMap> listMap=new ArrayList<HashMap>();
		List<TrafficTransStatistic> bandwidthList = new ArrayList<TrafficTransStatistic>();
		Map maxStatTime = ntcTotalReportDao.getMaxStatTime();
		if(maxStatTime!=null && maxStatTime.get("statTime")!=null) {
			Date stat = (Date) maxStatTime.get("statTime");
			HashMap m = new HashMap();
			bandwidthList=ntcTotalReportDao.getBandwidthTrans(stat, addrType, transType);
			List timeList = new ArrayList();
			List linkList = new ArrayList();
			List gbpsList = new ArrayList();
			List ppsList = new ArrayList();
			if(bandwidthList!=null&&bandwidthList.size()>0) {
				for (TrafficTransStatistic tt : bandwidthList) {
					if(tt.getTime()!=null) {
						timeList.add(tt.getTime());
						if(tt.getLinkNum()!=null) {
							linkList.add(tt.getLinkNum());
						}
						if(tt.getPps()!=null) {
							ppsList.add(tt.getPps());
						}
						if(tt.getGbps()!=null) {
							gbpsList.add(tt.getGbps());
						}
					}
				}
			}
			m.put("linkNum",linkList);
			m.put("gbps",gbpsList);
			m.put("pps",ppsList);
			m.put("statTime",timeList);
			listMap.add(m);
		}
		return listMap;
	}
	/**
	 * 根据service 动作查询近五分钟变化趋势
	 * entrance 默认为1,2 
	 */
	public List<HashMap> getActionTrans(String serviceType) {
		String sql ="";
		String sqlBlock="((service>=16 and service<=40) or (service>=258 and service<=273) or (service=576))"; //阻断
		String sqlMonitor="((service>=128 and service<=152) or (service>=384 and service<=513) or (service=592) or (service>=848 and service<=1030) or (service=1152))"; //监测
		// 区分动作
		if(StringUtils.isNotBlank(serviceType)) {
			if(serviceType.equalsIgnoreCase("block")) {
				sql=sqlBlock;
			}
			if(serviceType.equalsIgnoreCase("monitor")) {
				sql=sqlMonitor;
			}
		}
		ArrayList<HashMap> listMap=new ArrayList<HashMap>();
		List<NtcEntranceReport> entrance1 = new ArrayList<NtcEntranceReport>();
		List<NtcEntranceReport> entrance2 = new ArrayList<NtcEntranceReport>();
		Map maxReportTime = ntcTotalReportDao.getEntranceMaxReportTime();
		if(maxReportTime!=null && maxReportTime.get("reportTime")!=null) {
			Date reportTime = (Date) maxReportTime.get("reportTime");
			HashMap m1 = new HashMap();
			HashMap m2 = new HashMap();
			entrance1=ntcTotalReportDao.getActionTrans(reportTime, 1, sql);
			entrance2=ntcTotalReportDao.getActionTrans(reportTime, 2, sql);
			List timeList = new ArrayList();
			List sumList1 = new ArrayList();
			List sumList2 = new ArrayList();
			// entrance 为1的趋势
			if(entrance1!=null&&entrance1.size()>0) {
				for (NtcEntranceReport tt : entrance1) {
					if(tt.getTime()!=null&&tt.getSum()!=null) {
						timeList.add(tt.getTime());
						sumList1.add(tt.getSum());
						
					}
				}
			}
			// entrance 为2的趋势
			if(entrance2!=null&&entrance2.size()>0) {
				for (NtcEntranceReport tt : entrance2) {
					if(tt.getTime()!=null&&tt.getSum()!=null) {
//						timeList.add(tt.getTime());
						sumList2.add(tt.getSum());
					}
				}
			}
			m1.put("count",sumList1);//[{time:[],link:[],link2:[]}]
			m1.put("statTime",timeList);
			m1.put("entranceId",1);
			m2.put("count",sumList2);//[{link1:[],time:[],entrance:"1"},]
			m2.put("statTime",timeList);
			m2.put("entranceId",2);
			listMap.add(m1);
			listMap.add(m2);
		}
		return listMap;
	}
	
	/**
	 * 最近活跃端口时间五分钟数据
	 * @return List
	 */
	public List<Map> getPortActiveList(){
		List<Map> list = new ArrayList<Map>();
		TrafficPortActiveStatistic maxStatTime = trafficPortActiveStatisticDao.getMaxStatTime();
		if(maxStatTime!=null&&maxStatTime.getStatTime()!=null) {
			Date statTime = maxStatTime.getStatTime();
			List<TrafficPortActiveStatistic> portActiveList = trafficPortActiveStatisticDao.getPortActiveList(statTime);
			//上个时间五分钟数据
			if(portActiveList!=null&&portActiveList.size()>0) {
				for (TrafficPortActiveStatistic port : portActiveList) {
					if(port.getPort()!=null) {
						Map map = new HashMap();
						map.put("port", port.getPort());
						map.put("sum", port.getSum());
						TrafficPortActiveStatistic portActiveOld = trafficPortActiveStatisticDao.getPortActiveOld(port.getPort(),statTime);
						if(portActiveOld!=null&&portActiveOld.getSum()!=null){
							map.put("preSum",portActiveOld.getSum());
						}else{
							map.put("preSum",0);
						}
						list.add(map);
					}
				}
			}
		}
		return list;
	}
	

	/**
	 * 活跃IP最近五分钟数据TOP10
	 * @return
	 */
	public List<HashMap> ipActiveFiveMinute(){
		TrafficIpActiveStatistic maxStatTime = trafficIpActiveStatisticDao.getMaxStatTime();
		ArrayList<HashMap> listMap=new ArrayList<HashMap>();
		if(maxStatTime!=null&&maxStatTime.getStatTime()!=null) {
			Date statTime = maxStatTime.getStatTime();
			// 查询最近五分钟TOP10
			ArrayList<LinkedHashMap> list = trafficIpActiveStatisticDao.ipActiveChart(statTime);
			if(list!=null&&list.size()>0) {
				for (LinkedHashMap map : list) {
					HashMap m = new HashMap();
					if(map.get("ipAddr")!=null) {
						String ipAddr = (String) map.get("ipAddr");
						m.put("ipAddr", ipAddr);
						// 根据五分钟TOP10IP，查询TOP10中每个IP最近一小时的变化
						ArrayList<TrafficIpActiveStatistic> ipList = trafficIpActiveStatisticDao.ipActiveFiveMinute(ipAddr,statTime);
						List linkList = new ArrayList();
						List timeList = new ArrayList();
						if(ipList!=null&&ipList.size()>0) {
							for (TrafficIpActiveStatistic ip : ipList) {
								if(ip.getLinkNum()!=null&&ip.getTime()!=null) {
									linkList.add(ip.getLinkNum());
									timeList.add(ip.getTime());
								}
							}
						}
						m.put("linkNum",linkList);
						m.put("statTime",timeList);
						listMap.add(m);
					}
				}
			}
		}
		
		return listMap;
	}
	/**
	 * 根据活跃IP最近五分钟TOP10，查询近1小时最大值
	 * @return
	 */
	public List<HashMap> ipActiveOneHour(){
		TrafficIpActiveStatistic maxStatTime = trafficIpActiveStatisticDao.getMaxStatTime();
		ArrayList<HashMap> listMap=new ArrayList<HashMap>();
		if(maxStatTime!=null&&maxStatTime.getStatTime()!=null) {
			Date statTime = maxStatTime.getStatTime();
			ArrayList<LinkedHashMap> list = trafficIpActiveStatisticDao.ipActiveChart(statTime);
			if(list!=null&&list.size()>0) {
				for (LinkedHashMap map : list) {
					if(map.get("ipAddr")!=null) {
						String ipAddr = (String) map.get("ipAddr");
						ArrayList<HashMap> iplList = trafficIpActiveStatisticDao.ipActiveOneHour(ipAddr,statTime);
						listMap.add(iplList.get(0));
					}
				}
			}
		}
		
		return listMap;
	}
	@SuppressWarnings("rawtypes")
	public List<LinkedHashMap> ipActiveChart(){
		TrafficIpActiveStatistic maxStatTime = trafficIpActiveStatisticDao.getMaxStatTime();
		ArrayList<LinkedHashMap> list = new ArrayList<>();
		if(maxStatTime!=null&&maxStatTime.getStatTime()!=null) {
			Date statTime = maxStatTime.getStatTime();
			list = trafficIpActiveStatisticDao.ipActiveChart(statTime);
			if(list!=null&&list.size()>0) {
				for (LinkedHashMap map : list) {
					if(map.get("ipAddr")!=null) {
						Map m = new LinkedHashMap();
						map.put("pktNum",0);
						map.put("byteLen",0);
					}
				}
			}
		}
		return list;
	}

	/**
	 * 协议统计
	 * @return
	 */
	public List<Map> protocolChart() {
		TrafficProtocolStatistic maxStatTime = trafficProtocolStatisticDao.getMaxStatTime();
		List<Map> list=new ArrayList<Map>();
		if(maxStatTime!=null&&maxStatTime.getStatTime()!=null) {
			Date statTime = maxStatTime.getStatTime();
			list = trafficProtocolStatisticDao.protocolChart(statTime);
			//当不查询包，字节时 设为0
			if(list!=null&&list.size()>0) {
				Map map = new HashMap();
				map.put("pktNum",0);
				map.put("byteLen",0);
			}
		}
		return list;
	}

	/**
	 * 协议统计报表
	 * @return
	 */
	public List<Map> getProtocolList(){
		TrafficProtocolStatistic maxStatTime = trafficProtocolStatisticDao.getMaxStatTime();
		List<Map> list=new ArrayList<Map>();
		if(maxStatTime!=null&&maxStatTime.getStatTime()!=null) {
			Date statTime = maxStatTime.getStatTime();
			list = trafficProtocolStatisticDao.getProtocolList(statTime);
		}	
		return list;
	}
	
	/**
	 * app应用top10 图
	 * @return
	 */
	public List<Map> appChart() {
		List<Map> list=new ArrayList<Map>();
		TrafficAppStatistic maxStatTime = trafficAppStatisticDao.getMaxStatTime();
		if(maxStatTime!=null&&maxStatTime.getStatTime()!=null) {
			Date statTime = maxStatTime.getStatTime();
			list = trafficAppStatisticDao.appChart(statTime);
			//当不查询包，字节时 设为0
			if(list!=null&&list.size()>0) {
				Map map = new HashMap();
				map.put("pktNum",0);
				map.put("byteLen",0);
			}
		}
		return list;
	}
	/**
	 * App统计报表
	 * @return
	 */
	public List<Map> getAppList(){
		TrafficAppStatistic maxStatTime = trafficAppStatisticDao.getMaxStatTime();
		List<Map> list=new ArrayList<Map>();
		if(maxStatTime!=null&&maxStatTime.getStatTime()!=null) {
			Date statTime = maxStatTime.getStatTime();
			list = trafficAppStatisticDao.getAppList(statTime);
		}	
		return list;
	}
//	操作系统列表 排名TOP10
	@SuppressWarnings({ "unchecked", "rawtypes" })
	public List<Map> systemList(){
		List<Map> result = new ArrayList<Map>();
		TrafficUaStatistic maxStatTime = trafficUaStatisticDao.getMaxStatTime();
			if(maxStatTime!=null) {
				Date statTime = maxStatTime.getStatTime();
				List<TrafficUaStatistic> list = trafficUaStatisticDao.systemList(statTime);
				Long preCount=0l;
				for (TrafficUaStatistic ua : list) {
					Map map = new HashMap();
					map.put("osType",ua.getOsType());
					map.put("count",ua.getCount());
					map.put("pktNum",0);
					map.put("byteLen",0);
					preCount = trafficUaStatisticDao.preSystemListCount(ua.getOsType(),statTime);//上个时段的量 用于与现在对比
					if(preCount!=null){
						map.put("preCount",preCount);
					}else{
						map.put("preCount",0);
					}
					result.add(map);
				}
		}
		return result;
	}
	public List<Map> getBrowserBySystem(Integer osType ){
		List<Map> list = new ArrayList<Map>();
		TrafficUaStatistic maxStatTime = trafficUaStatisticDao.getMaxStatTime();
			if(maxStatTime!=null) {
				Date statTime = maxStatTime.getStatTime();
				list = trafficUaStatisticDao.getBrowserBySystem(osType,statTime);
				List bsType = new ArrayList();
				//查新固定操系统下的除了TOP10以外的others
				if(list!=null&& list.size()>0){
					for (Map map : list) {
						bsType.add(map.get("bsType"));
					}
					if(list.size()>10) {
						Map others = new HashMap();
						others = trafficUaStatisticDao.systemOthers(bsType,osType,statTime);
						if(others!=null&&others.size()>0){
							others.put("bsType", "-1");
							list.add(others);
						}
					}
				}
		}
		return list;
	}

	public List<Map> browserList() {
		List<Map> result = new ArrayList<Map>();
		TrafficUaStatistic maxStatTime = trafficUaStatisticDao.getMaxStatTime();
		if(maxStatTime!=null) {
			Date statTime = maxStatTime.getStatTime();
			List<TrafficUaStatistic> list = trafficUaStatisticDao.browserList(statTime);
			Long preCount=0l;
			if(list!=null&&list.size()>0){
					for (TrafficUaStatistic ua : list) {
						Map map = new HashMap();
						map.put("bsType",ua.getBsType());
						map.put("count",ua.getCount());
						map.put("pktNum",0);
						map.put("byteLen",0);
						preCount = trafficUaStatisticDao.preBrowserListCount(ua.getBsType(),statTime);//上个时段的量 用于与现在对比
						if(preCount!=null){
							map.put("preCount",preCount);
						}else{
							map.put("preCount",0);
						}
						result.add(map);
					}
				}
			}
		return result;
	}
	public List<Map> getSystemBybrowser(Integer bsType ){
		List<Map> list = new ArrayList<Map>();
		TrafficUaStatistic maxStatTime = trafficUaStatisticDao.getMaxStatTime();
		if(maxStatTime!=null) {
			Date statTime = maxStatTime.getStatTime();
			list = trafficUaStatisticDao.getSystemBybrowser(bsType,statTime);
			List osType = new ArrayList();
			//查询固定操系统下的除了TOP10以外的others
			if(list!=null&& list.size()>0){
				for (Map map : list) {
					osType.add(map.get("osType"));
				}
				if(list.size()>10) {
					Map others = new HashMap();
					others = trafficUaStatisticDao.browserOthers(osType,bsType,statTime);
					if(others!=null&&others.size()>0){
						others.put("osType", "-1");
						list.add(others);
					}
				}
			}	
		}
		return list;
	}
	public List<Map> websiteList() {
		List<Map> result = new ArrayList<Map>();
		TrafficHttpStatistic maxStatTime = trafficHttpStatisticDao.getMaxStatTime();
		if(maxStatTime!=null) {
			List<TrafficHttpStatistic> list = trafficHttpStatisticDao.websiteList(maxStatTime.getStatTime());
			Long preCount=0l;
			if(list!=null&&list.size()>0){
				for (TrafficHttpStatistic website : list) {
					Map map = new HashMap();
					map.put("webId",website.getWebId());
					map.put("count",website.getCount());
					map.put("pktNum",0);
					map.put("byteLen",0);
					preCount = trafficHttpStatisticDao.preWebsiteListCount(website.getWebId(),maxStatTime.getStatTime());//上个时段的量 用于与现在对比
					if(preCount!=null){
						map.put("preCount",preCount);
					}else{
						map.put("preCount",0);
					}
					result.add(map);
				}
			}	
		}
		return result;
	}
	/**
	 * 根据网站服务查询子域名
	 * @param websiteServiceId
	 * @return list
	 */
	public List<Map> getDomainByWebsiteServiceId(Integer websiteServiceId ){
		List<Map> list = new ArrayList<Map>();
		TrafficHttpStatistic maxStatTime = trafficHttpStatisticDao.getMaxStatTime();
		if(maxStatTime!=null) {
			Date statTime = maxStatTime.getStatTime();
			list = trafficHttpStatisticDao.getDomainByWebsiteServiceId(websiteServiceId,statTime);
			List webIdList = new ArrayList();
			//查询固定网站下的域名除了TOP10以外的others域名
			if(list!=null&& list.size()>0){
				for (Map map : list) {
					webIdList.add(map.get("webId"));
				}
				if(list.size()>10) {
					Map others = new HashMap();
					others = trafficHttpStatisticDao.websiteDomainOthers(webIdList,websiteServiceId,statTime);
					if(others!=null&&others.size()>0){
						others.put("webId", "-1");
						list.add(others);
					}
				}
			}	
		}
		return list;
	}
	
	/**
		获取网站列表
	 * @return
	 */
	@SuppressWarnings("unchecked")
	public List<Map> getDomainByWebsiteList(){
		List<Map> result = new ArrayList<Map>();
		TrafficHttpStatistic maxStatTime = trafficHttpStatisticDao.getMaxStatTime();
		if(maxStatTime!=null) {
			Date statTime = maxStatTime.getStatTime();
			List<TrafficHttpStatistic> websiteList = trafficHttpStatisticDao.getDomainByWebsiteList(statTime);
			if(websiteList!=null&&websiteList.size()>0){
				Long preCount=0l;
				for (TrafficHttpStatistic website : websiteList) {
					Map map = new HashMap();
					map.put("websiteServiceId",website.getWesiteServiceId());
					map.put("count",website.getCount());
					map.put("pktNum",0);
					map.put("byteLen",0);
					preCount = trafficHttpStatisticDao.preWebsiteListCount(website.getWesiteServiceId(),statTime);//上个时段的量 用于与现在对比
					if(preCount!=null){
						map.put("preCount",preCount);
					}else{
						map.put("preCount",0);
					}
					result.add(map);
				}
			}
		}
		return result;
	}
	/**
	 *主题网站分类，域名 
	 * 
	 * **/
	public List<Map> getTopicAndDomainList(){
		List<Map> topicList = new ArrayList<Map>();
		TrafficHttpStatistic maxStatTime = trafficHttpStatisticDao.getMaxStatTime();
		if(maxStatTime!=null) {
			Date statTime = maxStatTime.getStatTime();
			//按照主题分类 获得主题
			topicList = trafficHttpStatisticDao.getDomainByTopicList(statTime);
			if(topicList!=null&&topicList.size()>0){
				for (Map m : topicList) {
					List<Map> domainList= new ArrayList<Map>();
					if(m!=null&&m.get("topicId")!=null){
						domainList = trafficHttpStatisticDao.getDomainByTopicId(m.get("topicId"),statTime);
						m.put("domainData", domainList);
					}
				}
			}	
		}
		return topicList;
	}
}