GO 批量修改 + 事务

package main

import (
"database/sql"
"fmt"
_ "github.com/go-sql-driver/mysql" // 引入你的数据库驱动
)

func main() {
// 建立数据库连接
db, err := sql.Open("mysql", "user:password@/dbname")
if err != nil {
panic(err)
}
defer db.Close()

// 确保连接可用
if err := db.Ping(); err != nil {
panic(err)
}

// 准备批量更新
tx, err := db.Begin()
if err != nil {
panic(err)
}

stmt, err := tx.Prepare("UPDATE users SET name = ? WHERE id = ?")
if err != nil {
panic(err)
}
defer stmt.Close()

// 模拟批量更新数据
names := map[int]string{
1: "Alice",
2: "Bob",
3: "Charlie",
}

for id, name := range names {
if _, err := stmt.Exec(name, id); err != nil {
tx.Rollback()
panic(err)
}
}

// 提交事务
if err := tx.Commit(); err != nil {
panic(err)
}

fmt.Println("Batch update successful")
}

posted @ 2024-03-19 15:35  minch  阅读(72)  评论(0)    收藏  举报