// VulcanizeDB // Copyright © 2018 Vulcanize // This program is free software: you can redistribute it and/or modify // it under the terms of the GNU Affero General Public License as published by // the Free Software Foundation, either version 3 of the License, or // (at your option) any later version. // This program is distributed in the hope that it will be useful, // but WITHOUT ANY WARRANTY; without even the implied warranty of // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the // GNU Affero General Public License for more details. // You should have received a copy of the GNU Affero General Public License // along with this program. If not, see . package flip import ( "fmt" log "github.com/sirupsen/logrus" "github.com/vulcanize/vulcanizedb/pkg/core" "github.com/vulcanize/vulcanizedb/pkg/datastore/postgres" "github.com/vulcanize/vulcanizedb/pkg/transformers/shared" "github.com/vulcanize/vulcanizedb/pkg/transformers/shared/constants" ) type CatFileFlipRepository struct { db *postgres.DB } func (repository CatFileFlipRepository) Create(headerID int64, models []interface{}) error { tx, dBaseErr := repository.db.Begin() if dBaseErr != nil { return dBaseErr } for _, model := range models { flip, ok := model.(CatFileFlipModel) if !ok { rollbackErr := tx.Rollback() if rollbackErr != nil { log.Error("failed to rollback ", rollbackErr) } return fmt.Errorf("model of type %T, not %T", model, CatFileFlipModel{}) } _, execErr := repository.db.Exec( `INSERT into maker.cat_file_flip (header_id, ilk, what, flip, tx_idx, log_idx, raw_log) VALUES($1, $2, $3, $4, $5, $6, $7) ON CONFLICT (header_id, tx_idx, log_idx) DO UPDATE SET ilk = $2, what = $3, flip = $4, raw_log = $7;`, headerID, flip.Ilk, flip.What, flip.Flip, flip.TransactionIndex, flip.LogIndex, flip.Raw, ) if execErr != nil { rollbackErr := tx.Rollback() if rollbackErr != nil { log.Error("failed to rollback ", rollbackErr) } return execErr } } checkHeaderErr := shared.MarkHeaderCheckedInTransaction(headerID, tx, constants.CatFileFlipChecked) if checkHeaderErr != nil { rollbackErr := tx.Rollback() if rollbackErr != nil { log.Error("failed to rollback ", rollbackErr) } return checkHeaderErr } return tx.Commit() } func (repository CatFileFlipRepository) MarkHeaderChecked(headerID int64) error { return shared.MarkHeaderChecked(headerID, repository.db, constants.CatFileFlipChecked) } func (repository CatFileFlipRepository) MissingHeaders(startingBlockNumber, endingBlockNumber int64) ([]core.Header, error) { return shared.MissingHeaders(startingBlockNumber, endingBlockNumber, repository.db, constants.CatFileFlipChecked) } func (repository CatFileFlipRepository) RecheckHeaders(startingBlockNumber, endingBlockNumber int64) ([]core.Header, error) { return shared.RecheckHeaders(startingBlockNumber, endingBlockNumber, repository.db, constants.CatFileFlipChecked) } func (repository *CatFileFlipRepository) SetDB(db *postgres.DB) { repository.db = db }