1. Socket 程式設計簡介

注意，**Socket 程式設計與 Http 請求不同，Socket 程式設計當連線完成後，就可以一直給另一方傳送資料，只要連線沒有斷開，就可以一直髮送資料。而 Http 請求是連線、傳送資料、斷開。每次傳送資料都需要重複上面的過程，每次都需要連線。**這是非常重要的一點。

2. 基於 Socket 的簡單聊天程式

2.1 伺服器端

import socket
import threading

# socket.AF_INET 可以理解成 IPv4
# socket.AF_INET6 可以理解成 IPv6
# socket.SOCK_STREAM TCP 通訊
 

# socket.SOCK_DGRAM UDP 通訊
server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
# 注意，必須傳入一個 tuple
# 這兒不能傳入127.0.0.1，要不然就訪問不到了
server.bind(('0.0.0.0', 8000))
server.listen()

#獲取從客戶端傳送的資料
#一次獲取1k的資料
def handle_sock(sock, addr):
    while True:
        data = sock.recv(1024)
        data = data.decode("utf8"
 
)
        if data.lower() == 'exit' or data.lower() == 'bye':
            break
        # 獲取一個伺服器端的輸入，傳送給客戶端
        re_data = input()
        sock.send(re_data.encode("utf8"))
    sock.close()

while True:
    sock, addr = server.accept()

    # 用執行緒去處理新接收的連線(使用者)，這樣就可以同時接收多個客服端的請求
    # target 傳的是函式的名稱，不是函式的呼叫
 

    client_thread = threading.Thread(target=handle_sock, args=(sock, addr))
    client_thread.start()

2.2 客戶端

import socket
client = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
client.connect(('127.0.0.1', 8000))
while True:
    re_data = input()
    client.send(re_data.encode("utf8"))
    data = client.recv(1024)
    print(data.decode("utf8"))

3. 使用 Socket 模擬 Http 請求

# requests -> urlib -> socket
import socket
from urllib.parse import urlparse


def get_url(url):
    # 通過socket請求html
    url = urlparse(url)
    # 獲取主機名
    host = url.netloc
    # 請求路徑
    path = url.path
    # 當請求為空時，特殊處理
    if path == "":
        path = "/"

    # 建立socket連線
    client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # client.setblocking(False)
    client.connect((host, 80))  # 阻塞不會消耗cpu

    # 不停的詢問連線是否建立好， 需要while迴圈不停的去檢查狀態
    # 做計算任務或者再次發起其他的連線請求

    # 注意這兒的編碼
    client.send(
        "GET {} HTTP/1.1\r\nHost:{}\r\nConnection:close\r\n\r\n".format(path, host).encode("utf8"))

    # 注意這兒是 byte 型別
    data = b""
    while True:
        d = client.recv(1024)
        if d:
            data += d
        else:
            break

    data = data.decode("utf8")
    # 只保留 HTML 內容，去除頭部內容
    html_data = data.split("\r\n\r\n")[1]
    print(html_data)
    client.close()


if __name__ == "__main__":
    import time
    start_time = time.time()
    for url in range(20):
        url = "http://shop.projectsedu.com/goods/{}/".format(url)
        get_url(url)
    print(time.time()-start_time)