go语言将数据库中单表转成csv文件输出
2021/8/19 19:06:50
本文主要是介绍go语言将数据库中单表转成csv文件输出,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
直接贴出代码
package main import ( "database/sql" "encoding/csv" "fmt" _ "github.com/go-sql-driver/mysql" "os" ) // 从Mysql中导出数据到CSV文件。 var ( tables = []string{"user_info"} count = len(tables) ch = make(chan bool, count) ) func main() { db, err := sql.Open("mysql", "root@tcp(192.168.202.90:4000)/tidb_test?charset=utf8mb4&parseTime=True") // defer db.Close() if err != nil { panic(err.Error()) } for _, table := range tables { go querySQL(db, table, ch) } for i := 0; i < count; i++ { <-ch } fmt.Println("Successfully Done!!!") } func querySQL(db *sql.DB, table string, ch chan bool) { fmt.Println("开始处理:", table) rows, _ := db.Query(fmt.Sprintf("SELECT * from %s", table)) columns, err := rows.Columns() if err != nil { panic(err.Error()) } //values:一行的所有值,长度==列数 values := make([]sql.RawBytes, len(columns)) // print(len(values)) scanArgs := make([]interface{}, len(values)) for i := range values { scanArgs[i] = &values[i] } totalValues := [][]string{} for rows.Next() { var s []string err = rows.Scan(scanArgs...) //把每行的内容添加到scanArgs,也添加到了values if err != nil { panic(err.Error()) } for _, v := range values { s = append(s, string(v)) // print(len(s)) } totalValues = append(totalValues, s) } if err = rows.Err(); err != nil { panic(err.Error()) } writeToCSV(table+".csv", columns, totalValues) ch <- true } func writeToCSV(file string, columns []string, totalValues [][]string) { // fmt.Println(columns) f, err := os.Create(file) if err != nil { panic(err) } f.WriteString("\xEF\xBB\xBF") defer f.Close() w := csv.NewWriter(f) for a, i := range totalValues { if a == 0 { w.Write(columns) w.Write(i) } else { // fmt.Println(i) w.Write(i) } } w.Flush() fmt.Println("处理完毕:", file) }
这篇关于go语言将数据库中单表转成csv文件输出的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-06-26解决google chrome helper 内存占用较高!
- 2024-04-01got an unexpected keyword argument
- 2024-03-30维多利亚的秘密 golang入坑系统
- 2024-03-29mongodb sort by date
- 2024-03-29go swagger
- 2024-03-25mongodb cdc
- 2024-03-25how to use go in vscode
- 2024-03-22mongooseserverselectionerror: connect econnrefused ::1:27017
- 2024-03-21pymongo insert_many
- 2024-03-18projection mongodb