golang学习笔记16 beego orm 数据库操作

beego ORM 是一个强大的 Go 语言 ORM 框架。她的灵感主要来自 Django ORM 和 SQLAlchemy。

目前该框架仍处于开发阶段，可能发生任何导致不兼容的改动。

官方文档：https://beego.me/docs/mvc/model/overview.md

已支持数据库驱动：

以上数据库驱动均通过基本测试，但我们仍需要您的反馈。

ORM 特性：

  • 支持 Go 的所有类型存储
  • 轻松上手，采用简单的 CRUD 风格
  • 自动 Join 关联表
  • 跨数据库兼容查询
  • 允许直接使用 SQL 查询／映射
  • 严格完整的测试保证 ORM 的稳定与健壮

更多特性请在文档中自行品读。

安装 ORM：

go get github.com/astaxie/beego/orm
 简单示例：
package main

import (

    "fmt"

    "github.com/astaxie/beego/orm"

    _ "github.com/go-sql-driver/mysql" // import your used driver

)

// Model Struct

type User struct {

    Id   int

    Name string `orm:"size(100)"`

}

func init() {

    // set default database

    orm.RegisterDataBase("default", "mysql", "username:password@tcp(127.0.0.1:3306)/db_name?charset=utf8", 30)

    // register model

    orm.RegisterModel(new(User))

    // create table

    orm.RunSyncdb("default", false, true)

}

func main() {

    o := orm.NewOrm()

    user := User{Name: "slene"}

    // insert

    id, err := o.Insert(&user)

    fmt.Printf("ID: %d, ERR: %v\n", id, err)

    // update

    user.Name = "astaxie"

    num, err := o.Update(&user)

    fmt.Printf("NUM: %d, ERR: %v\n", num, err)

    // read one

    u := User{Id: user.Id}

    err = o.Read(&u)

    fmt.Printf("ERR: %v\n", err)

    // delete

    num, err = o.Delete(&u)

    fmt.Printf("NUM: %d, ERR: %v\n", num, err)

}

　　关联查询

type Post struct {

    Id    int    `orm:"auto"`

    Title string `orm:"size(100)"`

    User  *User  `orm:"rel(fk)"`

}

var posts []*Post

qs := o.QueryTable("post")

num, err := qs.Filter("User__Name", "slene").All(&posts)

　　SQL 查询

当您无法使用 ORM 来达到您的需求时，也可以直接使用 SQL 来完成查询／映射操作。

var maps []orm.Params

num, err := o.Raw("SELECT * FROM user").Values(&maps)

for _,term := range maps{

    fmt.Println(term["id"],":",term["name"])

}

　　事务处理

o.Begin()

...

user := User{Name: "slene"}

id, err := o.Insert(&user)

if err == nil {

    o.Commit()

} else {

    o.Rollback()

}

　　在开发环境下，您可以使用以下指令来开启查询调试模式：

func main() {

    orm.Debug = true

...

　　

开启后将会输出所有查询语句，包括执行、准备、事务等。

例如：

[ORM] - 2013-08-09 13:18:16 - [Queries/default] - [    db.Exec /     0.4ms] -   [INSERT INTO `user` (`name`) VALUES (?)] - `slene`

...

注意：我们不建议您在部署产品后这样做。

 实战例子：
model部分
// UpdateUser updates User by Id and returns error if

// the record to be updated doesn't exist

func UpdateUserById(m *User) (err error) {

	o := orm.NewOrm()

	v := User{Id: m.Id}

	// ascertain id exists in the database

	if err = o.Read(&v); err == nil {

		var num int64

		if num, err = o.Update(m); err == nil {

			fmt.Println("Number of records updated in database:", num)

		}

	}

	return

}

// GetUserByEmailOrMobileAndPassword

func GetUserByEmailOrMobileAndPassword(email string, password string) (maps []orm.Params, err error) {

	//orm.Debug = true

	o := orm.NewOrm()

	//var maps []orm.Params

	num, err := o.Raw("select * FROM user WHERE step>0 and (email=? or mobile=?) and password=? ", email, email,password).Values(&maps)

	//beego.Debug("row nums: ", num)

	if err == nil && num > 0 {

		return maps, nil

	}

	return nil, err

}

// GetUserByEmail

func GetUserByEmail(email string) (maps []orm.Params, err error) {

	o := orm.NewOrm()

	//var maps []orm.Params

	var num int64

	num, err = o.Raw("select * FROM user WHERE email=? ", email).Values(&maps)

	if err == nil && num > 0 {

		//fmt.Println("maps:", maps[0])

		return maps, nil

	}

	return nil, err

}

// UpdateUser step

func UpdateUserStepById(m *User) (err error) {

	//orm.Debug = true

	o := orm.NewOrm()

	v := User{Id: m.Id}

	// ascertain id exists in the database

	if err = o.Read(&v); err == nil {

		var num int64

		if num, err = o.Update(m,"Step", "UpdateTime"); err == nil {

			fmt.Println("Number of records updated in database:", num)

		}

	}

	return

}

　　Control部分（maps []orm.Params 返回的map是个interface对象，需要对里面的值做强制转换才能使用）

type RegisterReq struct {

	Email string

	Password string

}

// @Title Register

// @Description Register User

// @Param	body		body 	controllers.RegisterReq	true		"body for User register"

// @Success 201 {int} models.User

// @Failure 403 body is empty

// @router /register [post]

func (c *UserController) Register() {

	var v RegisterReq

	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &v); err == nil {

		md5pwd := c.GetMd5String(v.Password + beego.AppConfig.String("MD5_SALT"))

		var u models.User

		u.Email = v.Email

		u.Password = md5pwd

		u.Step = 0

		u.Status = 0

		u.Level = 0

		u.Role = 0

		u.Nickname = strings.Split(v.Email, "@")[0]

		u.CreateTime = time.Now()

		u.UpdateTime = time.Now()

		if _, err := models.AddUser(&u); err == nil {

			//c.Ctx.Output.SetStatus(201)

			c.Data["json"] = u

			utils.SendmailForVerify(v.Email)

		} else {

			//"Error 1062: Duplicate entry 'xxx' for key 'email'"

			c.Data["json"] = err.Error()

		}

	} else {

		c.Data["json"] = err.Error()

	}

	c.ServeJSON()

}

type LoginReq struct {

	LoginId string 	`description:"Email or Phone"`

	Password string

}

// @Title Login

// @Description Login

// @Param	body		body 	controllers.LoginReq	true		"body for User login"

// @Success 201 {int} models.User

// @Failure 403 body is empty

// @router /login [post]

func (c *UserController) Login() {

	var v LoginReq

	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &v); err == nil {

		md5pwd := c.GetMd5String(v.Password + beego.AppConfig.String("MD5_SALT"))

		user, _ := models.GetUserByEmailOrMobileAndPassword(v.LoginId,md5pwd)

		if user != nil {

			// get token uid nickname

			id, _  := strconv.Atoi(user[0]["id"].(string))

			nickname := user[0]["nickname"].(string)

			tokenString := utils.GetToken(id, v.LoginId, nickname)

			c.Data["json"] = map[string]interface{}{"success": 0, "msg": "登录成功","token":tokenString,"email":v.LoginId,"nickname":nickname,"id":id}

		} else {

			c.Data["json"] = map[string]interface{}{"success": -1, "msg": "账号密码不对或邮箱未验证激活"}

		}

	} else {

		c.Data["json"] = err.Error()

	}

	c.ServeJSON()

}

type ChangePasswordReq struct {

	OldPassword string

	NewPassword string

}

// @Title Change Password

// @Description Change Password

// @Security mySecurityApiKey

// @Param	body		body 	controllers.ChangePasswordReq	true		"body for Change Password"

// @Success 201 {int} models.User

// @Failure 403 body is empty

// @router /change_password [put]

func (c *UserController) ChangePassword() {

	email := c.GetUserMailByToken()

	var v ChangePasswordReq

	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &v); err == nil {

		md5pwd := c.GetMd5String(v.OldPassword + beego.AppConfig.String("MD5_SALT"))

		user, _ := models.GetUserByEmailOrMobileAndPassword(email, md5pwd)

		if user != nil {

			u, _ := models.GetUserByFilter("email", email)

			u.Password = c.GetMd5String(v.NewPassword + beego.AppConfig.String("MD5_SALT"))

			models.UpdateUserById(u)

			c.Data["json"] = Response{0, "success.", nil}

		} else {

			c.Data["json"] = map[string]interface{}{"success": -1, "msg": "账号密码不对"}

		}

	} else {

		c.Data["json"] = err.Error()

	}

	c.ServeJSON()

}

// Put ...

// @Title Put

// @Description update the User

// @Param	id		path 	string	true		"The id you want to update"

// @Param	body		body 	models.User	true		"body for User content"

// @Success 200 {object} models.User

// @Failure 403 :id is not int

// @router /:id [put]

func (c *UserController) Put() {

	idStr := c.Ctx.Input.Param(":id")

	id, _ := strconv.Atoi(idStr)

	v := models.User{Id: id}

	if err := json.Unmarshal(c.Ctx.Input.RequestBody, &v); err == nil {

		if err := models.UpdateUserById(&v); err == nil {

			c.Data["json"] = "OK"

		} else {

			c.Data["json"] = err.Error()

		}

	} else {

		c.Data["json"] = err.Error()

	}

	c.ServeJSON()

}

// @Title Get user profile

// @Description get user profile

// @Security mySecurityApiKey

// @Success 200 {object} models.User

// @router /profile [get]

func (c *UserController) Profile() {

	uid := c.GetUserIdByToken()

	v, err := models.GetUserById(uid)

	if err != nil {

		c.Data["json"] = err.Error()

	} else {

		c.Data["json"] = v

	}

	c.ServeJSON()

}

// Get Funds ...

// @Get My Funds

// @Security mySecurityApiKey

// @Description get my Funds

// @Success 200 {object} []models.Fund

// @Failure 403

// @router /funds [get]

func (c *UserController) Funds() {

	uid := int(c.GetUserIdByToken())

	fund, _ := models.GetAllAccountByUserId(uid)

	c.Data["json"] = fund

	c.ServeJSON()

}

　　

最新文章

  1. 探索javascript----拖拽
  2. IBM Domino 9 出现 Domino Designer 您正在试图升级多用户安装。请获取正确的安装包以完成升级。 解决方案
  3. C#中的那些全局异常捕获
  4. js promise 风格编程
  5. VS2010+Selenium测试脚本设计
  6. nyoj 139 我排第几个--康拓展开
  7. phpcms v9 下拉菜单 二级 三级子栏目调用方法
  8. STM32的USB速度，终于确定了传输极限，为以后的产品设计提供了数据。
  9. HttpURLConnection&HttpClient网络通信
  10. NSString截取字符串
  11. LINUX USB MASS STORAGE DRIVER流程图
  12. Android Intent 解析之二
  13. jquery 动态生成元素 事件
  14. 【web前端开发】浏览器兼容性处理
  15. 第一章 CLR执行模型
  16. 尚硅谷springboot学习35-启动原理
  17. js 获取getElementsTagName()方法返回值的内容
  18. C# 3.0 / C# 3.5 隐式（推断）类型 var
  19. hibernate--DetachedCriteria（离线条件查询）
  20. CentOS7 mini安装后没有ifconfig命令的解决办法

热门文章

  1. urllib2 python3错误？用from urllib import request来代替！
  2. WINDOWS SERVER 2008 R2安装指南
  3. iframe子父窗口相互操作方法或元素
  4. element-dialog封装成子组件
  5. 定义结构体和table type
  6. Tx.Origin 用作身份验证
  7. 截取字符串后几位用 length
  8. Case when then esle end
  9. SDN概述：简介、工具、环境部署
  10. [LeetCode] 154. Find Minimum in Rotated Sorted Array II_Hard