gorm自定义Model(2.0)

已于 2022-03-29 15:57:59 修改
阅读量1.2k 收藏
点赞数
分类专栏： gorm golang集成showdoc 文章标签： mysql go sql
于 2021-03-04 16:16:35 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_38544577/article/details/114368785
版权

将下面代码copy下来  以下是 string写法  如需改为int64可按官网写法  也可将下面的Deleted改为sql.NullInt64 将几处设置时间处改为int64即可(有更新 修复了批量创建后面数据无时间 修复当struct中无创建时间修改时间字段时panic)  

gitee.com/yanggit123/tool 此包已经包含gorm1.0、gorm2.0、gorm2.0主从初始化mysql 包含初始化redis 初始化validator 自动生成showdoc、markdown文档 gin后置拦截器 后续功能慢慢增加
/**
* @Auther:gy
* @Date:2020/10/23 11:01
 */

package tool

import (
	"errors"
	"fmt"
	"gorm.io/driver/mysql"
	"gorm.io/driver/postgres"
	"gorm.io/gorm"
	"gorm.io/gorm/clause"
	"gorm.io/gorm/logger"
	"gorm.io/gorm/schema"
	"gorm.io/plugin/dbresolver"
	"reflect"
	"strings"
	"time"
)

/*
tdb.Use(dbresolver.Register(dbresolver.Config{
		// `db2` 作为 sources，`db3`、`db4` 作为 replicas
		Sources:  []gorm.Dialector{mysql.Open("db2_dsn")},
		Replicas: []gorm.Dialector{mysql.Open("db3_dsn"), mysql.Open("db4_dsn")},
		// sources/replicas 负载均衡策略
		Policy: dbresolver.RandomPolicy{},
	}))
*/
func EnableDBResolver2(dbType string, writeconf []MysqlConf, readconf []MysqlConf, policy dbresolver.Policy) (*gorm.DB, error) {
	var db *gorm.DB
	var err error
	conf := writeconf[0]
	dialector, err := getDialector(dbType, conf)
	if err != nil {
		return nil, err
	}
	if dialector == nil {
		return nil, errors.New("类型错误")
	}
	logLevel := logger.Info
	if !conf.IsLogMode {
		logLevel = logger.Error
	}
	db, err = gorm.Open(dialector, &gorm.Config{
		//SkipDefaultTransaction: true,为了确保数据一致性，GORM 会在事务里执行写入操作（创建、更新、删除）。如果没有这方面的要求，您可以在初始化时禁用它。
		NamingStrategy: schema.NamingStrategy{ //GORM 允许用户通过覆盖默认的命名策略更改默认的命名约定，这需要实现接口 Namer
			TablePrefix:   conf.Prefix,     // 表名前缀，`User` 的表名应该是 `t_users`
			SingularTable: conf.IsSingular, // 使用单数表名，启用该选项，此时，`User` 的表名应该是 `t_user`
		},
		Logger:                                   logger.Default.LogMode(logLevel), //允许通过覆盖此选项更改 GORM 的默认 logger
		DisableForeignKeyConstraintWhenMigrating: true,                             //注意 AutoMigrate 会自动创建数据库外键约束，您可以在初始化时禁用此功能
	})
	if err != nil {
		return nil, err
	}
	sources, replicas := []gorm.Dialector{}, []gorm.Dialector{}
	for i := 1; i < len(writeconf); i++ {
		if conf.SSLCertRepeat {
			writeconf[i].tLSConfig = conf.tLSConfig
		}
		dialector, err := getDialector(dbType, writeconf[i])
		if err != nil {
			return nil, err
		}
		sources = append(sources, dialector)
	}
	for i := 0; i < len(readconf); i++ {
		if conf.SSLCertRepeat {
			readconf[i].tLSConfig = conf.tLSConfig
		}
		dialector, err := getDialector(dbType, readconf[i])
		if err != nil {
			return nil, err
		}
		replicas = append(replicas, dialector)
	}
	if policy == nil {
		policy = dbresolver.RandomPolicy{}
	}
	db.Use(dbresolver.Register(dbresolver.Config{
		// `db2` 作为 sources，`db3`、`db4` 作为 replicas
		Sources:  sources,
		Replicas: replicas,
		// sources/replicas 负载均衡策略
		Policy: policy,
	}).SetConnMaxIdleTime(time.Hour).
		SetConnMaxLifetime(time.Hour).
		SetMaxIdleConns(conf.MaxIdleConns).
		SetMaxOpenConns(conf.MaxOpenConns))
	if !conf.NotReplace {
		//自己定义的回调方法Register名字随意
		db.Callback().Create().Before("gorm:create").Register("gorm:update_time_stamp", updateTimeStampForCreateCallback2)
		db.Callback().Update().Before("gorm:update").Register("gorm:update_time_stamp", updateTimeStampForUpdateCallback2)
		//替换删除方法
		db.Callback().Delete().Replace("gorm:delete", deleteCallback2)
	}
	return db, nil
}

func getDialector(dbType string, conf MysqlConf) (dialector gorm.Dialector, err error) {
	switch dbType {
	case "mysql":
		dsn, err := getMysqlDsn(conf)
		if err != nil {
			return nil, err
		}
		return mysql.New(mysql.Config{
			DSN:                       dsn,   // DSN data source name
			DefaultStringSize:         256,   // string 类型字段的默认长度
			DisableDatetimePrecision:  true,  // 禁用 datetime 精度，MySQL 5.6 之前的数据库不支持
			DontSupportRenameIndex:    true,  // 重命名索引时采用删除并新建的方式，MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引
			DontSupportRenameColumn:   true,  // 用 `change` 重命名列，MySQL 8 之前的数据库和 MariaDB 不支持重命名列
			SkipInitializeWithVersion: false, // 根据版本自动配置
		}), nil
	case "pgsql":
		ipHosts := strings.Split(conf.Address, ":")
		if len(ipHosts) < 2 {
			return nil, nil
		}
		dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%s sslmode=disable TimeZone=Asia/Shanghai",
			ipHosts[0],
			conf.Username,
			conf.Password,
			conf.DbName,
			ipHosts[1])
		return postgres.Open(dsn), nil
	default:
		return nil, nil
	}

}

func EnableMysql2(conf MysqlConf) (*gorm.DB, error) {
	var db *gorm.DB
	var err error
	// 参考 https://github.com/go-sql-driver/mysql#dsn-data-source-name 获取详情
	/*newLogger := logger.New(
		log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer
		logger.Config{
			SlowThreshold: time.Second,   // 慢 SQL 阈值
			LogLevel:      logger.Silent, // Log level
			Colorful:      false,         // 禁用彩色打印
		},
	)
	newLogger.LogMode(logger.Silent)*/
	logLevel := logger.Info
	if !conf.IsLogMode {
		logLevel = logger.Error
	}
	dsn := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
		conf.Username,
		conf.Password,
		conf.Address,
		conf.DbName)
	db, err = gorm.Open(mysql.New(mysql.Config{
		DSN:                       dsn,   // DSN data source name
		DefaultStringSize:         256,   // string 类型字段的默认长度
		DisableDatetimePrecision:  true,  // 禁用 datetime 精度，MySQL 5.6 之前的数据库不支持
		DontSupportRenameIndex:    true,  // 重命名索引时采用删除并新建的方式，MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引
		DontSupportRenameColumn:   true,  // 用 `change` 重命名列，MySQL 8 之前的数据库和 MariaDB 不支持重命名列
		SkipInitializeWithVersion: false, // 根据版本自动配置
	}), &gorm.Config{
		//SkipDefaultTransaction: true,为了确保数据一致性，GORM 会在事务里执行写入操作（创建、更新、删除）。如果没有这方面的要求，您可以在初始化时禁用它。
		NamingStrategy: schema.NamingStrategy{ //GORM 允许用户通过覆盖默认的命名策略更改默认的命名约定，这需要实现接口 Namer
			TablePrefix:   conf.Prefix,     // 表名前缀，`User` 的表名应该是 `t_users`
			SingularTable: conf.IsSingular, // 使用单数表名，启用该选项，此时，`User` 的表名应该是 `t_user`
		},
		Logger:                                   logger.Default.LogMode(logLevel), //允许通过覆盖此选项更改 GORM 的默认 logger
		DisableForeignKeyConstraintWhenMigrating: true,                             //注意 AutoMigrate 会自动创建数据库外键约束，您可以在初始化时禁用此功能
	})
	if err != nil {
		return nil, err
	}

	sqlDB, err := db.DB()
	if err != nil {
		return nil, err
	}
	if !conf.NotReplace {
		//自己定义的回调方法Register名字随意
		db.Callback().Create().Before("gorm:create").Register("gorm:update_time_stamp", updateTimeStampForCreateCallback2)
		db.Callback().Update().Before("gorm:update").Register("gorm:update_time_stamp", updateTimeStampForUpdateCallback2)
		//替换删除方法
		db.Callback().Delete().Replace("gorm:delete", deleteCallback2)
	}
	// SetMaxIdleConns 设置空闲连接池中连接的最大数量
	sqlDB.SetMaxIdleConns(conf.MaxIdleConns)
	// SetMaxOpenConns 设置打开数据库连接的最大数量
	sqlDB.SetMaxOpenConns(conf.MaxOpenConns)
	// SetConnMaxLifetime 设置了连接可复用的最大时间
	sqlDB.SetConnMaxLifetime(time.Hour)
	return db, nil
}

type Model2 struct {
	ID        uint           `gorm:"primary_key;comment:'数据id'" json:"id" req:"-"`
	CreatedAt string         `json:"created_at" gorm:"not null;default:'';type:varchar(30)" comment:"创建时间" req:"-"`
	UpdatedAt string         `json:"updated_at" gorm:"not null;default:'';type:varchar(30)" comment:"修改时间" req:"-"`
	DeletedAt gorm.DeletedAt `json:"deleted_at" gorm:"index" comment:"删除时间" req:"-" resp:"-"`
}
type Model2NotDeleted struct {
	ID        uint   `gorm:"primary_key;comment:'数据id'" json:"id" req:"-"`
	CreatedAt string `json:"created_at" gorm:"not null;default:'';type:varchar(30)" comment:"创建时间" req:"-"`
	UpdatedAt string `json:"updated_at" gorm:"not null;default:'';type:varchar(30)" comment:"修改时间" req:"-"`
}

func updateTimeStampForCreateCallback2(db *gorm.DB) {
	if db.Statement.Schema != nil {
		currentTime := getCurrentTime()
		switch db.Statement.ReflectValue.Kind() {
		case reflect.Slice, reflect.Array:
			for i := 0; i < db.Statement.ReflectValue.Len(); i++ {
				rv := reflect.Indirect(db.Statement.ReflectValue.Index(i))
				field1 := db.Statement.Schema.FieldsByDBName["updated_at"]
				if field1 != nil {
					field1.Set(rv, currentTime)
				}
				field := db.Statement.Schema.FieldsByDBName["created_at"]
				if field != nil {
					field.Set(rv, currentTime)
				}
			}
		case reflect.Struct:
			if db.Statement.Schema.FieldsByDBName["created_at"] != nil {
				db.Statement.SetColumn("created_at", currentTime)
			}
			if db.Statement.Schema.FieldsByDBName["updated_at"] != nil {
				db.Statement.SetColumn("updated_at", currentTime)
			}

		}
	}
}

func updateTimeStampForUpdateCallback2(db *gorm.DB) {
	if db.Statement.Schema != nil {
		currentTime := getCurrentTime()
		db.Statement.SetColumn("UpdatedAt", currentTime)
	}
}
func deleteCallback2(db *gorm.DB) {
	if db.Error == nil {
		if db.Statement.Schema != nil && !db.Statement.Unscoped {
			for _, c := range db.Statement.Schema.DeleteClauses {
				db.Statement.AddClause(c)
			}
		}

		if db.Statement.SQL.String() == "" {
			db.Statement.SQL.Grow(100)
			db.Statement.AddClauseIfNotExists(clause.Delete{})

			if db.Statement.Schema != nil {
				_, queryValues := schema.GetIdentityFieldValuesMap(db.Statement.ReflectValue, db.Statement.Schema.PrimaryFields)
				column, values := schema.ToQueryValues(db.Statement.Table, db.Statement.Schema.PrimaryFieldDBNames, queryValues)

				if len(values) > 0 {
					db.Statement.AddClause(clause.Where{Exprs: []clause.Expression{clause.IN{Column: column, Values: values}}})
				}

				if db.Statement.ReflectValue.CanAddr() && db.Statement.Dest != db.Statement.Model && db.Statement.Model != nil {
					_, queryValues = schema.GetIdentityFieldValuesMap(reflect.ValueOf(db.Statement.Model), db.Statement.Schema.PrimaryFields)
					column, values = schema.ToQueryValues(db.Statement.Table, db.Statement.Schema.PrimaryFieldDBNames, queryValues)

					if len(values) > 0 {
						db.Statement.AddClause(clause.Where{Exprs: []clause.Expression{clause.IN{Column: column, Values: values}}})
					}
				}
			}

			db.Statement.AddClauseIfNotExists(clause.From{})
			db.Statement.Build("DELETE", "FROM", "WHERE")
		}

		if _, ok := db.Statement.Clauses["WHERE"]; !db.AllowGlobalUpdate && !ok {
			db.AddError(gorm.ErrMissingWhereClause)
			return
		}

		if !db.DryRun && db.Error == nil {
			if db.Statement.Schema.FieldsByDBName["deleted_at"] != nil {
				db.Statement.SetColumn("deleted_at", time.Now().Format("2006-01-02 15:04:05"))
			}
			result, err := db.Statement.ConnPool.ExecContext(db.Statement.Context, db.Statement.SQL.String(), db.Statement.Vars...)

			if err == nil {
				db.RowsAffected, _ = result.RowsAffected()
			} else {
				db.AddError(err)
			}
		}
	}
}

func getCurrentTime() string {
	return time.Now().Format("2006-01-02 15:04:05")
}

qq_38544577
关注
专栏目录
gorm v2.0简单使用总结
fan_nlnl的博客
08-15 1987
数据库连接 import ( "gorm.io/driver/mysql" "gorm.io/gorm" "testing" ) ​ // 测试连接mysql func TestConnectMysql(t *testing.T) { dsn := "root:1234@(127.0.0.1:3306)/SimpleGorm?charset=utf8mb4&parseTime=True&loc=Local" db, err := gorm.Open(my...
GORM自定义Gorm.Model实现自动添加时间戳
alir66898的博客
12-21 5309
废话不说直接开始 官网(http://gorm.io)有给出一套默认的gorm.Model模型，定义如下 package gorm import "time" // Model base model definition, including fields `ID`, `CreatedAt`, `UpdatedAt`, `DeletedAt`, which could...
GORM模型(Model)定义
jiangshanghe的博客
04-25 4967
在使用ORM工具时，通常我们需要在代码中定义模型（Models）与数据库中的数据表进行映射，在GORM中模型（Models）通常是正常定义的结构体、基本的go类型或它们的指针。 同时也支持sql.Scanner及driver.Valuer接口（interfaces）。 GORM 倾向于约定，而不是配置。默认情况下，GORM 使用ID作为主键，使用结构体名的蛇形复数作为表名，字段名的蛇形作为列名，并使用CreatedAt、UpdatedAt字段追踪创建、更新时间 遵循 GORM 已有的约定...
go使用gorm自动生成model
最新发布
qq_34631220的博客
08-12 481
【代码】go使用gorm自动生成model。
将gorm.Model改为自定义Model
qq_38544577的博客
05-13 2068
我相信有很多人都想把 type Model struct { ID uint `gorm:"primary_key"` CreatedAt time.Time UpdatedAt time.Time DeletedAt *time.Time `sql:"index"` } 把时间类型改为字符串或者int64来记录时间 弄了几次数据库DeletedAt保存的空串导致失败(原因在于DeletedAt必须定义指针类型,这样创建的时候就不会往数据库插入空串),下面以st.
gorm如何巧妙的定义model模型
猛犸象
10-12 6661
在接口开发中，很多时候返回的数据是基于一个数据表的，但是又不仅限与该表的字段，由于golang是强类型的语言，于是这个model就显得不好设计了。 比如想要返回类似于这样的数据： { "code": 0, "data": { "id": 0, "name": "TP-01", "votestart": 1599580800000, "voteend": 1606751999000, "start_url": ""
G0第23章：GORM基本示例、GORM Model定义、主键、表名、列名的约定
tianyi520jx的博客
05-28 2586
在使用GORM工具时，通常我们需要再代码中定义模型（model）与数据库中的数据表进行映射，在GORM中模型（Models）通常是正常定义的结构体、基本的go类型或他们的指针。同时也支持 sql.Scanner 及 driver.Valuer 接口。unique_index"` Role string `gorm:"size:255"` // 设置字段大小为255 MemberNumber * string `gorm:"unique;
gorm 自定义预加载 preload加条件
疯狂的鸭血
07-21 1590
gorm 自定义预加载 preload加条件 // 首页活动模块区域分类表模型 type LabelType struct { Model Id int `json:"id";gorm:"primary_key"` MallId int `json:"mall_id"` Status int `json:"-"` Name string `json:"name"` Weigh
GORM 字段使用自定义类型
热门推荐
有关心情
02-17 1万+
文章目录起步方法1：类型别名场景 1场景 2方法2：定义结构体场景 3结合源码分析Scan 与 Value 方法从何而来？Valuer 接口的注意事项 起步 想在使用 GORM 时使用自定义类型必然事出有因，一般可有以下两种方式： 方法 1： type MyString string 方法 2： type MyString struct { string } 当需求比较简单时，...
GORM GEN 生成代码如何自定义方法和表名
qqxhb 资源共享
04-03 1875
相信你在日常使用gorm gen开发的过程中也有这种场景：通过gen的等方式，从数据库或者sql文件生成了对应的gorm model，已经满足了基本的使用；某些时候希望model上能有一些通用的方法，比如IsEmpty()，判断model是否为空；如果生成之后在修改model，添加自定义方法，下次生成的时候会被覆盖，因此gen提供了WithMethod写一个通用方法的模版这个模版定义了两个方法，你可以把他放到所有需要的model上生成代码指定需要的自定义方法，可以是一个也可以是全部。
Go操作MySQL的一篇垃圾文章
weixin_43795939的博客
02-07 1193
go操作mysq数据库 && gorm的使用
golang数据库连接池参数设置
HONG的个人博客
09-20 2957
可用查看mysql服务器上的连接信息，Command表示连接的当前状态，Command为Sleep时表示休眠、空闲状态，Time表示此状态的已持续时长；当连接持续空闲时长达到maxIdleTime后，该连接就会被关闭并从连接池移除，哪怕当前空闲连接数已经小于SetMaxIdleConns(maxIdleConns)设置的值。maxLifeTime必须要比mysql服务器设置的wait_timeout小，否则会导致golang侧连接池依然保留已被mysql服务器关闭了的连接。连接池里面的连接最大空闲时长。
Gorm 读写分离(转)
Lucky小黄人的博客
07-03 2226
在gorm v2版本中，我们可以看到已经支持读写分离，那读写分离是基于主从复制的，所以我们要先配置mysql的主从复制 1、为什么需要主从复制？ 在业务复杂的系统中，有这么一个情景：如果有一句sql语句需要锁表，导致暂时不能使用读的服务，那么就很影响运行中的业务。使用主从复制，让主库负责写，从库负责读，这样，即使主库出现了锁表的情景，通过读从库也可以保证业务的正常运作。 做数据的热备 业务量越来......
【GORM框架】ORM介绍、GORM简单连接和高级配置详解
Yaoyao2024的博客
05-01 1732
ORM的全称是：Object Relational Mapping(对象关系映射）其中对象：指的是程序中对象/实例（在Go程序中就是结构体实例）关系：指的是关系型数据库（如Mysql)映射：也就是指程序中的对象、实例在数据库中的映射（这个实例映射到数据库中是某个实例）一个类/结构体，对应数据库的一张表类/结构体的实例化，对应这张表的一条记录。
【从java到Go】搭建Go的ORM框架Gorm
幽灵雾的专栏
09-21 1119
基于java基础，快速学习Golang 从零搭建go的orm服务
【Go】GORM手册（一）
如雨随行的博客
03-10 379
GORM基本使用
GO操作MYSQL
aiduo4911的博客
06-19 344
驱动包 MySql驱动 Go语言中的database/sql包不包含数据库驱动，使用时必须注入一个数据库驱动。 下载依赖 go get -u github.com/go-sql-driver/mysql 使用mysql驱动 语法: func Open(driverName, dataSourceName string) (*DB, error) 示例代码： impor...
为什么在连接mysql时，设置 SetConnMaxIdleTime 没有作用
翔云
02-25 1595
在使用golang 连接 mysql时，为了节省连接资源，在连接使用过后，希望在指定长度时间不再使用后，自动关闭连接。这时，经常会使用，设置最大连接有效时间，使用，设置最大空闲连接时间 max idle time。这两个函数的作用听起来差不多，理论上，都能达到相同效果。但是在实际使用中，却有些出人意料。首先看下测试代码，在以下代码中，设置 max idle time 为传入的参数值，设置 max life time 为传入的参数值，设置 max open conn 为1，
GORM框架学习笔记
qzcsu的博客
09-20 4436
简介 gorm是面向golang语言的一种ORM(持久层)框架，支持多种数据库的接入，例如MySQL，PostgreSQL，SQLite，SQL Server，Clickhouse。此框架的特点，弱化了开发者对于sql语言的掌握程度，使用提供的API进行底层数据库的访问。 笔记将会使用mysql数据库作为示例进行演示 使用步骤 一般而言对于数据库的使用步骤如下 创建连接。这个步骤一般会有引入一个数据库驱动的概念，我们的代码就是通过这个驱动去操作底层的数据库。 利用连接执行sql语句，操作数据库。 获取
GORM自定义数据结构
08-25
当需要使用自定义数据结构时，可以通过GORM的标签来指定数据库表的字段和约束。 下面是一个示例，展示如何使用GORM定义一个自定义数据结构： ```go type User struct { gorm.Model Name string Age int Email ...
评论 2
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值