Golang-访问MySQL数据库

忘是亡心i 2022-11-02 01:00 146阅读 0赞

go自身提供了`database/sql`这个包来做数据库访问层的抽象，相当于接口层。但是要访问具体的数据库，需要安装该数据库对应的驱动，对于MySQL来说，执行以下命令安装驱动：

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

有了驱动，后面访问数据库就很简单了。

## 连接MySQL ##

通过`sql.open`设置连接参数，该函数并不会真正的访问数据库。

db, err := sql.Open("mysql", "username:password@(127.0.0.1:3306)/dbname?parseTime=true")

通过`db.ping`来访问数据库，检查连接是否正常，如果连接参数错误 ，变量 err 将被赋值。

err := db.Ping()

## 访问MySQL并进行CURD操作 ##

### 1 创建表 ###

假定我们的创建一个用户表，其建表语句为：

CREATE TABLE users (
        id INT AUTO_INCREMENT,
        username TEXT NOT NULL,
        password TEXT NOT NULL,
        created_at DATETIME,
        PRIMARY KEY (id)
    );

那么go实现创建表的操作为如下所示，可以通过err判断建表是否出现问题

query := ` CREATE TABLE users ( id INT AUTO_INCREMENT, username TEXT NOT NULL, password TEXT NOT NULL, created_at DATETIME, PRIMARY KEY (id) );`
    
    // Executes the SQL query in our database. Check err to ensure there was no error.
    _, err := db.Exec(query)

### 2 插入数据 ###

如果需要插入日期数据，则需要引入`time`这个包

import "time"
    
    username := "johndoe"
    password := "secret"
    createdAt := time.Now()
    
    // Inserts our data into the users table and returns with the result and a possible error.
    // The result contains information about the last inserted id (which was auto-generated for us) and the count of rows this query affected.
    result, err := db.Exec(`INSERT INTO users (username, password, created_at) VALUES (?, ?, ?)`, username, password, createdAt)

通过`result`变量可以返回插入时数据库的变动信息，包括插入数据行的id，影响行数。

userID, err := result.LastInsertId()

### 3 查询数据 ###

#### 3.1 查询单条数据 ####

通过`db.QueryRow`可以实现单条数据的查询，`var()`封装一组变量，这种简洁的写法可避免每个变量都要做`var`声明

var (
        id        int
        username  string
        password  string
        createdAt time.Time
    )
    
    // Query the database and scan the values into out variables. Don't forget to check for errors.
    query := `SELECT id, username, password, created_at FROM users WHERE id = ?`
    err := db.QueryRow(query, 1).Scan(&id, &username, &password, &createdAt)

#### 3.2 查询多条数据 ####

查询多条数据和单条数据很相似，区别在于这里调用`db.Query`函数，它可以返回多条记录，而`db.QueryRow`最多只返回1条记录。

对于多条数据的查询结果，我们需要定义一个结构体，实现go字段和数据库字段的一一映射，然后迭代扫描每条行记录并封装到结构体成员中。

type user struct { 
        id        int
        username  string
        password  string
        createdAt time.Time
    }
    
    rows, err := db.Query(`SELECT id, username, password, created_at FROM users`) // check err
    defer rows.Close()
    
    var users []user
    for rows.Next() { 
        var u user
        err := rows.Scan(&u.id, &u.username, &u.password, &u.createdAt) // check err
        users = append(users, u)
    }
    err := rows.Err()

如果数据库有数据，那么users结构体数组将类似于下面这种形式

users { 
        user { 
            id:        1,
            username:  "johndoe",
            password:  "secret",
            createdAt: time.Time{ wall: 0x0, ext: 63701044325, loc: (*time.Location)(nil)},
        },
        user { 
            id:        2,
            username:  "alice",
            password:  "bob",
            createdAt: time.Time{ wall: 0x0, ext: 63701044622, loc: (*time.Location)(nil)},
        },
    }

### 4 删除数据 ###

执行`db.Exec`可以实现删除操作

_, err := db.Exec(`DELETE FROM users WHERE id = ?`, 1) // check err

## 完整代码 ##

package main
    
    import (
    	"database/sql"
    	"fmt"
    	"log"
    	"time"
    
    	_ "github.com/go-sql-driver/mysql"
    )
    
    func main() { 
    	db, err := sql.Open("mysql", "root:mysql520@(127.0.0.1:3306)/demo?parseTime=true")
    	if err != nil { 
    		log.Fatal(err)
    	}
    	if err := db.Ping(); err != nil { 
    		log.Fatal(err)
    	}
    
    	{  // Create a new table
    		query := ` CREATE TABLE users ( id INT AUTO_INCREMENT, username TEXT NOT NULL, password TEXT NOT NULL, created_at DATETIME, PRIMARY KEY (id) );`
    
    		if _, err := db.Exec(query); err != nil { 
    			log.Fatal(err)
    		}
    	}
    
    	{  // Insert a new user
    		username := "johndoe"
    		password := "secret"
    		createdAt := time.Now()
    
    		result, err := db.Exec(`INSERT INTO users (username, password, created_at) VALUES (?, ?, ?)`, username, password, createdAt)
    		if err != nil { 
    			log.Fatal(err)
    		}
    
    		id, err := result.LastInsertId()
    		fmt.Println(id)
    	}
    
    	{  // Query a single user
    		var (
    			id        int
    			username  string
    			password  string
    			createdAt time.Time
    		)
    
    		query := "SELECT id, username, password, created_at FROM users WHERE id = ?"
    		if err := db.QueryRow(query, 1).Scan(&id, &username, &password, &createdAt); err != nil { 
    			log.Fatal(err)
    		}
    
    		fmt.Println(id, username, password, createdAt)
    	}
    
    	{  // Query all users
    		type user struct { 
    			id        int
    			username  string
    			password  string
    			createdAt time.Time
    		}
    
    		rows, err := db.Query(`SELECT id, username, password, created_at FROM users`)
    		if err != nil { 
    			log.Fatal(err)
    		}
    		defer rows.Close()
    
    		var users []user
    		for rows.Next() { 
    			var u user
    
    			err := rows.Scan(&u.id, &u.username, &u.password, &u.createdAt)
    			if err != nil { 
    				log.Fatal(err)
    			}
    			users = append(users, u)
    		}
    		if err := rows.Err(); err != nil { 
    			log.Fatal(err)
    		}
    
    		fmt.Printf("%#v", users)
    	}
    
    	{ 
    		_, err := db.Exec(`DELETE FROM users WHERE id = ?`, 1)
    		if err != nil { 
    			log.Fatal(err)
    		}
    	}
    }