19 Go的全能ORM簡單入門

gorm

昨天我的ldap賬戶改了以后，openfalcon(v2.1)-dashboard竟然無法登陸了！顯然，沒有把我的密碼同步到本地數據庫里面，怎么辦？只能改openfalcon用戶認證的源碼了，把ldap密碼同步到數據庫里面，在這里改動的當中，發現openfalcon使用的gorm來操作數據庫，那么我就簡單的了解下gorm作為低級入門了。

話不多說，我們先看下openfalcon用戶登錄這塊的源碼：

1. 導入包與定義結構體

import (
    "github.com/jinzhu/gorm"
)

type DBPool struct {
	Falcon    *gorm.DB
	Graph     *gorm.DB
	Uic       *gorm.DB
	Dashboard *gorm.DB
	Alarm     *gorm.DB
}

2. 定義變量

var db config.DBPool

3. http請求處理與使用orm操作數據庫

	password := utils.HashIt(inputs.Passwd)

	var user uic.User
	user = uic.User{
		Name:   inputs.Name,
		Passwd: password,
		Cnname: inputs.Cnname,
		Email:  inputs.Email,
		Phone:  inputs.Phone,
		IM:     inputs.IM,
		QQ:     inputs.QQ,
	}
    db.Uic.Table("user").Where("name = ?", inputs.Name).Scan(&user)
    
    // 以下代碼 針對剛才說的bug，我自己改過了的，
	if user.ID != 0 { //  update the user's person info
		//h.JSONR(c, http.StatusBadRequest, "name is already existing")
		dt := db.Uic.Table("user").Where("name = ?", inputs.Name).Update("passwd", password)
		if dt.Error != nil {
			h.JSONR(c, http.StatusBadRequest, dt.Error)
			return
		}
	} else { //
		//for create a root user during the first time
		if inputs.Name == "root" {
			user.Role = 2
		}

		dt := db.Uic.Table("user").Create(&user)
		if dt.Error != nil {
			h.JSONR(c, http.StatusBadRequest, dt.Error)
			return
		}
	}

上面的流程說明了，下面就結合openfalcon的源碼看看gorm的使用吧

gorm的簡單使用

查詢用戶信息

先提前定義好用戶信息的結構體，基本與mysql表字段一致

db.Uic.Table("user").Where("name = ?", inputs.Name).Scan(&user)

數據更新

1. 更新一個字段用Update
2. 更新多個字段用Updates

更新一個字段

更新一個字段可以直接寫Update(字段名,新值)

dt := db.Uic.Table("user").Where("name = ?" , inputs.Name).Update("passwd", password)
if dt.Error != nil {
	return
}

更新多個字段

1.使用map
更新多個字段，那么就要寫成map形式了作為Updates的參數

dt := db.Uic.Table("user")..Where("name = ?" , inputs.Name).Updates(map[string]interface{}{"name": "hello", "im": xxx, "Phone": 182****0534})

2.使用struct來更新多個字段,使用struct更新多個屬性，只會更新這些更改的和非空白字段,也就是說當使用struct更新時，FORM將僅更新具有非空值的字段

dt := db.Uic.Table("user")..Where("name = ?" , inputs.Name).Updates(User{Name: "hello", cnname: xxx})

更新全部字段

使用Save方法來更新所有字段，傳入一個結構體的內存地址，凡是這個結構體的字段，都給更新了。

dt.Save(&user)

好了，到此為止，現學現用，如果還想學期其他功能的同學，可以查看 GORM 中文文檔教程