mirror of
https://gitlab.com/mbugroup/lti-api.git
synced 2026-05-20 05:21:57 +00:00
63 lines
1.9 KiB
Go
63 lines
1.9 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
|
|
"gitlab.com/mbugroup/lti-api.git/internal/common/repository"
|
|
entity "gitlab.com/mbugroup/lti-api.git/internal/entities"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type PaymentRepository interface {
|
|
repository.BaseRepository[entity.Payment]
|
|
BankExists(ctx context.Context, bankId uint) (bool, error)
|
|
CustomerExists(ctx context.Context, customerId uint) (bool, error)
|
|
SupplierExists(ctx context.Context, supplierId uint) (bool, error)
|
|
SupplierCategory(ctx context.Context, supplierId uint) (string, error)
|
|
NextPaymentSequence(ctx context.Context) (int64, error)
|
|
}
|
|
|
|
type PaymentRepositoryImpl struct {
|
|
*repository.BaseRepositoryImpl[entity.Payment]
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewPaymentRepository(db *gorm.DB) PaymentRepository {
|
|
return &PaymentRepositoryImpl{
|
|
BaseRepositoryImpl: repository.NewBaseRepository[entity.Payment](db),
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
func (r *PaymentRepositoryImpl) BankExists(ctx context.Context, bankId uint) (bool, error) {
|
|
return repository.Exists[entity.Bank](ctx, r.db, bankId)
|
|
}
|
|
|
|
func (r *PaymentRepositoryImpl) CustomerExists(ctx context.Context, customerId uint) (bool, error) {
|
|
return repository.Exists[entity.Customer](ctx, r.db, customerId)
|
|
}
|
|
|
|
func (r *PaymentRepositoryImpl) SupplierExists(ctx context.Context, supplierId uint) (bool, error) {
|
|
return repository.Exists[entity.Supplier](ctx, r.db, supplierId)
|
|
}
|
|
|
|
func (r *PaymentRepositoryImpl) SupplierCategory(ctx context.Context, supplierId uint) (string, error) {
|
|
var supplier entity.Supplier
|
|
if err := r.db.WithContext(ctx).
|
|
Select("id", "category").
|
|
First(&supplier, supplierId).Error; err != nil {
|
|
return "", err
|
|
}
|
|
return supplier.Category, nil
|
|
}
|
|
|
|
func (r *PaymentRepositoryImpl) NextPaymentSequence(ctx context.Context) (int64, error) {
|
|
var next int64
|
|
if err := r.db.WithContext(ctx).
|
|
Raw("SELECT nextval('payments_code_seq')").
|
|
Scan(&next).Error; err != nil {
|
|
return 0, err
|
|
}
|
|
return next, nil
|
|
}
|