使用Go语言进行MySQL数据库的数据导入的方法
MySQL是当前非常流行的关系型数据库，越来越多的开发者选择使用Go语言与MySQL进行数据交互，本文将介绍如何使用Go语言进行MySQL数据库的数据导入。
准备数据源在Go语言中，可以使用CSV、JSON、XML等格式的文件作为数据源进行导入。下面以CSV文件为例进行介绍：
id,name,age,gender
1,张三,18,男
2,李四,20,男
3,王五,22,女
登录后复制
通过上面的示例可以看到，每行数据以逗号分隔，第一行为字段名，后面的行为具体的数据记录。
go-sql-driver/mysql库，使用Open()函数连接MySQL数据库。示例代码：
import (
"database/sql"
_ "github.com/go-sql-driver/mysql"
)
db, err := sql.Open("mysql", "user:password@tcp(your-mysql-ip:port)/your-database-name")
if err != nil {
panic(err.Error())
}
defer db.Close() // 注意关闭数据库连接
登录后复制
示例代码：
sqlStr := "INSERT INTO your-table-name(id, name, age, gender) VALUES (%d, '%s', %d, '%s')"
登录后复制
csv包，可以轻松读取CSV文件中的数据。使用NewReader()函数创建一个CSV文件对象，使用Read()函数读取每一行数据。示例代码：
import (
"bufio"
"encoding/csv"
"os"
)
csvFile, err := os.Open("your-csv-file.csv")
if err != nil {
panic(err)
}
defer csvFile.Close()
reader := csv.NewReader(bufio.NewReader(csvFile))
for {
line, err := reader.Read()
if err == io.EOF {
break
} else if err != nil {
panic(err)
}
// TODO: 将读取到的数据进行插入操作
}
登录后复制
Exec()函数执行SQL语句，完成数据的插入操作。示例代码：
query := fmt.Sprintf(sqlStr, id, name, age, gender)
_, err = db.Exec(query)
if err != nil {
panic(err)
}
登录后复制
完整代码下面是通过以上步骤编写的完整代码：
import (
"bufio"
"database/sql"
"encoding/csv"
"fmt"
"io"
"os"
_ "github.com/go-sql-driver/mysql"
)
func main() {
db, err := sql.Open("mysql", "user:password@tcp(your-mysql-ip:port)/your-database-name")
if err != nil {
panic(err.Error())
}
defer db.Close()
sqlStr := "INSERT INTO your-table-name(id, name, age, gender) VALUES (%d, '%s', %d, '%s')"
csvFile, err := os.Open("your-csv-file.csv")
if err != nil {
panic(err)
}
defer csvFile.Close()
reader := csv.NewReader(bufio.NewReader(csvFile))
for {
line, err := reader.Read()
if err == io.EOF {
break
} else if err != nil {
panic(err)
}
id := line[0]
name := line[1]
age := line[2]
gender := line[3]
query := fmt.Sprintf(sqlStr, id, name, age, gender)
_, err = db.Exec(query)
if err != nil {
panic(err)
}
}
}
登录后复制
使用以上代码即可实现从CSV文件中导入数据到MySQL数据库中。需要注意的是，在使用Exec()函数执行SQL语句时，需要对插入的数据进行数据类型转换，并进行错误处理，以避免数据类型错误导致数据插入失败。同时也需要注意防止SQL注入攻击，使用动态生成SQL语句时需要进行转义处理。
以上就是使用Go语言进行MySQL数据库的数据导入的方法的详细内容，更多请关注每日运维网(www.mryunwei.com)其它相关文章！
