Beego ORM框架是一个独立的ORM模块,主要用于数据库操作。
说明: 对象-关系映射(Object/Relation Mapping,简称ORM), 在Go语言中就是将struct类型和数据库记录进行映射。
下面介绍如何操作mysql数据库。
1.安装包
因为beego orm是独立的模块,所以需要单独安装包。
// 安装beego orm包
go get github.com/astaxie/beego/orm
安装mysql驱动
go get github.com/go-sql-driver/mysql
beego orm包操作什么数据库,就需要单独安装对应的数据库驱动。
2.导入包
import (
// 导入orm包
"github.com/astaxie/beego/orm"
// 导入mysql驱动
_ "github.com/go-sql-driver/mysql"
)
3.连接mysql数据库
操作数据库之前首先需要配置好mysql数据库连接参数,通常在beego项目中,我们都会在main.go文件,对数据库进行配置,方便整个项目操作数据库。
例子:
package main
import (
_ "beegodemo/routers"
"github.com/astaxie/beego"
// 导入orm包
"github.com/astaxie/beego/orm"
// 导入mysql驱动
_ "github.com/go-sql-driver/mysql"
)
// 通过init函数配置mysql数据库连接信息
func init() {
// 这里注册一个default默认数据库,数据库驱动是mysql.
// 第三个参数是数据库dsn, 配置数据库的账号密码,数据库名等参数
// dsn参数说明:
// username - mysql账号
// password - mysql密码
// db_name - 数据库名
// 127.0.0.1:3306 - 数据库的地址和端口
orm.RegisterDataBase("default", "mysql", "username:password@tcp(127.0.0.1:3306)/db_name?charset=utf8&parseTime=true&loc=Local")
// 打开调试模式,开发的时候方便查看orm生成什么样子的sql语句
orm.Debug = true
}
func main() {
beego.Run()
}
4.定义模型(Model)
orm操作通常都是围绕struct对象进行,我们先定义一个表结构,方便后面演示数据库操作。
4.1.定义表结构
这里我们创建一个简单的订单表。
CREATE TABLE `orders` (
`id` int(10) unsigned NOT NULL AUTO_INCREMENT COMMENT '自增ID',
`shop_id` int(10) unsigned NOT NULL COMMENT '店铺id',
`customer_id` int(10) unsigned NOT NULL COMMENT '用户id',
`nickname` varchar(20) DEFAULT NULL COMMENT '用户昵称',
`address` varchar(200) NOT NULL DEFAULT '' COMMENT '用户地址',
`init_time` datetime NOT NULL COMMENT '创建订单的时间',
PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8
4.2.定义模型
所谓模型(Model)指的就是关联数据库表的struct类型。
这里我们定义个Order结构体, 他们的字段对应着上面的orders表结构。
// 默认情况struct字段名按照下面规则转换成mysql表字段名:
// 规则: 以下滑线分割首字母大写的单词,然后转换成小写字母。
type Order struct {
// 对应表字段名为: id
Id int
// 对应表字段名为: shop_id , 下面字段名转换规则以此类推。
ShopId int
// struct字段名跟表字段名不一样,通过orm标签指定表字段名为customer_id
Uid int`orm:"column(customer_id)"`
Nickname string
Address string
// 数据库init_time字段是datetime类型,支持自动转换成Go的time.Time类型,但是数据库连接参数必须设置参数parseTime=true
InitTime time.Time
}
// 指定Order结构体默认绑定的表名
func (o *Order) TableName() string {
return "orders"
}
// 注册模型
orm.RegisterModel(new(Order))
5.插入数据
例子1:
// 创建orm对象, 后面都是通过orm对象操作数据库
o := orm.NewOrm()
// 创建一个新的订单
order := Order{}
// 对order对象赋值
order.ShopId = 1
order.Uid = 1002
order.Nickname = "大锤"
order.Address = "深圳南山区"
order.InitTime = time.Now()
// 调用orm的Insert函数插入数据
// 等价sql: INSERT INTO `orders` (`shop_id`, `customer_id`, `nickname`,
`address`, `init_time`) VALUES (1, 1002, '大锤', '深圳南山区', '2019-06-24 23:08:57')
id, err := o.Insert(&order)
if err != nil {
fmt.Println("插入失败")
} else {
// 插入成功会返回插入数据自增字段,生成的id
fmt.Println("新插入数据的id为:", id)
}
例子2 批量插入数据:
o := orm.NewOrm()
orders := []Order{
{ShopId:1, Uid:1001, Nickname:"大锤1", Address:"深圳南山区", InitTime: time.Now()},
{ShopId:1, Uid:1002, Nickname:"大锤2", Address:"深圳南山区", InitTime: time.Now()},
{ShopId:1, Uid:1003, Nickname:"大锤3", Address:"深圳南山区", InitTime: time.Now()},
}
// 调用InsertMulti函数批量插入, 第一个参数指的是要插入多少数据
nums, err := o.InsertMulti(3, orders)
6.更新数据
orm的Update函数是根据主键id进行更新数据的,因此需要预先对id赋值。
例子:
o := orm.NewOrm()
// 需要更新的order对象
order := Order{}
// 先对主键id赋值, 更新数据的条件就是where id=2
order.Id = 2
// 对需要更新的数据进行赋值
order.Nickname = "小锤"
order.Address = "深圳宝安区"
// 调用Update函数更新数据, 默认Update根据struct字段,更新所有字段值,如果字段值为空也一样更新。
// 等价sql: update orders set shop_id=0, customer_id=0, nickname='小锤', address='深圳宝安区', init_time='0000:00:00' where id = 2
num, err := o.Update(&order)
if err != nil {
fmt.Println("更新失败")
} else {
fmt.Println("更新数据影响的行数:", num)
}
// 上面Update直接更新order结构体的所有字段,如果只想更新指定字段,可以这么写
num, err := o.Update(&order, "Nickname", "Address")
// 这里只是更新Nickname和Address两个字段
7.查询数据
默认orm的Read函数也是通过主键id查询数据。
例子:
o := orm.NewOrm()
// 定义order
order := Order{}
// 先对主键id赋值, 查询数据的条件就是where id=2
order.Id = 2
// 通过Read函数查询数据
// 等价sql: select id, shop_id, customer_id, nickname, address, init_time from orders where id = 2
err := o.Read(&order)
if err == orm.ErrNoRows {
fmt.Println("查询不到")
} else if err == orm.ErrMissPK {
fmt.Println("找不到主键")
} else {
fmt.Println(order.Id, order.Nickname)
}
// 通过ReadOrCreate函数,先尝试根据主键id查询数据,如果数据不存在则插入一条数据
created, id, err := o.ReadOrCreate(&order, "Id")
// ReadOrCreate返回三个参数,第一个参数表示是否插入了一条数据,第二个参数表示插入的id
8.删除数据
orm的Delete函数根据主键id删除数据。
例子:
o := orm.NewOrm()
// 定义order
order := Order{}
// 先对主键id赋值, 删除数据的条件就是where id=2
order.Id = 2
if num, err := o.Delete(&order); err != nil {
fmt.Println("删除失败")
} else {
fmt.Println("删除数据影响的行数:", num)
}