Go语言中将任意结构体安全导出为CSV文件的通用方案-Golang-PHP中文网

Go语言中将任意结构体安全导出为CSV文件的通用方案

本文介绍如何在go中设计一个类型安全、可扩展的通用函数，将任意json解析后的结构体（通过interface{}传入）自动转换并写入csv文件，重点讲解基于接口约束的优雅实现方式及反射方案的取舍。

本文介绍如何在go中设计一个类型安全、可扩展的通用函数，将任意json解析后的结构体（通过interface{}传入）自动转换并写入csv文件，重点讲解基于接口约束的优雅实现方式及反射方案的取舍。

在Go开发中，当需要将多种不同结构的JSON响应统一导出为CSV时，直接使用 interface{} 虽然提供了灵活性，但也牺牲了类型安全与编译期检查。最惯用且推荐的做法是放弃空接口，转而定义明确的行为契约——即通过自定义接口抽象导出能力。

✅ 推荐方案：基于接口的类型安全导出

我们定义一个 CsvWriter 接口，要求实现两个核心方法：

type CsvWriter interface {
    CSVHeaders() []string
    CSVRecord() []string // 单条记录（适用于单结构体）
    CSVRecords() [][]string // 多条记录（适用于切片类型，如 []Location）
}

以你提供的 Location 类型为例，可为其添加实现：

type Location struct {
    Name   string `json:"Name"`
    Region string `json:"Region"`
    Type   string `json:"Type"`
}

// 注意：原问题中 Location 是 []struct，建议拆分为原子类型 + 切片处理更清晰
func (l Location) CSVHeaders() []string { return []string{"Name", "Region", "Type"} }
func (l Location) CSVRecord() []string  { return []string{l.Name, l.Region, l.Type} }

// 对于切片类型，单独封装写入逻辑（见下文）

然后重构主函数，将 interface{} 替换为接口约束：

Go语言（Golang）1.26.0

Go语言（Golang）1.26.0版本官方下载，版本号 1.26.0，适合旧项目维护、兼容性测试和指定版本开发环境搭建。

下载

func decode_and_csv(my_response *http.Response, writer CsvWriter, filename string) error {
    f, err := os.Create(filename)
    if err != nil {
        return fmt.Errorf("failed to create file: %w", err)
    }
    defer f.Close()

    w := csv.NewWriter(f)
    defer w.Flush()

    // 写入表头
    if headers := writer.CSVHeaders(); len(headers) > 0 {
        if err := w.Write(headers); err != nil {
            return fmt.Errorf("failed to write headers: %w", err)
        }
    }

    // 写入数据行：支持单记录或批量记录
    switch v := writer.(type) {
    case interface{ CSVRecord() []string }:
        if record := v.CSVRecord(); len(record) > 0 {
            if err := w.Write(record); err != nil {
                return fmt.Errorf("failed to write record: %w", err)
            }
        }
    case interface{ CSVRecords() [][]string }:
        for _, record := range v.CSVRecords() {
            if err := w.Write(record); err != nil {
                return fmt.Errorf("failed to write record: %w", err)
            }
        }
    default:
        return fmt.Errorf("writer does not implement CSVRecord or CSVRecords")
    }

    return nil
}

调用示例（处理 []Location）：

var locations []Location
if err := json.NewDecoder(my_response.Body).Decode(&locations); err != nil {
    log.Fatal(err)
}

// 包装切片为可写入类型
type LocationSlice []Location

func (ls LocationSlice) CSVHeaders() []string { return []string{"Name", "Region", "Type"} }
func (ls LocationSlice) CSVRecords() [][]string {
    records := make([][]string, 0, len(ls))
    for _, l := range ls {
        records = append(records, []string{l.Name, l.Region, l.Type})
    }
    return records
}

err := decode_and_csv(my_response, LocationSlice(locations), "locations.csv")