110 lines
2.7 KiB
Go
110 lines
2.7 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"github.com/Caknoooo/go-gin-clean-starter/database/entities"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type (
|
|
TenantRepository interface {
|
|
Create(ctx context.Context, tx *gorm.DB, tenant entities.Tenant) (entities.Tenant, error)
|
|
GetById(ctx context.Context, tx *gorm.DB, tenantId string) (entities.Tenant, error)
|
|
GetByName(ctx context.Context, tx *gorm.DB, name string) (entities.Tenant, error)
|
|
// CheckEmail(ctx context.Context, tx *gorm.DB, email string) (entities.Tenant, bool, error)
|
|
CheckName(ctx context.Context, tx *gorm.DB, name string) (entities.Tenant, bool, error)
|
|
Update(ctx context.Context, tx *gorm.DB, tenant entities.Tenant) (entities.Tenant, error)
|
|
Delete(ctx context.Context, tx *gorm.DB, tenantId string) error
|
|
}
|
|
|
|
tenantRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
)
|
|
|
|
func NewTenantRepository(db *gorm.DB) TenantRepository {
|
|
return &tenantRepository{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
func (r *tenantRepository) Create(ctx context.Context, tx *gorm.DB, tenant entities.Tenant) (entities.Tenant, error) {
|
|
if tx == nil {
|
|
tx = r.db
|
|
}
|
|
|
|
if err := tx.WithContext(ctx).Create(&tenant).Error; err != nil {
|
|
return entities.Tenant{}, err
|
|
}
|
|
|
|
return tenant, nil
|
|
}
|
|
|
|
func (r *tenantRepository) GetById(ctx context.Context, tx *gorm.DB, tenantId string) (entities.Tenant, error) {
|
|
if tx == nil {
|
|
tx = r.db
|
|
}
|
|
|
|
var tenant entities.Tenant
|
|
if err := tx.WithContext(ctx).Where("id = ?", tenantId).Take(&tenant).Error; err != nil {
|
|
return entities.Tenant{}, err
|
|
}
|
|
|
|
return tenant, nil
|
|
}
|
|
|
|
func (r *tenantRepository) GetByName(ctx context.Context, tx *gorm.DB, name string) (entities.Tenant, error) {
|
|
if tx == nil {
|
|
tx = r.db
|
|
}
|
|
|
|
var tenant entities.Tenant
|
|
if err := tx.WithContext(ctx).Where("name = ?", name).Take(&tenant).Error; err != nil {
|
|
return entities.Tenant{}, err
|
|
}
|
|
|
|
return tenant, nil
|
|
}
|
|
|
|
func (r *tenantRepository) Update(ctx context.Context, tx *gorm.DB, tenant entities.Tenant) (entities.Tenant, error) {
|
|
if tx == nil {
|
|
tx = r.db
|
|
}
|
|
|
|
if err := tx.WithContext(ctx).Updates(&tenant).Error; err != nil {
|
|
return entities.Tenant{}, err
|
|
}
|
|
|
|
return tenant, nil
|
|
}
|
|
|
|
func (r *tenantRepository) Delete(ctx context.Context, tx *gorm.DB, tenantId string) error {
|
|
if tx == nil {
|
|
tx = r.db
|
|
}
|
|
|
|
if err := tx.WithContext(ctx).Delete(&entities.Tenant{}, "id = ?", tenantId).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r *tenantRepository) CheckName(ctx context.Context, tx *gorm.DB, name string) (entities.Tenant, bool, error) {
|
|
if tx == nil {
|
|
tx = r.db
|
|
}
|
|
|
|
var tenant entities.Tenant
|
|
if err := tx.WithContext(ctx).Where("name = ?", name).Take(&tenant).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return entities.Tenant{}, false, nil
|
|
}
|
|
return entities.Tenant{}, false, err
|
|
}
|
|
|
|
return tenant, true, nil
|
|
}
|