golang msql类封装

Golang封装MySQL类

在Golang开发中，与数据库交互是常见的需求。而MySQL作为一种广泛使用的关系型数据库，其在Golang中的使用也是十分常见的。本文将介绍如何封装一个能够简化MySQL操作的类。

MySQL连接管理

在使用MySQL之前，首先需要建立与数据库的连接。Golang中有很多第三方包可以用来连接和操作MySQL数据库，比如go-sql-driver/mysql、mysql驱动等。其中，go-sql-driver/mysql是一个支持database/sql接口的MySQL驱动程序。

通过创建一个MySQL类，我们可以将连接的操作封装起来，提供统一的API供应用程序调用。下面是一个示例：

```go package mysql import ( "database/sql" "fmt" _ "github.com/go-sql-driver/mysql" ) type MySQL struct { db *sql.DB } func NewMySQL(host, port, user, password, dbName string) (*MySQL, error) { dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8", user, password, host, port, dbName) db, err := sql.Open("mysql", dsn) if err != nil { return nil, err } return &MySQL{db}, nil } func (m *MySQL) Close() error { return m.db.Close() } ```

基本操作

在连接建立后，我们可以进行数据库的增删改查操作。下面是一些基本操作的示例。

查询操作

查询操作是对数据库进行读取的操作，我们可以封装一个用于查询的方法：

```go func (m *MySQL) Query(sql string, args ...interface{}) (*sql.Rows, error) { return m.db.Query(sql, args...) } ```

在这个方法中，我们使用了db.Query方法来执行查询操作，并将结果返回给调用者。

插入操作

插入操作是对数据库进行写入的操作，我们可以封装一个用于插入的方法：

```go func (m *MySQL) Insert(sql string, args ...interface{}) (int64, error) { result, err := m.db.Exec(sql, args...) if err != nil { return 0, err } return result.LastInsertId() } ```

在这个方法中，我们使用了db.Exec方法来执行插入操作，并通过result.LastInsertId获取插入的自增ID。

更新操作

更新操作是对数据库中已有记录进行修改的操作，我们可以封装一个用于更新的方法：

```go func (m *MySQL) Update(sql string, args ...interface{}) (int64, error) { result, err := m.db.Exec(sql, args...) if err != nil { return 0, err } return result.RowsAffected() } ```

在这个方法中，我们使用了db.Exec方法来执行更新操作，并通过result.RowsAffected获取受影响的行数。

删除操作

删除操作是对数据库中已有记录进行删除的操作，我们可以封装一个用于删除的方法：

```go func (m *MySQL) Delete(sql string, args ...interface{}) (int64, error) { result, err := m.db.Exec(sql, args...) if err != nil { return 0, err } return result.RowsAffected() } ```

在这个方法中，我们使用了db.Exec方法来执行删除操作，并通过result.RowsAffected获取受影响的行数。

使用示例

使用封装好的MySQL类进行数据库操作十分简单。下面是一个示例：

```go package main import ( "fmt" "log" "github.com/your-username/your-package/mysql" ) func main() { db, err := mysql.NewMySQL("localhost", "3306", "root", "password", "test") if err != nil { log.Fatalf("Failed to connect to database: %v", err) } defer db.Close() rows, err := db.Query("SELECT * FROM users") if err != nil { log.Fatalf("Failed to query from users table: %v", err) } defer rows.Close() for rows.Next() { var id int var name string err := rows.Scan(&id, &name) if err != nil { log.Fatalf("Failed to scan row: %v", err) } fmt.Printf("User ID: %d, Name: %s\n", id, name) } insertID, err := db.Insert("INSERT INTO users (name) VALUES (?)", "John") if err != nil { log.Fatalf("Failed to insert user: %v", err) } fmt.Printf("Inserted User ID: %d\n", insertID) updateRows, err := db.Update("UPDATE users SET name = ? WHERE id = ?", "Alice", insertID) if err != nil { log.Fatalf("Failed to update user: %v", err) } fmt.Printf("Updated Rows: %d\n", updateRows) deleteRows, err := db.Delete("DELETE FROM users WHERE id = ?", insertID) if err != nil { log.Fatalf("Failed to delete user: %v", err) } fmt.Printf("Deleted Rows: %d\n", deleteRows) } ```

在这个示例中，我们首先创建了一个MYSQL实例，然后通过Query方法查询users表的数据，并使用Scan方法将每一行的结果赋值给变量。接下来通过Insert方法向表中插入一条记录，并返回插入的自增ID。之后使用Update方法更新刚插入的记录的名字，并使用Delete方法删除该记录。

总结

通过封装一个MySQL类，我们能够方便地进行数据库操作，将数据库连接的建立和关闭封装起来，提供了统一的API供应用程序调用。同时，通过封装各种操作的方法，使得数据库的增删改查操作更加简洁明了。希望本文能够帮助到你在Golang开发中使用MySQL数据库。