109 lines
3.0 KiB
Go
109 lines
3.0 KiB
Go
// 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 <http://www.gnu.org/licenses/>.
|
|
|
|
package repositories
|
|
|
|
import (
|
|
"context"
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"database/sql"
|
|
|
|
"github.com/vulcanize/vulcanizedb/pkg/core"
|
|
"github.com/vulcanize/vulcanizedb/pkg/datastore/postgres"
|
|
)
|
|
|
|
type LogRepository struct {
|
|
*postgres.DB
|
|
}
|
|
|
|
func (logRepository LogRepository) CreateLogs(lgs []core.Log, receiptId int64) error {
|
|
tx, _ := logRepository.DB.BeginTx(context.Background(), nil)
|
|
for _, tlog := range lgs {
|
|
_, err := tx.Exec(
|
|
`INSERT INTO logs (block_number, address, tx_hash, index, topic0, topic1, topic2, topic3, data, receipt_id)
|
|
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)
|
|
`,
|
|
tlog.BlockNumber, tlog.Address, tlog.TxHash, tlog.Index, tlog.Topics[0], tlog.Topics[1], tlog.Topics[2], tlog.Topics[3], tlog.Data, receiptId,
|
|
)
|
|
if err != nil {
|
|
err = tx.Rollback()
|
|
if err != nil {
|
|
logrus.Error("CreateLogs: could not perform rollback: ", err)
|
|
}
|
|
return postgres.ErrDBInsertFailed
|
|
}
|
|
}
|
|
err := tx.Commit()
|
|
if err != nil {
|
|
err = tx.Rollback()
|
|
if err != nil {
|
|
logrus.Error("CreateLogs: could not perform rollback: ", err)
|
|
}
|
|
return postgres.ErrDBInsertFailed
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (logRepository LogRepository) GetLogs(address string, blockNumber int64) ([]core.Log, error) {
|
|
logRows, err := logRepository.DB.Query(
|
|
`SELECT block_number,
|
|
address,
|
|
tx_hash,
|
|
index,
|
|
topic0,
|
|
topic1,
|
|
topic2,
|
|
topic3,
|
|
data
|
|
FROM logs
|
|
WHERE address = $1 AND block_number = $2
|
|
ORDER BY block_number DESC`, address, blockNumber)
|
|
if err != nil {
|
|
return []core.Log{}, err
|
|
}
|
|
return logRepository.loadLogs(logRows)
|
|
}
|
|
|
|
func (logRepository LogRepository) loadLogs(logsRows *sql.Rows) ([]core.Log, error) {
|
|
var lgs []core.Log
|
|
for logsRows.Next() {
|
|
var blockNumber int64
|
|
var address string
|
|
var txHash string
|
|
var index int64
|
|
var data string
|
|
var topics core.Topics
|
|
err := logsRows.Scan(&blockNumber, &address, &txHash, &index, &topics[0], &topics[1], &topics[2], &topics[3], &data)
|
|
if err != nil {
|
|
logrus.Error("loadLogs: Error scanning a row in logRows: ", err)
|
|
return []core.Log{}, err
|
|
}
|
|
lg := core.Log{
|
|
BlockNumber: blockNumber,
|
|
TxHash: txHash,
|
|
Address: address,
|
|
Index: index,
|
|
Data: data,
|
|
}
|
|
for i, topic := range topics {
|
|
lg.Topics[i] = topic
|
|
}
|
|
lgs = append(lgs, lg)
|
|
}
|
|
return lgs, nil
|
|
}
|