使用Go语言：如何通过官方mongoexport包导出MongoDB文档？

你的现有代码仅从第一个文档提取键作为CSV表头，当集合中文档字段不一致时，后续文档里的额外键会被忽略，导致导出数据不完整。以下是优化方案：

核心思路 #

先遍历所有文档，收集所有唯一的键作为完整表头，再进行CSV导出，确保每个字段都能被包含，缺失字段自动留空。

优化后的代码 #

package main

import (
	"log"
	"os"
	"sort"

	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/x/mongo/driver/mongoexport"
)

func main() {
	// 假设docs是从MongoDB查询得到的[]*bson.D文档集合
	var docs []*bson.D

	// 第一步：遍历所有文档，收集所有唯一键
	keySet := make(map[string]struct{})
	for _, doc := range docs {
		for _, elem := range *doc {
			keySet[elem.Key] = struct{}{}
		}
	}

	// 将keySet转为有序切片（按字母排序保证表头顺序稳定）
	var keys []string
	for key := range keySet {
		keys = append(keys, key)
	}
	sort.Strings(keys)

	// 创建CSV文件
	export, err := os.Create("export.csv")
	if err != nil {
		log.Fatalf("创建导出文件失败: %v", err)
	}
	defer export.Close()

	// 初始化CSV导出器，使用完整的键集合
	csvExport := mongoexport.NewCSVExportOutput(keys, false, export)
	defer csvExport.Flush()

	// 写入表头
	if err := csvExport.WriteHeader(); err != nil {
		log.Fatalf("写入表头失败: %v", err)
	}

	// 遍历文档写入数据，单个文档失败时跳过而非中断任务
	for idx, doc := range docs {
		if err := csvExport.ExportDocument(*doc); err != nil {
			log.Printf("导出第%d个文档失败: %v", idx+1, err)
			continue
		}
	}

	log.Println("CSV导出完成")
}

额外优化建议 #

• 内存控制：如果文档数量极大，一次性遍历所有文档收集键会占用过多内存，可以分批次读取文档，每批收集键后合并到全局keySet中。
• 自定义排序：如果需要按业务逻辑指定表头顺序，可替换sort.Strings(keys)为自定义排序规则。
• 空值处理：若需要对缺失字段填充默认值（而非空字符串），可以在导出前手动给文档补充缺失键的默认值。

内容的提问来源于stack exchange，提问作者RamelHenderson