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

View File

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