Types

ilovepixelart · May 9, 2024 · 26e8a4b · 26e8a4b
1 parent 72e2ae5
commit 26e8a4b
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/src/cache/Cache.ts b/src/cache/Cache.ts
@@ -44,7 +44,7 @@ class CacheEngine {
     return cacheEntry
   }
 
-  async set(key: string, value: Record<string, unknown> | Record<string, unknown>[], ttl: string | null): Promise<void> {
+  async set(key: string, value: IData, ttl: string | null): Promise<void> {
     const actualTTL = ttl ? ms(ttl) : this.#defaultTTL
     await this.#engine.set(key, value, actualTTL)
     if (this.#debug) {

diff --git a/src/cache/engine/RedisCacheEngine.ts b/src/cache/engine/RedisCacheEngine.ts
@@ -24,7 +24,7 @@ class RedisCacheEngine implements ICacheEngine {
     return EJSON.parse(value) as Promise<Record<string, unknown> | Record<string, unknown>[]>
   }
 
-  async set(key: string, value: unknown, ttl = Infinity): Promise<void> {
+  async set(key: string, value: IData, ttl = Infinity): Promise<void> {
     const serializedValue = EJSON.stringify(convertToObject(value))
     await this.#client.setex(key, Math.ceil(ttl / 1000), serializedValue)
   }

diff --git a/src/version.ts b/src/version.ts
@@ -1,9 +1,10 @@
 import { satisfies } from 'semver'
 import mongoose from 'mongoose'
+import type IData from './interfaces/IData'
 
 export const isMongooseLessThan7 = satisfies(mongoose.version, '<7')
 
-export const convertToObject = (value: unknown): unknown => {
+export const convertToObject = <T>(value: T & { toObject?: () => IData } | undefined): IData => {
   if (isMongooseLessThan7) {
     if (value != null && typeof value === 'object' && !Array.isArray(value) && value.toObject) {
       // eslint-disable-next-line @typescript-eslint/no-unsafe-call