f

cmd/qygl_report_pdf/main.go

@@ -0,0 +1,59 @@
+package main
+
+import (
+	"context"
+	"encoding/json"
+	"flag"
+	"fmt"
+	"os"
+	"path/filepath"
+
+	"go.uber.org/zap"
+
+	"hyapi-server/internal/shared/pdf"
+)
+
+// 一个本地调试用的小工具：
+// 从 JSON 文件（企业报告.json）读取 QYGL 聚合结果，使用 gofpdf 生成企业全景报告 PDF，输出到当前目录。
+func main() {
+	var (
+		jsonPath string
+		outPath  string
+	)
+	flag.StringVar(&jsonPath, "json", "企业报告.json", "企业报告 JSON 数据源文件路径")
+	flag.StringVar(&outPath, "out", "企业全景报告_gofpdf.pdf", "输出 PDF 文件路径")
+	flag.Parse()
+
+	logger, _ := zap.NewDevelopment()
+	defer logger.Sync()
+
+	absJSON, _ := filepath.Abs(jsonPath)
+	fmt.Printf("读取 JSON 数据源：%s\n", absJSON)
+
+	data, err := os.ReadFile(jsonPath)
+	if err != nil {
+		fmt.Printf("读取 JSON 文件失败: %v\n", err)
+		os.Exit(1)
+	}
+
+	var report map[string]interface{}
+	if err := json.Unmarshal(data, &report); err != nil {
+		fmt.Printf("解析 JSON 失败: %v\n", err)
+		os.Exit(1)
+	}
+
+	fmt.Println("开始使用 gofpdf 生成企业全景报告 PDF...")
+	pdfBytes, err := pdf.GenerateQYGLReportPDF(context.Background(), logger, report)
+	if err != nil {
+		fmt.Printf("生成 PDF 失败: %v\n", err)
+		os.Exit(1)
+	}
+
+	if err := os.WriteFile(outPath, pdfBytes, 0644); err != nil {
+		fmt.Printf("写入 PDF 文件失败: %v\n", err)
+		os.Exit(1)
+	}
+
+	absOut, _ := filepath.Abs(outPath)
+	fmt.Printf("PDF 生成完成：%s\n", absOut)
+}