Files
lndhub.go/lib/service/balance.go
2022-01-19 16:10:55 +01:00

25 lines
791 B
Go

package service
import (
"context"
"github.com/getAlby/lndhub.go/db/models"
)
func (svc *LndhubService) CurrentBalance(ctx context.Context, userId int64) (int64, error) {
var balance int64
account, err := svc.AccountFor(ctx, "current", userId)
if err != nil {
return balance, err
}
err = svc.DB.NewSelect().Table("account_ledgers").ColumnExpr("sum(account_ledgers.amount) as balance").Where("account_ledgers.account_id = ?", account.ID).Scan(context.TODO(), &balance)
return balance, err
}
func (svc *LndhubService) AccountFor(ctx context.Context, accountType string, userId int64) (models.Account, error) {
account := models.Account{}
err := svc.DB.NewSelect().Model(&account).Where("user_id = ? AND type= ?", userId, accountType).Limit(1).Scan(ctx)
return account, err
}