Comment quitter une boucle de multitraitement Python?

Question

J'ai un code multi-traitement que la base de données de recherche enregistre pour une certaine condition, quand il atteint la fin de la base de données, la boucle doit s'arrêter, comment puis-je faire cela? Voici le code:

import psycopg2 
from multiprocessing import Pool 

conn = psycopg2.connect(a database) 
query=conn.cursor() 
query.execute(some query) 

def scanfile(): 
    try: row=query.fetchone() 
    except: return False #here the loop must stop 
    #do something with row... 

if __name__=='__main__': 
    pool=Pool() 
    while True: 
     pool.apply_async(scanfile) 
    pool.close() 
    pool.join() 

Answer 1

Déplacez votre requête/curseur comme itérable et laissez le multiprocessing distribuer le travail pour vous.

import psycopg2 
from multiprocessing import Pool 


def scanfile(row): 
    #do something with row... 


if __name__ == '__main__': 
    conn = psycopg2.connect(a database) 
    query = conn.cursor() 
    query.execute(some query) 
    pool = Pool() 
    pool.map_async(scanfile, query).wait() 
    pool.close() 
    pool.join()