refactor database schema and models to include transaction ID for user transactions

Author: Armin

.gitignore

@@ -1,32 +1,2 @@
-# If you prefer the allow list template instead of the deny list, see community template:
-# https://github.com/github/gitignore/blob/main/community/Golang/Go.AllowList.gitignore
-#
-# Binaries for programs and plugins
-*.exe
-*.exe~
-*.dll
-*.so
-*.dylib
-
-# Test binary, built with `go test -c`
-*.test
-
-# Code coverage profiles and other test artifacts
-*.out
-coverage.*
-*.coverprofile
-profile.cov
-
-# Dependency directories (remove the comment below to include it)
-# vendor/
-
-# Go workspace file
-go.work
-go.work.sum
-
-# env file
-.env
-
-# Editor/IDE
-# .idea/
-# .vscode/
+.idea
+.vscode

internal/balance/db/db.sql

@@ -1,23 +1,27 @@
 CREATE TABLE user_balances (
-    user_id BIGINT PRIMARY KEY,
+    id BIGINT PRIMARY KEY AUTO_INCREMENT,
+    user_id BIGINT NOT NULL,
     balance BIGINT NOT NULL DEFAULT 0,
-    last_updated TIMESTAMPTZ NOT NULL DEFAULT CURRENT_TIMESTAMP
-)ENGINE=InnoDB;
+    last_updated DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP
+) ENGINE=InnoDB;
 
 CREATE TABLE user_transactions (
-        user_id BIGINT NOT NULL,
-        amount BIGINT NOT NULL,  -- Positive for deposit, negative for withdrawal
-        transaction_type VARCHAR(50) NOT NULL,  -- e.g., 'deposit', 'withdrawal', 'transfer'
-        description TEXT,
-        created_at DATETIME(6) NOT NULL DEFAULT CURRENT_TIMESTAMP(6),
+    id BIGINT PRIMARY KEY AUTO_INCREMENT,
+    user_id BIGINT NOT NULL,
+    amount BIGINT NOT NULL,
+    transaction_type VARCHAR(50) NOT NULL,
+    description TEXT,
+    transaction_id VARCHAR(50) NOT NULL,
+    created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP
 ) ENGINE=InnoDB;
 
 CREATE TABLE sms_status (
-        user_id BIGINT NOT NULL,
-        status VARCHAR(50) NOT NULL,
-        type  VARCHAR(50) NOT NULL,
-        recipient VARCHAR(20) NOT NULL,
-        Provider VARCHAR(50)  NOT NULL default '',
-        created_at DATETIME(6) NOT NULL DEFAULT CURRENT_TIMESTAMP(6),
-        updated_at DATETIME(6) NOT NULL DEFAULT CURRENT_TIMESTAMP(6)
-) ENGINE=InnoDB;
+    id BIGINT PRIMARY KEY AUTO_INCREMENT,
+    user_id BIGINT NOT NULL,
+    status VARCHAR(50) NOT NULL,
+    type VARCHAR(50) NOT NULL,
+    recipient VARCHAR(20) NOT NULL,
+    provider VARCHAR(50) NOT NULL DEFAULT '',
+    created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    updated_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP
+) ENGINE=InnoDB;

internal/balance/service.go

@@ -7,6 +7,8 @@ import (
 	"fmt"
 	"sms-gateway/app"
 	"sms-gateway/internal/model"
+
+	"github.com/google/uuid"
 )
 
 type transactionType string
@@ -46,7 +48,7 @@ type DeductBalanceRequest struct {
 func DeductBalance(ctx context.Context, req DeductBalanceRequest) (err error) {
 	price := calculatePrice(req.Type, req.Quantity)
 
-	tx, err := app.DB.DB.BeginTxx(ctx, nil)
+	tx, err := app.DB.BeginTxx(ctx, nil)
 	if err != nil {
 		return err
 	}
@@ -70,13 +72,15 @@ func DeductBalance(ctx context.Context, req DeductBalanceRequest) (err error) {
 		return errors.New("insufficient balance")
 	}
 
-	const insertTransactionQuery = `INSERT INTO user_transactions (user_id, amount, transaction_type, description) VALUES (?, ?, ?, ?)`
+	txID := uuid.NewString()
+	const insertTransactionQuery = `INSERT INTO user_transactions (user_id, amount, transaction_type, description, transaction_id) VALUES (?, ?, ?, ?, ?)`
 	if _, err = tx.ExecContext(ctx,
 		insertTransactionQuery,
 		req.CustomerID,
 		-price,
 		Withdrawal,
-		descriptionGenerator(req.Type, req.Quantity)); err != nil {
+		descriptionGenerator(req.Type, req.Quantity),
+		txID); err != nil {
 		return err
 	}
 
@@ -92,10 +96,11 @@ type UserTransaction struct {
 	Amount          int64           `db:"amount"`
 	TransactionType transactionType `db:"transaction_type"`
 	Description     string          `db:"description"`
+	TransactionID   string          `db:"transaction_id" json:"transaction_id"`
 }
 
 func GetUserTransactions(ctx context.Context, userID string) ([]UserTransaction, error) {
-	const query = `SELECT user_id, amount, transaction_type, description FROM user_transactions WHERE user_id = ?`
+	const query = `SELECT user_id, amount, transaction_type, description, transaction_id FROM user_transactions WHERE user_id = ?`
 
 	var transactions []UserTransaction
 	if err := app.DB.SelectContext(ctx, &transactions, query, userID); err != nil {
@@ -174,8 +179,9 @@ func AddBalance(ctx context.Context, req AddBalanceRequest) (err error) {
 		description = fmt.Sprintf("افزایش موجودی به میزان %d", req.Amount)
 	}
 
-	const insertTransactionQuery = `INSERT INTO user_transactions (user_id, amount, transaction_type, description) VALUES (?, ?, ?, ?)`
-	if _, err = tx.ExecContext(ctx, insertTransactionQuery, req.CustomerID, req.Amount, Deposit, description); err != nil {
+	txID := uuid.NewString()
+	const insertTransactionQuery = `INSERT INTO user_transactions (user_id, amount, transaction_type, description, transaction_id) VALUES (?, ?, ?, ?, ?)`
+	if _, err = tx.ExecContext(ctx, insertTransactionQuery, req.CustomerID, req.Amount, Deposit, description, txID); err != nil {
 		return err
 	}
 

internal/model/model.go

@@ -8,8 +8,9 @@ const (
 )
 
 type SMS struct {
-	CustomerID int64    `json:"customer_id"`
-	Text       string   `json:"text"`
-	Recipients []string `json:"recipients"`
-	Type       Type     `json:"type"`
+	CustomerID    int64    `json:"customer_id"`
+	Text          string   `json:"text"`
+	Recipients    []string `json:"recipients"`
+	Type          Type     `json:"type"`
+	TransactionID string   `json:"transaction_id"`
 }