feat: check repetitive transaction (#31)

database/database.go

@@ -1,6 +1,8 @@
 package database
 
 import (
+	"errors"
+
 	"github.com/PACZone/wrapto/types/order"
 	"github.com/glebarez/sqlite"
 	"gorm.io/gorm"
@@ -97,6 +99,21 @@ func (db *DB) GetOrder(id string) (*Order, error) {
 	return ord, nil
 }
 
+func (db *DB) IsOrderExist(txHash string) (bool, error) {
+	var ord Order
+	err := db.Where("tx_hash = ?", txHash).First(&ord).Error
+	if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
+		return false, nil
+	} else if err != nil {
+		return false, DBError{
+			TableName: "Orders",
+			Reason:    err.Error(),
+		}
+	}
+
+	return true, nil
+}
+
 func (db *DB) GetOrderWithLogs(id string) (*Order, error) {
 	var ord *Order
 	if err := db.Preload("Logs").First(&ord, "id = ?", id).Error; err != nil {

database/database_test.go

@@ -8,6 +8,7 @@ import (
 	"github.com/PACZone/wrapto/types/order"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
+	"gorm.io/gorm"
 )
 
 func setup(t *testing.T) *database.DB {
@@ -132,3 +133,23 @@ func TestGetOrderLogs(t *testing.T) {
 	assert.Equal(t, l.Trace, "traceAbcd")
 	assert.Equal(t, l.OrderID, ordID)
 }
+
+func TestIsOrderExist(t *testing.T) {
+	db := setup(t)
+
+	ord, err := order.NewOrder("aaa", "sendet", "rec", 20e9)
+	assert.NoError(t, err)
+
+	ordID, err := db.AddOrder(ord)
+	assert.NoError(t, err)
+
+	isExist, err := db.IsOrderExist("bbb")
+	assert.Equal(t, isExist, false)
+	assert.Error(t, gorm.ErrRecordNotFound, err)
+
+	isExist, err = db.IsOrderExist("aaa")
+	assert.Equal(t, isExist, true)
+	assert.NoError(t, err)
+
+	assert.Equal(t, ord.ID, ordID)
+}

sides/pactus/listener.go

@@ -98,6 +98,16 @@ func (l *Listener) processBlocks() error {
 			continue
 		}
 
+		isExist, err := l.db.IsOrderExist(txHash)
+		if err != nil {
+			return err
+		}
+		if isExist {
+			logger.Warn("error repetitive transaction", "actor", l.bypassName, "txHash", txHash)
+
+			continue
+		}
+
 		ord, err := order.NewOrder(txHash, sender, destInfo.Addr, amt)
 		if err != nil {
 			logger.Error("error while making new order", "actor", l.bypassName, "err", err,

sides/polygon/listener.go

@@ -70,13 +70,24 @@ func (l *Listener) processOrder() error {
 		return nil
 	}
 
+	id := strconv.FormatUint(uint64(l.nextOrderNumber), 10)
+
+	isExist, err := l.db.IsOrderExist(id)
+	if err != nil {
+		return err
+	}
+	if isExist {
+		logger.Warn("error repetitive transaction", "actor", l.bypassName, "txHash", id)
+
+		return nil
+	}
+
 	l.nextOrderNumber++
 
 	logger.Info("processing new message on listener", "actor", l.bypassName, "orderNumber", l.nextOrderNumber)
 
 	amt, _ := o.Amount.Float64()
 	sender := o.Sender.Hex()
-	id := strconv.FormatUint(uint64(l.nextOrderNumber), 10)
 	ord, err := order.NewOrder(id, sender, o.DestinationAddress, amt)
 	if err != nil {
 		dbErr := l.db.UpdateOrderStatus(ord.ID, order.FAILED)