partial repository layer added

pkg/repository/transaction.go

@@ -0,0 +1,268 @@
+package repository
+
+import (
+	"context"
+	"fmt"
+	"github.com/go-pg/pg/v10/orm"
+	"wallet-api/pkg/filter"
+	"wallet-api/pkg/model"
+	"wallet-api/pkg/utl/common"
+
+	"github.com/go-pg/pg/v10"
+)
+
+type TransactionRepository struct {
+	db                          *pg.DB
+	subscriptionRepository      *SubscriptionRepository
+	transactionStatusRepository *TransactionStatusRepository
+}
+
+func NewTransactionRepository(db *pg.DB, ss *SubscriptionRepository, tsr *TransactionStatusRepository) *TransactionRepository {
+	return &TransactionRepository{
+		db:                          db,
+		subscriptionRepository:      ss,
+		transactionStatusRepository: tsr,
+	}
+}
+
+/*
+New row into transaction table
+
+Inserts
+
+	   	Args:
+			context.Context: Application context
+			*model.NewTransactionBody: Transaction body object
+		Returns:
+			*model.Transaction: Transaction object
+			*model.Exception: Exception payload.
+*/
+func (as *TransactionRepository) New(ctx context.Context, tm *model.Transaction) (*model.Transaction, error) {
+	db := as.db.WithContext(ctx)
+
+	tx, _ := db.Begin()
+	defer tx.Rollback()
+
+	_, err := tx.Model(tm).Insert()
+	if err != nil {
+		return nil, err
+	}
+	tx.Commit()
+
+	return tm, nil
+}
+
+/*
+GetAll
+
+Gets all rows from subscription type table.
+   	Args:
+		context.Context: Application context
+		string: Relations to embed
+	Returns:
+		*model.Exception: Exception payload.
+*/
+// Gets filtered rows from transaction table.
+func (as *TransactionRepository) GetAll(ctx context.Context, filtered *model.FilteredResponse, flt *filter.TransactionFilter) *model.Exception {
+	db := as.db.WithContext(ctx)
+
+	exceptionReturn := new(model.Exception)
+	wm := new([]model.Transaction)
+	transactionStatus := new(model.TransactionStatus)
+
+	tx, _ := db.Begin()
+	defer tx.Rollback()
+
+	tsFlt := filter.NewTransactionStatusFilter(model.Params{})
+	tsFlt.Status = "completed"
+	_, err := as.transactionStatusRepository.GetTx(tx, transactionStatus, tsFlt)
+
+	if err != nil {
+		exceptionReturn.StatusCode = 400
+		exceptionReturn.ErrorCode = "400117"
+		exceptionReturn.Message = fmt.Sprintf("Error selecting row in \"transactionStatus\" table: %s", err)
+		return exceptionReturn
+	}
+
+	flt.TransactionStatusId = transactionStatus.Id
+
+	query := tx.Model(wm)
+
+	as.OnBeforeGetTransactionFilter(query, flt)
+	err = FilteredResponse(query, wm, filtered)
+	if err != nil {
+		exceptionReturn.StatusCode = 400
+		exceptionReturn.ErrorCode = "400118"
+		exceptionReturn.Message = fmt.Sprintf("Error selecting row(s) in \"transaction\" table: %s", err)
+		return exceptionReturn
+	}
+
+	tx.Commit()
+	return nil
+}
+
+/*
+Check
+
+Checks subscriptions and create transacitons.
+   	Args:
+		context.Context: Application context
+		string: Relations to embed
+	Returns:
+		*model.Exception: Exception payload.
+*/
+// Gets filtered rows from transaction table.
+func (as *TransactionRepository) Check(ctx context.Context, filtered *model.FilteredResponse, flt *filter.TransactionFilter) *model.Exception {
+	db := as.db.WithContext(ctx)
+
+	wm := new([]model.Transaction)
+	sm := new([]model.Subscription)
+	transactionStatus := new(model.TransactionStatus)
+	exceptionReturn := new(model.Exception)
+
+	tx, _ := db.Begin()
+	defer tx.Rollback()
+
+	tsFlt := filter.NewTransactionStatusFilter(model.Params{})
+	tsFlt.Status = "pending"
+	_, err := as.transactionStatusRepository.GetTx(tx, transactionStatus, tsFlt)
+	if err != nil {
+		exceptionReturn.StatusCode = 400
+		exceptionReturn.ErrorCode = "400119"
+		exceptionReturn.Message = fmt.Sprintf("Error selecting row in \"transactionStatus\" table: %s", err)
+		return exceptionReturn
+	}
+	flt.TransactionStatusId = transactionStatus.Id
+
+	smFlt := filter.NewSubscriptionFilter(model.Params{})
+	smFlt.Id = flt.Id
+	smFlt.WalletId = flt.WalletId
+	as.subscriptionRepository.GetAllTx(tx, sm, smFlt)
+
+	for _, sub := range *sm {
+		if sub.HasNew() {
+			as.subscriptionRepository.SubToTrans(&sub, tx)
+		}
+	}
+
+	qry := tx.Model(wm)
+	as.OnBeforeGetTransactionFilter(qry, flt)
+	err = FilteredResponse(qry, wm, filtered)
+	if err != nil {
+		exceptionReturn.StatusCode = 400
+		exceptionReturn.ErrorCode = "400120"
+		exceptionReturn.Message = fmt.Sprintf("Error selecting row in \"transaction\" table: %s", err)
+		return exceptionReturn
+	}
+
+	tx.Commit()
+	return nil
+}
+
+/*
+Edit
+
+Updates row in transaction table by id.
+
+	   	Args:
+			context.Context: Application context
+			*model.TransactionEdit: Object to edit
+			string: id to search
+		Returns:
+			*model.Transaction: Transaction object from database.
+			*model.Exception: Exception payload.
+*/
+func (as *TransactionRepository) Edit(ctx context.Context, tm *model.Transaction) (*model.Transaction, error) {
+	db := as.db.WithContext(ctx)
+
+	tx, _ := db.Begin()
+	defer tx.Rollback()
+
+	_, err := tx.Model(tm).WherePK().UpdateNotZero()
+
+	if err != nil {
+		return nil, err
+	}
+
+	err = tx.Model(tm).WherePK().Select()
+	if err != nil {
+		return nil, err
+	}
+
+	tx.Commit()
+
+	return tm, nil
+}
+
+/*
+Bulk Edit
+
+Updates row in transaction table by id.
+
+	   	Args:
+			context.Context: Application context
+			?[]model.Transaction Bulk Edit: Object to edit
+			string: id to search
+		Returns:
+			*model.Transaction: Transaction object from database.
+			*model.Exception: Exception payload.
+*/
+func (as *TransactionRepository) BulkEdit(ctx context.Context, transactions *[]model.Transaction) (*[]model.Transaction, error) {
+	db := as.db.WithContext(ctx)
+	tx, _ := db.Begin()
+	defer tx.Rollback()
+
+	_, err := tx.Model(transactions).WherePK().UpdateNotZero()
+	if err != nil {
+		return nil, err
+	}
+
+	tx.Commit()
+
+	return transactions, nil
+}
+
+/*
+Get
+
+Gets row from transaction table by id.
+
+	   	Args:
+			context.Context: Application context
+			*model.Auth: Authentication object
+			string: id to search
+			*model.Params: url query parameters
+		Returns:
+			*model.Transaction: Transaction object from database.
+			*model.Exception: Exception payload.
+*/
+func (as *TransactionRepository) Get(ctx context.Context, flt *filter.TransactionFilter) (*model.Transaction, error) {
+	db := as.db.WithContext(ctx)
+	wm := new(model.Transaction)
+
+	tx, _ := db.Begin()
+	defer tx.Rollback()
+
+	qry := tx.Model(wm)
+	as.OnBeforeGetTransactionFilter(qry, flt)
+	err := common.GenerateEmbed(qry, flt.Embed).WherePK().Select()
+	if err != nil {
+		return nil, err
+	}
+
+	tx.Commit()
+
+	return wm, nil
+}
+
+func (as *TransactionRepository) OnBeforeGetTransactionFilter(qry *orm.Query, flt *filter.TransactionFilter) {
+	if flt.WalletId != "" {
+		qry.Where("? = ?", pg.Ident("wallet_id"), flt.WalletId)
+	}
+	if flt.Id != "" {
+		qry.Relation("Wallet").Where("wallet.? = ?", pg.Ident("user_id"), flt.Id)
+	}
+	if flt.NoPending && flt.TransactionStatusId != "" {
+		qry.Where("? = ?", pg.Ident("transaction_status_id"), flt.TransactionStatusId)
+	}
+}