mysql-load-data.go

package main

import (
	"database/sql"
	"fmt"

	_ "github.com/go-sql-driver/mysql"
)

func main() {
	db, err := sql.Open("mysql", "root:@/gotest")
	if err != nil {
		panic(err.Error())
	}
	defer db.Close()

	tx, err := db.Begin()
	if err != nil {
		panic(err.Error())
	}

	if _, err = db.Exec("DROP TABLE IF EXISTS loadtest;"); err != nil {
		panic(err.Error())
	}
	if _, err = db.Exec("CREATE TABLE loadtest ( id INT, value1 VARCHAR(20),value2 VARCHAR(20) );"); err != nil {
		panic(err.Error())
	}

	var stmt *sql.Stmt
	//stmt, err = tx.Prepare("INSERT INTO loadtest (id, value1, value2 ) VALUES (?, ?, ?);")
	stmt, err = tx.Prepare("LOAD DATA LOCAL INFILE 'Data::Data' INTO TABLE loadtest (id, value1, value2);")
	if err != nil {
		panic(err.Error())
	}

	for i := 0; i < 10000; i++ {
		if _, err = stmt.Exec(i, "test1", "test2"); err != nil {
			panic(err.Error())
		}
	}
	stmt.Close()

	var count int
	if err = tx.QueryRow("SELECT COUNT(*) FROM loadtest;").Scan(&count); err != nil {
		panic(err.Error())
	}
	fmt.Printf("loadtest table count: %d\n", count)
	if err = tx.Commit(); err != nil {
		panic(err.Error())
	}
}