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

55 阅读2分钟
	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" "brain/internal/util/client/mysql" "github.com/sirupsen/logrus" )

type UserMapper struct { mapper.BaseMapper[entity.User] }

func init() { err := mysql.Client.AutoMigrate(&entity.User{}) if err != nil { logrus.Errorf("MySQL 用户表迁移失败: %+v", err) } }


然后在别的地方就可以直接使用



users := mapper.UserMapper{}.List(user)


调用对应的数据库方法啦




![img](https://p6-xtjj-sign.byteimg.com/tos-cn-i-73owjymdk6/a7af73fd34324055b27337e8ce1f603c~tplv-73owjymdk6-jj-mark-v1:0:0:0:0:5o6Y6YeR5oqA5pyv56S-5Yy6IEAg5py65Zmo5a2m5Lmg5LmL5b-DQUk=:q75.awebp?rk3s=f64ab15b&x-expires=1771250436&x-signature=FV5K5P4XGx8vT9DvBn2HstVjpmQ%3D)
![img](https://p6-xtjj-sign.byteimg.com/tos-cn-i-73owjymdk6/c7f1419fe14c4be3b0aa257fa039285e~tplv-73owjymdk6-jj-mark-v1:0:0:0:0:5o6Y6YeR5oqA5pyv56S-5Yy6IEAg5py65Zmo5a2m5Lmg5LmL5b-DQUk=:q75.awebp?rk3s=f64ab15b&x-expires=1771250436&x-signature=W%2FtW9YG10gznlBAIBqb84xvpmFE%3D)

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

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


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