db.go 1.24 KB
Newer Older
Hamdi Allam's avatar
Hamdi Allam committed
1
// Database module defines the data DB struct which wraps specific DB interfaces for L1/L2 block headers, contract events, bridging schemas.
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
package database

import (
	"gorm.io/driver/postgres"
	"gorm.io/gorm"
)

type DB struct {
	gorm *gorm.DB

	Blocks         BlocksDB
	ContractEvents ContractEventsDB
	Bridge         BridgeDB
}

func NewDB(dsn string) (*DB, error) {
	gorm, err := gorm.Open(postgres.Open(dsn), &gorm.Config{
Hamdi Allam's avatar
Hamdi Allam committed
19
		// The indexer will explicitly manage the transaction
20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
		// flow processing blocks
		SkipDefaultTransaction: true,
	})

	if err != nil {
		return nil, err
	}

	db := &DB{
		gorm:           gorm,
		Blocks:         newBlocksDB(gorm),
		ContractEvents: newContractEventsDB(gorm),
		Bridge:         newBridgeDB(gorm),
	}

	return db, nil
}
37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53

// Transaction executes all operations conducted with the supplied database in a single
// transaction. If the supplied function errors, the transaction is rolled back.
func (db *DB) Transaction(fn func(db *DB) error) error {
	return db.gorm.Transaction(func(tx *gorm.DB) error {
		return fn(dbFromGormTx(tx))
	})
}

func dbFromGormTx(tx *gorm.DB) *DB {
	return &DB{
		gorm:           tx,
		Blocks:         newBlocksDB(tx),
		ContractEvents: newContractEventsDB(tx),
		Bridge:         newBridgeDB(tx),
	}
}