python 跨进程通信

时间:2022-05-23 JulyLi2019 人气:0

前言

项目中总会遇到数据需要跨进程通信的问题，今天就给大家带来一套简单的跨进程通信代码。代码分为服务端与客户端两部分。

一、server端

import multiprocessing
import time

def do_socket(conn, addr, ):
    try:

        while True:
            if conn.poll(1) == False:
                time.sleep(0.5)
                continue
            data = conn.recv()  # 等待接受数据
            conn.send('sucess')
            # ***********************
            # 要执行的程序写在这里
            # ***********************
            print(data)

    except Exception as e:
        print('Socket Error', e)

    finally:
        try:
            conn.close()
            print('Connection close.', addr)
        except:
            print('close except')


def run_server(host, port):
    from multiprocessing.connection import Listener
    server_sock = Listener((host, port))

    print("Sever running...", host, port)

    pool = multiprocessing.Pool(10)
    while True:
        # 接受一个新连接:

        conn = server_sock.accept()
        addr = server_sock.last_accepted
        print('Accept new connection', addr)

        # 创建进程来处理TCP连接:
        pool.apply_async(func=do_socket, args=(conn, addr,))


if __name__ == '__main__':
    server_host = '127.0.0.1'
    server_port = 8000
    run_server(server_host, server_port)

二、client端

import time

from multiprocessing.connection import Client

client = Client(('127.0.0.1', 8000))

while True:
    data = 'send data'
    client.send(data)
    data = client.recv()  # 等待接受数据
    print(data)
    time.sleep(1)

这里只是给了client端的示例代码，实际通信的数据类型可以是多样的（字符串、数字、图片等）

三、运行效果

先运行server端，再运行client端。

server端

在这里插入图片描述

client端

在这里插入图片描述

总结

本文简单给大家提供了一套简单封装的跨进程通信代码，方便大家快速上手实现跨进程通信，有需要的伙伴可以根据自己的需求调整代码。

加载全部内容