Gorm中修改mysql主键的方法

一、为什么要修改mysql主键策略

1、我们创建mysql数据表的时候正常操作都是采用id自增类型,但是往往会造成以下几个问题 让别人可以猜到你数据库的数据量多少,甚至可以根据当前看到的id可以手动的修改浏览器上id来访问下一条数据 如果分表后会造成主键id是一样的 2、正常的做法可以修改mysql主键策略 直接使用uuid来作为主键,但是这样不好的地方是无序的,不会根据从前往后排列 使用雪花算法生成一个唯一的id

二、gorm中修改mysql表主键的方法

1、直接使用数据表的hook方法,官网地址，直接在hook里面修改主键的值 2、使用gorm插件的方式全局修改,官网地址，官网上的案例似乎有点问题，自行验证

三、使用gorm插件来自定义主键

1、选用的雪花算法第三方库,链接地址

2、创建一个plugin.go的文件，我也不废话，直接贴上我自己定义的代码

package common

import (
 "fmt"
 "gin-admin-api/utils"
 "github.com/bwmarrin/snowflake"
 "gorm.io/gorm"
)

type DbFieldPlugin struct{}

func (op *DbFieldPlugin) Name() string {
 return "dbFieldPlugin"
}

func (op *DbFieldPlugin) Initialize(db *gorm.DB) (err error) {
 // 创建字段的时候雪花算法生成id
 db.Callback().Create().Before("gorm:create").Replace("id", func(db *gorm.DB) {
  node, _ := snowflake.NewNode(1)
  id := node.Generate()
  db.Statement.SetColumn("id", fmt.Sprintf("%d", id))
  accountId := db.Statement.Context.Value("accountId")
  //fmt.Println(accountId, "????")
  db.Statement.SetColumn("created_by", accountId)
 })
 // 创建人
 db.Callback().Create().Before("gorm:create").Replace("created_by", func(db *gorm.DB) {
  if db.Statement.Schema != nil {
   fmt.Println("创建的钩子函数")
   // 获取到上下文中数据
   accountId := db.Statement.Context.Value("accountId")
   fmt.Println(accountId, "????")
   db.Statement.SetColumn("created_by", accountId)
  }
 })
 // 更新人
 db.Callback().Update().Before("gorm:before_update").Replace("updated_by", func(db *gorm.DB) {
  if db.Statement.Schema != nil {
   fmt.Println("修改")
   accountId := db.Statement.Context.Value("accountId")
   db.Statement.SetColumn("updated_by", accountId)
  }
 })
 db.Callback().Update().Before("gorm:after_update").Replace("updated_by1", func(db *gorm.DB) {
  fmt.Println("更新后")
  fmt.Println(db.Statement.Model, "当前数据11")
  fmt.Println(utils.MapToJson(db.Statement.Dest), "当前数据21")
 })
 // 删除人
 db.Callback().Delete().Before("gorm:delete").Replace("deleted_by", func(db *gorm.DB) {
  if db.Statement.Schema != nil {
   accountId := db.Statement.Context.Value("accountId")
   fmt.Println("删除1111", accountId)
   db.Statement.SetColumn("deleted_by", accountId)
  }
 })
 return
}

3、使用插件

db.Use(&DbFieldPlugin{})