工作用Go: GORM篇

2023-01-17  本文已影响0人  daydaygo

快速熟悉掌握一个框架/库:

  1. 常备文档, 快速查询: gorm文档 https://github.com/go-gorm/gorm.io , 文档在 pages/docs/ 目录下
  2. 按照官方文档快速过一遍, 建立索引, 方便进行第一步的 快速查询, 详细见后续的 index 章节
    1. 小技巧: 文档添加到 vscode, 可通过 cmd-p 通过文件名快速查询&打开
  3. 常备test环境, 随时随地coding进行细节验证, 并记录
  4. 持续步骤3, 最终就能沉淀出 框架/库 的使用指南和最佳实践

本篇 blog 即为 GORM 学习/实践/总结 的成果

资源

https://gorm.io/docs

go get -u gorm.io/gorm
go get -u gorm.io/driver/sqlite
go get -u gorm.io/plugin/soft_delete
go package github
gorm.io/gorm github.com/go-gorm/gorm
gorm.io/driver/sqlite github.com/go-gorm/sqlite
gorm.io/plugin/soft_delete github.com/go-gorm/soft_delete
新 旧.弃用
gorm.io/gorm github.com/jinzhu/gorm

index

model 定义

gorm mysql 示例
数值 ✅ ✅
字符串 ✅ ✅
指针 null *int sql.NullTime gorm.DeleteAt
string datetime
time.Time timesatmp
string timesatmp
time.Time datetime
string date 2021-08-12T00:00:00+08:00
time.Time date {time.Time} 2023-01-03 20:55:09 +0800
string time 22:50:08 
-- 建表语句推荐
CREATE TABLE `tdb_person` (
    -- 默认字段
    `id`          bigint unsigned NOT NULL AUTO_INCREMENT primary key,
    `create_at`   datetime                 DEFAULT CURRENT_TIMESTAMP, -- 使用数据库自动更新
    `update_at`   datetime                 DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
    `delete_at`   datetime,
    `is_del`      tinyint not null default 0,

    -- 基础数据类型
    `username`    varchar(32)     NOT NULL DEFAULT '' COMMENT '用户名',
    `sex`         tinyint         NOT NULL DEFAULT '0' COMMENT '性别',
    `email`       varchar(255)    NOT NULL DEFAULT '' COMMENT '邮箱',
    `person_type` tinyint                  DEFAULT NULL, -- 验证null字段, 数据库不推荐null

    -- 日期时间: 默认都为null, 需要显式设置
    `in_date`     date            NOT NULL COMMENT '进入日期',
    `in_time`     time            NOT NULL COMMENT '进入时间',
    `in_dt`       datetime        NOT NULL COMMENT '进入日期时间',
    `in_ts`       int             NOT NULL COMMENT '进入时间戳',

    UNIQUE KEY `uni_username` (`username`),
    KEY `idx_create_at` (`create_at`),
    KEY `idx_update_at` (`update_at`)
) COMMENT ='tdb-person';

import (
    "time"

    "gorm.io/gorm"
    "gorm.io/plugin/soft_delete"
)

type Model struct {
    Id       uint      `gorm:"primarykey" json:"id"`
    CreateAt time.Time `gorm:"->" json:"create_at"`
    UpdateAt string    `gorm:"->" json:"update_at"`
    DeleteAt gorm.DeletedAt `json:"delete_at"`
    IsDel    soft_delete.DeletedAt `gorm:"softDelete:flag,DeletedAtField:DeleteAt" json:"is_del"` // use `1` `0`
}


type PersonModel struct {
    Model
    Username   string
    Sex        int
    Email      string
    PersonType *int

    InDate time.Time
    InDt   time.Time
    InTime string
    InTs   int64

  More string `gorm:"-"` // 数据库没有字段
}

// gorm.Model
type Model struct {
    ID        uint `gorm:"primarykey"`
    CreatedAt time.Time
    UpdatedAt time.Time
    DeletedAt DeletedAt `gorm:"index"`
}

读

db.
Model(&M{}). // 查询结果指定T可省略model; *T/T 指定id查询;  Table() ❌ 不推荐
Select(). // 推荐使用 string; 根据定义的struct返回
Where(). // 避免fmt导致sql注入
Order().
Limit().Offset()
.Take() // 获取结果 -> 主键/where

func (d *D) Get(id) (*T, error) {
  var row T
  err := db.Take(&row, 1).Error
  if err != nil {
    if errors.Is(err, gorm.ErrRecordNotFound) {
      return nil, err
    }
    return nil, err
  }
  return &row, nil
}

func (d *D) List(id) (*T, error) {
  var row []*T
  err := db.Find(&row).Error
  if err != nil {
    if errors.Is(err, gorm.ErrRecordNotFound) {
      return nil, err
    }
    return nil, err
  }
  return &row, nil
}

写

user := User{Name: "Jinzhu", Age: 18, Birthday: time.Now()}

result := db.Create(&user) // pass pointer of data to Create

user.ID             // returns inserted data's primary key
result.Error        // returns error
result.RowsAffected // returns inserted records count

more

db.Transaction(func(tx *gorm.DB) error { // begin
  if err := tx.Create(&Animal{Name: "Giraffe"}).Error; err != nil {
    return err // rollback
  }
  if err := tx.Create(&Animal{Name: "Lion"}).Error; err != nil {
    return err
  }
  return nil // commit
})

db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})

db.Where("name = ?", "jinzhu").Where("age = ?", 18).Find(&users)
// SELECT * FROM users WHERE name = 'jinzhu' AND age = 18;
db.Where("name = ?", "jinzhu2").Where("age = ?", 20).Find(&users)
// SELECT * FROM users WHERE name = 'jinzhu2' AND age = 20;
db.Find(&users)
// SELECT * FROM users;

tx := db.Where("name = ?", "jinzhu")
tx.Where("age = ?", 18).Find(&users)
// SELECT * FROM users WHERE name = 'jinzhu' AND age = 18;
tx.Where("age = ?", 28).Find(&users)
// SELECT * FROM users WHERE name = 'jinzhu' AND age = 18 AND age = 28; ❌

// session
tx := db.Where("name = ?", "jinzhu").Session(&gorm.Session{})
tx := db.Where("name = ?", "jinzhu").WithContext(context.Background())
tx := db.Where("name = ?", "jinzhu").Debug()
tx.Where("age = ?", 18).Find(&users)
// SELECT * FROM users WHERE name = 'jinzhu' AND age = 18;
tx.Where("age = ?", 28).Find(&users)
// SELECT * FROM users WHERE name = 'jinzhu' AND age = 28;

func TableOfYear(user *User, year int) func(db *gorm.DB) *gorm.DB {
  return func(db *gorm.DB) *gorm.DB {
    tableName := user.TableName() + strconv.Itoa(year)
    return db.Table(tableName)
  }
}

DB.Scopes(TableOfYear(user, 2019)).Find(&users)
// SELECT * FROM users_2019;
上一篇 下一篇

猜你喜欢

热点阅读