Gin篇:gorm 使用
连接数据库
下面演示如果连接数据库,使用gorm.Open()
方法可以初始化并返回一个gorm.DB
结构体,这个结构体封装了GORM框架所有的数据库操作方法,下面是gorm.Open()
方法的定义:
func Open(dialect string, args ...interface{}) (db *DB, err error)
示例代码:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
package main import "github.com/jinzhu/gorm" //DSN const DSN = "root:123456@tcp(localhost:3306)/test?charset=utf8&parseTime=True&loc=Local" //指定驱动 const DRIVER = "mysql" func init() { var err error db,err = gorm.Open(DRIVER,DSN) if err != nil{ panic(err) } } func main(){ defer db.Close() //退出前执行关闭 //调用db执行具体的逻辑 } |
在上面的例子中,我们在init方法中初始化gorm.DB
结构体,这样在下面的例子可以直接使用变量db
直接进行数据库操作。
基本操作
使用gorm.Open()
函数返回一个gorm.DB
结构体后,我们可以使用gorm.DB
结构体提供的方法操作数据库,下面我们演示如何使用gorm.DB
进行创建、查询、更新、删除等最基本的操作。
其实gorm.DB
是在Go语言的database/sql
库中的sql.DB
结构体上再封装,因为gorm.DB
提供许多和sql.DB
一样的方法,如下所示:
func (s *DB) Exec(sql string, values ...interface{}) *DB func (s *DB) Row() *sql.Row func (s *DB) Rows() (*sql.Rows, error) func (s *DB) Scan(dest interface{}) *DB
另外,使用gorm.DB
结构体中的DB()
方法,可以返回一个sql.DB
对象,如下:
func (s *DB) DB() *sql.DB
下面演示的是使用gorm.DB
结构体中一些更简便的方法进行数据库基本操作,不过,在演示之前,我们需要先定义一个模型,如下:
type User struct { Id int //对应数据表的自增id Username string Password string Email string Phone string }
我们定义了一个名称为User
的结构体,GROM
支持将结构体按规则映射为某个数据表的一行,结构体的每个字段表示数据表的列,结构体的字段首字母必须是大写的。
创建
使用gorm.DB
中的Create()
方法,GORM
会根据传给Create()
方法的模型,向数据表插入一行。
func (s *DB) Create(value interface{}) *DB //创建一行 func (s *DB) NewRecord(value interface{}) bool //根据自增id判断主键是否存在
1
2
3
4
5
6
7
8
9
10
|
func main() { defer db.Close() //具体的逻辑 u := &User{Username: "test_one" , Password: "testOne123456" , Email: "test_one@163.com" , Phone: "13711112222" } db.Create(u) if db.NewRecord(u){ fmt.Println( "写入失败" ) } else { fmt.Println( "写入成功" ) } |
查询
GROM
框架在sql/database
包的原生基础上封装了简便的方法,可以直接调用便将数据映射到对应的结构体模型中,用起来非常简单,如下面这几个方法:
1
2
3
4
5
6
7
8
9
|
//返回第一条 func (s *DB) First(out interface {}, where ... interface {}) *DB //返回最后一条 func (s *DB) Last(out interface {}, where ... interface {}) *DB //返回符合条件的内容 func (s *DB) Find(out interface {}, where ... interface {}) *DB //返回Count(*)结果 func (s *DB) Count(value interface {}) *DB 复制代码 |
示例代码
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
//Find方法示例 func find() { var users = make([]*User, 0) db.Model(&User2{}).Find(&users) fmt.Println(users) } //First方法示例 func first() { var user1,user2 User db.First(&user1) fmt.Println(user1) db.First(&user2, "id = ?" ,20) fmt.Println(user2) } //Last方法示例 func last() { var user1,user2 User db.Last(&user1) fmt.Println(user1) db.First(&user2, "id = ?" ,19) fmt.Println(user2) } //Count方法示例 func count() { var count int db.Model(&User{}).Count(&count) fmt.Println(count) } |
更新
更新数据可以使用gorm.DB
的Save()
或Update()
,UpdateColumn()
,UpdateColumns()
,Updates()
等方法,后面这四个方法需要与Model()
方法一起使用。
1
2
3
4
5
6
7
8
|
func (s *DB) Save(value interface {}) *DB func (s *DB) Model(value interface {}) *DB //下面的方法需要与Model方法一起使用,通过Model方法指定更新数据的条件 func (s *DB) Update(attrs ... interface {}) *DB func (s *DB) UpdateColumn(attrs ... interface {}) *DB func (s *DB) UpdateColumns(values interface {}) *DB func (s *DB) Updates(values interface {}, ignoreProtectedAttrs ...bool) *DB |
代码示例
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
//Save()方法示例 func save(){ u := &User{} db.First(u) u.Email = "test@163.com" db.Save(u) fmt.Println(u) } //Update方法示例 func update() { u := &User{} db.First(u) db.Model(u).Update( "username" , "hello" ) } //Updates方法示例 func updates() { u := &User{} db.First(u) db.Model(&u).Updates( map [string] interface {}{ "username" : "hello2" }) } |
删除
使用gorm.DB
的Delete()
方法可以很简单地删除满足条件的记录,下面是Delete()
方法的定义:
1
2
|
//value如果有主键id,则包含在判断条件内,通过where可以指定其他条件 func (s *DB) Delete(value interface {}, where ... interface {}) *DB |
示例代码
1
2
3
4
5
6
|
func delete(){ defer db.Close() u := &User{Id: 16} db.Delete(u) //根据id db.Delete(&User{}, "username = ? " , "test_one" ) //根据额外条件删除 } |
综合使用
创建 main.go
代码
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
package main import ( _ "github.com/go-sql-driver/mysql" //加载mysql "fmt" ) //注意如果 定义成小写username 引用时 无法调用 type User struct { ID int64 // 列名为 `id` Username string // 列名为 `username` Password string // 列名为 `password` } //设置表名 func (User) TableName() string { return "users" } func main() { db, err := gorm.Open( "mysql" , "数据库用户名:数据库密码@tcp(ip地址:端口号)/数据库名? charset=utf8&parseTime=True&loc=Local&timeout=10ms" ) defer db.Close() if err != nil { fmt.Printf( "mysql connect error %v" , err) } //执行迁移文件 生成数据表:如果数据库表已经存在,可以不做这一步 db.CreateTable(&User{}) //添加数据 user := User{Username: "root" , Password: "root" } result := db.Create(&user) if result.Error != nil { fmt.Printf( "insert row err %v" , result.Error) return } fmt.Println(user.ID) //返回id //查询单条数据 getUser := User{} //SELECT id, first FROM users WHERE id = 1 LIMIT 1; db.Select([]string{ "id" , "username" }).First(&getUser, 1) fmt.Println(getUser) //打印查询数据 //修改数据 user.Username = "update username" user.Password = "update password" db.Save(&user) //查询列表数据 users := []User{} db.Find(&users) fmt.Println(&users) //获取所有数据 //删除数据 db.Delete(&user) } |
执行
gin+gorm+router 快速搭建 crud restful API 接口
apis/apis/user.go
package apis import ( "github.com/gin-gonic/gin" model "api/models" "net/http" "strconv" ) //列表数据 func Users(c *gin.Context) { var user model.User user.Username = c.Request.FormValue("username") user.Password = c.Request.FormValue("password") result, err := user.Users() if err != nil { c.JSON(http.StatusOK, gin.H{ "code": -1, "message": "抱歉未找到相关信息", }) return } c.JSON(http.StatusOK, gin.H{ "code": 1, "data": result, }) } //添加数据 func Store(c *gin.Context) { var user model.User user.Username = c.Request.FormValue("username") user.Password = c.Request.FormValue("password") id, err := user.Insert() if err != nil { c.JSON(http.StatusOK, gin.H{ "code": -1, "message": "添加失败", }) return } c.JSON(http.StatusOK, gin.H{ "code": 1, "message": "添加成功", "data": id, }) } //修改数据 func Update(c *gin.Context) { var user model.User id, err := strconv.ParseInt(c.Param("id"), 10, 64) user.Password = c.Request.FormValue("password") result, err := user.Update(id) if err != nil || result.ID == 0 { c.JSON(http.StatusOK, gin.H{ "code": -1, "message": "修改失败", }) return } c.JSON(http.StatusOK, gin.H{ "code": 1, "message": "修改成功", }) } //删除数据 func Destroy(c *gin.Context) { var user model.User id, err := strconv.ParseInt(c.Param("id"), 10, 64) result, err := user.Destroy(id) if err != nil || result.ID == 0 { c.JSON(http.StatusOK, gin.H{ "code": -1, "message": "删除失败", }) return } c.JSON(http.StatusOK, gin.H{ "code": 1, "message": "删除成功", }) }
database/mysql.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
package database import ( _ "github.com/go-sql-driver/mysql" //加载mysql "github.com/jinzhu/gorm" "fmt" ) func init() { var err error Eloquent, err = gorm.Open( "mysql" , "root:root@tcp(127.0.0.1:3306)/test?charset=utf8&parseTime=True&loc=Local&timeout=10ms" ) if err != nil { fmt.Printf( "mysql connect error %v" , err) } if Eloquent.Error != nil { fmt.Printf( "database error %v" , Eloquent.Error) } } |
models/user.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
package models import ( orm "api/database" ) type User struct { ID int64 `json: "id" ` // 列名为 `id` Username string `json: "username" ` // 列名为 `username` Password string `json: "password" ` // 列名为 `password` } var Users []User //添加 func (user User) Insert() (id int64, err error) { //添加数据 result := orm.Eloquent.Create(&user) id =user.ID if result.Error != nil { err = result.Error return } return } //列表 func (user *User) Users() (users []User, err error) { if err = orm.Eloquent.Find(&users).Error; err != nil { return } return } //修改 func (user *User) Update(id int64) (updateUser User, err error) { if err = orm.Eloquent.Select([]string{ "id" , "username" }).First(&updateUser, id).Error; err != nil { return } //参数1:是要修改的数据 //参数2:是修改的数据 if err = orm.Eloquent.Model(&updateUser).Updates(&user).Error; err != nil { return } return } //删除数据 func (user *User) Destroy(id int64) (Result User, err error) { if err = orm.Eloquent.Select([]string{ "id" }).First(&user, id).Error; err != nil { return } if err = orm.Eloquent.Delete(&user).Error; err != nil { return } Result = *user return } |
router/router.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
|
package main import ( _ "api/database" "api/router" orm "api/database" ) func main() { defer orm.Eloquent.Close() router := router.InitRouter() router.Run( ":8000" ) } |
执行 go run main.go
访问地址
1
2
3
4
|
POST localhost:8006/user 添加 GET localhost:8006/users 列表 DELETE localhost:8006/user/id 删除 PUT localhost:8006/user/id 修改 |
原文作者:yourself
转自链接:https://learnku.com/articles/23548/gingormrouter-quickly-build-crud-restful-api-interface
作者:漏网的鱼
链接:https://www.csev.cn/share/code/20240404153.html
来源:彩色动力-测试分享
版权声明:本文欢迎任何形式转载,转载时完整保留本声明信息(包含原文链接、原文出处、原文作者、版权声明)即可。本文后续所有修改都会第一时间在原始地址更新。
共有 0 条评论