简体   繁体   中英

Blocking other Threads while one Thread is running

Let's say I have two types of threads,

  1. single thread that run every x min. let's call it A thread

  2. multi threads run all the time. B threads when A thread do_something() i want All B threads to wait till A finish then resume them. i can't figure it out what to use.

I try to use threading.Condition , wait() / notifyAll() but it did not work as I want. once i put Condition in, it process 1 by 1 like synco threads or something. I want them to run freely.

This is the sample code I try to put them wait() , then notify them but it do 1 by 1 like join() . No idea what to us.

class ...
check = True
def xxx(self,g,con):
  for i in range(3):
    with con:
      if self.check:
        con.wait()
      self.check = False
      time.sleep(3)
      print(g)

con = threading.Condition()
threading.Thread(target=xxx,args=('a',con,)).start()
threading.Thread(target=xxx,args=('b',con,)).start()
threading.Thread(target=xxx,args=('c',con,)).start()
time.sleep(2)
con.notifyAll()

Question : Blocking other Threads while one Thread is running

Instead of using threading.Condition() , this example uses threading.Barrier(...) .


Used modules from docs.python.org :


import time, threading
from threading import BrokenBarrierError

def worker_A(g, terminate, barrier):
    # Counter to simulate conditional workload
    do_something = 3

    while not terminate.is_set():
        if do_something == 0:
            # Reset the barrier and wait until n_waiting == 2
            barrier.reset()
            while not terminate.is_set() and barrier.n_waiting < 2:
                time.sleep(0.5)

            # Now the other Threads waiting at the barrier
            # Simulate worklaod ...
            print('worker_A barrier.broken={} n_waiting={}'
                .format(barrier.broken, barrier.n_waiting))
            time.sleep(3)

            # Call the third barrier.wait to release the barrier
            try:
                barrier.wait()
            except BrokenBarrierError:
                pass

            # Reset counter to restart simulate conditional workload
            do_something = 3
        else:
            # Count down and give the other threads a timeslice
            do_something -= 1
            time.sleep(0.5)

def worker_B(g, terminate, barrier):
    while not terminate.is_set():
        # Simulate workload ...
        print('worker_B({})'.format(g))
        time.sleep(1)

        # Block at barrier.wait() if the barrier is NOT in the broken state
        try:
            barrier.wait()
        except BrokenBarrierError:
            pass

if __name__ == "__main__":
    # Event to terminate all Threads save
    terminate = threading.Event()

    # Barrier to block worker_B Threads
    # We use 3 Threads, therefore init with parties=3
    barrier = threading.Barrier(3)
    barrier.abort()

    # Create and start the Threads
    threads = []
    for t in [(worker_A, 'a'), (worker_B, 'b'), (worker_B, 'c'), ]:
        threads.append(threading.Thread(target=t[0], args=(t[1], terminate, barrier,)))
        threads[-1].start()
        time.sleep(0.2)

    # Simulating MAIN Thread
    time.sleep(20)

    # Set the `terminate` Event to True, 
    # and abort the barrier to force all Threads to terminate
    print('Terminate...')
    terminate.set()
    barrier.abort()

    # Wait until all Threads terminated
    for t in threads:
        t.join()

    print('EXIT MAIN')

Tested with Python: 3.5

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM