feat: token 缓存，不用每次都生成

Author: winily

dist/core/jwt.cjs

@@ -35,8 +35,13 @@ __export(jwt_exports, {
 module.exports = __toCommonJS(jwt_exports);
 var import_jsonwebtoken = __toESM(require("jsonwebtoken"), 1);
 var API_TOKEN_TTL_SECONDS = 3 * 60;
-var generateToken = (apiSecretKey) => {
+var CACHE_TTL_SECONDS = API_TOKEN_TTL_SECONDS - 30;
+var tokenCache = {};
+var generateToken = (apiSecretKey, cache = true) => {
   try {
+    if (tokenCache[apiSecretKey] && Date.now() - tokenCache[apiSecretKey].createAt < CACHE_TTL_SECONDS * 1e3) {
+      return tokenCache[apiSecretKey].token;
+    }
     const [apiKey, secret] = apiSecretKey.split(".");
     const payload = {
       "api_key": apiKey,
@@ -47,6 +52,12 @@ var generateToken = (apiSecretKey) => {
       algorithm: "HS256",
       header: { alg: "HS256", sign_type: "SIGN" }
     });
+    if (cache) {
+      tokenCache[apiSecretKey] = {
+        token: ret,
+        createAt: Date.now()
+      };
+    }
     return ret;
   } catch (e) {
     throw "invalid api_key";

dist/core/jwt.cjs.map

@@ -1,3 +1,3 @@
-declare const generateToken: (apiSecretKey: string) => never;
+declare const generateToken: (apiSecretKey: string, cache?: boolean) => string;
 
 export { generateToken };

dist/core/jwt.d.cts

@@ -1,3 +1,3 @@
-declare const generateToken: (apiSecretKey: string) => never;
+declare const generateToken: (apiSecretKey: string, cache?: boolean) => string;
 
 export { generateToken };

dist/core/jwt.d.ts

@@ -69,8 +69,13 @@ var import_assert = __toESM(require("assert"), 1);
 // lib/core/jwt.ts
 var import_jsonwebtoken = __toESM(require("jsonwebtoken"), 1);
 var API_TOKEN_TTL_SECONDS = 3 * 60;
-var generateToken = (apiSecretKey) => {
+var CACHE_TTL_SECONDS = API_TOKEN_TTL_SECONDS - 30;
+var tokenCache = {};
+var generateToken = (apiSecretKey, cache = true) => {
   try {
+    if (tokenCache[apiSecretKey] && Date.now() - tokenCache[apiSecretKey].createAt < CACHE_TTL_SECONDS * 1e3) {
+      return tokenCache[apiSecretKey].token;
+    }
     const [apiKey, secret] = apiSecretKey.split(".");
     const payload = {
       "api_key": apiKey,
@@ -81,6 +86,12 @@ var generateToken = (apiSecretKey) => {
       algorithm: "HS256",
       header: { alg: "HS256", sign_type: "SIGN" }
     });
+    if (cache) {
+      tokenCache[apiSecretKey] = {
+        token: ret,
+        createAt: Date.now()
+      };
+    }
     return ret;
   } catch (e) {
     throw "invalid api_key";
@@ -253,13 +264,15 @@ var ZhipuAI = class {
     __publicField(this, "images");
     __publicField(this, "embeddings");
     __publicField(this, "files");
+    var _a;
     if (!options.apiKey)
       options.apiKey = process.env["ZHIPUAI_API_KEY"] || "";
     import_assert.default.ok(options.apiKey, "\u672A\u63D0\u4F9Bapi_key\uFF0C\u8BF7\u901A\u8FC7\u53C2\u6570\u6216\u73AF\u5883\u53D8\u91CF\u63D0\u4F9B");
     if (!options.baseUrl)
       options.baseUrl = process.env["ZHIPUAI_BASE_URL"] || "";
     if (!options.baseUrl)
       options.baseUrl = "https://open.bigmodel.cn/api/paas/v4";
+    options.cacheToken = (_a = options.cacheToken) != null ? _a : true;
     this.request = new Request({
       timeout: options.timeout,
       headers: options.customHeaders,
@@ -296,7 +309,7 @@ var ZhipuAI = class {
     });
   }
   authHeaders() {
-    const token = generateToken(this.options.apiKey);
+    const token = generateToken(this.options.apiKey, this.options.cacheToken);
     return { "Authorization": token };
   }
 };