可以将最后返回的base64编码的图片改为使用Node.js的文件流传输图片。具体实现方法如下:

  1. 替换掉返回值中的isBase64Encoded: true,因为这个属性是告诉API Gateway返回的内容是否是base64编码的,我们需要用文件流传输图片,所以不需要这个属性。

  2. 将返回值中的body属性替换为一个带有文件流的对象。这个对象包含了文件流的内容,以及文件流的类型,并且不需要使用base64编码。具体实现方法如下:

const imageStream = fs.createReadStream(`/tmp/${output}`);
return {
  statusCode: 200,
  headers: {
    'Content-Type': 'image/png',
  },
  body: imageStream,
};
  1. 将Lambda函数的返回类型从JSON改为二进制类型,这样API Gateway就会正确地将文件流传输给客户端。具体实现方法是在Lambda函数的配置中将返回类型修改为"binary/octet-stream"。

完整的Lambda函数代码如下:

const fs = require('fs');
const path = require('path');
const puppeteer = require('puppeteer');

const launchBrowser = async () => {
  const browser = await puppeteer.launch({
    args: [
      '--disable-gpu',
      '--disable-dev-shm-usage',
      '--disable-setuid-sandbox',
      '--no-first-run',
      '--no-zygote',
      '--no-sandbox'
    ],
    headless: true,
  });

  return browser;
};

const autoScroll = async (page) => {
  return page.evaluate(() => {
    return new Promise((resolve, reject) => {
      var totalHeight = 0;
      var distance = 100;
      var timer = setInterval(() => {
        var scrollHeight = document.body.scrollHeight;
        window.scrollBy(0, distance);
        totalHeight += distance;
        if (totalHeight >= scrollHeight) {
          clearInterval(timer);
          resolve();
        }
      }, 100);
    });
  });
};

exports.handler = async (event, context) => {
  const input = JSON.parse(event.body).input;
  console.log('Input:', input);

  const fileinput = `${Date.now()}.mmd`;
  fs.writeFileSync(`/tmp/${fileinput}`, input);

  const output = `${Date.now()}.png`;
  const browser = await launchBrowser();
  const page = await browser.newPage();
  await page.emulateTimezone('Asia/Shanghai');

  try {
    const html = fs.readFileSync(path.join(__dirname, 'template.html'), 'utf8');
    const renderedHtml = html.replace('{{mermaid}}', input);
    const filehtml = `${Date.now()}.html`;
    fs.writeFileSync(`/tmp/${filehtml}`, renderedHtml);
    let url = 'file://' + path.join('/tmp', filehtml);
    await page.goto(url, { waitUntil: 'networkidle2' });
    await page.setViewport({ width: 1200, height: 800 });
    await autoScroll(page);
    await page.screenshot({ path: `/tmp/${output}`, fullPage: true, type: 'png' });

    console.log('Image generated:', output);
    const imageStream = fs.createReadStream(`/tmp/${output}`);
    return {
      statusCode: 200,
      headers: {
        'Content-Type': 'image/png',
      },
      body: imageStream,
    };
  } catch (err) {
    console.error(err);
    return {
      statusCode: 500,
      body: 'Error generating image',
    };
  } finally {
    await browser.close();
  }
};
``
可以把这个函数计算返回最后改为图片传输而不是bse64吗const fs = requirefs;const path = requirepath;const puppeteer = requirepuppeteer;const launchBrowser = async = const browser = await puppeteerlaunch args --disa

原文地址: https://www.cveoy.top/t/topic/gKON 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录