90 lines
1.6 KiB
Go
90 lines
1.6 KiB
Go
|
package database
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
|
||
|
|
"gorm.io/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
type BaseDAO[T any] struct{}
|
||
|
|
|
||
|
|
func (b BaseDAO[T]) Migrate(db *gorm.DB, e T) error {
|
||
|
|
return db.Set("gorm:table_options", "CHARSET=utf8mb4").AutoMigrate(e)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (BaseDAO[T]) GetAll(db *gorm.DB, e T) ([]T, error) {
|
||
|
|
var r []T
|
||
|
|
if err := db.Find(&r, e).Error; err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return r, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (BaseDAO[T]) GetOne(db *gorm.DB, e T) (T, error) {
|
||
|
|
var r T
|
||
|
|
if err := db.First(&r, e).Error; err != nil {
|
||
|
|
return r, err
|
||
|
|
}
|
||
|
|
return r, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (BaseDAO[T]) GetSome(db *gorm.DB, e T, limit, offset int) ([]T, error) {
|
||
|
|
var r []T
|
||
|
|
if err := db.Find(&r, e).Limit(limit).Offset(offset).Error; err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return r, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (BaseDAO[T]) Create(db *gorm.DB, e T) (T, error) {
|
||
|
|
if err := db.Create(&e).Error; err != nil {
|
||
|
|
return e, err
|
||
|
|
}
|
||
|
|
return e, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (BaseDAO[T]) FirstOrCreate(db *gorm.DB, e T) (T, error) {
|
||
|
|
if err := db.FirstOrCreate(&e).Error; err != nil {
|
||
|
|
return e, err
|
||
|
|
}
|
||
|
|
return e, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (BaseDAO[T]) Update(db *gorm.DB, e T) (T, error) {
|
||
|
|
if err := db.Updates(&e).Error; err != nil {
|
||
|
|
return e, err
|
||
|
|
}
|
||
|
|
return e, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b BaseDAO[T]) UpdateOrCreate(db *gorm.DB, e T) (T, error) {
|
||
|
|
e, err := b.Update(db, e)
|
||
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
|
|
return b.Create(db, e)
|
||
|
|
}
|
||
|
|
return e, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (BaseDAO[T]) Delete(db *gorm.DB, e T) error {
|
||
|
|
if err := db.Delete(e).Error; err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type IBaseDAO interface {
|
||
|
|
Migrate()
|
||
|
|
|
||
|
|
GetAll()
|
||
|
|
GetOne()
|
||
|
|
GetSome()
|
||
|
|
|
||
|
|
Create()
|
||
|
|
FirstOrCreate()
|
||
|
|
|
||
|
|
Update()
|
||
|
|
UpdateOrCreate()
|
||
|
|
|
||
|
|
Delete()
|
||
|
|
}
|