update

src/main/java/com/yutou/tools/utils/Tools.java

@@ -1,15 +1,167 @@
 package com.yutou.tools.utils;
 
+import com.alibaba.fastjson.JSONArray;
+
+import javax.annotation.Resource;
 import javax.servlet.http.Cookie;
 import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.io.InputStream;
+import java.net.HttpURLConnection;
+import java.net.URL;
+import java.net.URLEncoder;
+import java.util.Random;
 
 public class Tools {
-    public static Cookie getCookie(HttpServletRequest request,String key){
-        for (Cookie cookie : request.getCookies()) {
-            if(cookie.getName().equals(key)){
+    /**
+     *  设置Cookie
+     * @param response
+     * @param key
+     * @param value
+     * @param time
+     */
+    public static void setCookie(HttpServletResponse response, String key,String value,int time) {
+        Cookie cookie = new Cookie(key, value);
+        if(time!=-1) {
+            cookie.setMaxAge(time);
+        }
+        response.addCookie(cookie);
+    }
+    /**
+     * 设置Cookie
+     * @param request
+     * @param response
+     * @param key
+     * @param time 生命周期，为0时即为删除
+     * @return
+     */
+    private static String setCookie(HttpServletRequest request, HttpServletResponse response, String key,int time) {
+        Cookie name = new Cookie("uname", key);
+        Cookie session = new Cookie("session", request.getSession().getId());
+        if(time!=-1) {
+            name.setMaxAge(time);
+            session.setMaxAge(time);
+        }
+        response.addCookie(name);
+        response.addCookie(session);
+        return request.getSession().getId();
+    }
+    /**
+     * 获取Cookie
+     * @param request
+     * @param key
+     * @return
+     */
+    public static Cookie getCookie(HttpServletRequest request,String key) {
+        Cookie[] cookies = request.getCookies();
+        for (Cookie cookie : cookies) {
+            if (key!=null&&cookie.getName().equals(key)) {
                 return cookie;
             }
         }
         return null;
     }
+    /**
+     * 删除Cookie
+     * @param request
+     * @param response
+     * @param key
+     * @return
+     */
+    public static String deleteCookie(HttpServletRequest request, HttpServletResponse response, String key) {
+        return setCookie(request, response, key, 0);
+    }
+    public static void sendServer(String title,String msg){
+        try{
+            System.out.println("title="+title+" msg="+msg);
+            /*HttpURLConnection connection= (HttpURLConnection) new URL("https://sc.ftqq.com/SCU64034T5adf5c5940dcecc016e0e9d0cf9b1e725da126ff47475.send?text="
+                    + URLEncoder.encode(title,"UTF-8")+"&desp="+URLEncoder.encode(msg,"UTF-8")).openConnection();
+            connection.connect();
+            InputStream inputStream=connection.getInputStream();
+            int i=inputStream.read();
+            inputStream.close();
+            connection.disconnect();*/
+        }catch (Exception e){
+            e.printStackTrace();
+        }
+    }
+
+    /**
+     * 获取项目路径
+     * @param request
+     * @return
+     */
+    public static String getPath(HttpServletRequest request) {
+        return request.getServletContext().getRealPath("/") + "/";
+    }
+    /**
+     * 获取客户端IP
+     *
+     * @param request
+     * @return
+     */
+    public static String getRemoteAddress(HttpServletRequest request) {
+        String ip = request.getHeader("x-forwarded-for");
+        if (ip == null || ip.length() == 0 || ip.equalsIgnoreCase("unknown")) {
+            ip = request.getHeader("Proxy-Client-IP");
+        }
+        if (ip == null || ip.length() == 0 || ip.equalsIgnoreCase("unknown")) {
+            ip = request.getHeader("WL-Proxy-Client-IP");
+        }
+        if (ip == null || ip.length() == 0 || ip.equalsIgnoreCase("unknown")) {
+            ip = request.getRemoteAddr();
+        }
+        return ip;
+    }
+    /**
+     * N以内的不重复随机数
+     *
+     * @param min
+     *            最小值
+     * @param max
+     *            最大值
+     * @param n
+     * @return
+     */
+    public static int[] randomCommon(int min, int max, int n) {
+        int len = max - min + 1;
+        if (max < min || n > len) {
+            return new int[0];
+        }
+        // 初始化给定范围的待选数组
+        int[] source = new int[len];
+        for (int i = min; i < min + len; i++) {
+            source[i - min] = i;
+        }
+        int[] result = new int[n];
+        Random rd = new Random();
+        int index = 0;
+        for (int i = 0; i < result.length; i++) {
+            // 待选数组0到(len-2)随机一个下标
+            index = Math.abs(rd.nextInt() % len--);
+            // 将随机到的数放入结果集
+            result[i] = source[index];
+            // 将待选数组中被随机到的数，用待选数组(len-1)下标对应的数替换
+            source[index] = source[len];
+        }
+        return result;
+    }
+    public static int checkWebLogin(HttpServletRequest request,RedisTools redisTools){
+        JSONArray array=new JSONArray();
+        if(redisTools.get("bean")!=null){
+            array=JSONArray.parseArray(redisTools.get("bean"));
+        }
+        if(array.contains(Tools.getRemoteAddress(request))){
+            System.out.println("IP已被封禁");
+            return -100;
+        }
+        Cookie cookie = Tools.getCookie(request, "user");
+        if (cookie == null) {
+            return -1;
+        }
+        if (redisTools.get(cookie.getValue()).equals("ok")) {
+           return 1;
+        }
+        return 0;
+    }
 }