diff --git a/libraries/shared/watcher.go b/libraries/shared/watcher.go index 70277cfd..3b1cff20 100644 --- a/libraries/shared/watcher.go +++ b/libraries/shared/watcher.go @@ -71,9 +71,7 @@ func (watcher *Watcher) Execute() error { chunkedLogs := watcher.Chunker.ChunkLogs(logs) for _, transformer := range watcher.Transformers { - // TODO delegate log chunks to respective transformers - // Need to get the transformer name... :/ - logChunk := chunkedLogs["transformerName"] + logChunk := chunkedLogs[transformer.Name()] err = transformer.Execute(logChunk, header) } } diff --git a/pkg/transformers/factories/log_note_transformer.go b/pkg/transformers/factories/log_note_transformer.go index 663b9dc0..849d9619 100644 --- a/pkg/transformers/factories/log_note_transformer.go +++ b/pkg/transformers/factories/log_note_transformer.go @@ -59,3 +59,7 @@ func (transformer LogNoteTransformer) Execute(logs []types.Log, header core.Head } return nil } + +func (transformer LogNoteTransformer) Name() string { + return transformer.Config.TransformerName +} diff --git a/pkg/transformers/factories/transformer.go b/pkg/transformers/factories/transformer.go index 28ed3953..eae4fd7f 100644 --- a/pkg/transformers/factories/transformer.go +++ b/pkg/transformers/factories/transformer.go @@ -66,3 +66,7 @@ func (transformer Transformer) Execute(logs []types.Log, header core.Header) err return nil } + +func (transformer Transformer) Name() string { + return transformer.Config.TransformerName +} diff --git a/pkg/transformers/shared/repository.go b/pkg/transformers/shared/repository.go index a1d5c8db..0380d50a 100644 --- a/pkg/transformers/shared/repository.go +++ b/pkg/transformers/shared/repository.go @@ -26,7 +26,6 @@ func MarkHeaderCheckedInTransaction(headerID int64, tx *sql.Tx, checkedHeadersCo } // Treats a header as missing if it's not in the headers table, or not checked for some log type -// TODO Revisit definition of "checked header func MissingHeaders(startingBlockNumber, endingBlockNumber int64, db *postgres.DB, notCheckedSQL string) ([]core.Header, error) { var result []core.Header var query string diff --git a/pkg/transformers/shared/transformer.go b/pkg/transformers/shared/transformer.go index 8b5c8c49..a5ff5835 100644 --- a/pkg/transformers/shared/transformer.go +++ b/pkg/transformers/shared/transformer.go @@ -24,6 +24,7 @@ import ( type Transformer interface { Execute(logs []types.Log, header core.Header) error + Name() string } type TransformerInitializer func(db *postgres.DB) Transformer