Python Language
Wielowątkowość

Wprowadzenie

Podstawy wielowątkowości

Za pomocą threading modułu nowym wątku wykonania może być uruchomiony przez utworzenie nowego threading.Thread i przypisując mu funkcję do wykonania:

import threading

def foo():
  print "Hello threading!"

my_thread = threading.Thread(target=foo)

Parametr target odwołuje się do funkcji (lub obiektu na żądanie), który ma zostać uruchomiony. Wątek nie rozpocznie wykonywania, dopóki nie zostanie wywołany start na obiekcie Thread .

Rozpoczęcie wątku

my_thread.start() # prints 'Hello threading!'

Teraz, gdy my_thread uruchomił się i zakończył, wywołanie start ponownie spowoduje wygenerowanie RuntimeError . Jeśli chcesz uruchomić swój wątek jako demon, przekazanie daemon=True kwarg lub ustawienie my_thread.daemon na True przed wywołaniem start() , powoduje, że Twój Thread działa w tle jako demon.

Dołączanie do wątku

W przypadkach, w których dzielisz jedno duże zadanie na kilka małych i chcesz je uruchomić jednocześnie, ale musisz poczekać, aż wszystkie z nich zakończą się, zanim przejdziesz dalej, Thread.join() jest metodą, której szukasz.

Załóżmy na przykład, że chcesz pobrać kilka stron witryny i skompilować je w jedną stronę. Zrobiłbyś to:

import requests
from threading import Thread
from queue import Queue

q = Queue(maxsize=20)
def put_page_to_q(page_num):
    q.put(requests.get('http://some-website.com/page_%s.html' % page_num)

def compile(q):
    # magic function that needs all pages before being able to be executed
    if not q.full():
        raise ValueError
    else:
        print("Done compiling!")

threads = []
for page_num in range(20):
     t = Thread(target=requests.get, args=(page_num,))
     t.start()
     threads.append(t)

# Next, join all threads to make sure all threads are done running before
# we continue. join() is a blocking call (unless specified otherwise using 
# the kwarg blocking=False when calling join)
for t in threads:
    t.join()

# Call compile() now, since all threads have completed
compile(q)

Bliższe spojrzenie na sposób działania join() można znaleźć tutaj .

Utwórz niestandardową klasę wątków

Za pomocą klasy threading.Thread możemy podklasować nową niestandardową klasę Thread. musimy zastąpić metodę run w podklasie.

from threading import Thread
import time

class Sleepy(Thread):

    def run(self):
        time.sleep(5)
        print("Hello form Thread")

if __name__ == "__main__":
    t = Sleepy()
    t.start()      # start method automatic call Thread class run method.
    # print 'The main program continues to run in foreground.'
    t.join()
    print("The main program continues to run in the foreground.")

Komunikacja między wątkami

W twoim kodzie jest wiele wątków i musisz bezpiecznie się między nimi komunikować.

Możesz użyć Queue z biblioteki queue .

from queue import Queue
from threading import Thread

# create a data producer 
def producer(output_queue):
    while True:
        data = data_computation()
        
        output_queue.put(data)

# create a consumer
def consumer(input_queue):
    while True:
        # retrieve data (blocking)
        data = input_queue.get()

        # do something with the data

        # indicate data has been consumed
        input_queue.task_done()

Tworzenie wątków producenta i konsumenta za pomocą wspólnej kolejki

q = Queue()
t1 = Thread(target=consumer, args=(q,))
t2 = Thread(target=producer, args=(q,))
t1.start()
t2.start()

Tworzenie puli pracowników

Używanie threading i queue :

from socket import socket, AF_INET, SOCK_STREAM
from threading import Thread
from queue import Queue
    
def echo_server(addr, nworkers):
    print('Echo server running at', addr)
    # Launch the client workers
    q = Queue()
    for n in range(nworkers):
        t = Thread(target=echo_client, args=(q,))
        t.daemon = True
        t.start()

    # Run the server
    sock = socket(AF_INET, SOCK_STREAM)
    sock.bind(addr)
    sock.listen(5)
    while True:
        client_sock, client_addr = sock.accept()
        q.put((client_sock, client_addr))

echo_server(('',15000), 128)

Korzystanie z concurrent.futures.Threadpoolexecutor :

from socket import AF_INET, SOCK_STREAM, socket
from concurrent.futures import ThreadPoolExecutor

def echo_server(addr):
    print('Echo server running at', addr)
    pool = ThreadPoolExecutor(128)
    sock = socket(AF_INET, SOCK_STREAM)
    sock.bind(addr)
    sock.listen(5)
    while True:
        client_sock, client_addr = sock.accept()
        pool.submit(echo_client, client_sock, client_addr)

echo_server(('',15000))

Python Cookbook, wydanie trzecie, David Beazley i Brian K. Jones (O'Reilly). Prawa autorskie 2013 David Beazley i Brian Jones, 978-1-449-34037-7.

Zaawansowane wykorzystanie wielowątkowości

Ta sekcja będzie zawierać niektóre z najbardziej zaawansowanych przykładów zrealizowanych przy użyciu wielowątkowości.

Zaawansowana drukarka (rejestrator)

Wątek, który drukuje wszystko, jest odbierany i modyfikuje dane wyjściowe zgodnie z szerokością terminala. Zaletą jest to, że również „już zapisane” wyjście jest modyfikowane, gdy zmienia się szerokość terminala.

#!/usr/bin/env python2

import threading
import Queue
import time
import sys
import subprocess
from backports.shutil_get_terminal_size import get_terminal_size

printq = Queue.Queue()
interrupt = False
lines = []

def main():

    ptt = threading.Thread(target=printer) # Turn the printer on
    ptt.daemon = True
    ptt.start()

    # Stupid example of stuff to print
    for i in xrange(1,100):
        printq.put(' '.join([str(x) for x in range(1,i)]))           # The actual way to send stuff to the printer
        time.sleep(.5)

def split_line(line, cols):
    if len(line) > cols:
        new_line = ''
        ww = line.split()
        i = 0
        while len(new_line) <= (cols - len(ww[i]) - 1):
            new_line += ww[i] + ' '
            i += 1
            print len(new_line)
        if new_line == '':
            return (line, '')

        return (new_line, ' '.join(ww[i:]))
    else:
        return (line, '')


def printer():

    while True:
        cols, rows = get_terminal_size() # Get the terminal dimensions
        msg = '#' + '-' * (cols - 2) + '#\n' # Create the
        try:
            new_line = str(printq.get_nowait())
            if new_line != '!@#EXIT#@!': # A nice way to turn the printer
                                         # thread out gracefully
                lines.append(new_line)
                printq.task_done()
            else:
                printq.task_done()
                sys.exit()
        except Queue.Empty:
            pass

        # Build the new message to show and split too long lines
        for line in lines:
            res = line          # The following is to split lines which are
                                # longer than cols.
            while len(res) !=0:
                toprint, res = split_line(res, cols)
                msg += '\n' + toprint

        # Clear the shell and print the new output
        subprocess.check_call('clear') # Keep the shell clean
        sys.stdout.write(msg)
        sys.stdout.flush()
        time.sleep(.5)

Zatrzymywalny wątek z krótką pętlą

import threading
import time

class StoppableThread(threading.Thread):
    """Thread class with a stop() method. The thread itself has to check
    regularly for the stopped() condition."""

    def __init__(self):
        super(StoppableThread, self).__init__()
        self._stop_event = threading.Event()

    def stop(self):
        self._stop_event.set()

    def join(self, *args, **kwargs):
        self.stop()
        super(StoppableThread,self).join(*args, **kwargs)

    def run()
        while not self._stop_event.is_set():
            print("Still running!")
            time.sleep(2)
        print("stopped!"

Na podstawie tego pytania .