@milkio/redis 1.0.0-alpha.42

package/.publish/publish.json

@@ -0,0 +1,7 @@
+{
+  "enableReleases": false,
+  "githubOwner": "akirarika",
+  "githubRepo": "milkio",
+  "giteeOwner": "akirarika",
+  "giteeRepo": "milkio"
+}

package/README.md

@@ -0,0 +1,15 @@
+# milkio-redis
+
+To install dependencies:
+
+```bash
+bun install
+```
+
+To run:
+
+```bash
+bun run index.ts
+```
+
+This project was created using `bun init` in bun v1.1.30. [Bun](https://bun.sh) is a fast all-in-one JavaScript runtime.

package/index.ts

@@ -0,0 +1,110 @@
+import type { RedisClientOptions } from 'redis'
+import { TSON } from '@southern-aurora/tson'
+
+export async function createRedis<Options extends RedisClientOptions>(options: Options) {
+  const NodeRedis = await import('redis')
+  const redis = await NodeRedis.default.createClient(options).connect()
+
+  const milkioRedis = {
+    raw: redis,
+    useCache: <T>(key: string, defaultValue: T | undefined = undefined) => ({
+      set: async (value: T, expireMs: number): Promise<T> => {
+        await redis.PSETEX(key, expireMs, TSON.stringify(value))
+        return value
+      },
+      get: async (): Promise<undefined | T> => {
+        const result = await redis.GET(key)
+        if (result === null) return defaultValue
+        return TSON.parse(result)
+      },
+      pull: async () => {
+        const resultRaw = await redis.MULTI().GET(key).DEL(key).EXEC()
+        const result = resultRaw[0]
+        if (result === null) return defaultValue
+        return TSON.parse(result as string)
+      },
+      has: async (): Promise<boolean> => {
+        const result = await redis.GET(key)
+        return result !== null
+      },
+      del: async () => {
+        await redis.DEL(key)
+      },
+    }),
+    useCount: (key: string) => ({
+      get: async (): Promise<number> => {
+        const result = await redis.GET(key)
+        return result ? Number(result) : 0
+      },
+      add: async (amount: number, expireMs?: number): Promise<number> => {
+        if (!expireMs) {
+          const result = await redis.INCRBY(key, amount)
+          return result
+        }
+        else {
+          const result = await redis.MULTI().INCRBY(key, amount).PEXPIRE(key, expireMs).EXEC()
+          return Number(result[0])
+        }
+      },
+      sub: async (key: string, amount: number): Promise<number> => {
+        const result = await redis.DECRBY(key, amount)
+        return result
+      },
+    }),
+    useResultCache: async <Handler extends () => unknown | Promise<unknown>>(key: string, expireMs: number, handler: Handler, options?: { realExpireMs?: number, lockInterval?: number }): Promise<Awaited<ReturnType<Handler>>> => {
+      const resultRaw = await redis.get(key)
+      if (resultRaw) {
+        const result: { T: number, R: any } = TSON.parse(resultRaw)
+        if (result.T > new Date().getTime()) return result.R
+        const lock = await redis.GET(`${key}:lock`)
+        if (lock === '1') return result.R
+        await redis.PSETEX(`${key}:lock`, options?.lockInterval ?? 6000, '1')
+      }
+      const result = { R: (await handler()) as Awaited<ReturnType<Handler>>, T: new Date().getTime() + expireMs }
+      await redis.PSETEX(key, expireMs + (options?.realExpireMs ?? expireMs + Math.floor(expireMs * Math.random())) + (options?.lockInterval ?? 6000), TSON.stringify(result))
+
+      return result.R
+    },
+    useClockIn: (key: string, cleanDate: Date) => ({
+      clockIn: async (offset: number): Promise<void> => {
+        await redis.MULTI().SETBIT(key, offset, 1).PEXPIREAT(key, cleanDate.getTime()).EXEC()
+      },
+      check: async (offset: number): Promise<boolean> => {
+        const result = await redis.GETBIT(key, offset)
+        return result === 1
+      },
+      firstClockIn: async (): Promise<number> => {
+        const result = await redis.BITPOS(key, 1)
+        return result
+      },
+      lastClockIn: async (): Promise<number> => {
+        const result = await redis.BITPOS(key, 1, -1)
+        return result
+      },
+      toArray: async (length: number): Promise<boolean[]> => {
+        const resultRaw = await redis.BITFIELD(key, [
+          {
+            operation: 'GET',
+            encoding: `u${length}`,
+            offset: `#0`,
+          },
+        ])
+        const result = Number.parseInt(`${resultRaw}`).toString(2).split('')
+        const fill = []
+        for (let i = 0; i < length - result.length; i++) fill.push('0')
+        return [...fill, ...result].map(v => (v === '1'))
+      },
+      count: async (): Promise<number> => {
+        const result = await redis.BITCOUNT(key)
+        return result
+      },
+      clean: async (): Promise<void> => {
+        await redis.DEL(key)
+      },
+    }),
+  }
+
+  return milkioRedis
+}
+
+export type Redis = Awaited<ReturnType<typeof createRedis>>

package/package.json

@@ -0,0 +1,11 @@
+{
+  "name": "@milkio/redis",
+  "type": "module",
+  "version": "1.0.0-alpha.42",
+  "module": "index.ts",
+  "devDependencies": {
+    "@southern-aurora/tson": "*",
+    "@types/bun": "latest",
+    "redis": "^4.7.0"
+  }
+}

package/tsconfig.json

@@ -0,0 +1,27 @@
+{
+  "compilerOptions": {
+    "target": "ESNext",
+    "jsx": "react-jsx",
+    // Enable latest features
+    "lib": ["ESNext", "DOM"],
+    "moduleDetection": "force",
+    "module": "ESNext",
+
+    // Bundler mode
+    "moduleResolution": "bundler",
+    "allowImportingTsExtensions": true,
+    "allowJs": true,
+
+    // Best practices
+    "strict": true,
+    "noFallthroughCasesInSwitch": true,
+
+    "noPropertyAccessFromIndexSignature": false,
+    // Some stricter flags (disabled by default)
+    "noUnusedLocals": false,
+    "noUnusedParameters": false,
+    "noEmit": true,
+    "verbatimModuleSyntax": true,
+    "skipLibCheck": true
+  }
+}