ossSts添加缓存

tone-page-server/src/oss/oss.service.ts

@@ -9,10 +9,40 @@ export class OssService {
         accessKeySecret: process.env.ALIYUN_ACCESS_KEY_SECRET,
     });
 
+    private stsCache: {
+        [session: string]: {
+            credentials: {
+                AccessKeyId: string;
+                AccessKeySecret: string;
+                SecurityToken: string;
+                Expiration: string;
+            };
+            expireTime: number; // 时间戳，单位为毫秒
+        }
+    } = {};
+
+    /** @todo 该方法存在缓存穿透问题，待优化 */
     async getStsToken(session: string) {
+        if (this.stsCache[session]) {
+            const cached = this.stsCache[session];
+            // 检查缓存是否过期
+            if (cached.expireTime > Date.now()) {
+                return cached.credentials;
+            } else {
+                // 如果过期，删除缓存
+                delete this.stsCache[session];
+            }
+        }
+
         return this.sts.assumeRole(
             process.env.ALIYUN_OSS_STS_ROLE_ARN, ``, 3600, `${session}`,
         ).then((res) => {
+            // 缓存
+            this.stsCache[session] = {
+                credentials: res.credentials,
+                expireTime: new Date(res.credentials.Expiration).getTime() - 5 * 60 * 1000, // 提前5分钟过期,
+            };
+
             return res.credentials;
         }).catch(err => {
             console.error('获取STS Token失败:', err);