naise_pos/db/sqlc/store.go
2023-03-12 11:01:43 +07:00

129 lines
3.0 KiB
Go

package db
import (
"context"
"database/sql"
"fmt"
"github.com/google/uuid"
)
type Store interface {
Querier
PurchaseOrderTx(ctx context.Context, arg PurchasoOrderTxParams) (PurchaseOrderTxResult, error)
}
type SQLStore struct {
*Queries
db *sql.DB
}
func NewStore(db *sql.DB) Store {
return &SQLStore{
db: db,
Queries: New(db),
}
}
func (store *SQLStore) execTx(ctx context.Context, fn func(*Queries) error) error {
tx, err := store.db.BeginTx(ctx, nil)
if err != nil {
return err
}
q := New(tx)
err = fn(q)
if err != nil {
if rbErr := tx.Rollback(); rbErr != nil {
return fmt.Errorf("tx err: %v, rb err : %v", err, rbErr)
}
return err
}
return tx.Commit()
}
type PurchaseOrderProduct struct {
ProductID uuid.UUID `json:"product_id"`
Quantity float64 `json:"quantity"`
Sub_total float64 `json:"sub_total"`
Price float64 `json:"price"`
}
type PurchasoOrderTxParams struct {
MerchantID uuid.UUID `json:"merchant_id"`
SupplierID uuid.UUID `json:"supplier_id"`
Code sql.NullString `json:"code"`
IsPaid bool `json:"is_paid"`
Total float64 `json:"total"`
PaidNominal float64 `json:"paid_nominal"`
Note sql.NullString `json:"note"`
Products []PurchaseOrderProduct `json:"products"`
}
type PurchaseOrderTxResult struct {
PurchaseOrder PurchaseOrder `json:"purchase_order"`
PurchaseOrderDetail []PurchaseOrderDetail `json:"detail"`
}
func (store *SQLStore) PurchaseOrderTx(ctx context.Context, arg PurchasoOrderTxParams) (PurchaseOrderTxResult, error) {
var result PurchaseOrderTxResult
err := store.execTx(ctx, func(q *Queries) error {
var err error
// create purchase order
result.PurchaseOrder, err = q.CreatePurchaseOrder(ctx, CreatePurchaseOrderParams{
MerchantID: arg.MerchantID,
SupplierID: arg.SupplierID,
Code: arg.Code,
IsPaid: arg.IsPaid,
Total: arg.Total,
PaidNominal: arg.PaidNominal,
Note: arg.Note,
})
if err != nil {
return err
}
// create purchase order detail for each products in purchase order
for i := 0; i < len(arg.Products); i++ {
purchaseOrderDetail, err := q.CreatePurchaseOrderDetail(ctx, CreatePurchaseOrderDetailParams{
PurchaseOrderID: result.PurchaseOrder.ID,
MerchantID: arg.MerchantID,
ProductID: arg.Products[i].ProductID,
Quantity: arg.Products[i].Quantity,
SubTotal: arg.Products[i].Sub_total,
ProductPrice: arg.Products[i].Price,
})
if err != nil {
return err
}
result.PurchaseOrderDetail = append(result.PurchaseOrderDetail, purchaseOrderDetail)
product, err := q.GetStockForUpdateStock(ctx, arg.Products[i].ProductID)
if err != nil {
return err
}
err = q.UpdateProductStock(ctx, UpdateProductStockParams{
ID: product.ID,
Stock: product.Stock + arg.Products[i].Quantity,
})
if err != nil {
return err
}
}
return nil
})
return result, err
}