Files
SQLBot/g2-ssr/app.js

78 lines
2.0 KiB
JavaScript
Raw Permalink Normal View History

2025-09-08 16:35:29 +08:00
const http = require('http');
const url = require("url");
const util = require('util');
const { createChart } = require('@antv/g2-ssr');
const port = 3000;
const { getPieOptions } = require('./charts/pie.js');
const { getLineOptions } = require('./charts/line.js');
const { getColumnOptions } = require('./charts/column.js');
const { getBarOptions } = require('./charts/bar.js');
http.createServer((req, res) => {
res.statusCode = 200,
res.setHeader('Content-Type', 'text/plain;charset=utf-8');
if (req.method === 'GET') {
toGet(req, res);
} else if (req.method === 'POST') {
toPost(req, res);
}
}).listen(port, () => {
console.info(`Server listening on: http://localhost:${port}`);
});
function getOptions(type, axis, data) {
const base_options = {
width: 640,
height: 480,
imageType: 'png',
theme: {
view: {
viewFill: '#FFFFFF',
},
}
}
switch (type) {
case 'bar':
return getBarOptions(base_options, axis, data);
case 'column':
return getColumnOptions(base_options, axis, data);
case 'line':
return getLineOptions(base_options, axis, data);
case 'pie':
return getPieOptions(base_options, axis, data);
}
return base_options
}
// 创建 Chart 和配置
async function GenerateCharts(obj) {
const options = getOptions(obj.type, JSON.parse(obj.axis), JSON.parse(obj.data));
const chart = await createChart(options);
// 导出
chart.exportToFile(obj.path || 'chart');
// -> chart.png
chart.toBuffer();
}
// -> get buffer
//获取GET请求内容
function toGet(req, res) {
let data = 'GET请求内容\n' + util.inspect(url.parse(req.url));
res.end(data);
}
//获取POST请求内容、cookie
function toPost(req, res) {
req.on('data', async function (chunk) {
await GenerateCharts(JSON.parse(chunk))
res.end('complete');
});
}