@domain.js/main 0.3.14 → 0.3.15

package/dist/deps/frequency/index.d.ts

@@ -6,9 +6,10 @@ interface Cnf {
 }
 export declare const Deps: readonly ["redis"];
 interface Deps {
-    redis: Pick<Redis, "hincrby" | "expire">;
+    redis: Pick<Redis, "hincrby" | "expire" | "hget" | "hdel">;
 }
 export declare function Main(cnf: Cnf, deps: Deps): {
+    control: (filed: string, ms: number, limit: number) => Promise<void>;
     check: (filed: string, ms: number, limit: number) => Promise<void>;
 };
 export {};

package/dist/deps/frequency/index.js

@@ -6,23 +6,51 @@ function Main(cnf, deps) {
     const { redis } = deps;
     const { frequency: { key: KEY }, } = cnf;
     /**
-     * 检测频次控制，超过显示过抛出异常
+     * 计算应该的key
+     * @param ms 频率控制长度，单位毫秒
+     */
+    const getKey = (ms) => {
+        const now = Date.now();
+        const t = Math.floor(now / ms);
+        return `${KEY}-${ms}-${t}`;
+    };
+    /**
+     * 计算过期时间
+     * @param ms 频率控制长度，单位毫秒
+     */
+    const expire = (ms) => {
+        const now = Date.now();
+        const t = Math.floor(now / ms);
+        return Math.floor((ms * (t + 1) - now) / 1000);
+    };
+    /**
+     * 频次控制，超过显示过抛出异常
      * @param filed 控频字段key
      * @param ms 控频周期长度 毫秒
      * @param limit 控频极限次数
      */
-    const check = async (filed, ms, limit) => {
-        const now = Date.now();
-        const t = Math.floor(now / ms);
-        const key = `${KEY}-${ms}-${t}`;
+    const control = async (filed, ms, limit) => {
+        const key = getKey(ms);
         const val = await redis.hincrby(key, filed, 1);
         if (val === 1)
-            await redis.expire(key, Math.floor((ms * (t + 1) - now) / 1000));
+            await redis.expire(key, expire(ms));
         if (val > limit)
             throw Error("Too many request");
     };
-    return {
-        check,
+    /**
+     * 检测频次控制，超过显示过抛出异常
+     * @param filed 控频字段key
+     * @param ms 控频周期长度 毫秒
+     * @param limit 控频极限次数
+     */
+    const check = async (filed, ms, limit) => {
+        const key = getKey(ms);
+        const val = await redis.hget(key, filed);
+        if (!val)
+            return;
+        if ((Number(val) | 0) > limit)
+            throw Error("Too many request");
     };
+    return { control, check };
 }
 exports.Main = Main;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@domain.js/main",
-  "version": "0.3.14",
+  "version": "0.3.15",
   "description": "DDD framework",
   "main": "dist/index.js",
   "bin": {