/**
 * 任務(wù)調(diào)度調(diào)用攔截器
 */
public class TaskControlInterceptor implements Interceptor {

  //存放本機(jī)IP地址列表(包括ipv4和ipv6)
  private static Set<String> localHostList = new HashSet<>();

  @Override
  public void intercept(Invocation inv) {
   Controller controller = inv.getController();
   HttpServletRequest request = controller.getRequest();
   /*
   獲取本機(jī)的IP地址列表
   請(qǐng)求的時(shí)候判斷來(lái)源IP地址是否在該列表中，如果不在的話則不予通過(guò)
    */
   if (localHostList.size() == 0) {
     localHostList = getIpAddress();
   }
   if (StringUtils.isNotBlank(request.getRemoteAddr())) {
     if (!localHostList.contains(request.getRemoteAddr())) {
      Result result = new Result(new Error("450","非法的請(qǐng)求，請(qǐng)求來(lái)源IP地址不是本機(jī)"));
      controller.getResponse().setStatus(450);
      controller.renderJson(result);
      return;
     }
   } else {
     Result result = new Result(new Error("450","非法的請(qǐng)求，請(qǐng)求來(lái)源IP地址為空"));
     controller.getResponse().setStatus(450);
     controller.renderJson(result);
     return;
   }
   inv.invoke();
  }

  /**
  * 獲取本機(jī)的IP地址（包括ipv4和ipv6）
  * <br>包含回環(huán)地址127.0.0.1和0:0:0:0:0:0:0:1
  */
  private static Set<String> getIpAddress() {
   Set<String> ipList = new HashSet<>();
   try {
     Enumeration<NetworkInterface> allNetInterfaces = NetworkInterface.getNetworkInterfaces();
     InetAddress ip = null;
     while (allNetInterfaces.hasMoreElements()) {
      NetworkInterface netInterface = (NetworkInterface) allNetInterfaces.nextElement();
      //排除虛擬接口和沒有啟動(dòng)運(yùn)行的接口
      if (netInterface.isVirtual() || !netInterface.isUp()) {
        continue;
      } else {
        Enumeration<InetAddress> addresses = netInterface.getInetAddresses();
        while (addresses.hasMoreElements()) {
         ip = addresses.nextElement();
         if (ip != null && (ip instanceof Inet4Address || ip instanceof Inet6Address)) {
           ipList.add(ip.getHostAddress());
         }
        }
      }
     }
   } catch (Exception e) {
     e.printStackTrace();
   }
   return ipList;
  }

}