Python 多进程操作同一个对列加锁
时间: 2023-11-05 07:00:24 浏览: 103
在 Python 中,可以使用 `multiprocessing` 模块来实现多进程操作同一个队列加锁。具体步骤如下:
1. 导入 `multiprocessing` 和 `queue` 模块:
```python
import multiprocessing
import queue
```
2. 创建一个队列和一个锁对象:
```python
queue = multiprocessing.Queue()
lock = multiprocessing.Lock()
```
3. 定义一个函数,用来往队列中添加数据。在函数中使用锁对象来保证多进程访问队列时的数据安全:
```python
def add_data(data):
lock.acquire()
queue.put(data)
lock.release()
```
4. 定义一个函数,用来从队列中取出数据。在函数中同样使用锁对象来保证多进程访问队列时的数据安全:
```python
def get_data():
lock.acquire()
data = queue.get()
lock.release()
return data
```
5. 创建两个进程,一个用来往队列中添加数据,另一个用来从队列中取出数据:
```python
p1 = multiprocessing.Process(target=add_data, args=('data1',))
p2 = multiprocessing.Process(target=get_data)
```
6. 启动进程:
```python
p1.start()
p2.start()
```
7. 等待进程结束:
```python
p1.join()
p2.join()
```
完整代码示例:
```python
import multiprocessing
import queue
queue = multiprocessing.Queue()
lock = multiprocessing.Lock()
def add_data(data):
lock.acquire()
queue.put(data)
lock.release()
def get_data():
lock.acquire()
data = queue.get()
lock.release()
return data
p1 = multiprocessing.Process(target=add_data, args=('data1',))
p2 = multiprocessing.Process(target=get_data)
p1.start()
p2.start()
p1.join()
p2.join()
```
阅读全文