package com.terra.proxy.controller;
|
|
|
import cn.hutool.json.JSONUtil;
|
import com.alibaba.fastjson.JSONObject;
|
import com.terra.proxy.bean.BlackRule;
|
import com.terra.proxy.bean.VistorBean;
|
import com.terra.proxy.service.Impl.LogServiceImpl;
|
import com.terra.proxy.util.Result;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Controller;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.servlet.ModelAndView;
|
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.util.*;
|
|
@Controller
|
@RequestMapping("/log")
|
@CrossOrigin
|
public class LoggerController {
|
|
|
private Logger log = LoggerFactory.getLogger(LoggerController.class);
|
/**
|
* 查询每日访问日志统计图
|
*
|
* @param 时间,为空取今日
|
*/
|
|
@Autowired
|
LogServiceImpl logservice;
|
|
@RequestMapping(value = "/getDailyEchart", method = RequestMethod.GET)
|
@ResponseBody
|
public String getDailyEchart(HttpServletRequest request, String starttime, String endtime) {
|
Map<String, Object> param = new HashMap<>();
|
Map<String, Object> result = new HashMap<>();
|
param.put("starttime", starttime);
|
param.put("endtime", endtime);
|
List<Map> list = logservice.getTotalCountPerHour(param);
|
List<Map> resultlist = new ArrayList<>();
|
List<Map> infolist = logservice.getVisitInfoPerHourByip(param);
|
JSONObject json = new JSONObject();
|
|
for (Map map : infolist) {
|
Map<String, Object> tempmap = new HashMap<>();
|
tempmap.put(map.get("requestip").toString(), map.get("count").toString());
|
if (result.containsKey(map.get("caltime").toString())) {
|
List<Map> tlist = (List<Map>) result.get(map.get("caltime").toString());
|
tlist.add(tempmap);
|
} else {
|
List<Map> templist = new ArrayList<Map>();
|
templist.add(tempmap);
|
result.put(map.get("caltime").toString(), templist);
|
}
|
}
|
|
for (Map map : list) {
|
Map<String, Object> tmap = new HashMap<>();
|
tmap.put("name", map.get("caltime").toString());
|
Object[] obj = new Object[3];
|
obj[0] = map.get("caltime").toString();
|
obj[1] = map.get("sum").toString();
|
|
tmap.put("value", obj);
|
if (result.containsKey(map.get("caltime").toString())) {
|
tmap.put("info", result.get(map.get("caltime").toString()));
|
}
|
resultlist.add(tmap);
|
}
|
return JSONUtil.toJsonStr(resultlist);
|
}
|
|
|
@RequestMapping(value = "/querylog", method = RequestMethod.GET)
|
@ResponseBody
|
public String querylog(HttpServletRequest request, HttpServletResponse httpresponse, String ip, Integer id, String result,
|
Date starttime, Date endtime, String servername, Integer resourceid, String fromsys, Integer page, Integer limit) {
|
Map<String, Object> params = new HashMap<>();
|
Map<String, Object> resultmap = new HashMap<>();
|
params.put("ip", ip);
|
params.put("id", id);
|
params.put("result", result);
|
params.put("servername", servername);
|
params.put("resourceid", resourceid);
|
params.put("fromsys", fromsys);
|
params.put("starttime", starttime);
|
params.put("endtime", endtime);
|
params.put("offSet", (page - 1) * limit);
|
params.put("limit", limit);
|
List<Map> list = logservice.querylog(params);
|
int total = logservice.queryTotalLog(params);
|
resultmap.put("data", list);
|
resultmap.put("total", total);
|
return JSONUtil.toJsonStr(resultmap);
|
}
|
|
@RequestMapping(value = "/error", method = RequestMethod.GET)
|
@ResponseBody
|
public ModelAndView error(String result) {
|
ModelAndView mv = new ModelAndView();
|
JSONObject json = JSONObject.parseObject(result);
|
|
mv.addObject("message", json.get("msg"));
|
mv.addObject("status", json.get("code"));
|
|
mv.setViewName("error");
|
return mv;
|
}
|
|
@RequestMapping(value = "/queryBlackLists")
|
@ResponseBody
|
public Result queryBlackLists(Integer limit, Integer page, String ip, String status) {
|
int total = 0;
|
|
HashMap<String, Object> param = new HashMap<>();
|
param.put("limit", limit);
|
param.put("page", (page - 1) * limit);
|
param.put("ip", ip == "" ? null : ip);
|
param.put("status", status == "" ? null : status);
|
List<VistorBean> data = logservice.queryBlackLists(param);
|
total = logservice.getTotalBlackList(param);
|
return Result.ok().put("data", data).put("total", total).put("totalpage", Math.ceil((double) total / (double) limit));
|
}
|
|
@RequestMapping(value = "/delBlack")
|
@ResponseBody
|
public Result delBlackById(VistorBean vistorBean) {
|
logservice.delBlackById(vistorBean);
|
return Result.ok();
|
}
|
|
@RequestMapping(value = "/updateBlackRule")
|
@ResponseBody
|
public Result updateBlackRule(@RequestBody BlackRule blackRule) {
|
logservice.updateRules(blackRule);
|
return Result.ok();
|
}
|
|
@RequestMapping(value = "/selectBalckRule")
|
@ResponseBody
|
public Result selectBalckRule() {
|
BlackRule blackRule = logservice.selectBlackRule();
|
return Result.ok().put("data", blackRule);
|
}
|
|
@RequestMapping(value = "/selectWhiteList")
|
@ResponseBody
|
public Result selectWhiteList(Integer limit, Integer page, String ip, String status) {
|
int total = 0;
|
page = page == null ? 0 : page;
|
HashMap<String, Object> param = new HashMap<>();
|
param.put("limit", limit);
|
param.put("page", (page - 1) * limit);
|
param.put("ip", ip == "" ? null : ip);
|
param.put("status", status == "" ? null : status);
|
List<VistorBean> data = logservice.selectWhiteList(param);
|
total = logservice.getTotalWhiteList(param);
|
return Result.ok().put("data", data).put("total", total).put("totalpage", Math.ceil((double) total / (double) limit));
|
}
|
|
@RequestMapping(value = "/updateWhite")
|
@ResponseBody
|
public Result updateWhite(VistorBean vistorBean) {
|
//先移除黑名单 status 1,2
|
logservice.delBlackById(vistorBean);
|
//再加入白名单 status 3,4
|
logservice.updateWhite(vistorBean);
|
|
return Result.ok();
|
}
|
}
|
