node-karin 0.0.3 → 0.1.0

package/lib/db/redis_level.js

@@ -1,290 +1,290 @@
-import { Level } from 'level'
-export default class RedisLevel {
-  #level
-  #expireMap
-  id
-  constructor () {
-    const path = process.cwd() + '/data/db/RedisLevel'
-    this.#level = new Level(path, { valueEncoding: 'json' })
-    /**
-         * @type {'RedisLevel'} 唯一标识符 用于区分不同的数据库
-         */
-    this.id = 'RedisLevel'
-    /**
-         * 过期时间映射表
-         * @type {Map<string, number>} 键: 值 (过期时间)
-         */
-    this.#expireMap = new Map()
-    /**
-         * 开启过期时间处理
-         */
-    this.#expireHandle()
+import { Level } from 'level'
+export default class RedisLevel {
+  #level
+  #expireMap
+  id
+  constructor () {
+    const path = process.cwd() + '/data/db/RedisLevel'
+    this.#level = new Level(path, { valueEncoding: 'json' })
+    /**
+         * @type {'RedisLevel'} 唯一标识符 用于区分不同的数据库
+         */
+    this.id = 'RedisLevel'
+    /**
+         * 过期时间映射表
+         * @type {Map<string, number>} 键: 值 (过期时间)
+         */
+    this.#expireMap = new Map()
+    /**
+         * 开启过期时间处理
+         */
+    this.#expireHandle()
   }
-
-  /**
-     * 过期时间处理 每分钟检查一次
-     */
-  async #expireHandle () {
-    setInterval(async () => {
-      const now = Date.now()
-      // 获取代理对象的键值对数组
-      const entries = Array.from(this.#expireMap.entries())
-      for (const [key, expire] of entries) {
-        if (expire < now) {
-          await this.#level.del(key)
-          this.#expireMap.delete(key) // 通过代理的方式删除键值对
-        }
-      }
-    }, 60000)
-    /**
-         * 对expireMap进行代理 实现持久化 每次触发都保存到数据库
-         */
-    const handler = {
-      get: function (target, prop, receiver) {
-        if (prop === 'get' || prop === 'entries') {
-          return function (...args) {
-            const reflect = Reflect.get(target, prop).apply(target, args)
-            return typeof reflect === 'function' ? reflect.bind(target) : reflect
-          }
-        }
-        const reflect = Reflect.get(target, prop, receiver)
-        return typeof reflect === 'function' ? reflect.bind(target) : reflect
-      },
-      set: function (target, key, value) {
-        target.set(key, value)
-        // 修改后持久化到数据库
-        const reflect = Reflect.set(target, key, value)
-        return typeof reflect === 'function' ? reflect.bind(target) : reflect
-      },
-      deleteProperty: function (target, key) {
-        Reflect.deleteProperty(target, key)
-        // 删除后持久化到数据库
-        const reflect = Reflect.deleteProperty(target, key)
-        return typeof reflect === 'function' ? reflect.bind(target) : reflect
-      },
-    }
-    this.#expireMap = new Proxy(this.#expireMap, handler)
+
+  /**
+     * 过期时间处理 每分钟检查一次
+     */
+  async #expireHandle () {
+    setInterval(async () => {
+      const now = Date.now()
+      // 获取代理对象的键值对数组
+      const entries = Array.from(this.#expireMap.entries())
+      for (const [key, expire] of entries) {
+        if (expire < now) {
+          await this.#level.del(key)
+          this.#expireMap.delete(key) // 通过代理的方式删除键值对
+        }
+      }
+    }, 60000)
+    /**
+         * 对expireMap进行代理 实现持久化 每次触发都保存到数据库
+         */
+    const handler = {
+      get: function (target, prop, receiver) {
+        if (prop === 'get' || prop === 'entries') {
+          return function (...args) {
+            const reflect = Reflect.get(target, prop).apply(target, args)
+            return typeof reflect === 'function' ? reflect.bind(target) : reflect
+          }
+        }
+        const reflect = Reflect.get(target, prop, receiver)
+        return typeof reflect === 'function' ? reflect.bind(target) : reflect
+      },
+      set: function (target, key, value) {
+        target.set(key, value)
+        // 修改后持久化到数据库
+        const reflect = Reflect.set(target, key, value)
+        return typeof reflect === 'function' ? reflect.bind(target) : reflect
+      },
+      deleteProperty: function (target, key) {
+        Reflect.deleteProperty(target, key)
+        // 删除后持久化到数据库
+        const reflect = Reflect.deleteProperty(target, key)
+        return typeof reflect === 'function' ? reflect.bind(target) : reflect
+      },
+    }
+    this.#expireMap = new Proxy(this.#expireMap, handler)
   }
-
-  /**
-     * get 获取数据
-     * @param {string} key 键
-     * @returns {Promise<string>|Error} 数据
-     */
-  async get (key) {
-    try {
-      /** 先查过期时间 */
-      const expire = this.#expireMap.get(key)
-      if (expire && expire < Date.now()) {
-        await this.#level.del(key)
-        this.#expireMap.delete(key)
-        return null
-      }
-      return await this.#level.get(key)
-    } catch (error) {
-      if (error.notFound) { return null }
-      throw error
-    }
+
+  /**
+     * get 获取数据
+     * @param {string} key 键
+     * @returns {Promise<string>|Error} 数据
+     */
+  async get (key) {
+    try {
+      /** 先查过期时间 */
+      const expire = this.#expireMap.get(key)
+      if (expire && expire < Date.now()) {
+        await this.#level.del(key)
+        this.#expireMap.delete(key)
+        return null
+      }
+      return await this.#level.get(key)
+    } catch (error) {
+      if (error.notFound) { return null }
+      throw error
+    }
   }
-
-  /**
-     * set 设置数据
-     * @param {string} key 键
-     * @param {string} value 值
-     * @param {object} [options] 选项
-     * @param [options.EX] 过期时间 单位秒
-     * @returns {Promise<void>|Error}
-     */
-  async set (key, value, options) {
-    if (options && options.EX) {
-      this.#expireMap.set(key, Date.now() + options.EX * 1000)
-    }
-    return await this.#level.put(key, value)
+
+  /**
+     * set 设置数据
+     * @param {string} key 键
+     * @param {string} value 值
+     * @param {object} [options] 选项
+     * @param [options.EX] 过期时间 单位秒
+     * @returns {Promise<void>|Error}
+     */
+  async set (key, value, options) {
+    if (options && options.EX) {
+      this.#expireMap.set(key, Date.now() + options.EX * 1000)
+    }
+    return await this.#level.put(key, value)
   }
-
-  /**
-     * del 删除数据
-     * @param {string} key 键
-     * @returns {Promise<void>|Error}
-     */
-  async del (key) {
-    this.#expireMap.delete(key)
-    return await this.#level.del(key)
+
+  /**
+     * del 删除数据
+     * @param {string} key 键
+     * @returns {Promise<void>|Error}
+     */
+  async del (key) {
+    this.#expireMap.delete(key)
+    return await this.#level.del(key)
   }
-
-  /**
-     * keys 获取所有键
-     * @param {string} [prefix] 前缀
-     * @returns {Promise<string[]>|Error} 键列表
-     */
-  async keys (prefix = '') {
-    const keys = []
-    for await (const key of this.#level.keys({ gte: prefix, lt: prefix + '\xFF' })) {
-      // Check if the key has expired
-      const expire = this.#expireMap.get(key)
-      if (expire && expire < Date.now()) {
-        await this.#level.del(key)
-        this.#expireMap.delete(key)
-      } else {
-        keys.push(key)
-      }
-    }
-    return keys
+
+  /**
+     * keys 获取所有键
+     * @param {string} [prefix] 前缀
+     * @returns {Promise<string[]>|Error} 键列表
+     */
+  async keys (prefix = '') {
+    const keys = []
+    for await (const key of this.#level.keys({ gte: prefix, lt: prefix + '\xFF' })) {
+      // Check if the key has expired
+      const expire = this.#expireMap.get(key)
+      if (expire && expire < Date.now()) {
+        await this.#level.del(key)
+        this.#expireMap.delete(key)
+      } else {
+        keys.push(key)
+      }
+    }
+    return keys
   }
-
-  /**
-     * incr 递增数据
-     * @param {string} key 键
-     * @returns {Promise<number>|Error}
-     */
-  async incr (key) {
-    let value = Number(await this.get(key))
-    if (value === null) {
-      value = 0
-    } else {
-      value = Number(value)
-      if (isNaN(value)) { throw new Error('Value is not a number') }
-    }
-    value += 1
-    await this.set(key, value.toString())
-    return value
+
+  /**
+     * incr 递增数据
+     * @param {string} key 键
+     * @returns {Promise<number>|Error}
+     */
+  async incr (key) {
+    let value = Number(await this.get(key))
+    if (value === null) {
+      value = 0
+    } else {
+      value = Number(value)
+      if (isNaN(value)) { throw new Error('Value is not a number') }
+    }
+    value += 1
+    await this.set(key, value.toString())
+    return value
   }
-
-  /**
-     * decr 递减数据
-     * @param {string} key 键
-     * @returns {Promise<number>|Error}
-     */
-  async decr (key) {
-    let value = Number(await this.get(key))
-    if (value === null) {
-      value = 0
-    } else {
-      value = Number(value)
-      if (isNaN(value)) { throw new Error('Value is not a number') }
-    }
-    value -= 1
-    await this.set(key, value.toString())
-    return value
+
+  /**
+     * decr 递减数据
+     * @param {string} key 键
+     * @returns {Promise<number>|Error}
+     */
+  async decr (key) {
+    let value = Number(await this.get(key))
+    if (value === null) {
+      value = 0
+    } else {
+      value = Number(value)
+      if (isNaN(value)) { throw new Error('Value is not a number') }
+    }
+    value -= 1
+    await this.set(key, value.toString())
+    return value
   }
-
-  /**
-     * expire 设置过期时间
-     * @param {string} key 键
-     * @param seconds 过期时间 单位秒
-     * @returns {Promise<number>|Error}
-     */
-  async expire (key, seconds) {
-    this.#expireMap.set(key, Date.now() + seconds * 1000)
-    return seconds
+
+  /**
+     * expire 设置过期时间
+     * @param {string} key 键
+     * @param seconds 过期时间 单位秒
+     * @returns {Promise<number>|Error}
+     */
+  async expire (key, seconds) {
+    this.#expireMap.set(key, Date.now() + seconds * 1000)
+    return seconds
   }
-
-  /**
-     * ttl 获取过期时间
-     * @param {string} key 键
-     * @returns {Promise<number>|Error}
-     */
-  async ttl (key) {
-    const expire = this.#expireMap.get(key)
-    if (expire) {
-      return Math.ceil((expire - Date.now()) / 1000)
-    }
-    return -2
+
+  /**
+     * ttl 获取过期时间
+     * @param {string} key 键
+     * @returns {Promise<number>|Error}
+     */
+  async ttl (key) {
+    const expire = this.#expireMap.get(key)
+    if (expire) {
+      return Math.ceil((expire - Date.now()) / 1000)
+    }
+    return -2
   }
-
-  /**
-     * setEx 设置数据并设置过期时间
-     * @param {string} key 键
-     * @param seconds 过期时间 单位秒
-     * @param {string} value 值
-     * @returns {Promise<void>|Error}
-     */
-  async setEx (key, seconds, value) {
-    this.#expireMap.set(key, Date.now() + seconds * 1000)
-    return await this.#level.put(key, value)
+
+  /**
+     * setEx 设置数据并设置过期时间
+     * @param {string} key 键
+     * @param seconds 过期时间 单位秒
+     * @param {string} value 值
+     * @returns {Promise<void>|Error}
+     */
+  async setEx (key, seconds, value) {
+    this.#expireMap.set(key, Date.now() + seconds * 1000)
+    return await this.#level.put(key, value)
   }
-
-  /**
-     * exists 判断键是否存在
-     * @param {string} key 键
-     * @returns {Promise<number>|Error}
-     */
-  async exists (key) {
-    const value = await this.get(key)
-    return value === null ? 0 : 1
+
+  /**
+     * exists 判断键是否存在
+     * @param {string} key 键
+     * @returns {Promise<number>|Error}
+     */
+  async exists (key) {
+    const value = await this.get(key)
+    return value === null ? 0 : 1
   }
-
-  /**
-     * zAdd 有序集合添加元素
-     * @param {string} key 键
-     * @param {object} data 数据
-     * @param data.score 分数
-     * @param {string} data.value 值
-     */
-  async zAdd (key, { score, value }) {
-    const set = await this.get(key)
-    const arr = (set ? JSON.parse(set) : [])
-    arr.push({ score, value })
-    arr.sort((a, b) => a.score - b.score)
-    await this.set(key, JSON.stringify(arr))
+
+  /**
+     * zAdd 有序集合添加元素
+     * @param {string} key 键
+     * @param {object} data 数据
+     * @param data.score 分数
+     * @param {string} data.value 值
+     */
+  async zAdd (key, { score, value }) {
+    const set = await this.get(key)
+    const arr = (set ? JSON.parse(set) : [])
+    arr.push({ score, value })
+    arr.sort((a, b) => a.score - b.score)
+    await this.set(key, JSON.stringify(arr))
   }
-
-  /**
-     * zRem 有序集合删除元素
-     * @param {string} key 键
-     * @param {string} value 值
-     */
-  async zRem (key, value) {
-    const set = await this.get(key)
-    if (set === null) { return }
-    let arr = JSON.parse(set)
-    arr = arr.filter((item) => item.value !== value)
-    await this.set(key, JSON.stringify(arr))
+
+  /**
+     * zRem 有序集合删除元素
+     * @param {string} key 键
+     * @param {string} value 值
+     */
+  async zRem (key, value) {
+    const set = await this.get(key)
+    if (set === null) { return }
+    let arr = JSON.parse(set)
+    arr = arr.filter((item) => item.value !== value)
+    await this.set(key, JSON.stringify(arr))
   }
-
-  /**
-     * zIncrBy 有序集合分数递增
-     * @param {string} key 键
-     * @param increment 递增值
-     * @param {string} value 值
-     * @returns {Promise<number>|Error}
-     */
-  async zIncrBy (key, increment, value) {
-    const set = await this.get(key)
-    if (set === null) { throw new Error('Set does not exist') }
-    const arr = JSON.parse(set)
-    const item = arr.find((item) => item.value === value)
-    if (item) { item.score += increment }
-    arr.sort((a, b) => a.score - b.score)
-    await this.set(key, JSON.stringify(arr))
-    return item.score
+
+  /**
+     * zIncrBy 有序集合分数递增
+     * @param {string} key 键
+     * @param increment 递增值
+     * @param {string} value 值
+     * @returns {Promise<number>|Error}
+     */
+  async zIncrBy (key, increment, value) {
+    const set = await this.get(key)
+    if (set === null) { throw new Error('Set does not exist') }
+    const arr = JSON.parse(set)
+    const item = arr.find((item) => item.value === value)
+    if (item) { item.score += increment }
+    arr.sort((a, b) => a.score - b.score)
+    await this.set(key, JSON.stringify(arr))
+    return item.score
   }
-
-  /**
-     * zRangeByScore 有序集合根据分数范围获取元素
-     * @param {string} key 键
-     * @param min 最小分数
-     * @param max 最大分数
-     * @returns {Promise<string[]>|Error}
-     */
-  async zRangeByScore (key, min, max) {
-    const set = await this.get(key)
-    if (set === null) { return [] }
-    const arr = JSON.parse(set)
-    return arr.filter((item) => item.score >= min && item.score <= max).map((item) => item.value)
+
+  /**
+     * zRangeByScore 有序集合根据分数范围获取元素
+     * @param {string} key 键
+     * @param min 最小分数
+     * @param max 最大分数
+     * @returns {Promise<string[]>|Error}
+     */
+  async zRangeByScore (key, min, max) {
+    const set = await this.get(key)
+    if (set === null) { return [] }
+    const arr = JSON.parse(set)
+    return arr.filter((item) => item.score >= min && item.score <= max).map((item) => item.value)
   }
-
-  /**
-     * zScore 有序集合获取元素分数
-     * @param {string} key 键
-     * @param {string} value 值
-     * @returns {Promise<number>|Error}
-     */
-  async zScore (key, value) {
-    const set = await this.get(key)
-    if (set === null) { return null }
-    const arr = JSON.parse(set)
-    const item = arr.find((item) => item.value === value)
-    return item ? item.score : null
-  }
-}
+
+  /**
+     * zScore 有序集合获取元素分数
+     * @param {string} key 键
+     * @param {string} value 值
+     * @returns {Promise<number>|Error}
+     */
+  async zScore (key, value) {
+    const set = await this.get(key)
+    if (set === null) { return null }
+    const arr = JSON.parse(set)
+    const item = arr.find((item) => item.value === value)
+    return item ? item.score : null
+  }
+}