102 lines
3.0 KiB
Go
102 lines
3.0 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"github.com/google/uuid"
|
|
|
|
"apskel-pos-be/internal/entities"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type OrganizationRepositoryImpl struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewOrganizationRepositoryImpl(db *gorm.DB) *OrganizationRepositoryImpl {
|
|
return &OrganizationRepositoryImpl{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
func (r *OrganizationRepositoryImpl) Create(ctx context.Context, org *entities.Organization) error {
|
|
return r.db.WithContext(ctx).Create(org).Error
|
|
}
|
|
|
|
func (r *OrganizationRepositoryImpl) GetByID(ctx context.Context, id uuid.UUID) (*entities.Organization, error) {
|
|
var org entities.Organization
|
|
err := r.db.WithContext(ctx).First(&org, "id = ?", id).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &org, nil
|
|
}
|
|
|
|
func (r *OrganizationRepositoryImpl) GetWithOutlets(ctx context.Context, id uuid.UUID) (*entities.Organization, error) {
|
|
var org entities.Organization
|
|
err := r.db.WithContext(ctx).Preload("Outlets").First(&org, "id = ?", id).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &org, nil
|
|
}
|
|
|
|
func (r *OrganizationRepositoryImpl) GetByPlanType(ctx context.Context, planType string) ([]*entities.Organization, error) {
|
|
var organizations []*entities.Organization
|
|
err := r.db.WithContext(ctx).Where("plan_type = ?", planType).Find(&organizations).Error
|
|
return organizations, err
|
|
}
|
|
|
|
func (r *OrganizationRepositoryImpl) UpdatePlanType(ctx context.Context, id uuid.UUID, planType string) error {
|
|
return r.db.WithContext(ctx).Model(&entities.Organization{}).
|
|
Where("id = ?", id).
|
|
Update("plan_type", planType).Error
|
|
}
|
|
|
|
func (r *OrganizationRepositoryImpl) Update(ctx context.Context, org *entities.Organization) error {
|
|
return r.db.WithContext(ctx).Save(org).Error
|
|
}
|
|
|
|
func (r *OrganizationRepositoryImpl) Delete(ctx context.Context, id uuid.UUID) error {
|
|
return r.db.WithContext(ctx).Delete(&entities.Organization{}, "id = ?", id).Error
|
|
}
|
|
|
|
func (r *OrganizationRepositoryImpl) List(ctx context.Context, filters map[string]interface{}, limit, offset int) ([]*entities.Organization, int64, error) {
|
|
var organizations []*entities.Organization
|
|
var total int64
|
|
|
|
query := r.db.WithContext(ctx).Model(&entities.Organization{})
|
|
|
|
for key, value := range filters {
|
|
query = query.Where(key+" = ?", value)
|
|
}
|
|
|
|
if err := query.Count(&total).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
err := query.Limit(limit).Offset(offset).Find(&organizations).Error
|
|
return organizations, total, err
|
|
}
|
|
|
|
func (r *OrganizationRepositoryImpl) Count(ctx context.Context, filters map[string]interface{}) (int64, error) {
|
|
var count int64
|
|
query := r.db.WithContext(ctx).Model(&entities.Organization{})
|
|
|
|
for key, value := range filters {
|
|
query = query.Where(key+" = ?", value)
|
|
}
|
|
|
|
err := query.Count(&count).Error
|
|
return count, err
|
|
}
|
|
|
|
func (r *OrganizationRepositoryImpl) GetByEmail(ctx context.Context, email string) (*entities.Organization, error) {
|
|
var org entities.Organization
|
|
err := r.db.WithContext(ctx).First(&org, "email = ?", email).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &org, nil
|
|
}
|