gorm.open mysql_Go语言 orm库之gorm

GORM是一个比较流行且功能齐全的orm库，主要有以下特性全功能 ORM

关联 (Has One，Has Many，Belongs To，Many To Many，多态，单表继承)

Create，Save，Update，Delete，Find 中钩子方法

支持 Preload、Joins 的预加载

事务，嵌套事务，Save Point，Rollback To Saved Point

Context，预编译模式，DryRun 模式

批量插入，FindInBatches，Find/Create with Map，使用 SQL 表达式、Context Valuer 进行 CRUD

SQL 构建器，Upsert，数据库锁，Optimizer/Index/Comment Hint，命名参数，子查询

复合主键，索引，约束

Auto Migration

自定义 Logger

灵活的可扩展插件 API：Database Resolver(多数据库，读写分离)、Prometheus…

每个特性都经过了测试的重重考验

开发者友好

依赖的库

数据库转struct工具

mysql driver

gorm 库

测试库

新建一个数据库testdb，建一个members表CREATE TABLE `members` (

`id` bigint(20) NOT NULL AUTO_INCREMENT,

`name` varchar(50) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NOT NULL DEFAULT '' COMMENT '姓名',

`state` tinyint(4) NOT NULL DEFAULT '0' COMMENT '状态(1:有效，0无效，9删除)',

`ctime` int(10) NOT NULL DEFAULT '0' COMMENT '创建时间',

`mtime` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,

PRIMARY KEY (`id`)

) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci;

安装gorm tool 工具go get -u -v github.com/xxjwxc/gormt@master

通过如下命令查查找gormt文件地址which gormt

和gormt执行文件同目录有个config.yml文件，配置数据库信息out_dir : "./model" # out dir

url_tag : json # web url tag(json,db(https://github.com/google/go-querystring))

language : # language(English,中 文)

db_tag : gorm # DB tag(gorm,db)

singular_table : true # Table name plural (big Camel-Case):gorm.SingularTable

simple : false #simple output

is_out_sql : false # Whether to output sql

is_out_func : true # Whether to output function

is_foreign_key : true # Whether to mark foreign key or not

is_gui : false # Whether to operate on gui

is_table_name : false # Whether to out GetTableName function

is_null_to_point : false # database is 'DEFAULT NULL' then set element type as point

is_web_tag: false

is_web_tag_pk_hidden: false

db_info :

host : "127.0.0.1"

port : 3306

username : "root"

password : "123456"

database : "testdb"

type: 0 # database type (0:mysql , 1:sqlite , 2:mssql)

执行./gormt -g=true

安装依赖库

这里以mysql库为例go get -u gorm.io/gorm

go get -u gorm.io/driver/mysql

创建main.gopackage main

import (

"demo/model"

"errors"

"fmt"

"gorm.io/driver/mysql"

"gorm.io/gorm"

"time"

)

func main() {

//连接数据库

// https://github.com/go-sql-driver/mysql

//dsn := "root:123456@tcp(localhost:3306)/testdb?charset=utf8mb4&parseTime=True&loc=Local"

//db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})

db, err := gorm.Open(mysql.New(mysql.Config{

DSN: "root:123456@tcp(localhost:3306)/testdb?charset=utf8mb4&parseTime=True&loc=Local", // data source name, refer https://github.com/go-sql-driver/mysql#dsn-data-source-name

DefaultStringSize: 256, // add default size for string fields, by default, will use db type `longtext` for fields without size, not a primary key, no index defined and don't have default values

DisableDatetimePrecision: true, // disable datetime precision support, which not supported before MySQL 5.6

DontSupportRenameIndex: true, // drop & create index when rename index, rename index not supported before MySQL 5.7, MariaDB

DontSupportRenameColumn: true, // use change when rename column, rename rename not supported before MySQL 8, MariaDB

SkipInitializeWithVersion: false, // smart configure based on used version

}), &gorm.Config{})

if err != nil {

panic("failed to connect database")

}

//设置连接池

// 获取通用数据库对象 sql.DB ，然后使用其提供的功能

sqlDB, err := db.DB()

// SetMaxIdleConns 用于设置连接池中空闲连接的最大数量。

sqlDB.SetMaxIdleConns(10)

// SetMaxOpenConns 设置打开数据库连接的最大数量。

sqlDB.SetMaxOpenConns(100)

// SetConnMaxLifetime 设置了连接可复用的最大时间。

sqlDB.SetConnMaxLifetime(time.Hour)

//开启debug 控制台输出sql

db=db.Debug()

memberMgr:=model.MembersMgr(db)

//user := model.Members{Name: "haha", Ctime: int(time.Now().Unix()), Mtime: time.Now()}

//result := db.Create(&user) // 通过数据的指针来创建

//fmt.Println(result)

//user.ID // 返回插入数据的主键

//result.Error // 返回 error

//result.RowsAffected // 返回插入记录的条数

//批量插入

//var users = []model.Members{{Name: "haha1",Mtime: time.Now()}, {Name: "haha2",Mtime: time.Now()}, {Name: "haha3",Mtime: time.Now()}}

//db.Create(&users)

//获取单条数据

rows,err:=memberMgr.Get()

fmt.Println(rows)

//获取所有数据

rowArr,err:=memberMgr.Gets()

for _,v:=range rowArr {

fmt.Println(v)

}

members :=&model.Members{}

// 获取第一条记录(主键升序)

db.Where("name = ?", "haha2").First(members)

fmt.Println(members)

//更新

members1:=model.Members{}

// 通过 `RowsAffected` 得到更新的记录数

result := db.Model(members1).Where("name = ?", "haha1").Updates(model.Members{Name: "hello", Mtime: time.Now()})

fmt.Println(result.RowsAffected) // 更新的记录数

fmt.Println(result.Error) // 更新的错误

//删除

//db.Delete(&model.Members{ID:1})

//事务

db.Transaction(func(tx *gorm.DB) error {

// 在事务中执行一些 db 操作(从这里开始，您应该使用 'tx' 而不是 'db')

if err := tx.Create(&model.Members{Name: "Giraffe",Mtime: time.Now()}).Error; err != nil {

// 返回任何错误都会回滚事务

return err

}

if err := tx.Create(&model.Members{Name: "Giraffe",Mtime: time.Now()}).Error; err != nil {

return err

}

// 返回 nil 提交事务

return nil

})

//嵌套事务

user1:=model.Members{Name: "haha2",Mtime: time.Now()}

user2:=model.Members{Name: "haha2",Mtime: time.Now()}

user3:=model.Members{Name: "haha2",Mtime: time.Now()}

db.Transaction(func(tx *gorm.DB) error {

tx.Create(&user1)

tx.Transaction(func(tx2 *gorm.DB) error {

tx2.Create(&user2)

return errors.New("rollback user2") // Rollback user2

})

tx.Transaction(func(tx2 *gorm.DB) error {

tx2.Create(&user3)

return nil

})

return nil

})

}

参考资料

links