61 lines
1.4 KiB
Go
61 lines
1.4 KiB
Go
package main
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"flag"
|
||
"fmt"
|
||
"os"
|
||
"path/filepath"
|
||
|
||
"go.uber.org/zap"
|
||
|
||
"tyapi-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)
|
||
}
|
||
|