golang數據庫操作之gorm


人生苦短,let’s Go.

 

ORM 框架
ORM:Object Relational Mapping  —— 對象關系映射。

圖片

作用:

  • 通過操作結構體對象,來達到操作數據庫表的目的。

  • 通過結構體對象,來生成數據庫表。

優點:

  • SQL有可能 比較復雜。(Oracle --- 子查詢 -- 嵌套)ORM 操作數據庫,不需要使用 SQL

  • 不同開發者,書寫的 SQL 語句執行效率不同。

go語言支持的 ORM:gORM , xORM

 

[連接數據庫]

import (
 "github.com/jinzhu/gorm"
 _ "github.com/go-sql-driver/mysql"   //"_" 代碼不直接使用包, 底層鏈接要使用!
 "fmt"
)
// mysql: 數據庫的驅動名
// 鏈接數據庫 --格式: 用戶名:密碼@協議(IP:port)/數據庫名?xxx&yyy&
conn, err := gorm.Open("mysql", "root:123456@tcp(127.0.0.1:3306)/test")
if err != nil {
    fmt.Println("gorm.Open err:",err)
    return
}
defer conn.Close()

創建數據庫表。—— 不能使用gorm創建數據庫。 提前使用 SQL語句,創建好想要的數據庫。
AutoMigrate() 創建表。默認創建的表為復數類型。—— 自動添加“s”
在創建之前, 添加  conn.SingularTable(true) 可以創建非復數表名的表。

// 不要復數表名
conn.SingularTable(true)

// 借助 gorm 創建數據庫表.
fmt.Println(conn.AutoMigrate(new(Student)).Error)

 

 

MySQL 包的 init 方法

_ "github.com/go-sql-driver/mysql" 導入包時, “_” ,表示,驅使go系統,在main() 函數被調用之前,自動調用 init() 函數。

go語言中有兩個特殊函數: —— 首字母小寫,包外可見

    • main()   —— 項目的入口函數

    • init() —— 當導包,但沒有在程序中使用。在main() 調用之前,自動被調用。
      查看:光標置於 MySQL包的 “mysql” 上。使用 Ctrl-鼠標左鍵。看到源碼。在 driver.go 底部包含 init() 函數的 定義。

      init() 作用:實現注冊 MySQL 驅動。

 

 

gorm的連接池

-- 默認,gorm框架創建好的MySQL數據庫連接 conn ,就是一個連接池的句柄。
conn, err := gorm.Open("mysql", "root:123456@tcp(127.0.0.1:3306)/test")

初始化全局變量, 接收句柄

// 創建全局連接池句柄
var GlobalConn *gorm.DB
GlobalConn = conn

db.LogMode(false)  是否打印執行SQL

修改連接池初始屬性

// 初始數
GlobalConn.DB().SetMaxIdleConns(10)
// 最大數
GlobalConn.DB().SetMaxOpenConns(100)

 

使用連接池句柄
--- 對比redis連接池:不需要使用 Get() 方法,取一條連接。

// 不要復數表名
GlobalConn.SingularTable(true)
// 借助 gorm 創建數據庫表.
fmt.Println(GlobalConn.AutoMigrate(new(Student)).Error)

 

MySQL 8小時 時區問題

MySQL默認使用的時間 :美國 東 8 區 時間 。—— 北京時間 —— 差 8 小時。
在連接數據庫時,添加屬性:?parseTime=True&loc=Local

conn, err := gorm.Open("mysql”,"root:123456@tcp(127.0.0.1:3306)/test?parseTime=True&loc=Local")

再執行SQL語句、gorm訪問MySQL 使用 北京時間。

 

[SQL操作]

插入:

// insert into student(name, age) values('zhangsan', 100)

func InsertData()  {
 // 先創建數據 --- 創建對象
 var stu Student
 stu.Name = "zhangsan"
 stu.Age = 100

 // 插入(創建)數據
 fmt.Println(GlobalConn.Create(&stu).Error)
}

使用注意事項:

  • 插入數據時,使用的 create() 函數,傳參時,必須傳入 &對象。如果遺漏 “&” 會報錯

  • 要保證 ,在插入數據庫時,GlobalConn.SingularTable(true) 生效。代表不使用 復數表名。

 

 

查詢:

簡單查詢方法:
First(&user): 獲取 user 表中的第一條數據

func SearchData()  {
 var stu Student
 GlobalConn.First(&stu)
 fmt.Println(stu)
}

相當於SQL:SELECT * FROM student ORDER BY id LIMIT 1;

只查詢 nameage 不查詢其他值:

GlobalConn.Select("name, age").First(&stu)

Last(&user): 獲取 user 表中的最后一條數據
相當於SQL:SELECT * FROM users ORDER BY id DESC LIMIT 1

Find(&user): 獲取 user 表中的所有數據。

var stu []Student  // 改為切片
GlobalConn.Select("name, age").Find(&stu)   // Find() 查詢多條

相當於SQL:select name, age from student;

 

條件查詢:

select name, age from student where name = 'lisi';
GlobalConn.Select("name, age").Where("name = ?", "lisi").Find(&stu)


select name, age from student where name = 'lisi' and age = 22;
//方法1:
GlobalConn.Select("name, age").Where("name = ?", "lisi").
   Where("age = ?", 22).Find(&stu)
//方法2:
GlobalConn.Select("name, age").Where("name = ? and age = ?", "lisi", 22).Find(&stu)

有時候 數據查詢不到的情況err也是非nil,這個時候可以進行判斷

if gorm.IsRecordNotFoundError(err) {  // 記錄不存在
...
}

 

更新:

Save(): 根據主鍵更新。如果數據沒有指定主鍵,不更新,變為 “插入”操作。

func UpdateData()  {
    var stu Student
    stu.Name = "wangwu"
    stu.Age = 99 // 沒有指定 id -- 沒有主鍵! --- 插入
    fmt.Println(GlobalConn.Save(&stu).Error)
}

更新一個:

func UpdateData()  {
    var stu Student
    stu.Name = "wangwu"
    stu.Age = 77
    stu.Id = 4  //指定 id -- 更新操作!
    fmt.Println(GlobalConn.Save(&stu).Error)
}

Update(): 更新一個字段。

fmt.Println(GlobalConn.Model(new(Student)).Where("name = ?", "zhaoliu").
            Update("name", "lisi").Error)
// Model(new(Student): 指定更新 “student” 表

Updates(): 更新多個字段。

fmt.Println(GlobalConn.Model(new(Student)).Where("name = ?", "lisi").
            Updates(map[string]interface{}{"name":"liuqi", "age":24}).Error)

 

grom 刪除數據

刪除:物理刪除。真正的執行 Delete

軟刪除:邏輯刪除。不真正刪。不執行Delete

創建表時,在表中添加一個 “刪除字段” 。當需要刪除時,更新 “刪除字段”, 更新為true
查詢時,不查詢 “刪除字段” 為 null 的值。

創建表:

// 創建全局結構體
type Student struct {
      gorm.Model // go語言中, 匿名成員 --- 繼承! Student 繼承 Model
      Name string
      Age int
}
// 在“Model”上,Ctrl-B 跳轉至 Model 類定義。
type Model struct {
      ID        uint `gorm:"primary_key"`
      CreatedAt time.Time
      UpdatedAt time.Time
      DeletedAt *time.Time `sql:"index"`
}
// Model 表由 mysql自動維護,不需要我們手動維護。

 

執行軟刪除:

// 使用 Delete() 參數,指定要刪除的數據所在表的表名。
fmt.Println(GlobalConn.Where("name = ?", "lisi").Delete(new(Student)).Error)


db.Delete(Email{}, "email LIKE ?", "%jinzhu%")

驗證:
select * from student;  依然能看到 “lisi” 相關數據。但是 。delete_at 字段。被填入數據。

 

db.Debug().Delete(u)  debug模式, 打印執行的SQL語句

 

 

在 gorm 框架中,執行 查詢語句:

func SearchData()  {
    var stu []Student
 GlobalConn.Find(&stu)
 fmt.Println(stu)
}

--- 查詢結果為:[ ]  ---- "軟刪除" 成功!
想查詢“軟刪除”的數據:

GlobalConn.Unscoped().Find(&stu)

想 實現 “物理刪除”
--- 借助 Unscoped() 執行刪除。

GlobalConn.Unscoped().Where("name = ?", "lisi").Delete(new(Student))

 

gorm 設置表屬性

修改表字段大小

// 創建全局結構體
type Student struct {
 Id    int
        // string -- varchar。 默認大小255. 可以在創建表時,指定大小。
 Name  string `gorm:"size:100;default:'xiaoming'"`
 Age   int
 Class int    `gorm:"not null"`
}

 

結論:修改表屬性,只能在第一次建表的時候,有效!或者給表增加新字段的時候,有效!其他場景,修改表屬性 ,在 gorm 操作中,無效!

設置時間
默認MySQL數據庫 有 3 種時間:

  • date

  • datetime

  • timeStamp:時間戳。——  gorm 中,只有 timeStamp

如果必須使用 MySQL 數據庫特有的 “數據類型”, 使用 “type” 關鍵字來設置。

// 創建全局結構體
type Student struct {
 Id    int
 Name  string    `gorm:"size:100;default:'xiaoming'"`
 Age   int
 Class int       `gorm:"not null"`
 Join  time.Time `gorm:"type:timestamp"`// 創建 Student 表指定 timestamp類型。
}

 

預加載:  https://gorm.io/zh_CN/docs/preload.html

  [Preload]

type User struct {
  gorm.Model
  Username string
  Orders   []Order
}

type Order struct {
  gorm.Model
  UserID uint
  Price  float64
}

// 查找 user 時預加載相關 Order
db.Preload("Orders").Find(&users)
// SELECT * FROM users;
// SELECT * FROM orders WHERE user_id IN (1,2,3,4);

db.Preload("Orders").Preload("Profile").Preload("Role").Find(&users)
// SELECT * FROM users;
// SELECT * FROM orders WHERE user_id IN (1,2,3,4); // has many
// SELECT * FROM profiles WHERE user_id IN (1,2,3,4); // has one
// SELECT * FROM roles WHERE id IN (4,5,6); // belongs to

  [Joins]

db.Joins("Company").Joins("Manager").Joins("Account").First(&user, 1)
db.Joins("Company").Joins("Manager").Joins("Account").First(&user, "users.name = ?", "jinzhu")
db.Joins("Company").Joins("Manager").Joins("Account").Find(&users, "users.id IN ?", []int{1,2,3,4,5})

帶條件:

  

db.Joins("Company", DB.Where(&Company{Alive: true})).Find(&users)
// SELECT `users`.`id`,`users`.`name`,`users`.`age`,`Company`.`id` AS `Company__id`,`Company`.`name` AS `Company__name` FROM `users` LEFT JOIN `companies` AS `Company` ON `users`.`company_id` = `Company`.`id` AND `Company`.`alive` = true;

 

示例:

CREATE TABLE `user` (
  `id` int(11) NOT NULL AUTO_INCREMENT,
  `name` varchar(64) NOT NULL DEFAULT '',
  `c_sn` int(11) NOT NULL DEFAULT '0',
  `created_at` datetime(3) DEFAULT NULL,
  `updated_at` datetime(3) DEFAULT NULL,
  `deleted_at` datetime(3) DEFAULT NULL,
  PRIMARY KEY (`id`) USING BTREE,
  KEY `index_casusers_on_cas_uid` (`cas_uid`) USING BTREE,
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;


CREATE TABLE `company` (
  `id` int(11) NOT NULL AUTO_INCREMENT,
  `name` varchar(64) NOT NULL DEFAULT '',
  `c_sn` int(11) NOT NULL DEFAULT '0',
  PRIMARY KEY (`id`) USING BTREE,
  KEY `index_c_sn` (`c_sn`) USING BTREE,
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;

belongs to 會與另一個模型建立了一對一的連接。 這種模型的每一個實例都 “屬於” 另一個模型的一個實例。

例如,您的應用包含 user 和 company,並且每個 user 都可以分配給一個 company

主表與其它表建立關連(關鏈關系字段存到主表)

// `User` 屬於 `Company`,       `CompanyID` 是外鍵
type User struct {
  gorm.Model
  Name      string
  CompanyID int
  Company   Company
}

type Company struct {
  ID   int
  Name string
}

重寫外鍵: foreignKey:company_sn (指定user表里的company_sn)

User表外鍵默認關聯Company的主鍵id ,可以重寫成其它外鍵,如Company里的c_sn字段

重寫引用: references:c_sn (指定company_表里的c_sn)

指定user表里的company_sn字段關聯company_表里的c_sn

  1. 寫法一:和mysql里的字段字寫成一樣的

    gorm:"foreignKey:company_sn;references:c_sn"

  2. 寫法一:和結構體里key名寫成一樣的

    gorm:"foreignKey:CompanySn;references:CSn"

// `User` 屬於 `Company`,`CompanyID` 是外鍵
type User struct {
  gorm.Model
  Name      string `json:"name"`
  CompanySn string `json:"company_sn"` //關連接company表
  Company   Company `json:"casusers" gorm:"foreignKey:company_sn;references:c_sn"`
}

type Company struct {
  ID   int `json:"ID"`
  Name string `json:"name"`
  CSn string `json:"cSn"`
}

// 查找 user 時預加載相關 Company
db.Joins("Company").First(&user, 1)
// 查找 user 時預加載相關 Company
db.Preload("Company").Find(&users)

 

更多參考: https://www.cnblogs.com/haima/p/14879268.html

 

 

 

 


 

 

[分頁處理]

// Paginate 分頁封裝
func Paginate(current int,pageSize int) func(db *gorm.DB) *gorm.DB {
	return func (db *gorm.DB) *gorm.DB {
		if current == 0 {
			current = 1
		}
		switch {
		case pageSize > 100:
			pageSize = 100
		case pageSize <= 0:
			pageSize = 10
		}
		offset := (current - 1) * pageSize
		return db.Offset(offset).Limit(pageSize)
	}
}

// 參數中定義
type Pagination struct {
	Current  int `json:"current" form:"page"`      // 頁碼
	PageSize int `json:"pageSize" form:"pageSize"` // 每頁大小
}


使用:
func GetRoleList(query *request.Pagination) (err error, total int64, list []model.Role) {
	db := global.GvaDb.Model(&model.Role{})
	err = db.Where("deleted_at is null").Count(&total).Error
	if err != nil {
		return err, 0, nil
	}
	err = db.Scopes(utils.Paginate(query.Current, query.PageSize)).Order("created_at desc").Find(&list).Error
	return err, total, list
}

  

 

 

 注意點:

1.數據插入的時候, 如果存在created_at,則gorm會自動寫入該值,使用當前時間, 如果不想這樣的話, 在插入的時候, 使用Omit("UpdatedAt") 方法進行忽略

2.定義的model模型中,默認指定的字段是下划線,解析為蛇形字段, 否則的話需要再tag中執行gorm:"column:真實字段名稱" 即可

 

 

 

 

 更多使用參考:

 

 

 

 

 

 

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM