mirror of
https://github.com/FJurmanovic/wallet-go-api.git
synced 2026-02-06 06:08:16 +00:00
129 lines
3.3 KiB
Go
129 lines
3.3 KiB
Go
package services
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
"wallet-api/pkg/models"
|
|
"wallet-api/pkg/utl/common"
|
|
|
|
"github.com/go-pg/pg/v10"
|
|
)
|
|
|
|
type WalletService struct {
|
|
Db *pg.DB
|
|
}
|
|
|
|
func (as *WalletService) New(am *models.NewWalletBody) *models.Wallet {
|
|
|
|
walletModel := new(models.Wallet)
|
|
walletModel.Init()
|
|
walletModel.UserID = am.UserID
|
|
walletModel.Name = am.Name
|
|
as.Db.Model(walletModel).Insert()
|
|
return walletModel
|
|
}
|
|
|
|
func (as *WalletService) Get(am *models.Auth, embed string) *models.Wallet {
|
|
wm := new(models.Wallet)
|
|
|
|
query := as.Db.Model(wm).Where("? = ?", pg.Ident("user_id"), am.Id)
|
|
common.GenerateEmbed(query, embed).Select()
|
|
|
|
return wm
|
|
}
|
|
|
|
func (as *WalletService) GetAll(am *models.Auth, filtered *models.FilteredResponse) {
|
|
wm := new([]models.Wallet)
|
|
|
|
query := as.Db.Model(wm).Where("? = ?", pg.Ident("user_id"), am.Id)
|
|
FilteredResponse(query, wm, filtered)
|
|
}
|
|
|
|
func (as *WalletService) GetHeader(am *models.Auth, embed string, walletId string) *models.WalletHeader {
|
|
wm := new(models.WalletHeader)
|
|
var wallets []models.WalletTransactions
|
|
var wg sync.WaitGroup
|
|
transactions := new([]models.Transaction)
|
|
|
|
query := as.Db.Model(transactions).Relation("Wallet").Where("wallet.? = ?", pg.Ident("user_id"), am.Id).Relation("TransactionType")
|
|
|
|
if walletId != "" {
|
|
query.Where("? = ?", pg.Ident("wallet_id"), walletId)
|
|
}
|
|
|
|
query.Select()
|
|
|
|
currentBalance := 0
|
|
lastMonthBalance := 0
|
|
nextMonth := 0
|
|
|
|
now := time.Now()
|
|
|
|
currentYear, currentMonth, _ := now.Date()
|
|
currentLocation := now.Location()
|
|
|
|
firstOfMonth := time.Date(currentYear, currentMonth, 1, 0, 0, 0, 0, currentLocation)
|
|
firstOfNextMonth := time.Date(currentYear, currentMonth+1, 1, 0, 0, 0, 0, currentLocation)
|
|
firstOfMonthAfterNext := time.Date(currentYear, currentMonth+2, 1, 0, 0, 0, 0, currentLocation)
|
|
|
|
for _, trans := range *transactions {
|
|
addWhere(&wallets, trans.WalletID, trans)
|
|
}
|
|
|
|
for range wallets {
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
for _, trans := range *transactions {
|
|
if trans.TransactionDate.Before(firstOfNextMonth) && trans.TransactionDate.After(firstOfMonth) {
|
|
if trans.TransactionType.Type == "expense" {
|
|
currentBalance -= trans.Amount
|
|
} else {
|
|
currentBalance += trans.Amount
|
|
}
|
|
} else if trans.TransactionDate.Before(firstOfMonthAfterNext) && trans.TransactionDate.After(firstOfNextMonth) {
|
|
if trans.TransactionType.Type == "expense" {
|
|
nextMonth -= trans.Amount
|
|
} else {
|
|
nextMonth += trans.Amount
|
|
}
|
|
} else if trans.TransactionDate.Before(firstOfMonth) {
|
|
if trans.TransactionType.Type == "expense" {
|
|
lastMonthBalance -= trans.Amount
|
|
} else {
|
|
lastMonthBalance += trans.Amount
|
|
}
|
|
}
|
|
}
|
|
}()
|
|
}
|
|
|
|
wg.Wait()
|
|
|
|
wm.LastMonth = lastMonthBalance
|
|
wm.CurrentBalance = currentBalance + lastMonthBalance
|
|
wm.NextMonth = currentBalance + nextMonth
|
|
wm.Currency = "USD"
|
|
wm.WalletId = walletId
|
|
|
|
//common.GenerateEmbed(query, embed).Select()
|
|
|
|
return wm
|
|
}
|
|
|
|
func addWhere(s *[]models.WalletTransactions, walletId string, e models.Transaction) {
|
|
var exists bool
|
|
for _, a := range *s {
|
|
if a.WalletId == walletId {
|
|
a.Transactions = append(a.Transactions, e)
|
|
}
|
|
exists = true
|
|
}
|
|
if !exists {
|
|
var walletTransaction models.WalletTransactions
|
|
walletTransaction.WalletId = walletId
|
|
walletTransaction.Transactions = append(walletTransaction.Transactions, e)
|
|
*s = append(*s, walletTransaction)
|
|
}
|
|
}
|