From 50bb28e90b7093a0cf256ece4517d85110a87670 Mon Sep 17 00:00:00 2001
From: Rogee <rogeecn@qq.com>
Date: Tue, 10 Dec 2024 14:56:19 +0800
Subject: [PATCH] fix: buy history

---
 backend/modules/medias/service.go | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/backend/modules/medias/service.go b/backend/modules/medias/service.go
index e56f598..713939f 100644
--- a/backend/modules/medias/service.go
+++ b/backend/modules/medias/service.go
@@ -378,6 +378,31 @@ func (svc *Service) Checkout(ctx context.Context, tenantId, userId, mediaId int6
 		return errors.Wrap(err, "update user balance")
 	}
 
+	// add user balance
+	tblUserBalances := table.UserBalanceHistories
+	stmtUserBalances := tblUserBalances.
+		INSERT(
+			tblUserBalances.TenantID,
+			tblUserBalances.UserID,
+			tblUserBalances.Balance,
+			tblUserBalances.Target,
+			tblUserBalances.Type,
+		).
+		VALUES(
+			Int(tenantId),
+			Int(userId),
+			Int(-media.Price),
+			Json(pg.BalanceTarget{
+				ID:   media.ID,
+				Name: table.Medias.TableName(),
+			}.MustValue()),
+			String(pg.BalanceTypeConsume.String()),
+		)
+	log.Debug(stmtUserBalances.DebugSql())
+	if _, err := stmtUserBalances.ExecContext(ctx, tx); err != nil {
+		return errors.Wrap(err, "insert user balance")
+	}
+
 	if err := tx.Commit(); err != nil {
 		return errors.Wrap(err, "commit transaction")
 	}