在Go语言中,可以使用第三方库如go-sql-driver/mysql
来初始化MySQL连接并实现函数化。以下是一个示例代码:
package main
import (
"database/sql"
"fmt"
_ "github.com/go-sql-driver/mysql"
)
func main() {
db, err := InitDB()
if err != nil {
fmt.Println("Failed to connect to MySQL:", err)
return
}
defer db.Close()
err = DoSomethingWithDB(db)
if err != nil {
fmt.Println("Error occurred while working with MySQL:", err)
return
}
fmt.Println("Successfully executed MySQL operations.")
}
// 初始化数据库连接
func InitDB() (*sql.DB, error) {
db, err := sql.Open("mysql", "username:password@tcp(host:port)/database")
if err != nil {
return nil, err
}
err = db.Ping()
if err != nil {
return nil, err
}
return db, nil
}
// 在这里进行数据库操作的函数化实现
func DoSomethingWithDB(db *sql.DB) error {
stmt, err := db.Prepare("INSERT INTO users(name, age) VALUES(?, ?)")
if err != nil {
return err
}
defer stmt.Close()
result, err := stmt.Exec("John Doe", 30)
if err != nil {
return err
}
lastInsertID, _ := result.LastInsertId()
rowsAffected, _ := result.RowsAffected()
fmt.Printf("Last inserted ID: %d\nRows affected: %d\n", lastInsertID, rowsAffected)
return nil
}
请替换代码中的username
、password
、host
、port
和database
为您自己的MySQL连接信息。这样,您就可以通过调用InitDB()
来初始化数据库连接,并在DoSomethingWithDB()
中进行具体的数据库操作。
内容由零声教学AI助手提供,问题来源于学员提问