Я пытаюсь создать многоклиентный сервер http2 с помощью гипер-библиотеки Python.Завершает только первый запрос - PullRequest
Новая
       36

Я пытаюсь создать многоклиентный сервер http2 с помощью гипер-библиотеки Python.Завершает только первый запрос

0 голосов
/ 04 декабря 2018

Я могу обслуживать http2, используя это руководство.https://python -hyper.org / проекты / h2 / о / стабильный / базовый usage.html .Эта версия отлично работает.Но этот сервер может обслуживать только одного клиента за раз из-за блокирующих сокетов.Затем я искал многоклиентные серверные реализации, основанные на select() системном вызове.Этот урок показал мне, как это сделать.https://realpython.com/python-sockets/. Я объединил два учебника, чтобы создать сервер, обслуживающий несколько клиентов http2.Вот чем я закончил:

server.py 

import json
import socket

import h2.connection
import h2.events

import types
from thread import start_new_thread
try:
    import selectors
except ImportError:
    import selectors2 as selectors

sel = selectors.DefaultSelector()
h2conn = h2.connection.H2Connection(client_side=False)

def send_response(hyperConn, event):
    print ('sending respond')
    stream_id = event.stream_id
    response_data = json.dumps(dict(event.headers))+'KUKA MAKI'.encode('utf-8')

    hyperConn.send_headers(
        stream_id=stream_id,
        headers=[
            (':status', '200'),
            ('server', 'basic-h2-server/1.0'),
            ('content-length', str(len(response_data))),
            ('content-type', 'application/json'),
        ],
    )
    hyperConn.send_data(
        stream_id=stream_id,
        data=response_data,
        end_stream=True
    )

def accept_wrapper(sock):
    conn, addr = sock.accept()  # Should be ready to read
    print("accepted connection from", addr)
    conn.setblocking(False)

    data = lambda: None    # Dummy namespace
    data.addr = addr
    data.inb = b''
    data.outb = b''
    events = selectors.EVENT_READ | selectors.EVENT_WRITE
    sel.register(conn, events, data=data)
    h2InitConn = h2.connection.H2Connection(client_side=False)
    h2InitConn.initiate_connection()
    conn.sendall(h2InitConn.data_to_send())


def service_connection(key, mask):

    sock = key.fileobj
    data = key.data

    if mask & selectors.EVENT_READ:
        recv_data = sock.recv(65535)  # Should be ready to read
        if recv_data:
            # data.outb += recv_data
            print ('raw data: ', recv_data)
            events = h2conn.receive_data(recv_data)
            print 'events received:' , events
            for event in events:
                if isinstance(event, h2.events.RequestReceived):
                    print ('request recieved')
                    send_response(h2conn, event)

        else:
            print("closing connection to", data.addr)
            sel.unregister(sock)
            sock.close()
    if mask & selectors.EVENT_WRITE:
        data_to_send = h2conn.data_to_send()
        if data_to_send:
            print("replying with stuff to", data.addr)
            sock.sendall(data_to_send)

lsock = socket.socket()
lsock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
host, port = '0.0.0.0' , 8080
lsock.bind((host, port))
lsock.listen(5)
print("listening on", (host, port))
lsock.setblocking(False)


sel.register(lsock, selectors.EVENT_READ, data=None)
try:
    while True:
        events = sel.select(timeout=None)
        for key, mask in events:
            if key.data is None:
                accept_wrapper(key.fileobj)
            else:
                service_connection(key, mask)
except KeyboardInterrupt:
    print("caught keyboard interrupt, exiting")
finally:
    sel.close()

Если я запустил python server.py и в другом терминалеЯ запускаю инструмент гипер CLI, чтобы отправить запрос http2 (hyper --h2 GET http://localhost:8080/).Первый запрос выполнен успешно, я вижу сгенерированные события, а гиперинструмент показывает резонансы и корректно завершает работу.После этой отправки той же гиперкоманды снова нет событий, сгенерированных запросом, и инструмент CLI также зависает.Не могли бы вы помочь мне разобраться в проблеме?

1 Ответ

0 голосов
/ 04 декабря 2018

Мне удалось это выяснить.

Мне пришлось сохранить состояние соединения http2 в функции accept_wrapper в словарь, где ключом является адрес сокета клиента.В функции service_connection я проанализировал данные с соответствующим сохраненным объектом соединения http2, а затем произвел события.

в начале файла: 

h2conns = {}

конец функции accept_wrapperaccept_wrapper: 

h2conns[conn.getpeername()[1]] = h2InitConn

в функции servie_connection: 

h2conn = h2conns[sock.getpeername()[1]]
...