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)
调用对应的数据库方法啦


**网上学习资料一大堆,但如果学到的知识不成体系,遇到问题时只是浅尝辄止,不再深入研究,那么很难做到真正的技术提升。**
**[需要这份系统化的资料的朋友,可以添加戳这里获取](https://gitee.com/vip204888)**
**一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!**