wildfire chat server source initial submit

2019-03-16 17:04:23 +08:00
parent e59217bd10
commit cab3ca0916
1466 changed files with 528862 additions and 15919 deletions
@@ -0,0 +1,119 @@
+package win.liyufan.im;
+
+import java.util.HashMap;
+import java.util.Iterator;
+import java.util.Map;
+
+/**
+ * 漏桶算法
+ * capacity * 1000是为了更精确, 漏水的小洞更小^~^
+ */
+public class RateLimiter {
+    private static final int DEFAULT_LIMIT_TIME_SECOND = 5;
+    private static final int DEFAULT_LIMIT_COUNT = 100;
+    private static final long expire = 2 * 60 * 60 * 1000;
+    private double rate = (double) DEFAULT_LIMIT_COUNT / (DEFAULT_LIMIT_TIME_SECOND);
+    private long capacity = DEFAULT_LIMIT_COUNT * 1000;
+    private long lastCleanTime;
+    private Map<String, Long> requestCountMap = new HashMap<>();
+    private Map<String, Long> requestTimeMap = new HashMap<>();
+
+    private SpinLock lock = new SpinLock();
+
+    public RateLimiter() {
+
+    }
+
+    public RateLimiter(int limitTimeSecond, int limitCount) {
+        if (limitTimeSecond <= 0 || limitCount <= 0) {
+            throw new IllegalArgumentException();
+        }
+        this.capacity = limitCount * 1000;
+        this.rate = (double) limitCount / limitTimeSecond;
+    }
+
+    /**
+     * 漏桶算法，https://en.wikipedia.org/wiki/Leaky_bucket
+     */
+    public boolean isGranted(String userId) {
+        try {
+            lock.lock();
+            long current = System.currentTimeMillis();
+            cleanUp(current);
+            Long lastRequestTime = requestTimeMap.get(userId);
+            long count = 0;
+            if (lastRequestTime == null) {
+                count += 1000;
+                requestTimeMap.put(userId, current);
+                requestCountMap.put(userId, count);
+                return true;
+            } else {
+                count = requestCountMap.get(userId);
+                lastRequestTime = requestTimeMap.get(userId);
+                count -= (current - lastRequestTime) * rate;
+                count = count > 0 ? count : 0;
+                requestTimeMap.put(userId, current);
+                if (count < capacity) {
+                    count += 1000;
+                    requestCountMap.put(userId, count);
+                    return true;
+                } else {
+                    requestCountMap.put(userId, count);
+                    return false;
+                }
+            }
+        } finally {
+            lock.unLock();
+        }
+    }
+
+    private void cleanUp(long current) {
+        if (current - lastCleanTime > expire) {
+            for (Iterator<Map.Entry<String, Long>> it = requestTimeMap.entrySet().iterator(); it.hasNext();) {
+                Map.Entry<String, Long> entry = it.next();
+                if (entry.getValue() < current - expire) {
+                    it.remove();
+                    requestCountMap.remove(entry.getKey());
+                }
+            }
+            lastCleanTime = current;
+        }
+    }
+
+    public static void main(String[] args) throws InterruptedException {
+        RateLimiter limiter = new RateLimiter(1, 10);
+        long start = System.currentTimeMillis();
+        for (int i = 0; i < 53; i++) {
+            if (!limiter.isGranted("test")) {
+                System.out.println("1 too frequency " + i);
+            }
+        }
+        Thread.sleep(1 * 1000);
+        System.out.println("sleep 1 s");
+        for (int i = 0; i < 53; i++) {
+            if (!limiter.isGranted("test")) {
+                System.out.println("2 too frequency " + i);
+            }
+        }
+
+        Thread.sleep(5 * 1000);
+        System.out.println("sleep 5 s");
+        for (int i = 0; i < 53; i++) {
+            if (!limiter.isGranted("test")) {
+                System.out.println("3 too frequency " + i);
+            }
+        }
+
+        Thread.sleep(5 * 1000);
+        System.out.println("sleep 5 s");
+        long second = System.currentTimeMillis();
+        for (int i = 0; i < 100; i++) {
+            if (!limiter.isGranted("test")) {
+                System.out.println("4 too frequency " + i);
+            }
+            Thread.sleep(50);
+        }
+        System.out.println("second: " + (System.currentTimeMillis() - second));
+        System.out.println("end: " + (System.currentTimeMillis() - start));
+    }
+}