完善设置

hydro/model/opcount.js

@@ -3,26 +3,26 @@ const db = require('../service/db.js');
 
 const coll = db.collection('opcount');
 
-module.exports = {
-    /**
-     * @param {string} op
-     * @param {string} ident
-     * @param {number} period_secs
-     * @param {number} max_operations
-     */
-    async inc(op, ident, periodSecs, maxOperations) {
-        const curTime = new Date().getTime();
-        const beginAt = new Date(curTime - (curTime % (periodSecs * 1000)));
-        const expireAt = new Date(beginAt.getTime() + periodSecs * 1000);
-        try {
-            await coll.findOneAndUpdate({
-                ident,
-                beginAt,
-                expireAt,
-                op: { $not: { $gte: maxOperations } },
-            }, { $inc: { op: 1 } }, { upsert: true });
-        } catch (e) {
-            throw new OpcountExceededError(op, periodSecs, maxOperations);
-        }
-    },
-};
+/**
+ * @param {string} op
+ * @param {string} ident
+ * @param {number} period_secs
+ * @param {number} max_operations
+ */
+async function inc(op, ident, periodSecs, maxOperations) {
+    const curTime = new Date().getTime();
+    const beginAt = new Date(curTime - (curTime % (periodSecs * 1000)));
+    const expireAt = new Date(beginAt.getTime() + periodSecs * 1000);
+    try {
+        await coll.findOneAndUpdate({
+            ident,
+            beginAt,
+            expireAt,
+            op: { $not: { $gte: maxOperations } },
+        }, { $inc: { op: 1 } }, { upsert: true });
+    } catch (e) {
+        throw new OpcountExceededError(op, periodSecs, maxOperations);
+    }
+}
+
+global.Hydro.model.opcount = module.exports = { inc };