Go连接Mysql执行crud

作者: 太阳上的雨天 分类: Go 发布时间: 2022-02-17 13:13

Go原生提供了连接数据库操作的支持,在用 Golang进行开发的时候,如果需要在和数据库交互,则可以使用database/sql包。这是一个对关系型数据库的通用抽象,它提供了标准的、轻量的、面向行的接口。

sql包提供了保证SQL或类SQL数据库的泛用接口。
使用sql包时必须注入(至少)一个数据库驱动。

使用的驱动为: github.com/go-sql-driver/mysql
导入的包

    "database/sql"
    _ "github.com/go-sql-driver/mysql"

在连接mysql之前,先准备一个数据库和一张表
数据库为test
表结构

create table users (
    `id` int PRIMARY KEY auto_increment,
    `username` VARCHAR(100) not null unique,
    `password` VARCHAR(100) not NULL,
    `email` VARCHAR(100) not null
)
  1. 连接数据库
    创建utils/db.go
package utils

import (
    "database/sql"
    "fmt"
    _ "github.com/go-sql-driver/mysql"
)

var (
    Db *sql.DB
    err error
)

func init() {
    Db, err = sql.Open("mysql", "root:root@tcp(127.0.0.1:3306)/test")
    if err != nil {
        fmt.Println("mysql connect failed err: ", err.Error())
        panic(err.Error())
    }
}

2.创建model/user.go

package model

import (
    "fmt"
    "go-web-book-shop/utils"
)

//User
type User struct {
    ID int
    Username string
    Password string
    Email string
}

//AddUser 添加用户 - 预编译
func (user *User) AddUser() error {
    // sql语句
    sql := "insert into users(username, password, email) values (?, ?, ?)"
    // 预编译
    insertStmt, err := utils.Db.Prepare(sql)
    if err != nil {
        fmt.Println("预编译出现异常: ", err)
        return err
    }
    // 执行
    _, err = insertStmt.Exec("admin3", "123456", "zhangsan@163.com")
    if err != nil {
        fmt.Println("数据写入失败: ", err)
        return err
    }
    return nil
}

//AddUser2 添加用户 Exec
func (user *User) AddUser2() error {
    // sql语句
    sql := "insert into users(username, password, email) values (?, ?, ?)"
    // 执行
    _, err := utils.Db.Exec(sql, "admin4", "123456", "lisi@163.com")
    if err != nil {
        fmt.Println("数据写入失败!: ", err)
        return err
    }

    return nil
}

//Del 根据ID删除用户
func (user *User) Del() error {
    sql := "delete from users where id = ?"
    stmt, err := utils.Db.Prepare(sql)
    if err != nil {
        fmt.Println("预编译失败...")
        return err
    }

    _, err = stmt.Exec(4)
    if err != nil {
        fmt.Println("执行删除语句失败")
        return err
    }

    return nil
}

//UpdateUserById 根据ID修改用户信息
func (user *User) UpdateUserById() error {
    sql := "update users set username = ?, password = ?, email = ? where id = ?"
    stmt, err := utils.Db.Prepare(sql)
    if err != nil {
        fmt.Println("预编sql失败: ", err)
        return err
    }
    _, err = stmt.Exec("admin", "123456", "123456@163.com", 4)
    if err != nil {
        fmt.Println("执行修改sql语句失败: ", err)
        return err
    }
    return nil
}

//QueryOne 查询一条记录
func (user *User) QueryOne() (*User, error) {
    // sql语句
    sql := "select * from users where id = ?"
    // 执行
    row := utils.Db.QueryRow(sql, user.ID)
    // 声明变量接受扫描出来的数据
    var id       int
    var username string
    var password string
    var email    string
    err := row.Scan(&id, &username, &password, &email)
    if err != nil {
        fmt.Println("查询失败:", err)
        return nil, err
    }
    u := &User{
        ID:       id,
        Username: username,
        Password: password,
        Email:    email,
    }
    return u, nil
}

//QueryAll 查询全部记录
func (user *User) QueryAll() ([]*User, error) {
    sql := "select * from users"
    rows, err := utils.Db.Query(sql)
    if err != nil {
        fmt.Println("执行sql语句失败: ", err)
        return nil, err
    }

    // 创建User 切片
    var users []*User

    for rows.Next() {
        var id       int
        var username string
        var password string
        var email    string
        err = rows.Scan(&id, &username, &password, &email)
        if err != nil {
            return nil, err
        }

        u := &User{
            ID:       id,
            Username: username,
            Password: password,
            Email:    email,
        }

        users = append(users, u)
    }
    return users, nil
}

3.创建model/user_test.go单元测试文件

package model

import (
    "fmt"
    "testing"
)

//TestMain 测试函数执行之前的函数
func TestMain(m *testing.M) {
    m.Run()
}

//TestUser 测试User 中的方法
func TestUser(t *testing.T) {
    fmt.Println("开始测试User中的相关的方法")

    // 通过t.Run()执行子测试函数
    t.Run("测试添加用户", testUser_QueryAll) // 
}

//testUser_AddUser 测试添加用户
func testUser_AddUser(t *testing.T) {
    fmt.Println("测试添加用户...")
    user := &User{}
    user.AddUser()
    user.AddUser2()
}

//testUser_De 测试删除用户
func testUser_Del(t *testing.T) {
    fmt.Println("测试删除用户...")
    user := &User{}
    user.Del()
}

//testUser_UpdateUserById 测试通过ID修改用户信息
func testUser_UpdateUserById(t *testing.T) {
    fmt.Println("测试通过ID修改用户信息...")
    user := &User{}
    user.UpdateUserById()
}

//testUser_QueryOne 测试查询一条数据
func testUser_QueryOne(t *testing.T) {
    fmt.Println("测试查询一条数据...")
    user := &User{
        ID: 15,
    }
    u, _ := user.QueryOne()
    fmt.Println("查询id为15的记录为: ", u)
}

//testUser_QueryAll 测试获取所有的数据
func testUser_QueryAll(t *testing.T) {
    fmt.Println("测试获取所有的数据...")
    user := User{}
    users, _ := user.QueryAll()
    for k, v := range users{
        fmt.Printf("第%d个用户是: %v: \n", k+1, v)
    }
}

4.执行单元测试 go test

发表回复

您的电子邮箱地址不会被公开。 必填项已用*标注