85 lines
1.4 KiB
Go
85 lines
1.4 KiB
Go
package repository
|
|
|
|
import (
|
|
"errors"
|
|
"repo-pattern/app/repository/smartfilter"
|
|
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
type ListMethod[T schema.Tabler] struct {
|
|
DbConn *gorm.DB
|
|
}
|
|
|
|
func (m *ListMethod[T]) Init(dbConn *gorm.DB) {
|
|
m.DbConn = dbConn
|
|
}
|
|
|
|
func (m ListMethod[T]) List(filter interface{}) (*[]T, error) {
|
|
var (
|
|
model T
|
|
models []T
|
|
)
|
|
|
|
query, err := smartfilter.ToQuery(model, filter, m.DbConn)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
query.Find(&models)
|
|
return &models, nil
|
|
}
|
|
|
|
type GetMethod[T schema.Tabler] struct {
|
|
DbConn *gorm.DB
|
|
}
|
|
|
|
func (m *GetMethod[T]) Init(dbConn *gorm.DB) {
|
|
m.DbConn = dbConn
|
|
}
|
|
|
|
func (m GetMethod[T]) Get(filter interface{}) (*T, error) {
|
|
var (
|
|
model T
|
|
)
|
|
|
|
query, err := smartfilter.ToQuery(model, filter, m.DbConn)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
result := query.First(&model)
|
|
if result.Error != nil {
|
|
return nil, result.Error
|
|
}
|
|
|
|
return &model, nil
|
|
}
|
|
|
|
type ExistsMethod[T schema.Tabler] struct {
|
|
DbConn *gorm.DB
|
|
}
|
|
|
|
func (m *ExistsMethod[T]) Init(dbConn *gorm.DB) {
|
|
m.DbConn = dbConn
|
|
}
|
|
|
|
func (m ExistsMethod[T]) Exists(filter interface{}) (bool, error) {
|
|
var (
|
|
model T
|
|
)
|
|
|
|
query := m.DbConn.Model(model)
|
|
|
|
query, err := smartfilter.ToQuery(model, filter, query)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
result := query.Select("*").First(&model)
|
|
|
|
exists := !errors.Is(result.Error, gorm.ErrRecordNotFound) && result.Error == nil
|
|
return exists, nil
|
|
}
|