头图

Golang可以通过Gorm包来操作数据库,所谓ORM,即Object Relational Mapping(数据关系映射),说白了就是通过模式化的语法来操作数据库的行对象或者表对象,对比相对灵活繁复的SQL语句,ORM上手简单,通用性较高,但是在性能层面略有损耗,Gorm的底层是结构体对象,关于结构体,请移玉步至:你有对象类,我有结构体,Go lang1.18入门精炼教程,由白丁入鸿儒,go lang结构体(struct)的使用EP06

Gorm的安装与配置

首先如果要使用Gorm操作数据库,得先有数据库才行,这里为了全平台统一标准,我们使用Docker来安装Mysql数据库,Docker的安装请参见:一寸宕机一寸血,十万容器十万兵|Win10/Mac系统下基于Kubernetes(k8s)搭建Gunicorn+Flask高可用Web集群,运行命令运行mysql容器:

docker run --name mysql -p 3306:3306 -e MYSQL_ROOT_PASSWORD=root -d mysql:8.0.19

注意这里如果宿主机已经有Mysql服务了,需要将:左边的端口号错开,改成3307或者其他什么端口。

随后在终端运行命令安装Gorm包:

go get -u github.com/jinzhu/gorm

这里-u参数的意思是为当前用户安装,并不局限于某个项目。

随后安装Mysql数据库链接驱动包:

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

接着在任意位置编写test.go脚本:

package main  
  
import (  
    "fmt"  
    "github.com/jinzhu/gorm"  
    _ "github.com/jinzhu/gorm/dialects/mysql"  
)  
  
func main() {  
    db, err := gorm.Open("mysql", "root:root@(localhost)/mytest?charset=utf8mb4&parseTime=True&loc=Local")  
  
    if err != nil {
                fmt.Println(err)
               fmt.Println("连接数据库出错")  
        return  
    }  
  
    defer db.Close()
        fmt.Println("链接Mysql成功")  
}

这里导入输出包和Gorm包,同时通过下划线的形式导入mysql驱动包,这样做的好处是mysql驱动的init()函数会在被导入时执行,因为我们并不需要驱动包的具体模块或者函数,而仅仅是用它连一下数据库而已。

随后,创建结构体变量db,注意Open函数对应的Mysql参数是否正确。

注意,结构体变量赋值过程中如果报错,需要判断err变量内容,并且使用return关键字提前结束逻辑,关于golang的错误处理,可参见:人非圣贤孰能无过,Go lang1.18入门精炼教程,由白丁入鸿儒,Go lang错误处理机制EP11

最后,使用defer关键字在所有逻辑执行后关闭Mysql数据库链接。

编译执行后,程序返回:

链接Mysql成功

当然Gorm并不仅仅只能操作Mysql,其他主流数据库也都支持,比方说Sqllite3,事实上,在成本有限或者缓存体系比较完备的情况下,Sqllite3完全可以替代Mysql,首先安装sqllite3驱动:

go get -u github.com/jinzhu/gorm/dialects/sqlite

然后修改test.go文件:

package main  
  
import (  
    "fmt"  
  
    "github.com/jinzhu/gorm"  
    //_ "github.com/jinzhu/gorm/dialects/mysql"  
    _ "github.com/jinzhu/gorm/dialects/sqlite"  
)  
  
func main() {  
    //db, err := gorm.Open("mysql", "root:root@(localhost)/mytest?charset=utf8mb4&parseTime=True&loc=Local")  
  
    db, err := gorm.Open("sqlite3", "/tmp/gorm.db")  
  
    if err != nil {  
        fmt.Println(err)  
        fmt.Println("连接数据库出错")  
        return  
    }  
  
    defer db.Close()  
  
    fmt.Println("链接sqllite3成功")  
}

编译执行后返回:

链接sqllite3成功

数据库操作

连接好数据库之后,我们就可以做一些数据库层面的操作了,比如程序层面的数据库迁移操作:

// 文章信息  
type ArticleInfo struct {  
    ID     uint  
    Title  string  
    Author string  
}

这里建立文章表的结构体数据,随后编写迁移逻辑:

//数据迁移  
db.AutoMigrate(&ArticleInfo{})  
fmt.Println("表创建成功")

进入Mysql命令行,输入命令:

MySQL [(none)]> use mytest;  
Database changed  
MySQL [mytest]> desc article_infos  
    -> ;  
+--------+------------------+------+-----+---------+----------------+  
| Field  | Type             | Null | Key | Default | Extra          |  
+--------+------------------+------+-----+---------+----------------+  
| id     | int(10) unsigned | NO   | PRI | NULL    | auto_increment |  
| title  | varchar(255)     | YES  |     | NULL    |                |  
| author | varchar(255)     | YES  |     | NULL    |                |  
+--------+------------------+------+-----+---------+----------------+  
3 rows in set (0.03 sec)  
  
MySQL [mytest]>

没有问题。

创建数据:

a1 := ArticleInfo{1, "iris", "iris"}  
a2 := ArticleInfo{2, "iris", "女"}  
// 创建记录  
db.Create(&a1)  
db.Create(&a2)

这里我们声明两个结构体变量,然后将其指针传递给db变量的Create函数,编译运行后,键入命令进行查询操作:

MySQL [mytest]> select * from article_infos\g  
+----+-------+--------+  
| id | title | author |  
+----+-------+--------+  
|  1 | iris  | iris   |  
|  2 | iris  | 女     |  
+----+-------+--------+

随后,将刚才入库的数据查询出来:

// 查询  
a := new(ArticleInfo)  
db.First(a)  
fmt.Println(a)

这里通过new关键字初始化结构体,然后使用First函数获取第一条记录。

程序返回:

链接mysql成功  
&{1 iris iris}

查出来的结构体指针可以直接用来做修改操作:

// 查询  
    a := new(ArticleInfo)  
    db.First(a)  
    fmt.Println(a)  
  
    //修改  
    db.Model(&a).Update("Author", "123")  
  
    fmt.Println(a)

程序返回:

链接mysql成功  
&{1 iris iris}  
&{1 iris 123}

非常方便。

最后,是删除操作:

// 删除  
db.Delete(&a)

这里通过指针传入Delete函数即可:

MySQL [mytest]> select * from article_infos\g  
+----+-------+--------+  
| id | title | author |  
+----+-------+--------+  
|  2 | iris  | 女     |

该条记录已经被物理删除。

执行原生SQL

如果我们需要执行原生的sql语句,Gorm也提供了对应的函数:

var articles []ArticleInfo  
//   查询 执行用Scan 和Find 一样  
db = db.Raw("select * from article_infos ").Scan(&articles)  
fmt.Println(articles)

这里只需要声明文章的结构体变量,然后执行Scan函数即可:

[{2 iris 女} {3 iris iris} {4 iris 女}]

这里会返回一个切片嵌套结构体的结果集。

除此之外,更新和删除操作:

//  更新和删除.插入用 Exec  
db = db.Exec("update article_infos set author='123' where id = 2")  
fmt.Println("更新了", db.RowsAffected, "条数据")  
db = db.Exec("delete from article_infos where id=4")  
fmt.Println("更新了", db.RowsAffected, "条数据")

程序返回:

[]main.ArticleInfo更新了 1 条数据  
更新了 1 条数据

结语

目前Golang的比较流行的ORM包除了Gorm,还有Xorm,对比Python数据库ORM的百花齐放,百家争鸣,Go lang还有很长的一段路需要走,真实环境下的数据库操作也不仅仅是增删改查,更多操作请移步Gorm官方文档:https://gorm.io/zh\_CN/docs/


刘悦的技术博客
51 声望10 粉丝