Golang數據庫編程之GORM庫入門

原創做者，公衆號【程序員讀書】，歡迎關注公衆號，轉載文章請註明出處哦。

在上一篇文章中咱們講解了使用Go語言的標準庫sql/database包操做數據庫的過程，雖然使用sql/database包操做數據也是挺方便的，可是須要本身寫每一條SQL語句，所以咱們可能會本身再度進行封裝，以便更好地使用，而使用現有Go語言開源ORM框架則是代替本身封裝的一個更好的方式。

ORM，即對象關係映射(Object Relational Mapping)，能夠簡單理解爲將關係型數據庫中的數據表映射爲編程語言中的具體的數據類型(如struct)，而GORM庫就是一個使用Go語言實現的且功能很是完善易使用的ORM框架。

下面一塊兒來探索一下如何使用GORM框架吧！

特性

• 關聯 (Has One, Has Many, Belongs To, Many To Many, 多態)
• 鉤子 (在建立/保存/更新/刪除/查找以前或以後)
• 預加載
• 事務
• 複合主鍵
• SQL 生成器
• 數據庫自動遷移
• 自定義日誌
• 可擴展性, 可基於 GORM 回調編寫插件

如何安裝

安裝GORM很是簡單，使用go get -u就能夠在GOPATH目錄下安裝最新GROM框架。

go get -u github.com/jinzhu/gorm
複製代碼

安裝以後，即可以使用import關鍵字導入GORM庫，開始使用啦！

import "github.com/jinzhu/gorm"
複製代碼

支持的數據庫

GORM框架支持MySQL,SQL Server,Sqlite3,PostgreSQL四種數據庫驅動，若是咱們要鏈接這些數據庫，則須要導入不一樣的驅動包及定義不一樣格式的DSN(Data Source Name)。

MySQL

1. 導入

import _ "github.com/jinzhu/gorm/dialects/mysql"
//或者//import _ "github.com/go-sql-driver/mysql"
複製代碼

2. DSN

//user指用戶名，password指密碼,dbname指數據庫名
"user:password@/dbname?charset=utf8&parseTime=True&loc=Local"
複製代碼

SQL Server

1. 導入

import _ "github.com/jinzhu/gorm/dialects/mssql"
複製代碼

2. DSN

//username指用戶名，password指密碼,host指主機地址，port指端口號，database指數據庫名
"sqlserver://username:password@host:port?database=dbname"
複製代碼

Sqlite3

1. 導入

import _ "github.com/jinzhu/gorm/dialects/sqlite"
複製代碼

2. DSN

鏈接Sqlite3數據庫的DSN只須要指定Sqlite3的數據庫文件的路徑便可，如：

//數據庫路徑
/tmp/gorm.db
複製代碼

PostgreSQL

1. 導入

import _ "github.com/jinzhu/gorm/dialects/postgres"
複製代碼

2. DSN

//host指主機地址,port指端口號,user指用戶名,dbname指數據庫名,password指密碼
host=myhost port=myport user=gorm dbname=gorm password=mypassword
複製代碼

鏈接數據庫

上面咱們定義了鏈接不一樣的數據庫的DSN，下面演示若是鏈接數據庫，使用gorm.Open()方法能夠初始化並返回一個gorm.DB結構體，這個結構體封裝了GORM框架全部的數據庫操做方法，下面是gorm.Open()方法的定義：

func Open(dialect string, args ...interface{}) (db *DB, err error)
複製代碼

示例代碼：

package main

import "github.com/jinzhu/gorm"
import _ "github.com/jinzhu/gorm/dialects/mysql"//導入鏈接MySQL數據庫的驅動包

//DSN
const DSN = "root:123456@tcp(localhost:3306)/test?charset=utf8&parseTime=True&loc=Local"
//指定驅動
const DRIVER = "mysql"

var db *gorm.DB

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判斷主鍵是否存在
複製代碼

示例

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包的原生基礎上封裝了簡便的方法，能夠直接調用便將數據映射到對應的結構體模型中，用起來很是簡單，以下面這幾個方法：

//返回第一條
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
複製代碼

示例代碼

//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()方法一塊兒使用。

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
複製代碼

代碼示例

//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()方法的定義：

//value若是有主鍵id，則包含在判斷條件內，經過where能夠指定其餘條件
func (s *DB) Delete(value interface{}, where ...interface{}) *DB
複製代碼

示例代碼

func delete(){
    defer db.Close()
    u := &User{Id: 16}
    db.Delete(u)//根據id
    db.Delete(&User{},"username = ? ","test_one")//根據額外條件刪除
}
複製代碼

小結

在這篇文章中咱們只是講解使用GROM框架如何鏈接和簡單操做數據庫而已，其實GROM框架還有許多更加高級功能，可讓咱們的開發變得更加簡潔，在以後的文章中，咱們再進行詳細講解吧。

---

你的關注，是我寫做路上最大的鼓勵！