updates for light sync transactions

libraries/shared/transactions/syncer.go

@@ -0,0 +1,55 @@
+package transactions
+
+import (
+	"github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/core/types"
+	"github.com/vulcanize/vulcanizedb/pkg/core"
+	"github.com/vulcanize/vulcanizedb/pkg/datastore"
+	"github.com/vulcanize/vulcanizedb/pkg/datastore/postgres"
+	"github.com/vulcanize/vulcanizedb/pkg/datastore/postgres/repositories"
+)
+
+type ITransactionsSyncer interface {
+	SyncTransactions(headerID int64, logs []types.Log) error
+}
+
+type TransactionsSyncer struct {
+	BlockChain core.BlockChain
+	Repository datastore.HeaderRepository
+}
+
+func NewTransactionsSyncer(db *postgres.DB, blockChain core.BlockChain) TransactionsSyncer {
+	repository := repositories.NewHeaderRepository(db)
+	return TransactionsSyncer{
+		BlockChain: blockChain,
+		Repository: repository,
+	}
+}
+
+func (syncer TransactionsSyncer) SyncTransactions(headerID int64, logs []types.Log) error {
+	transactionHashes := getUniqueTransactionHashes(logs)
+	transactions, transactionErr := syncer.BlockChain.GetTransactions(transactionHashes)
+	if transactionErr != nil {
+		return transactionErr
+	}
+	for _, transaction := range transactions {
+		writeErr := syncer.Repository.CreateTransaction(headerID, transaction)
+		if writeErr != nil {
+			return writeErr
+		}
+	}
+	return nil
+}
+
+func getUniqueTransactionHashes(logs []types.Log) []common.Hash {
+	seen := make(map[common.Hash]struct{}, len(logs))
+	var result []common.Hash
+	for _, log := range logs {
+		if _, ok := seen[log.TxHash]; ok {
+			continue
+		}
+		seen[log.TxHash] = struct{}{}
+		result = append(result, log.TxHash)
+	}
+	return result
+}

libraries/shared/transactions/syncer_test.go

@@ -0,0 +1,80 @@
+package transactions_test
+
+import (
+	"github.com/ethereum/go-ethereum/core/types"
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+	"github.com/vulcanize/vulcanizedb/libraries/shared/transactions"
+	"github.com/vulcanize/vulcanizedb/pkg/core"
+	"github.com/vulcanize/vulcanizedb/pkg/fakes"
+	"github.com/vulcanize/vulcanizedb/test_config"
+)
+
+var _ = Describe("Transaction syncer", func() {
+	It("fetches transactions for logs", func() {
+		db := test_config.NewTestDB(test_config.NewTestNode())
+		blockChain := fakes.NewMockBlockChain()
+		syncer := transactions.NewTransactionsSyncer(db, blockChain)
+
+		err := syncer.SyncTransactions(0, []types.Log{})
+
+		Expect(err).NotTo(HaveOccurred())
+		Expect(blockChain.GetTransactionsCalled).To(BeTrue())
+	})
+
+	It("only fetches transactions with unique hashes", func() {
+		db := test_config.NewTestDB(test_config.NewTestNode())
+		blockChain := fakes.NewMockBlockChain()
+		syncer := transactions.NewTransactionsSyncer(db, blockChain)
+
+		err := syncer.SyncTransactions(0, []types.Log{{
+			TxHash: fakes.FakeHash,
+		}, {
+			TxHash: fakes.FakeHash,
+		}})
+
+		Expect(err).NotTo(HaveOccurred())
+		Expect(len(blockChain.GetTransactionsPassedHashes)).To(Equal(1))
+	})
+
+	It("returns error if fetching transactions fails", func() {
+		db := test_config.NewTestDB(test_config.NewTestNode())
+		blockChain := fakes.NewMockBlockChain()
+		blockChain.GetTransactionsError = fakes.FakeError
+		syncer := transactions.NewTransactionsSyncer(db, blockChain)
+
+		err := syncer.SyncTransactions(0, []types.Log{})
+
+		Expect(err).To(HaveOccurred())
+		Expect(err).To(MatchError(fakes.FakeError))
+	})
+
+	It("passes transactions to repository for persistence", func() {
+		db := test_config.NewTestDB(test_config.NewTestNode())
+		blockChain := fakes.NewMockBlockChain()
+		blockChain.Transactions = []core.TransactionModel{{}}
+		syncer := transactions.NewTransactionsSyncer(db, blockChain)
+		mockHeaderRepository := fakes.NewMockHeaderRepository()
+		syncer.Repository = mockHeaderRepository
+
+		err := syncer.SyncTransactions(0, []types.Log{})
+
+		Expect(err).NotTo(HaveOccurred())
+		Expect(mockHeaderRepository.CreateTransactionCalled).To(BeTrue())
+	})
+
+	It("returns error if persisting transactions fails", func() {
+		db := test_config.NewTestDB(test_config.NewTestNode())
+		blockChain := fakes.NewMockBlockChain()
+		blockChain.Transactions = []core.TransactionModel{{}}
+		syncer := transactions.NewTransactionsSyncer(db, blockChain)
+		mockHeaderRepository := fakes.NewMockHeaderRepository()
+		mockHeaderRepository.CreateTransactionError = fakes.FakeError
+		syncer.Repository = mockHeaderRepository
+
+		err := syncer.SyncTransactions(0, []types.Log{})
+
+		Expect(err).To(HaveOccurred())
+		Expect(err).To(MatchError(fakes.FakeError))
+	})
+})

libraries/shared/transactions/transactions_suite_test.go

@@ -0,0 +1,13 @@
+package transactions_test
+
+import (
+	"testing"
+
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+)
+
+func TestTransactions(t *testing.T) {
+	RegisterFailHandler(Fail)
+	RunSpecs(t, "Shared Transactions Suite")
+}