go gorm レコードを作成する

1. モデルの作成

package mysqltest

import (
	"errors"
	"fmt"

	"gorm.io/gorm"
)

type Product struct {
    
    
	gorm.Model
	ID    uint    `gorm:"primarykey"`
	Name  string  `gorm:"column:name"`
	Price float64 `gorm:"column:price_value"`
	Test  string  `gorm:"-"`
}

type Tabler interface {
    
    
	TableName() string
}

// 修改表名
func (recv Product) TableName() string {
    
    
	return "product"
}

// 钩子
func (recv *Product) BeforeCreate(tx *gorm.DB) (err error) {
    
    
	if recv.Name == "haha" {
    
    
		fmt.Println("不合法哈")
		return errors.New("不合法哈")
	} else {
    
    
		return nil
	}
}

2. デヘルパー

package mysqltest

import (
	"fmt"

	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/clause"
)

var db *gorm.DB

func Initdb() {
    
    
	dsn := "ellis:ellis@tcp(192.168.214.134:3306)/go_db?charset=utf8mb4&parseTime=True&loc=Local"
	var err error
	db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
    
    })
	if err != nil {
    
    
		panic(err)
	}
}

func Create(value interface{
    
    }) {
    
    
	d := db.Create(value)
	fmt.Printf("d.RowsAffected: %v\n", d.RowsAffected)
}

func InitTable(table interface{
    
    }) {
    
    
	db.AutoMigrate(table)
}

// 根据选择的字段插入
func CreateBySelectFields(fields []string, value interface{
    
    }) {
    
    
	db.Select(fields).Create(value)
}

// 批量插入
func CreateMulti(value interface{
    
    }) {
    
    
	d := db.Create(value)
	fmt.Printf("d.RowsAffected: %v\n", d.RowsAffected)
}

// 批量插入
func CreateBatch(value interface{
    
    }) {
    
    
	d := db.CreateInBatches(value, 100)
	fmt.Printf("d.RowsAffected: %v\n", d.RowsAffected)
}

// 忽略钩子
func CreateIgnoreHook(value interface{
    
    }) {
    
    
	db.Session(&gorm.Session{
    
    SkipHooks: true}).Create(value)
}

// 通过map创建
func CreateByMap(value map[string]interface{
    
    }, model interface{
    
    }) {
    
    
	db.Model(model).Create(value)
}

// 冲突啥也不做
func DoNothingWhenInsert(value interface{
    
    }) {
    
    
	db.Clauses(clause.OnConflict{
    
    DoNothing: true}).Create(value)
}

3. 主要

package main

import (
	"ellis/mysqltest"
	"fmt"
)

func main() {
    
    
	mysqltest.Initdb()
	// 创建数据库
	mysqltest.InitTable(&mysqltest.Product{
    
    })
	// 添加单个数据
	mysqltest.Create(&mysqltest.Product{
    
    Name: "dafault"})
	// 添加数据忽略钩子
	mysqltest.CreateIgnoreHook(&mysqltest.Product{
    
    Name: "haha", Price: 100})

	// 选择字段进行插入
	var value *mysqltest.Product = &mysqltest.Product{
    
    Name: "twofield", Price: 100}
	mysqltest.CreateBySelectFields([]string{
    
    "Name", "Price"}, value)
	fmt.Printf("value: %v\n", value)

	// 批量插入
	var value1 []mysqltest.Product = []mysqltest.Product{
    
    mysqltest.Product{
    
    Name: "h1", Price: 1}, mysqltest.Product{
    
    Name: "h2", Price: 2}}
	mysqltest.CreateMulti(value1)
	fmt.Printf("value: %v\n", value1)

	// 批量插入
	var value3 []mysqltest.Product = []mysqltest.Product{
    
    mysqltest.Product{
    
    Name: "h1", Price: 1}, mysqltest.Product{
    
    Name: "h2", Price: 2}}
	mysqltest.CreateBatch(value3)
	fmt.Printf("value: %v\n", value3)
	// 通过map创建
	mysqltest.CreateByMap(map[string]interface{
    
    }{
    
    "Name": "ellis", "Price": "30000"}, mysqltest.Product{
    
    })
	// 冲突donothing
	mysqltest.DoNothingWhenInsert(&mysqltest.Product{
    
    ID: 1, Name: "ellis"})

}

参考