rui.jiang
/
performance-test


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166
							import puppeteer from "puppeteer";
import fs from "fs-extra";
import path from "path";
import FormData from "form-data";
import axios from "axios"; // 导入 axios
import { colorSchemes } from "../colors.js";
import { text } from "stream/consumers";

export const getYawErrorBarSumCharts = async (
  data,
  bucketName,
  objectName,
  analysisTypeCode
) => {
  try {
    // 提取静态偏航误差值和机组数量
    let a = [];
    let b = [];
    let c = [];
    data.map((item) => {
      if (item["[0,3]"] != "0.0") {
        a.push(item["[0,3]"]);
      }
      if (item["(3,5]"] != "0.0") {
        b.push(item["(3,5]"]);
      }
      if (item["(5, )"] != "0.0") {
        c.push(item["(5, )"]);
      }
    });

    const xData = ["(0,3]", "(3,5]", "(>5]"]; // 偏航误差区间
    const yData = [a.length, b.length, c.length]; // 各区间的机组数量

    // 每个柱子的颜色
    const colors = [
      "#8AC8BE", // (0, 3] 蓝色
      "#407DB3", // (3, 5] 绿色
      "#1B2973", // (5, ∞] 红色
    ];

    const trace = {
      x: xData, // 横坐标数据
      y: yData, // 纵坐标数据
      type: "bar", // 当前图表类型
      marker: {
        color: colors, // 为每个柱子分配不同的颜色
      },
      // hovertemplate:
      //   `偏航误差值:` + ` %{x} <br> ` + `台数:` + "%{y} <br> <extra></extra>",
    };

    const layout = {
      title: {
        text: "静态偏航误差的绝对值机组台数分布情况",
        font: {
          size: 16,
          weight: "bold",
        },
      }, // 图表标题
      xaxis: {
        title: { text: "静态偏航误差值（度）" }, // 横坐标标题
        gridcolor: "rgb(255,255,255)",
        tickcolor: "rgb(255,255,255)",
        backgroundcolor: "#e5ecf6",
      },
      yaxis: {
        title: {
          text: "台数",
        }, // 纵坐标标题
        gridcolor: "rgb(255,255,255)",
        tickcolor: "rgb(255,255,255)",
        backgroundcolor: "#e5ecf6",
      },
      margin: {
        l: 50,
        r: 50,
        t: 50,
        b: 50,
      },
      autosize: true, // 开启自适应
      // showlegend: true, // 显示图例
      plot_bgcolor: "#e5ecf6",
      gridcolor: "#fff",
      bgcolor: "#e5ecf6", // 设置背景颜色
    };

    // 创建临时目录
    const tempDir = path.join(process.cwd(), "images");
    await fs.ensureDir(tempDir);
    const tempFilePath = path.join(
      tempDir,
      `temp_yaw_error_bar_sum_chart_${Date.now()}.png`
    );
    // 获取 plotly.js 的绝对路径
    const plotlyPath = path.join(
      process.cwd(),
      "src",
      "public",
      "js",
      "plotly-3.0.1.min.js"
    );
    const plotlyContent = await fs.readFile(plotlyPath, "utf-8");

    // 使用 Puppeteer 生成图表的截图
    const browser = await puppeteer.launch({
      headless: "new",
      args: ["--no-sandbox", "--disable-setuid-sandbox"],
    });
    try {
      const page = await browser.newPage();
      const htmlContent = `
        <!DOCTYPE html>
        <html>
          <head>
            <meta charset="UTF-8">
            <title>静态偏航误差值</title>
            <script>${plotlyContent}</script>
            <style>
              body { margin: 0; }
              #chart { width: 800px; height: 600px; }
            </style>
          </head>
          <body>
            <div id="chart"></div>
            <script>
              window.onload = function() {
                Plotly.newPlot('chart', [${JSON.stringify(
                  trace
                )}], ${JSON.stringify(layout)}).then(() => {
                  window.chartRendered = true;
                });
              };
            </script>
          </body>
        </html>
      `;
      await page.setContent(htmlContent, { waitUntil: "networkidle0" });
      await page.waitForFunction(() => window.chartRendered === true, {
        timeout: 60000,
      });
      // 截图并保存到临时文件
      const chartElement = await page.$("#chart");
      await chartElement.screenshot({ path: tempFilePath, type: "png" });
      // 上传图片到服务器
      const formData = new FormData();
      formData.append("file", fs.createReadStream(tempFilePath));
      const response = await axios.post(
        `${process.env.API_BASE_URL}/examples/upload`,
        {
          filePath: tempFilePath,
          bucketName,
          objectName,
        }
      );
      return response.data.url;
      console.log("上传成功:", response.data);
    } catch (error) {
      console.error("生成图表失败:", error);
    } finally {
      await browser.close();
    }
  } catch (error) {
    console.error("发生错误:", error);
  }
};