Fix getting not enrich blocks (#165)

2023-03-16 19:42:02 +04:00
parent f7451c54eb
commit ef6ff919bd
2 changed files with 20 additions and 21 deletions
--- a/src/main/kotlin/io/emeraldpay/dshackle/cache/BlocksMemCache.kt
+++ b/src/main/kotlin/io/emeraldpay/dshackle/cache/BlocksMemCache.kt
@@ -35,13 +35,12 @@ open class BlocksMemCache(
    }

    open fun get(key: BlockId): BlockContainer? {
-        return mapping.getIfPresent(key)
+        return mapping.getIfPresent(key)?.let {
+            return@let if (it.enriched) it else null
+        }
    }

    open fun add(block: BlockContainer) {
-//        if (!block.enriched) {
-//            return
-//        }
        mapping.put(block.hash, block)
    }