Comment gérer gracieusement les interruptions de clavier dans les pools multitraitements Python ?

Gestion gracieuse des interruptions de clavier dans les pools multitraitements Python

Lorsque vous travaillez avec le pool multitraitement de Python, la gestion des événements KeyboardInterrupt n'est pas toujours simple. Dans cet article, nous explorerons comment gérer de telles interruptions et garantir que les processus se terminent correctement.

L'exemple de code fourni illustre le défi. Malgré la présence d'un bloc catch pour KeyboardInterrupt, il ne s'exécute pas lorsque vous appuyez sur control-C. Au lieu de cela, le programme se bloque jusqu'à ce qu'il soit terminé en externe.

La racine de ce problème réside dans un bug Python lié au blocage d'une condition dans threading.Condition.wait(). Dans ce contexte, KeyboardInterrupt n’est jamais envoyé. En conséquence, cela empêche la gestion de l'interruption au sein du pool.

Une solution à ce problème consiste à spécifier un délai d'attente lors de l'attente des résultats. La méthode map_async() avec un paramètre timeout peut être utilisée à la place de map(). Cette approche permet à KeyboardInterrupt d'être reconnu et traité :

<code class="python">results = pool.map_async(slowly_square, range(40)).get(9999999)</code>

En définissant une valeur de délai d'attente élevée, nous disons essentiellement au Pool de continuer à attendre les résultats jusqu'à ce qu'il soit interrompu par l'utilisateur.

Il est important de noter que cette solution de contournement présente certaines limites. Si le délai d'attente expire avant que toutes les tâches ne soient terminées, les résultats incomplets seront ignorés. Par conséquent, il est crucial de choisir une valeur de délai d'attente appropriée qui équilibre la réactivité avec le risque de résultats incomplets.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!