aboutsummaryrefslogtreecommitdiffstats
path: root/python/sandcrawler/persist.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/sandcrawler/persist.py')
-rw-r--r--python/sandcrawler/persist.py17
1 files changed, 16 insertions, 1 deletions
diff --git a/python/sandcrawler/persist.py b/python/sandcrawler/persist.py
index 3b9cde9..ea54d6b 100644
--- a/python/sandcrawler/persist.py
+++ b/python/sandcrawler/persist.py
@@ -156,8 +156,23 @@ class PersistIngestFileResultWorker(SandcrawlerWorker):
if not batch:
return []
+ # need to ensure that we aren't trying to update same row multiple
+ # times in same batch (!)
results = [self.file_result_to_row(raw) for raw in batch]
- results = [r for r in results if r]
+ results.reverse()
+ clean_results = []
+ result_keys = []
+ for r in results:
+ if not r:
+ continue
+ key = (r['ingest_type'], r['base_url'])
+ if key in result_keys:
+ self.counts['skip-duplicate-result'] += 1
+ continue
+ result_keys.append(key)
+ clean_results.append(r)
+ results = clean_results
+
requests = [self.request_to_row(raw['request']) for raw in batch if raw.get('request')]
requests = [r for r in requests if r]