go语言中将json格式数据文件保存到csv文件

Go  /  管理员 发布于 2年前   1216

需求：将JSON编码数据的数据文件并将其保存到CSV文件

data.json文件 数据格式：

 [
 {"Name":"Adam","Age":36,"Job":"CEO"},
 {"Name":"Eve","Age":34,"Job":"CFO"},
 {"Name":"Mike","Age":38,"Job":"COO"}
 ]

输出至data.csv文件 数据格式：

 Adam,36,CEO
 Eve,34,CFO
 Mike,38,COO

代码示例：

package controllers

import (
	"encoding/csv"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"strconv"
	"os"
)

// TestController  is a test control
type TestController struct {
	//beego.Controller
	BaseController
}


type Employee struct {
	Name string
	Age  int
	Job  string
}

func (c *TestController) Json2csv() {
	c.Ctx.WriteString(`666`)

	//从logs/data.json文件中读取数据
	jsondatafromfile, err := ioutil.ReadFile("logs/data.json")

	if err != nil {
		fmt.Println(err)
	}

	//解析JSON数据
	var jsondata []Employee
	err = json.Unmarshal([]byte(jsondatafromfile), &jsondata)
	fmt.Println(jsondata)
	if err != nil {
		fmt.Println(err)
	}

	//创建空文件logs/data.csv
	csvdatafile, err := os.Create("logs/data.csv")
	if err != nil {
		fmt.Println(err)
	}
	defer csvdatafile.Close()

	//写入data.csv文件
	writer := csv.NewWriter(csvdatafile)

	for _, worker := range jsondata {
		var record []string
		record = append(record, worker.Name)
		record = append(record, strconv.Itoa(worker.Age))
		record = append(record, worker.Job)
		writer.Write(record)
	}

	//如上的操作还是在缓冲区中写数据，数据还在内存中
	writer.Flush() //将内存中的数据刷回到硬盘保存

}

效果：