index receipts by the contract address

pkg/config/subscription.go

@@ -44,6 +44,7 @@ type TrxFilter struct {
 
 type ReceiptFilter struct {
 	Off     bool
+	Contracts []string
 	Topic0s []string
 }
 

pkg/ipfs/converter.go

@@ -86,7 +86,8 @@ func (pc *Converter) Convert(payload statediff.Payload) (*IPLDPayload, error) {
 		}
 		// Extract topic0 data from the receipt's logs for indexing
 		rctMeta := &ReceiptMetaData{
-			Topic0s: make([]string, 0, len(gethReceipt.Logs)),
+			Topic0s:         make([]string, 0, len(gethReceipt.Logs)),
+			ContractAddress: gethReceipt.ContractAddress.Hex(),
 		}
 		for _, log := range gethReceipt.Logs {
 			if len(log.Topics[0]) < 1 {

pkg/ipfs/publisher.go

@@ -170,6 +170,7 @@ func (pub *Publisher) publishReceipts(receipts types.Receipts, receiptMeta []*Re
 	if len(receiptsCids) != len(receipts) {
 		return nil, errors.New("expected one CID for each receipt")
 	}
+	// Keep receipts associated with their transaction
 	mappedRctCids := make(map[common.Hash]*ReceiptMetaData, len(receiptsCids))
 	for i, rct := range receipts {
 		mappedRctCids[rct.TxHash] = receiptMeta[i]

pkg/ipfs/repository.go

@@ -106,8 +106,8 @@ func (repo *Repository) indexTransactionAndReceiptCIDs(tx *sqlx.Tx, payload *CID
 }
 
 func (repo *Repository) indexReceiptCID(tx *sqlx.Tx, cidMeta *ReceiptMetaData, txID int64) error {
-	_, err := tx.Exec(`INSERT INTO public.receipt_cids (tx_id, cid, topic0s) VALUES ($1, $2, $3)`,
-		txID, cidMeta.CID, pq.Array(cidMeta.Topic0s))
+	_, err := tx.Exec(`INSERT INTO public.receipt_cids (tx_id, cid, contract, topic0s) VALUES ($1, $2, $3, $4)`,
+		txID, cidMeta.CID, cidMeta.ContractAddress, pq.Array(cidMeta.Topic0s))
 	return err
 }
 

pkg/ipfs/retreiver.go

@@ -198,11 +198,16 @@ func (ecr *EthCIDRetriever) retrieveRctCIDs(tx *sqlx.Tx, streamFilters config.Su
 			AND header_cids.block_number = $1`
 	args = append(args, blockNumber)
 	if len(streamFilters.ReceiptFilter.Topic0s) > 0 {
-		pgStr += ` AND (receipt_cids.topic0s && $2::VARCHAR(66)[]`
+		pgStr += ` AND ((receipt_cids.topic0s && $2::VARCHAR(66)[]`
 		args = append(args, pq.Array(streamFilters.ReceiptFilter.Topic0s))
 	}
+	if len(streamFilters.ReceiptFilter.Contracts) > 0 {
+		pgStr += ` AND receipt_cids.contract = ANY($3::VARCHAR(66)[])`
+	} else {
+		pgStr += `)`
+	}
 	if len(trxIds) > 0 {
-		pgStr += ` OR receipt_cids.tx_id = ANY($3::INTEGER[]))`
+		pgStr += ` OR receipt_cids.tx_id = ANY($4::INTEGER[]))`
 		args = append(args, pq.Array(trxIds))
 	} else {
 		pgStr += `)`

pkg/ipfs/screener.go

@@ -128,7 +128,7 @@ func checkTransactions(wantedSrc, wantedDst []string, actualSrc, actualDst strin
 func (s *Screener) filerReceipts(streamFilters config.Subscription, response *ResponsePayload, payload IPLDPayload, trxHashes []common.Hash) error {
 	if !streamFilters.ReceiptFilter.Off && checkRange(streamFilters.StartingBlock.Int64(), streamFilters.EndingBlock.Int64(), payload.BlockNumber.Int64()) {
 		for i, receipt := range payload.Receipts {
-			if checkReceipts(receipt, streamFilters.ReceiptFilter.Topic0s, payload.ReceiptMetaData[i].Topic0s, trxHashes) {
+			if checkReceipts(receipt, streamFilters.ReceiptFilter.Topic0s, payload.ReceiptMetaData[i].Topic0s, streamFilters.ReceiptFilter.Contracts, payload.ReceiptMetaData[i].ContractAddress, trxHashes) {
 				receiptForStorage := (*types.ReceiptForStorage)(receipt)
 				receiptBuffer := new(bytes.Buffer)
 				err := receiptForStorage.EncodeRLP(receiptBuffer)
@@ -142,23 +142,42 @@ func (s *Screener) filerReceipts(streamFilters config.Subscription, response *Re
 	return nil
 }
 
-func checkReceipts(rct *types.Receipt, wantedTopics, actualTopics []string, wantedTrxHashes []common.Hash) bool {
-	// If we aren't filtering for any topics, all topics are a go
-	if len(wantedTopics) == 0 {
+func checkReceipts(rct *types.Receipt, wantedTopics, actualTopics, wantedContracts []string, actualContract string, wantedTrxHashes []common.Hash) bool {
+	// If we aren't filtering for any topics or contracts, all topics are a go
+	if len(wantedTopics) == 0 && len(wantedContracts) == 0 {
 		return true
 	}
+	// No matter what filters we have, we keep receipts for the trxs we are interested in
 	for _, wantedTrxHash := range wantedTrxHashes {
 		if bytes.Equal(wantedTrxHash.Bytes(), rct.TxHash.Bytes()) {
 			return true
 		}
 	}
-	for _, wantedTopic := range wantedTopics {
-		for _, actualTopic := range actualTopics {
-			if wantedTopic == actualTopic {
-				return true
+
+	if len(wantedContracts) == 0 {
+		// We keep all receipts that have logs we are interested in
+		for _, wantedTopic := range wantedTopics {
+			for _, actualTopic := range actualTopics {
+				if wantedTopic == actualTopic {
+					return true
+				}
+			}
+		}
+	} else {
+		// We only keep receipts with logs of interest if we are interested in that contract
+		for _, wantedTopic := range wantedTopics {
+			for _, actualTopic := range actualTopics {
+				if wantedTopic == actualTopic {
+					for _, wantedContract := range wantedContracts {
+						if wantedContract == actualContract {
+							return true
+						}
+					}
+				}
 			}
 		}
 	}
+
 	return false
 }
 

pkg/ipfs/types.go

@@ -141,8 +141,9 @@ type StorageNodeCID struct {
 
 // ReceiptMetaData wraps some additional data around our receipt CIDs for indexing
 type ReceiptMetaData struct {
-	CID     string
-	Topic0s []string
+	CID             string
+	Topic0s         []string
+	ContractAddress string
 }
 
 // TrxMetaData wraps some additional data around our transaction CID for indexing