python - 多线程写入文件

Question

前几天刚开始使用 python，对多线程的整个概念还很陌生。我在多线程时写入文件时遇到问题。如果我按照常规方式执行此操作，它会不断覆盖正在写入的内容。

使用 5 个线程写入文件的正确方法是什么？

Answer 1

不降低性能的最佳方法是在所有线程之间使用队列，每个线程都会排队一个项目，而主线程只是简单地将一个项目出队并将其写入文件，队列是线程安全的，并且在以下情况下会阻塞它是空的，或者如果可能的话更好，只需从 5 个线程返回所有值，然后将其写入文件，IO 往往是我们可以执行的更昂贵的操作之一，因此最好尽可能限制它可以。

另请注意，Python 中的线程不利用多核，这是 GIL 的原因。而是使用 multiprocessing如果您想利用多个处理引擎。

这是一个简单的示例:

from multiprocessing import Process, Queue

def test_1(q):
    for i in range(10):
        q.put('test_1: ' + str(i))

def test_2(q):
    for i in range(10):
        q.put('test_2: ' + str(i))

q = Queue()
p1 = Process(target=test_1, args=(q,))
p2 = Process(target=test_2, args=(q,))
p1.start()
p2.start()

with open('test.txt', 'w') as file:
    while p1.is_alive() or p2.is_alive() or not q.empty():
        try:
            value = q.get(timeout = 1)
            file.write(value + '\n')
        except Exception as qe:
            print "Empty Queue or dead process"
p1.join()
p2.join()

以及test.txt的内容:

test_1: 0
test_1: 1
test_1: 2
test_1: 3
test_1: 4
test_2: 0
test_1: 5
test_2: 1
test_1: 6
test_2: 2
test_1: 7
test_2: 3
test_1: 8
test_2: 4
test_1: 9
test_2: 5
test_2: 6
test_2: 7
test_2: 8
test_2: 9