MySQL 数据库操作
概述
数据库的相关使用, 我们一般推荐使用 goctl 直接生成 model 代码，同时会自动生成 golang 结构体，CURD操作方法，缓存等，可以参考 goctl model。
但是针对特殊情况需要自行直接链接数据库的，我们也可以直接自己初始化 sql conn。
任务目标
- 了解 github.com/zeromicro/go-zero/core/stores/sqlx 包的使用。
- 根据 sqlx 创建一个 sql 链接。
创建数据库
首先在的数据库中创建如下的表。
CREATE TABLE user (
id bigint AUTO_INCREMENT,
name varchar(255) NOT NULL DEFAULT '' COMMENT 'The username',
type tinyint(1) NULL DEFAULT 0 COMMENT 'The user type, 0:normal,1:vip, for test golang keyword',
create_at timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
update_at timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
PRIMARY KEY (id)
) ENGINE = InnoDB COLLATE utf8mb4_general_ci COMMENT 'user table';
连接到数据库
Mysql
package main
import (
"github.com/zeromicro/go-zero/core/stores/sqlx"
)
func main() {
// 参考 https://github.com/go-sql-driver/mysql#dsn-data-source-name 获取详情
// 需要自行将 dsn 中的 host，账号 密码配置正确
dsn := "user:pass@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
conn := sqlx.NewMysql(dsn)
_ = conn
}注意
想要正确的处理 time.Time ，您需要带上 parseTime 参数， 更多参数 要支持完整的 UTF-8 编码，您需要将 charset=utf8 更改为 charset=utf8mb4
自定义驱动
go-zero 允许通过 DriverName 选项自定义 MySQL 驱动，例如：
package main
import (
"github.com/zeromicro/go-zero/core/stores/sqlx"
)
func main() {
dsn := "user:pass@tcp(127.0.0.1:3307)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
conn := sqlx.NewSqlConn("mysql", dsn)
_ = conn
}现有的数据库连接
go-zero 允许通过 现有的数据库初始化 Sql 链接，例如：
package main
import (
"database/sql"
"github.com/zeromicro/go-zero/core/stores/sqlx"
)
func main() {
sqlDB, err := sql.Open("mysql", "mydb_dsn")
if err != nil {
panic(err)
}
conn := sqlx.NewSqlConnFromDB(sqlDB)
_ = conn
}
开始 CRUD
插入一条数据
我们使用上面的创建链接的方法得到一个链接之后，我们可以开始操作数据库。
package main
import (
"context"
"fmt"
"github.com/zeromicro/go-zero/core/stores/sqlx"
)
func main() {
// 参考 https://github.com/go-sql-driver/mysql#dsn-data-source-name 获取详情
// 需要自行确保 dsn 中的 host 账号 密码都配置正确
dsn := "user:pass@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
conn := sqlx.NewMysql(dsn)
r, err := conn.ExecCtx(context.Background(), "insert into user (type, name) values (?, ?)", 1, "test")
if err != nil {
panic(err)
}
fmt.Println(r.RowsAffected())
}
执行程序，我们会在user 中插入一条记录。
查询数据
我们需要先定义一个 User 结构体，接着再查询
package main
import (
"context"
"database/sql"
"time"
"github.com/zeromicro/go-zero/core/stores/sqlx"
)
type User struct {
Id int64 `db:"id"`
Name sql.NullString `db:"name"` // The username
Type int64 `db:"type"` // The user type, 0:normal,1:vip, for test golang keyword
CreateAt sql.NullTime `db:"create_at"`
UpdateAt time.Time `db:"update_at"`
}
func main() {
// 参考 https://github.com/go-sql-driver/mysql#dsn-data-source-name 获取详情
// 需要自行确保 dsn 中的 host 账号 密码都配置正确
dsn := "user:pass@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
conn := sqlx.NewMysql(dsn)
var u User
query := "select id, name, type, create_at, update_at from user where id=?"
err := conn.QueryRowCtx(context.Background(), &u, query, 1)
if err != nil {
panic(err)
}
}执行上述程序，我们会看到 我们刚刚插入进去的 user 信息
修改数据 我们继续修改代码
package main
import (
"context"
"fmt"
"github.com/zeromicro/go-zero/core/stores/sqlx"
)
func main() {
// 参考 https://github.com/go-sql-driver/mysql#dsn-data-source-name 获取详情
// 需要自行确保 dsn 中的 host 账号 密码都配置正确
dsn := "user:pass@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
conn := sqlx.NewMysql(dsn)
_, err := conn.ExecCtx(context.Background(), "update user set type = ? where name = ?", 2, "test")
if err != nil {
fmt.Println(err)
return
}
}运行上述代码，会发现数据库中的记录 type 变为 2了。
删除数据
package main
import (
"context"
"fmt"
"github.com/zeromicro/go-zero/core/stores/sqlx"
)
func main() {
// 参考 https://github.com/go-sql-driver/mysql#dsn-data-source-name 获取详情
// 需要自行确保 dsn 中的 host 账号 密码都配置正确
dsn := "user:pass@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
conn := sqlx.NewMysql(dsn)
_, err := conn.ExecCtx(context.Background(), "delete from user where `id` = ?", 1)
if err != nil {
fmt.Println(err)
return
}
}
运行上述代码，会发现数据库中的记录已经被删除了。
至此，你已经完成 mysql 数据库的基本使用。