使用Python實現(xiàn)TCP/IP客戶端和服務端通信功能

更新時間：2024年12月13日 10:09:52 作者：程序員黃同學

TCP/IP（傳輸控制協(xié)議/互聯(lián)網協(xié)議）是互聯(lián)網的基礎協(xié)議,用于在網絡中的計算機之間進行可靠的數據傳輸,在Python中,可以使用socket模塊來實現(xiàn)TCP/IP通信,本文給大家介紹了使用Python實現(xiàn)TCP/IP客戶端和服務端通信功能,需要的朋友可以參考下

1. TCP/IP通信基礎

TCP/IP（傳輸控制協(xié)議/互聯(lián)網協(xié)議）是互聯(lián)網的基礎協(xié)議，用于在網絡中的計算機之間進行可靠的數據傳輸。在Python中，可以使用socket模塊來實現(xiàn)TCP/IP通信。

2. 實現(xiàn)TCP/IP服務端

import socket
 
def start_server(host='127.0.0.1', port=65432):
    """
    啟動一個TCP/IP服務端
    :param host: 服務端監(jiān)聽的IP地址，默認為本地回環(huán)地址
    :param port: 服務端監(jiān)聽的端口
    """
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.bind((host, port))
        s.listen()
        print(f"Server listening on {host}:{port}")
        conn, addr = s.accept()
        with conn:
            print(f"Connected by {addr}")
            while True:
                data = conn.recv(1024)
                if not data:
                    break
                print(f"Received from client: {data.decode()}")
                conn.sendall(data)  # Echo back the received data
 
if __name__ == "__main__":
    start_server()

代碼解釋：

socket.socket(socket.AF_INET, socket.SOCK_STREAM)：創(chuàng)建一個TCP套接字。
s.bind((host, port))：綁定IP地址和端口。
s.listen()：開始監(jiān)聽連接。
conn, addr = s.accept()：接受一個連接，返回一個新的套接字對象和客戶端地址。
conn.recv(1024)：接收數據，最多1024字節(jié)。
conn.sendall(data)：將接收到的數據發(fā)送回客戶端。

3. 實現(xiàn)TCP/IP客戶端

import socket
 
def start_client(host='127.0.0.1', port=65432):
    """
    啟動一個TCP/IP客戶端
    :param host: 服務端的IP地址
    :param port: 服務端的端口
    """
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.connect((host, port))
        while True:
            message = input("Enter a message to send: ")
            s.sendall(message.encode())
            data = s.recv(1024)
            print(f"Received from server: {data.decode()}")
 
if __name__ == "__main__":
    start_client()

代碼解釋：

socket.socket(socket.AF_INET, socket.SOCK_STREAM)：創(chuàng)建一個TCP套接字。
s.connect((host, port))：連接到服務端。
s.sendall(message.encode())：發(fā)送數據到服務端。
s.recv(1024)：接收服務端發(fā)送的數據。

4. 日常開發(fā)中的合理化使用建議

錯誤處理：在實際應用中，網絡通信可能會遇到各種錯誤，如連接中斷、超時等。應使用try-except塊來捕獲和處理這些異常。
多線程/多進程：對于高并發(fā)場景，可以使用多線程或多進程來處理多個客戶端連接。
心跳機制：為了檢測客戶端的存活狀態(tài)，可以在通信中加入心跳機制。
數據加密：對于敏感數據的傳輸，應考慮使用SSL/TLS進行加密。

5. 實際開發(fā)過程中需要注意的點

資源管理：確保在通信結束后正確關閉套接字和其他資源，避免資源泄漏。
數據格式：在發(fā)送和接收數據時，應明確數據的格式（如JSON、XML等），并進行相應的序列化和反序列化操作。
超時設置：設置合理的超時時間，避免長時間等待導致的性能問題。
日志記錄：記錄通信過程中的關鍵信息，便于調試和故障排查。

6. 錯誤處理示例

import socket
 
def start_server(host='127.0.0.1', port=65432):
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.bind((host, port))
        s.listen()
        print(f"Server listening on {host}:{port}")
        while True:
            try:
                conn, addr = s.accept()
                with conn:
                    print(f"Connected by {addr}")
                    while True:
                        data = conn.recv(1024)
                        if not data:
                            break
                        print(f"Received from client: {data.decode()}")
                        conn.sendall(data)
            except Exception as e:
                print(f"Error: {e}")
 
if __name__ == "__main__":
    start_server()

代碼解釋：

使用try-except塊捕獲可能的異常，并打印錯誤信息。

7. 多線程示例

import socket
import threading
 
def handle_client(conn, addr):
    with conn:
        print(f"Connected by {addr}")
        while True:
            data = conn.recv(1024)
            if not data:
                break
            print(f"Received from client: {data.decode()}")
            conn.sendall(data)
 
def start_server(host='127.0.0.1', port=65432):
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.bind((host, port))
        s.listen()
        print(f"Server listening on {host}:{port}")
        while True:
            conn, addr = s.accept()
            thread = threading.Thread(target=handle_client, args=(conn, addr))
            thread.start()
 
if __name__ == "__main__":
    start_server()

代碼解釋：

使用threading.Thread為每個客戶端連接創(chuàng)建一個新線程，實現(xiàn)并發(fā)處理。

8. 心跳機制示例

import socket
import time
 
def start_client(host='127.0.0.1', port=65432):
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.connect((host, port))
        while True:
            message = input("Enter a message to send: ")
            s.sendall(message.encode())
            data = s.recv(1024)
            print(f"Received from server: {data.decode()}")
            time.sleep(5)  # Send a heartbeat every 5 seconds
            s.sendall(b"heartbeat")
 
if __name__ == "__main__":
    start_client()

代碼解釋：