python解决多线程输出混乱

python解决多线程输出混乱

定义一个加锁print()

_print = print
mutex = threading.Lock()
def print(text, *args, **kw):
    with mutex:
        _print(text, *args, **kw)

示例

import queue
import threading
import time

exitFlag = 0

_print = print
mutex = threading.Lock()
def print(text, *args, **kw):
    with mutex:
        _print(text, *args, **kw)

class myThread(threading.Thread):
    def __init__(self, threadID, name, q):
        threading.Thread.__init__(self)
        self.threadID = threadID
        self.name = name
        self.q = q

    def run(self):
        print("开启线程：" + self.name, "ID", self.threadID)
        process_data(self.name, self.q)
        print("退出线程：" + self.name, "ID", self.threadID)


def process_data(threadName, q):
    while not exitFlag:
        queueLock.acquire()
        if not workQueue.empty():
            data = q.get()
            queueLock.release()
            print("%s processing %s" % (threadName, data))
        else:
            queueLock.release()
        time.sleep(1)


def process_data(threadName, que):
    while not exitFlag:
        queueLock.acquire()
        if not workQueue.empty():
            data = que.get()
            print("%s processing %s" % (threadName, data))
            print('queue size',que.qsize())
        queueLock.release()
        time.sleep(1)


threadList = ["Thread-1", "Thread-2", "Thread-3"]
nameList = ["One", "Two", "Three", "Four", "Five"]

queueLock = threading.Lock()
workQueue = queue.Queue(10)
threads = []
threadID = 0

# 创建新线程
for tName in threadList:
    threadID += 1
    thread = myThread(threadID, tName, workQueue)
    thread.start()
    threads.append(thread)


# 填充队列
queueLock.acquire()
for word in nameList:
    workQueue.put(word)
queueLock.release()
print('queue size',workQueue.qsize())

# 等待队列清空
while not workQueue.empty():
    pass

# 通知线程是时候退出
exitFlag = 1

# 等待所有线程完成
for t in threads:
    t.join()

print("退出主线程")