From de9dbb26cca5126b7b26bdc1fd7573634bd04c4f Mon Sep 17 00:00:00 2001
From: snowtafir <fengxuehuayue@gmail.com>
Date: Mon, 26 Aug 2024 23:12:05 +0800
Subject: [PATCH] optimize render

---
 utils/puppeteer.render.ts | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/utils/puppeteer.render.ts b/utils/puppeteer.render.ts
index b6b55fc..a56504d 100644
--- a/utils/puppeteer.render.ts
+++ b/utils/puppeteer.render.ts
@@ -101,7 +101,7 @@ export class YukiPuppeteerRender extends Puppeteer {
       let numSun = 0;
       let start = Date.now();
       const ret = new Array<Buffer>();
-      let buff: string | false | Buffer;
+      let buff: string | false | Uint8Array;
 
       for (let i = 1; i <= num; i++) {
         if (i > 1) {
@@ -134,15 +134,13 @@ export class YukiPuppeteerRender extends Puppeteer {
         numSun++; // 增加截图次数
 
         if (buff !== false) {
-          if (!Buffer.isBuffer(buff)) {
-            buff = Buffer.from(buff)
-          }
+          let imgBuff: Buffer = !Buffer.isBuffer(buff) ? Buffer.from(buff) : buff;
           /** 计算图片大小 */
-          const kb = (buff?.length / 1024).toFixed(2) + "kb"; // 计算图片大小
+          const kb = (imgBuff?.length / 1024).toFixed(2) + "kb"; // 计算图片大小
 
           logger.mark(`[图片生成][${name}][${numSun}次] ${kb} ${logger.green(`${Date.now() - start}ms`)}`); // 记录日志
 
-          ret.push(buff); // 将截图结果添加到数组中
+          ret.push(imgBuff); // 将截图结果添加到数组中
         } else {
           logger.error(`[puppeteer]`, '截图失败');
         }