do not queue requests that can be answered using the cache
[electrum-server.git] / backends / abe / __init__.py
index 91435fd..78271d0 100644 (file)
@@ -67,12 +67,16 @@ class AbeStore(Datastore_class):
         self.tx_cache = {}
         self.bitcoind_url = 'http://%s:%s@%s:%s/' % ( config.get('bitcoind','user'), config.get('bitcoind','password'), config.get('bitcoind','host'), config.get('bitcoind','port'))
 
+        self.chunk_cache = {}
+
         self.address_queue = Queue()
 
-        self.dblock = thread.allocate_lock()
+        self.lock = threading.Lock()        # for the database
+        self.cache_lock = threading.Lock()  # for the cache
         self.last_tx_id = 0
         self.known_mempool_hashes = []
 
+
     
     def import_tx(self, tx, is_coinbase):
         tx_id = super(AbeStore, self).import_tx(tx, is_coinbase)
@@ -106,9 +110,11 @@ class AbeStore(Datastore_class):
                 continue
 
             address = hash_to_address(chr(self.addrtype), _hash)
-            if self.tx_cache.has_key(address):
-                print "cache: invalidating", address
-                self.tx_cache.pop(address)
+            with self.cache_lock:
+                if self.tx_cache.has_key(address):
+                    print "cache: invalidating", address
+                    self.tx_cache.pop(address)
+
             self.address_queue.put(address)
 
         outrows = self.get_tx_outputs(txid, False)
@@ -119,22 +125,24 @@ class AbeStore(Datastore_class):
                 continue
 
             address = hash_to_address(chr(self.addrtype), _hash)
-            if self.tx_cache.has_key(address):
-                print "cache: invalidating", address
-                self.tx_cache.pop(address)
+            with self.cache_lock:
+                if self.tx_cache.has_key(address):
+                    print "cache: invalidating", address
+                    self.tx_cache.pop(address)
+
             self.address_queue.put(address)
 
     def safe_sql(self,sql, params=(), lock=True):
 
         error = False
         try:
-            if lock: self.dblock.acquire()
+            if lock: self.lock.acquire()
             ret = self.selectall(sql,params)
         except:
             error = True
             traceback.print_exc(file=sys.stdout)
         finally:
-            if lock: self.dblock.release()
+            if lock: self.lock.release()
 
         if error: 
             raise BaseException('sql error')
@@ -266,11 +274,15 @@ class AbeStore(Datastore_class):
             raise BaseException('limit reached')
         return out
 
-    def get_history(self, addr):
 
-        cached_version = self.tx_cache.get( addr )
-        if cached_version is not None:
-            return cached_version
+
+    def get_history(self, addr, cache_only=False):
+        with self.cache_lock:
+            cached_version = self.tx_cache.get( addr )
+            if cached_version is not None:
+                return cached_version
+
+        if cache_only: return -1
 
         version, binaddr = decode_check_address(addr)
         if binaddr is None:
@@ -385,14 +397,27 @@ class AbeStore(Datastore_class):
         # cache result
         # do not cache mempool results because statuses are ambiguous
         if not address_has_mempool:
-            self.tx_cache[addr] = txpoints
+            with self.cache_lock:
+                self.tx_cache[addr] = txpoints
         
         return txpoints
 
+    def get_history2(self, addr, cache_only=False):
+        h = self.get_history(addr, cache_only)
+        if cache_only and h==-1: return -1
+
+        out = map(lambda x: {'tx_hash':x['tx_hash'], 'height':x['height']}, h)
+        out2 = []
+        for item in out:
+            if item not in out2: out2.append(item)
+        return out2
 
-    def get_status(self,addr):
+
+    def get_status(self, addr, cache_only=False):
         # get address status, i.e. the last block for that address.
-        tx_points = self.get_history(addr)
+        tx_points = self.get_history(addr, cache_only)
+        if cache_only and tx_points == -1: return -1
+
         if not tx_points:
             status = None
         else:
@@ -403,6 +428,17 @@ class AbeStore(Datastore_class):
                 status = status + ':%d'% len(tx_points)
         return status
 
+    def get_status2(self, addr, cache_only=False):
+        # for 0.5 clients
+        tx_points = self.get_history2(addr)
+        if cache_only and tx_points == -1: return -1
+
+        if not tx_points: return None
+        status = ''
+        for tx in tx_points:
+            status += tx.get('tx_hash') + ':%d:' % tx.get('height')
+        return hashlib.sha256( status ).digest().encode('hex')
+
 
     def get_block_header(self, block_height):
         out = self.safe_sql("""
@@ -430,6 +466,10 @@ class AbeStore(Datastore_class):
         
 
     def get_chunk(self, index):
+        with self.cache_lock:
+            msg = self.chunk_cache.get(index)
+            if msg: return msg
+
         sql = """
             SELECT
                 block_hash,
@@ -458,11 +498,24 @@ class AbeStore(Datastore_class):
             #print "hash", encode(Hash(msg.decode('hex')))
             #if h.get('block_height')==1:break
 
+        with self.cache_lock:
+            self.chunk_cache[index] = msg
         print "get_chunk", index, len(msg)
         return msg
 
 
 
+    def get_raw_tx(self, tx_hash, height):
+        postdata = dumps({"method": 'getrawtransaction', 'params': [tx_hash, 0, height], 'id':'jsonrpc'})
+        respdata = urllib.urlopen(self.bitcoind_url, postdata).read()
+        r = loads(respdata)
+        if r['error'] != None:
+            raise BaseException(r['error'])
+
+        hextx = r.get('result')
+        return hextx
+
+
     def get_tx_merkle(self, tx_hash):
 
         out = self.safe_sql("""
@@ -571,13 +624,15 @@ class AbeStore(Datastore_class):
         return out
 
 
-    def main_iteration(store):
-        with store.dblock:
-            store.catch_up()
-            store.memorypool_update()
-            height = store.get_block_number( store.chain_id )
+    def main_iteration(self):
+        with self.lock:
+            self.catch_up()
+            self.memorypool_update()
+            height = self.get_block_number( self.chain_id )
+            try: self.chunk_cache.pop(height/2016) 
+            except: pass
 
-        block_header = store.get_block_header( height )
+        block_header = self.get_block_header( height )
         return block_header
 
 
@@ -612,7 +667,14 @@ class BlockchainProcessor(Processor):
 
         threading.Timer(10, self.run_store_iteration).start()
 
-    def process(self, request):
+
+    def add_request(self, request):
+        # see if we can get if from cache. if not, add to queue
+        if self.process( request, cache_only=True) == -1:
+            self.queue.put(request)
+
+
+    def process(self, request, cache_only = False):
         #print "abe process", request
 
         message_id = request['id']
@@ -630,45 +692,80 @@ class BlockchainProcessor(Processor):
         elif method == 'blockchain.address.subscribe':
             try:
                 address = params[0]
-                result = self.store.get_status(address)
+                result = self.store.get_status(address, cache_only)
                 self.watch_address(address)
             except BaseException, e:
                 error = str(e) + ': ' + address
                 print "error:", error
 
-        elif method == 'blockchain.address.get_history':
+        elif method == 'blockchain.address.subscribe2':
             try:
                 address = params[0]
-                result = self.store.get_history( address ) 
+                result = self.store.get_status2(address, cache_only)
+                self.watch_address(address)
             except BaseException, e:
                 error = str(e) + ': ' + address
                 print "error:", error
 
-        elif method == 'blockchain.block.get_header':
+        elif method == 'blockchain.address.get_history':
             try:
-                height = params[0]
-                result = self.store.get_block_header( height ) 
+                address = params[0]
+                result = self.store.get_history( address, cache_only )
             except BaseException, e:
-                error = str(e) + ': %d'% height
+                error = str(e) + ': ' + address
                 print "error:", error
 
-        elif method == 'blockchain.block.get_chunk':
+        elif method == 'blockchain.address.get_history2':
             try:
-                index = params[0]
-                result = self.store.get_chunk( index ) 
+                address = params[0]
+                result = self.store.get_history2( address, cache_only )
             except BaseException, e:
-                error = str(e) + ': %d'% index
+                error = str(e) + ': ' + address
                 print "error:", error
 
+        elif method == 'blockchain.block.get_header':
+            if cache_only: 
+                result = -1
+            else:
+                try:
+                    height = params[0]
+                    result = self.store.get_block_header( height ) 
+                except BaseException, e:
+                    error = str(e) + ': %d'% height
+                    print "error:", error
+                    
+        elif method == 'blockchain.block.get_chunk':
+            if cache_only:
+                result = -1
+            else:
+                try:
+                    index = params[0]
+                    result = self.store.get_chunk( index ) 
+                except BaseException, e:
+                    error = str(e) + ': %d'% index
+                    print "error:", error
+                    
         elif method == 'blockchain.transaction.broadcast':
             txo = self.store.send_tx(params[0])
             print "sent tx:", txo
             result = txo 
 
         elif method == 'blockchain.transaction.get_merkle':
+            if cache_only:
+                result = -1
+            else:
+                try:
+                    tx_hash = params[0]
+                    result = self.store.get_tx_merkle(tx_hash ) 
+                except BaseException, e:
+                    error = str(e) + ': ' + tx_hash
+                    print "error:", error
+                    
+        elif method == 'blockchain.transaction.get':
             try:
                 tx_hash = params[0]
-                result = self.store.get_tx_merkle(tx_hash ) 
+                height = params[1]
+                result = self.store.get_raw_tx(tx_hash, height ) 
             except BaseException, e:
                 error = str(e) + ': ' + tx_hash
                 print "error:", error
@@ -676,6 +773,7 @@ class BlockchainProcessor(Processor):
         else:
             error = "unknown method:%s"%method
 
+        if cache_only and result == -1: return -1
 
         if error:
             response = { 'id':message_id, 'error':error }
@@ -722,7 +820,9 @@ class BlockchainProcessor(Processor):
                 break
             if addr in self.watched_addresses:
                 status = self.store.get_status( addr )
+                status2 = self.store.get_status2( addr )
                 self.push_response({ 'id': None, 'method':'blockchain.address.subscribe', 'params':[addr, status] })
+                self.push_response({ 'id': None, 'method':'blockchain.address.subscribe2', 'params':[addr, status2] })
 
         threading.Timer(10, self.run_store_iteration).start()