def __init__(self, generation, mgr_interval=5):
self.settings = Settings.getInstance()
self.debug = self.settings.debug
self.gen = generation
self.mgr_interval = mgr_interval
self.person_queue = Queue.Queue(maxsize=self.settings.person_cache_size)
self.person_id_set = set([]) # sync with queue, quick contains using id.
self.pubmap = {} # {id -> pub}
self.person_pub_map = {} # {person_id->[pub_id_list]} - person to pub_ids
self.pub_db_cache = {}
self.pub_lock = threading.Lock()
self.pub_dbcache_lock = threading.RLock()
self.running = True #sync ed with main running flag in mgr_interval_thread
self.blocked_pub_t = 0
# time sum
self.ppt_wait = 0
self.ppt_getlock = 0
self.ppt_get = 0
self.person_dao = PersonDao()
self.pub_dao = PublicationDao()
def __init__(self):
print "Task: extract paper's citation from schooler.google.com.\n"
self.settings = Settings.getInstance()
self.debug = self.settings.debug
# Configs
self.mgr_interval = 10 # seconds
self.max_person_thread = 2 # max threads used to extract person,
self.max_pub_thread = 2 # these 2 values can modified on the fly. diff in day or night
# Threads and configurations
self.t_mgr = None # MgrThread(self) # management thread, create
self.t_provider = None
self.person_thread_pool = [
] #= Queue.Queue(maxsize=self.max_person_thread)
self.pub_thread_pool = [] #= Queue.Queue(maxsize=self.max_pub_thread)
self.busy_semaphore = 0 # 用来监视是否所有的线程都处于Idle状态
self.busy_semaphore_lock = threading.Lock() # 用来监视是否所有的线程都处于Idle状态
# utils
self.store = None
# switchers & flags
self.running = True # If False, threads will stop after current task.
self.stopped = False # If MGRThread can stop.
self.pause = False # All works paused.
self.waiting_to_finish = False # No additional data. all added to queue.
self.num_report = 0
self.last_report_time = datetime.datetime.now() # 上次Interval的时间
self.restart_all_thread = False
self.detect_exit_wait = 0 # 当刚刚从pause模式退出来时，会有大量failed的任务，会导致立刻再次等待
self.generation = 0
self.dao = dbs()
self.personDao = PersonDao()
self.pubDao = PublicationDao()
if self.settings.save_pdflink:
self.pdfcache = PDFLinkSaver.getInstance()
# start
self.determineGereration()
