minor update

This commit is contained in:
Miroslav Stampar 2011-11-22 10:41:56 +00:00
parent b117c40aa5
commit 267d67b024
2 changed files with 6 additions and 0 deletions

View File

@ -14,6 +14,7 @@ import traceback
from thread import error as threadError
from lib.core.data import conf
from lib.core.data import kb
from lib.core.data import logger
from lib.core.datatype import AttribDict
@ -180,5 +181,7 @@ def runThreads(numThreads, threadFunction, cleanupFunction=None, forwardExceptio
kb.threadContinue = True
kb.threadException = False
conf.hashDB.flush()
if cleanupFunction:
cleanupFunction()

View File

@ -76,6 +76,9 @@ class HashDB(object):
self.flush()
def flush(self):
if not self._write_cache:
return
self._cache_lock.acquire()
items = self._write_cache.items()
self._write_cache.clear()