From e262e53bc5e77f3bc9ac0d5ea43b1997743065cd Mon Sep 17 00:00:00 2001
From: Jude Nelson <judecn@gmail.com>
Date: Fri, 11 Sep 2015 12:27:13 -0400
Subject: [PATCH] Cache the keyring once generated, so we don't waste time
 generating it again

---
 blockstore/lib/nameset/namedb.py | 40 ++++++++++++++++++++++++++++++--
 1 file changed, 38 insertions(+), 2 deletions(-)

diff --git a/blockstore/lib/nameset/namedb.py b/blockstore/lib/nameset/namedb.py
index 7488f0cb9..29def90fa 100644
--- a/blockstore/lib/nameset/namedb.py
+++ b/blockstore/lib/nameset/namedb.py
@@ -28,6 +28,7 @@ import hashlib
 import math
 import keychain
 import pybitcoin
+import os
 
 from collections import defaultdict
 from ..config import NAMESPACE_DEFAULT, MIN_OP_LENGTHS, OPCODES, MAGIC_BYTES, TESTSET, MAX_NAMES_PER_SENDER, \
@@ -388,6 +389,27 @@ class BlockstoreDB( virtualchain.StateEngine ):
       Generate all possible NAME_IMPORT addresses from the NAMESPACE_REVEAL public key 
       """
       
+      pubkey_addr = pybitcoin.BitcoinPublicKey( str(pubkey_hex) ).address()
+      
+      # do we have a cached one on disk?
+      cached_keychain = os.path.join( virtualchain.get_working_dir(), "%s.keychain" % pubkey_addr)
+      if os.path.exists( cached_keychain ):
+          
+          child_addrs = []
+          try:
+              lines = []
+              with open(cached_keychain, "r") as f:
+                  lines = f.readlines()
+              
+              child_attrs = [l.strip() for l in lines]
+              
+              log.debug("Loaded cached import keychain for '%s' (%s)" % (pubkey_hex, pubkey_addr))
+              return child_attrs 
+          
+          except Exception, e:
+              log.exception(e)
+              pass 
+      
       pubkey_hex = str(pubkey_hex)
       public_keychain = keychain.PublicKeychain.from_public_key( pubkey_hex )
       child_addrs = []
@@ -400,11 +422,25 @@ class BlockstoreDB( virtualchain.StateEngine ):
           
           if i % 20 == 0 and i != 0:
               log.debug("%s children..." % i)
-          
+      
       # include this address 
-      pubkey_addr = pybitcoin.BitcoinPublicKey( pubkey_hex ).address()
       child_addrs.append( pubkey_addr )
       
+      log.debug("Done building import keychain for '%s' (%s)" % (pubkey_hex, pubkey_addr))
+      
+      # cache 
+      try:
+          with open(cached_keychain, "w+") as f:
+              for addr in child_addrs:
+                  f.write("%s\n" % addr)
+                  
+              f.flush()
+              
+          log.debug("Cached keychain to '%s'" % cached_keychain)
+      except Exception, e:
+          log.exception(e)
+          log.error("Unable to cache keychain for '%s' (%s)" % (pubkey_hex, pubkey_addr))
+      
       return child_addrs