subscription controller, service and model

This commit is contained in:
Fran Jurmanović
2021-06-16 21:33:26 +02:00
parent ebcb507629
commit 150339628e
12 changed files with 347 additions and 0 deletions

View File

@@ -0,0 +1,45 @@
package services
import (
"time"
"wallet-api/pkg/models"
"github.com/go-pg/pg/v10"
)
type SubscriptionService struct {
Db *pg.DB
}
func (as *SubscriptionService) New(body *models.NewSubscriptionBody) *models.Subscription {
tm := new(models.Subscription)
amount, _ := body.Amount.Int64()
tm.Init()
tm.WalletID = body.WalletID
tm.TransactionTypeID = body.TransactionTypeID
tm.SubscriptionTypeID = body.SubscriptionTypeID
tm.CustomRange = body.CustomRange
tm.Description = body.Description
tm.StartDate = body.StartDate
tm.Amount = int(amount)
if body.StartDate.IsZero() {
tm.StartDate = time.Now()
}
as.Db.Model(tm).Insert()
return tm
}
func (as *SubscriptionService) GetAll(am *models.Auth, walletId string, filtered *models.FilteredResponse) {
wm := new([]models.Subscription)
query := as.Db.Model(wm).Relation("Wallet").Where("wallet.? = ?", pg.Ident("user_id"), am.Id)
if walletId != "" {
query = query.Where("? = ?", pg.Ident("wallet_id"), walletId)
}
FilteredResponse(query, wm, filtered)
}