init
This commit is contained in:
@@ -0,0 +1,101 @@
|
||||
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
|
||||
}
|
||||
Reference in New Issue
Block a user