Go语言实现MySQL数据库数据导入的方法

2025-01-14 22:41:03   小编

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执行语句并返回结果,通过LastInsertIdRowsAffected可以获取插入的自增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交互

欢迎使用万千站长工具!

Welcome to www.zzTool.com