golang 碎片整理之MySQL

MySQL是常用的關係型數據庫,本文介紹了go語言如何操作MySQL數據庫。

Go操作MySQL

連接

Go語言中的database/sql 包提供了保證SQL或類SQL數據庫的廣泛接口,並不提供具體的數據庫驅動。使用database/sql包時必須注入一個數據庫驅動。

下載依賴

go get -u github.com/go-sql-driver/mysql

使用MySQL驅動

func Open(driverName,dataSourceName string)(*DB,error)

Open打開一個driverName 指定的數據庫,dataSourceName指定數據源,一般包至少包括數據庫文件名和可能的連接信息。

package main

import (
    "database/sql"
)

func main() {
    dsn := "user:password@tcp(127.0.0.1:3306)/dbname"
    db, err := sql.Open("mysql", dsn)
    if err != nil {
        panic(err)
    }
    defer db.Close()
}

初始化連接

Open函數可能只是驗證其參數,而不創建與數據庫的連接。如果要檢查數據源的名稱的合法性,應該調用返回值的ping方法。
返回的DB可以安全的被多個goroutine同時調用,並會維護自身的閒置連接池。這樣一來,Open函數只需要調用一次。很少需要關閉DB.

var db *sql.DB

func initDB() (err error) {
    dsn := "user:password@tcp(127.0.0.1:3306)/test"
    db, err = sql.Open("mysql", dsn)
    if err != nil {
        return err
    }
    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
    }
}

其中的sql.DB 是一個數據庫操作句柄,代表一個具有零到讀個底層連接的連接池。它可以安全的被多個goroutine 同時使用。database/sql 包會自動創建和釋放連接;它也會維護一個閒置的連接池。

SetMaxOpenConns

func (db *DB) SetMaxOpenConns(n int)

SetMaxOpenConns 設置與數據庫建立連接的最大數目。如果n大於0且小於最大閒置連接數,會將最大閒置連接數減小到匹配最大開啓連接數的限制。如果n<=0,不會限制最大開啓連接數,默認是0.

SetMaxIdleConns

func (db *DB)SetMaxIdleConns(n int)

SetMaxIdleConns設置連接池中的最大閒置連接數。如果n大於最大開啓連接數,則新的最大閒置連接數會減小到匹配最大開啓連接數的限制。如果n<=0,不會保留限制連接。

CRUD

建庫建表

我們先在MySQL中創建一個名爲sql_test的數據庫。

CREATE DATABASE sql_test;

進入數據庫:

use sql_test;

執行下面命令創建一張用於測試的數據表:

CREATE TABLE `user` (
    `id` BIGINT(20) NOT NULL AUTO_INCREMENT,
    `name` VARCHAR(20) DEFAULT '',
    `age` INT(11) DEFAULT '0',
    PRIMARY KEY(`id`)
)ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8mb4;

查詢

單行查詢

單行查詢db.QueryRow()執行一次查詢,並期望返回最多一行結果(即Row)。 QueryRow總是返回非nil的值,直到返回值的Scan方法被調用時,纔會返回被延遲的錯誤(如:未找到結果)。

func (db *DB)QueryRow(query string, args ...interface{})*Row

具體示例代碼:

func queryRowDeme() {
    sqlstr := "select id ,name,age from user where id =?"
    var u user
    err := db.QueryRow(sqlstr, 1).Scan(&u.id, &u.name, &u.name)
    if err != nil {
        fmt.Printf("scan failed,err:%v\n", err)
        return
    }
    fmt.Printf("id:%d name:%s age:%d\n", u.id, u.name, u.age)

}

多行查詢

多行查詢db.Query()執行一次查詢,返回多行結果(即Rows),一般用於執行select命令,參數args表示query中的佔位參數。

func (db *DB) Query(query string, args ...interface{}) (*Rows, error)

具體示例代碼:

func queryMultiRowDemo() {
    sqlstr := "select id,name,age from user where id >?"
    rows, err := db.Query(sqlstr, 0)
    if err != nil {
        fmt.Printf("query failed,err:%v\n", err)
        return
    }
    defer rows.Close()
    for rows.Next() {
        var u user
        err := rows.Scan(&u.id, &u.name, &u.age)
        if err != nil {
            fmt.Printf("scan failed,err:%v\n", err)
            return
        }
        fmt.Printf("id:%d name:%s age:%d\n", u.id, u.name, u.age)
    }
}

插入數據

插入、更新和刪除操作都使用方法:

func (db *DB) Exec(query string, args ...interface{})(Result,err)

Exec執行一次命令(包括查詢,刪除,更新,插入等),返回的Result是對已執行的SQL命令的總結。參數args表示query中的佔位參數。
具體插入數據示例代碼如下:

func insertRowDemo(){
    sqlstr := "insert into user(name, age) values (?,?)"
    ret,err :db.Exec(sqlstr,"張一",18)
    if err != nil{
        fmt.Printf("insert failed,err:%v\n",err)
        return
    }

    theID,err := ret.LastInsertId()
    if err != nil{
        fmt.Printf("get lastinsert ID failed,err:%v\n",err)
        return 
    }
    fmt.Printf("insert success,the id is %d.\n",theID)
}

更新數據

具體更新數據示例代碼如下:

func updateRowDemo(){
    sqlstr := "update user set age=? where id =?"
    ret ,err := db.Exec(sqlstr,39,3)
    if err != nil{
        fmt.Printf("update failed,err:%v\n",err)
        return 
    }
    n, err := ret.RowsAffected()
    if err != nil{
        fmt.Printf("get Rowaffected failed,err:%v\n",err)
        return
    }
    fmt.Printf("update success,affected rows:%d\n",n)
}

刪除數據

具體刪除數據的示例代碼如下:

func deleteRowDemo(){
    sqlstr = "delete from user where id =?"
    ret,err := db.Exec(sqlstr,3)
    if err != nil{
        fmt.Printf("delete failed,err:%v\n",err)
        return 
    }
    n ,err := ret.RowsAffected()
    if err != nil{
        fmt.Printf("get RowsAffected failed,err:%v\n",err)
        return
    }
    fmt.Printf("delete success,affected rows:%d\n",n)
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章