Golang泛型实战 使用泛型结构体实现gorm baseMapper_gorm 通过反射创建 struct(2)

实例化数据库客户端

在这个例子中，我们使用MySQL数据库进行示范，首先我们需要实例化一个MySQL数据库客户端：

package mysql

import (
	"brain/config"  // 数据库参数保存在该配置文件中
	"github.com/sirupsen/logrus"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"time"
)

var (
	Client *gorm.DB
)

func init() {
	if Client == nil {
		var err error
		Client, err = gorm.Open(mysql.Open(config.BrainCfg.Mysql.Source), &gorm.Config{})
		db, _ := Client.DB()
		db.SetMaxOpenConns(config.BrainCfg.Mysql.MaxOpenConn)
		db.SetMaxIdleConns(config.BrainCfg.Mysql.MaxIdleConn)
		db.SetConnMaxLifetime(time.Duration(config.BrainCfg.Mysql.ConnMaxLifeTime) * time.Second)
		if err != nil {
			logrus.Errorf("mysql client create failed: %+v", err)
		} else {
			logrus.Info("mysql client create successed")
		}
	}
}

定义泛型结构体

我们定义一个泛型结构体BaseMapper，在该结构体中实现数据库的增删改查操作：

package mapper

import (
	"brain/internal/util/client/mysql"
	"errors"
	"github.com/sirupsen/logrus"
	"gorm.io/gorm"
)

type BaseMapper[T any] struct {
}

func (m BaseMapper[T]) Insert(entity T) bool {
	logrus.Infof("insert %+v", entity)
	err := mysql.Client.Create(&entity).Error
	if err != nil {
		logrus.Errorf("insert %+v failed, %+v", entity, err)
		return false
	}
	return true
}

func (m BaseMapper[T]) Delete(entity T) bool {
	logrus.Infof("delete %+v", entity)
	err := mysql.Client.Delete(*&entity).Error
	if err != nil {
		logrus.Errorf("delete %+v failed, %+v", entity, err)
		return false
	}
	return true
}

func (m BaseMapper[T]) Update(entity T) bool {
	logrus.Infof("update %+v", entity)
	err := mysql.Client.Updates(*&entity).Error
	if err != nil {
		logrus.Errorf("update %+v failed, %+v", entity, err)
		return false
	}
	return true
}

func (m BaseMapper[T]) List(entity T) (res []T) {
	logrus.Infof("list %+v", entity)
	err := mysql.Client.Find(&res, *&entity).Error
	if err != nil {
		logrus.Errorf("list %+v failed, %+v", entity, err)
	}
	return res
}

func (m BaseMapper[T]) Detail(entity T) (res T, err error) {
	logrus.Infof("detail %+v", entity)
	result := mysql.Client.First(&res, *&entity)
	if errors.Is(result.Error, gorm.ErrRecordNotFound) {
		return res, result.Error
	}
	return res, nil
}

泛型结构体的使用

假设我们新增一个用户表，包含如下对象：

package entity

import "time"

type User struct {
	ID       string    `gorm:"size:36;primaryKey" json:"id"`
	NickName string    `gorm:"size:128;not null" json:"nickName"`
	Avatar   string    `gorm:"size:2048; not null" json:"avatar"`
	CreateAt time.Time `gorm:"not null;autoCreateTime;type:timestamp" json:"createAt"`
	UpdateAt time.Time `gorm:"not null;autoUpdateTime;type:timestamp" json:"updateAt"`
}

我们要为用户表实现增删改查操作，仅需要如下操作即可：

package mapper

import (
	"brain/internal/common/mapper"
	"brain/internal/domain/user/entity"


![img](https://p3-xtjj-sign.byteimg.com/tos-cn-i-73owjymdk6/a22bb5f0ce844a918b1490d08bbc0b97~tplv-73owjymdk6-jj-mark-v1:0:0:0:0:5o6Y6YeR5oqA5pyv56S-5Yy6IEAg5py65Zmo5a2m5Lmg5LmL5b-DQUk=:q75.awebp?rk3s=f64ab15b&x-expires=1771856679&x-signature=RmfB%2BrimTsLfywyM6gTLioOLHu4%3D)
![img](https://p3-xtjj-sign.byteimg.com/tos-cn-i-73owjymdk6/0547b57d1baa4359a86851bb58c6bf61~tplv-73owjymdk6-jj-mark-v1:0:0:0:0:5o6Y6YeR5oqA5pyv56S-5Yy6IEAg5py65Zmo5a2m5Lmg5LmL5b-DQUk=:q75.awebp?rk3s=f64ab15b&x-expires=1771856679&x-signature=CsaHBU2BRO4mB%2B9uWhINtUpgih8%3D)

**网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。**

**[需要这份系统化的资料的朋友，可以添加戳这里获取](https://gitee.com/vip204888)**


**一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！**