加入SaveCsvCfg

csv.go

@@ -4,6 +4,7 @@ import (
 	"encoding/csv"
 	"log"
 	"os"
+	"reflect"
 	"strconv"
 )
 
@@ -116,3 +117,47 @@ func LoadCsvCfg(filename string, row int) *CsvTable {
 	}
 	return result
 }
+
+//SaveCsvCfg 导出CSV文件
+func SaveCsvCfg(table []map[string]interface{}, filename string) {
+	// 创建csv文件
+	file, err := os.Create(filename)
+	if err != nil {
+		log.Fatalln(err.Error())
+	}
+	defer file.Close()
+
+	// 创建csv writer
+	cw := csv.NewWriter(file)
+
+	var wKey []string
+
+	// 循环遍历传入的数据数组
+	for index, item := range table {
+		var wData []string
+		for key, dataitem := range item {
+			if index == 0 {
+				wKey = append(wKey, key)
+			}
+
+			switch reflect.TypeOf(dataitem).String() {
+			case "bool":
+				boolData := strconv.FormatBool(dataitem.(bool))
+				wData = append(wData, boolData)
+			case "int":
+				intData := strconv.Itoa(dataitem.(int))
+				wData = append(wData, intData)
+			case "float64":
+				floatData := strconv.FormatFloat(dataitem.(float64), 'f', -1, 64)
+				wData = append(wData, floatData)
+			default:
+				wData = append(wData, dataitem.(string))
+			}
+		}
+		if index == 0 {
+			cw.Write(wKey)
+		}
+		cw.Write(wData)
+	}
+	cw.Flush()
+}

go.mod

@@ -0,0 +1,3 @@
+module github.com/sersoong/go-csv
+
+go 1.13

test/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"encoding/json"
 	"fmt"
 
 	csvmgr "github.com/sersoong/go-csv"
@@ -14,11 +15,24 @@ type TestTable struct {
 	Text  string
 }
 
+//RespData RespData
+type RespData struct {
+	Data []map[string]interface{} `json:"data"`
+}
+
+var jsondata = `{"data":[{"id":0,"name":"张三","value":"aaaaa"},{"id":1,"name":"李四","value":"bbbbb"}]}`
 var gTestTable map[int]*TestTable
 
 func main() {
 	LoadTestTable()
-	fmt.Print(gTestTable[1])
+
+	var ret RespData
+	json.Unmarshal([]byte(jsondata), &ret)
+
+	csvmgr.SaveCsvCfg(ret.Data, "./out.csv")
+	for _, item := range gTestTable {
+		fmt.Println(item)
+	}
 }
 
 //LoadTestTable 载入测试数据表

test/out.csv

@@ -0,0 +1,3 @@
+id,name,value
+0,张三,aaaaa
+1,李四,bbbbb