【MySQL】——golang操作mysql数据库

小灰灰 2022-12-20 06:29 253阅读 0赞

## 一、前言 ##

golang适合写中间件，今年一直在做数据库相关的工作，用的也是golang，短平快的写过一波代码之后开始回过头来做一个总结。在golang中，提供了标准的数据库接口database/sql包。

## 二、database/sql简介 ##

SQL 是应用程序和数据库之间的中间层，应用程序在多数情况下都不需要关心底层数据库的实现，它们只关心 SQL 查询返回的数据。我们可以使用相同的 SQL 语言查询关系型数据库，所有关系型数据库的客户端都需要实现如下所示的驱动接口：

type Driver interface {
    	Open(name string) (Conn, error)
    }
    
    type Conn interface {
    	Prepare(query string) (Stmt, error)
    	Close() error
    	Begin() (Tx, error)
    }

[`database/sql/driver.Driver`][database_sql_driver.Driver] 接口中只包含一个 `Open` 方法，该方法接收一个数据库连接串作为输入参数并返回一个特定数据库的连接，作为参数的数据库连接串是数据库特定的格式，这个返回的连接仍然是一个接口，整个标准库中的全部接口可以构成如下所示的树形结构：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0OTMwMA_size_16_color_FFFFFF_t_70][]

Go语言中的`database/sql`包提供了保证SQL或类SQL数据库的泛用接口，并不提供具体的数据库驱动。使用`database/sql`包时必须注入相应的数据库驱动。常见的数据库驱动如下：

*  `Mysql: https://github.com/go-sql-driver/mysql`
 *  `MyMysql: https://github.com/ziutek/mymysql`
 *  `Postgres: https://github.com/lib/pq`
 *  `Tidb: https://github.com/pingcap/tidb`
 *  `SQLite: https://github.com/mattn/go-sqlite3`
 *  `MsSql: https://github.com/denisenkom/go-mssqldb`

## 三、代码分析 ##

### 1、连接数据库。 ###

package main
    
    import (
    	"database/sql"
    	"fmt"
    	_ "github.com/go-sql-driver/mysql"
    )
    // 定义一个全局对象db
    var db *sql.DB
    
    // 定义一个初始化数据库的函数
    func initDB() (err error) {
    	// DSN:Data Source Name
    	dsn := "user:password@tcp(127.0.0.1:3306)/test"
    	// 不会校验账号密码是否正确
    	// 注意！！！这里不要使用:=，我们是给全局变量赋值，然后在main函数中使用全局变量db
    	db, err = sql.Open("mysql", dsn)
    	if err != nil {
    		return err
    	}
    	// 尝试与数据库建立连接（校验dsn是否正确）
    	err = db.Ping()
    	if err != nil {
    		return err
    	}
    	return nil
    }
    
    func main() {
    	err := initDB() // 调用输出化数据库的函数
    	if err != nil {
    		fmt.Printf("init db failed,err:%v\n", err)
    		return
    	}
    }

调用db.Ping()没有问题之后，我们就可以使用db去操作了，注意的是，这个返回的db可以安全的被多个goroutine同时使用，并会维护自身的闲置连接池。这样一来，Open函数只需调用一次。很少需要关闭db。

准备数据：

新建test数据库，person、place 表

CREATE TABLE `person` (
        `user_id` int(11) NOT NULL AUTO_INCREMENT,
        `username` varchar(260) DEFAULT NULL,
        `sex` varchar(260) DEFAULT NULL,
        `email` varchar(260) DEFAULT NULL,
        PRIMARY KEY (`user_id`)
      ) ENGINE=InnoDB AUTO_INCREMENT=2 DEFAULT CHARSET=utf8;
    
    CREATE TABLE place (
        country varchar(200),
        city varchar(200),
        telcode int
    )ENGINE=InnoDB AUTO_INCREMENT=2 DEFAULT CHARSET=utf8;

### 2、insert代码 ###

3、select代码

4、update代码

5、delete代码

6、事务代码

golang MySQL事务应用：

1） import (“github.com/jmoiron/sqlx")
        2)  Db.Begin()        开始事务
        3)  Db.Commit()        提交事务
        4)  Db.Rollback()     回滚事务

package main
    
        import (
            "fmt"
    
            _ "github.com/go-sql-driver/mysql"
            "github.com/jmoiron/sqlx"
        )
    
        type Person struct {
            UserId   int    `db:"user_id"`
            Username string `db:"username"`
            Sex      string `db:"sex"`
            Email    string `db:"email"`
        }
    
        type Place struct {
            Country string `db:"country"`
            City    string `db:"city"`
            TelCode int    `db:"telcode"`
        }
    
        var Db *sqlx.DB
    
        func init() {
            database, err := sqlx.Open("mysql", "root:root@tcp(127.0.0.1:3306)/test")
            if err != nil {
                fmt.Println("open mysql failed,", err)
                return
            }
            Db = database
        }
    
        func main() {
            conn, err := Db.Begin()
            if err != nil {
                fmt.Println("begin failed :", err)
                return
            }
    
            r, err := conn.Exec("insert into person(username, sex, email)values(?, ?, ?)", "stu001", "man", "stu01@qq.com")
            if err != nil {
                fmt.Println("exec failed, ", err)
                conn.Rollback()
                return
            }
            id, err := r.LastInsertId()
            if err != nil {
                fmt.Println("exec failed, ", err)
                conn.Rollback()
                return
            }
            fmt.Println("insert succ:", id)
    
            r, err = conn.Exec("insert into person(username, sex, email)values(?, ?, ?)", "stu001", "man", "stu01@qq.com")
            if err != nil {
                fmt.Println("exec failed, ", err)
                conn.Rollback()
                return
            }
            id, err = r.LastInsertId()
            if err != nil {
                fmt.Println("exec failed, ", err)
                conn.Rollback()
                return
            }
            fmt.Println("insert succ:", id)
    
            conn.Commit()
        }

## 四、预处理 ##

普通SQL语句执行过程：

1.  客户端对SQL语句进行占位符替换得到完整的SQL语句。
2.  客户端发送完整SQL语句到MySQL服务端
3.  MySQL服务端执行完整的SQL语句并将结果返回给客户端。

预处理执行过程：

1.  把SQL语句分成两部分，命令部分与数据部分。
2.  先把命令部分发送给MySQL服务端，MySQL服务端进行SQL预处理。
3.  然后把数据部分发送给MySQL服务端，MySQL服务端对SQL语句进行占位符替换。
4.  MySQL服务端执行完整的SQL语句并将结果返回给客户端。

为什么要预处理？

1.  优化MySQL服务器重复执行SQL的方法，可以提升服务器性能，提前让服务器编译，一次编译多次执行，节省后续编译的成本。
2.  避免SQL注入问题。

mysql预处理代码：

package main
    
    import (
    	"database/sql"
    	"fmt"
    	_ "github.com/lib/pq"
    )
    
    var db *sql.DB
    var err error
    
    func initDB() {
    	if db, err = sql.Open("postgres", "postgres://postgres:zgghyys123@127.0.0.1:5432/postgres?sslmode=disable"); err != nil {
    		panic(err)
    	}
    
    	if err = db.Ping(); err != nil {
    		panic(err)
    	}
    }
    
    type heroes struct {
    	id int64
    	name string
    	age string
    	hp int64
    	attack string
    	role string
    	ultimate string
    	country string
    }
    
    func insertData(){
    	insertQuery := "insert into anime.overwatch(id, name, age) values($1, $2, $3)"
    	//Prepare方法会先将sql语句发送给MySQL服务端，返回一个准备好的状态用于之后的查询和命令。返回值可以同时执行多个查询和命令。
    	stmt, _ := db.Prepare(insertQuery)
    	defer stmt.Close()
    	//传入参数
    	ret, err := stmt.Exec(100, "hanser", 26)
    	if err != nil {
    		panic(err)
    	}
    
    	if rowCount, err := ret.RowsAffected(); err != nil {
    		panic(err)
    	} else {
    		fmt.Println(rowCount) //1
    	}
    }
    
    func main() {
    	initDB()
    	insertData()
    }
    //我们这里可以多次执行Exec，会多次插入记录

五、参考文章

1、[https://www.cnblogs.com/traditional/p/11905780.html][https_www.cnblogs.com_traditional_p_11905780.html]

2、[https://draveness.me/golang/docs/part4-advanced/ch09-stdlib/golang-database-sql/][https_draveness.me_golang_docs_part4-advanced_ch09-stdlib_golang-database-sql]

3、[http://www.topgoer.com/%E6%95%B0%E6%8D%AE%E5%BA%93%E6%93%8D%E4%BD%9C/go%E6%93%8D%E4%BD%9Cmysql/mysql%E4%BD%BF%E7%94%A8.html][http_www.topgoer.com_E6_95_B0_E6_8D_AE_E5_BA_93_E6_93_8D_E4_BD_9C_go_E6_93_8D_E4_BD_9Cmysql_mysql_E4_BD_BF_E7_94_A8.html]

[database_sql_driver.Driver]: https://github.com/golang/go/blob/b68fa57c599720d33a2d735782969ce95eabf794/src/database/sql/driver/driver.go#L84
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDQ0OTMwMA_size_16_color_FFFFFF_t_70]: /images/20221120/9f1085dcec934fb59099bcdd1ddf4d6c.png
[https_www.cnblogs.com_traditional_p_11905780.html]: https://www.cnblogs.com/traditional/p/11905780.html
[https_draveness.me_golang_docs_part4-advanced_ch09-stdlib_golang-database-sql]: https://draveness.me/golang/docs/part4-advanced/ch09-stdlib/golang-database-sql/
[http_www.topgoer.com_E6_95_B0_E6_8D_AE_E5_BA_93_E6_93_8D_E4_BD_9C_go_E6_93_8D_E4_BD_9Cmysql_mysql_E4_BD_BF_E7_94_A8.html]: http://www.topgoer.com/%E6%95%B0%E6%8D%AE%E5%BA%93%E6%93%8D%E4%BD%9C/go%E6%93%8D%E4%BD%9Cmysql/mysql%E4%BD%BF%E7%94%A8.html