persist_data: Add back code to retry in the case of locked database errors

Signed-off-by: Richard Purdie <richard.purdie@linuxfoundation.org>

diff --git a/lib/bb/persist_data.py b/lib/bb/persist_data.py
index 03adc7cc53dd99bbdd58f496c4724ebfe5f21717..551b58a2a90e60a70938bb5277e885a2c2574a07 100644 (file)
--- a/lib/bb/persist_data.py
+++ b/lib/bb/persist_data.py
@@ -55,7 +55,16 @@ class SQLTable(collections.MutableMapping):
                       % table)
 
     def _execute(self, *query):
-        return self.cursor.execute(*query)
+        """Execute a query, waiting to acquire a lock if necessary"""
+        count = 0
+        while True:
+            try:
+                return self.cursor.execute(*query)
+            except sqlite3.OperationalError as exc:
+                if 'database is locked' in str(exc) and count < 500:
+                    count = count + 1
+                    continue
+                raise
 
     def __enter__(self):
         self.cursor.__enter__()