技术文摘
Go语言实现MySQL数据库数据导入的方法
Go语言实现MySQL数据库数据导入的方法
在开发过程中,将数据导入MySQL数据库是一项常见任务。Go语言以其高效、简洁的特性,为我们提供了便利的方式来实现这一功能。
需要安装并引入必要的MySQL驱动。常用的驱动有go - sql - driver/mysql。通过命令go get - u github.com/go - sql - driver/mysql即可完成安装。接着,在代码中引入该驱动:import "github.com/go - sql - driver/mysql"。
连接MySQL数据库是数据导入的基础。使用如下代码来建立连接:
dsn := "user:password@tcp(127.0.0.1:3306)/database_name"
db, err := sql.Open("mysql", dsn)
if err!= nil {
log.Fatal(err)
}
defer db.Close()
err = db.Ping()
if err!= nil {
log.Fatal(err)
}
这里dsn包含了数据库连接所需的用户名、密码、地址和数据库名等信息。sql.Open打开数据库连接,db.Ping用于检查连接是否正常。
对于简单的数据导入,可以使用Exec方法。例如插入一条数据:
stmt, err := db.Prepare("INSERT INTO table_name (column1, column2) VALUES (?,?)")
if err!= nil {
log.Fatal(err)
}
defer stmt.Close()
result, err := stmt.Exec("value1", "value2")
if err!= nil {
log.Fatal(err)
}
lastInsertId, err := result.LastInsertId()
if err!= nil {
log.Fatal(err)
}
rowsAffected, err := result.RowsAffected()
if err!= nil {
log.Fatal(err)
}
Prepare方法用于预处理SQL语句,防止SQL注入。Exec执行语句并返回结果,通过LastInsertId和RowsAffected可以获取插入的自增ID和影响的行数。
若要导入大量数据,为提高效率,可使用事务。示例代码如下:
tx, err := db.Begin()
if err!= nil {
log.Fatal(err)
}
stmt, err := tx.Prepare("INSERT INTO table_name (column1, column2) VALUES (?,?)")
if err!= nil {
log.Fatal(err)
}
defer stmt.Close()
for _, data := range dataList {
_, err := stmt.Exec(data.column1Value, data.column2Value)
if err!= nil {
tx.Rollback()
log.Fatal(err)
}
}
err = tx.Commit()
if err!= nil {
log.Fatal(err)
}
通过Begin开启事务,在循环中批量执行插入操作,若出现错误则Rollback回滚事务,全部成功则Commit提交事务。
利用Go语言结合MySQL驱动,通过合理使用连接、预处理语句以及事务等技术,能高效、安全地实现MySQL数据库的数据导入功能。无论是少量数据的快速插入,还是大量数据的批量导入,都能轻松应对。
TAGS: GO语言 MySQL数据库 数据导入 Go与MySQL交互