ipld-eth-server/pkg/datastore/postgres/repositories/logs_repository.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
}