pg2csv/main.go

package main

import (
	"fmt"
	"pg2csv/utils"
)

func main() {
	// 测试加载表列表
	tables, err := utils.LoadTableList()
	if err != nil {
		fmt.Printf("加载表列表错误: %v\n", err)
	} else {
		fmt.Printf("加载的数据表列表: %v\n", tables)
	}

	// 测试加载数据库配置
	config, err := utils.LoadDBConfig()
	if err != nil {
		fmt.Printf("加载数据库配置错误: %v\n", err)
	} else {
		fmt.Printf("加载的数据库配置: %v\n", config)
	}

	if tables == nil || config == nil {
		fmt.Println("程序退出!")
		return
	}

	// 连接数据库
	db, err := utils.ConnectDB(config)
	if err != nil {
		fmt.Printf("数据库连接失败: %v\n", err)
		return
	}
	defer db.Close()

	fmt.Println("数据库连接成功!")

	// 导出所有表到CSV
	fmt.Println("开始导出数据表到CSV...")
	dbName := config["database"]
	if err := utils.ExportAllTables(db, dbName, tables); err != nil {
		fmt.Printf("导出数据失败: %v\n", err)
		return
	}

	fmt.Println("所有表导出完成!")
}