// Copyright 2018 Vulcanize // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. package ilk import ( "fmt" "github.com/vulcanize/vulcanizedb/pkg/core" "github.com/vulcanize/vulcanizedb/pkg/datastore/postgres" ) type PitFileIlkRepository struct { DB *postgres.DB } func (repository PitFileIlkRepository) Create(headerID int64, models []interface{}) error { tx, err := repository.DB.Begin() if err != nil { return err } for _, model := range models { pitFileIlk, ok := model.(PitFileIlkModel) if !ok { tx.Rollback() return fmt.Errorf("model of type %T, not %T", model, PitFileIlkModel{}) } _, err = tx.Exec( `INSERT into maker.pit_file_ilk (header_id, ilk, what, data, tx_idx, raw_log) VALUES($1, $2, $3, $4::NUMERIC, $5, $6)`, headerID, pitFileIlk.Ilk, pitFileIlk.What, pitFileIlk.Data, pitFileIlk.TransactionIndex, pitFileIlk.Raw, ) if err != nil { tx.Rollback() return err } } _, err = tx.Exec(`INSERT INTO public.checked_headers (header_id, pit_file_ilk_checked) VALUES ($1, $2) ON CONFLICT (header_id) DO UPDATE SET pit_file_ilk_checked = $2`, headerID, true) if err != nil { tx.Rollback() return err } return tx.Commit() } func (repository PitFileIlkRepository) MarkHeaderChecked(headerID int64) error { _, err := repository.DB.Exec(`INSERT INTO public.checked_headers (header_id, pit_file_ilk_checked) VALUES ($1, $2) ON CONFLICT (header_id) DO UPDATE SET pit_file_ilk_checked = $2`, headerID, true) return err } func (repository PitFileIlkRepository) MissingHeaders(startingBlockNumber, endingBlockNumber int64) ([]core.Header, error) { var result []core.Header err := repository.DB.Select( &result, `SELECT headers.id, headers.block_number FROM headers LEFT JOIN checked_headers on headers.id = header_id WHERE (header_id ISNULL OR pit_file_ilk_checked IS FALSE) AND headers.block_number >= $1 AND headers.block_number <= $2 AND headers.eth_node_fingerprint = $3`, startingBlockNumber, endingBlockNumber, repository.DB.Node.ID, ) return result, err } func (repository *PitFileIlkRepository) SetDB(db *postgres.DB) { repository.DB = db }