可以把这个函数计算返回最后改为图片传输而不是bse64吗const fs = requirefs;const path = requirepath;const puppeteer = requirepuppeteer;const launchBrowser = async = const browser = await puppeteerlaunch args --disa
可以将最后返回的base64编码的图片改为使用Node.js的文件流传输图片。具体实现方法如下:
-
替换掉返回值中的
isBase64Encoded: true,因为这个属性是告诉API Gateway返回的内容是否是base64编码的,我们需要用文件流传输图片,所以不需要这个属性。 -
将返回值中的
body属性替换为一个带有文件流的对象。这个对象包含了文件流的内容,以及文件流的类型,并且不需要使用base64编码。具体实现方法如下:
const imageStream = fs.createReadStream(`/tmp/${output}`);
return {
statusCode: 200,
headers: {
'Content-Type': 'image/png',
},
body: imageStream,
};
- 将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();
}
};
``
原文地址: https://www.cveoy.top/t/topic/gKON 著作权归作者所有。请勿转载和采集!