Use transformer factory for Drip drip

This commit is contained in:
Rob Mulholand
2018-10-26 15:35:18 -05:00
parent f3a82e6ba1
commit 1b680d2a4b
11 changed files with 123 additions and 311 deletions
+13 -12
View File
@@ -15,34 +15,31 @@
package drip_drip
import (
"fmt"
"github.com/vulcanize/vulcanizedb/pkg/core"
"github.com/vulcanize/vulcanizedb/pkg/datastore/postgres"
)
type Repository interface {
Create(headerID int64, models []DripDripModel) error
MarkHeaderChecked(headerID int64) error
MissingHeaders(startingBlockNumber, endingBlockNumber int64) ([]core.Header, error)
}
type DripDripRepository struct {
db *postgres.DB
}
func NewDripDripRepository(db *postgres.DB) DripDripRepository {
return DripDripRepository{db: db}
}
func (repository DripDripRepository) Create(headerID int64, models []DripDripModel) error {
func (repository DripDripRepository) Create(headerID int64, models []interface{}) error {
tx, err := repository.db.Begin()
if err != nil {
return err
}
for _, model := range models {
dripDrip, ok := model.(DripDripModel)
if !ok {
tx.Rollback()
return fmt.Errorf("model of type %T, not %T", model, DripDripModel{})
}
_, err = tx.Exec(
`INSERT into maker.drip_drip (header_id, ilk, log_idx, tx_idx, raw_log)
VALUES($1, $2, $3, $4, $5)`,
headerID, model.Ilk, model.LogIndex, model.TransactionIndex, model.Raw,
headerID, dripDrip.Ilk, dripDrip.LogIndex, dripDrip.TransactionIndex, dripDrip.Raw,
)
if err != nil {
tx.Rollback()
@@ -84,3 +81,7 @@ func (repository DripDripRepository) MissingHeaders(startingBlockNumber, endingB
)
return result, err
}
func (repository *DripDripRepository) SetDB(db *postgres.DB) {
repository.db = db
}