fix cache of sign (#183)

diff --git a/database/store.go b/database/store.go
index fa76fc0..e5e4134 100644 (file)
--- a/database/store.go
+++ b/database/store.go
@@ -293,9 +293,12 @@ func (s *Store) SaveBlockHeader(blockHeader *types.BlockHeader) error {
        }
 
        blockHash := blockHeader.Hash()
-       batch := s.db.NewBatch()
-       batch.Set(calcBlockHeaderKey(blockHeader.Height, &blockHash), binaryBlockHeader)
-       batch.Write()
+       s.db.Set(calcBlockHeaderKey(blockHeader.Height, &blockHash), binaryBlockHeader)
+
+       // updata blockheader cache
+       if _, ok := s.cache.getBlockHeader(&blockHash); ok {
+               s.cache.addBlockHeader(blockHeader)
+       }
 
        log.WithFields(log.Fields{
                "module":   logModule,