关于go:Golang泛型实战-使用泛型结构体实现gorm-baseMapper

1次阅读

共计 2645 个字符，预计需要花费 7 分钟才能阅读完成。

实现一个根底数据库操作构造体，在 gorm 中一个新的表中可间接应用增删改查操作

在这个例子中，咱们应用 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"
    "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)
    }
}

而后在别的中央就能够间接应用