mirror of
https://github.com/FJurmanovic/wallet-go-api.git
synced 2026-02-06 06:08:16 +00:00
46 lines
1.0 KiB
Go
46 lines
1.0 KiB
Go
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)
|
|
}
|