Python-сообщество

Всем привет.

Есть сервер. Сначала думал над тем, как вручную реализовать, потом наткнулся на http://docs.python.org/py3k/library/socketserver.html и решил, что это то, что мне надо.

Собственно, вся реализация сервера:

class ThreadedTCPServer(socketserver.ThreadingMixIn, socketserver.TCPServer):
    pass

class ThreadedTCPRequestHandler(socketserver.BaseRequestHandler):
    def handle(self):
        data = str(self.request.recv(1024), 'ascii')
        cur_thread = threading.current_thread()
        response = bytes("{}: {}".format(cur_thread.name, data), 'ascii')
        self.request.sendall(response)

tasks = queue.Queue()
ThreadedTCPRequestHandler.tasks = tasks

worker = Worker(tasks)

class ThreadedTCPRequestHandler(socketserver.BaseRequestHandler):
    def handle(self):
        data = str(self.request.recv(1024), 'ascii')
        ThreadedTCPRequestHandler.tasks.put(data)
        # что посылать в ответ??
        self.request.sendall(???)
class Worker:
    def __init__(self, tasks):
        self.tasks = tasks
    def do(self):
        while True:
            item = q.get()
            # делаю тут что-то, например, работаю с базой данных sqlite
            q.task_done()

можно сделать каждому потоку очередь в которую worker будет скидывать результат

def handle(self):
    data = str(self.request.recv(1024), 'ascii')
    myname = threading.current_thread().name
    ThreadedTCPRequestHandler.tasks.put((myname, data))
    result = ThreadedTCPRequestHandler.results[myname].get()
    self.request.sendall(result)

while True:
    name, item = q.get()
    # делаю тут что-то, например, работаю с базой данных sqlite
    ThreadedTCPRequestHandler.results[name].put(smth)
    q.task_done()

Отредактировано mindless (Окт. 19, 2012 02:35:52)

mindless
upd: почему подсветка по умолчанию не python?

Если вы хотите писать сами низкоуровневые вещи, рекомендую изучить хотя бы потоки.
Вот есть хороший пример сервера, который создает пул потоков-обработчиков для каждого клиента:
http://keysolutions.ru/articles/osnovy-raboty-s-potokami-v-python

Мм, спасибо за ссылку. Может подумаю о том, чтобы создать пул потоков.

С этим вроде бы разобрался (без пула), сделал так, чтобы всё отправлялось через один и тот же сокет, и через него же принималось.
Вкратце - есть поток, который прослушивает сокет, и если пришло сообщение - парсит его. Если это notification, кладёт его в одну очередь, а если response - в другую.
И другой поток (по сути главный поток) является клиентом, он может отправлять запросы (request) в этот же самый socket, но response (ответы) он оттуда не берёт, а вместо этого запрашивает их из очереди (в которую их кладёт первый поток).

Довольно гибкая система, хотя код IMHO всё равно получился ужасен не привык я пока к питону. Но работает. Как надо.

Осталось с GUI нормальным разобраться ну и с прочими тонкостями этого приложения. Но это уже совсем другая история.

battlecoder
Нюанс вот в чём. Мне нужно чтобы сервер обслуживал многих клиентов, а они могли выполнять какие-то операции. Например, пользователь хочет зарегистрироваться - для этого клиент посылает специальный запрос, сервер его обрабатывает, и он должен добавить запись в базу данных (использую sqlite).Но sqlite нельзя использовать из множества потоков!! Решить проблему можно так - выбрать один поток, который и будет писать в базу данных и выполнять все операции.

Отредактировано altRUist (Дек. 17, 2012 20:48:09)

Если запрос к базе занимает малую часть от общего времени обработки сообщения, то преимущество все равно будет. Да, это станет одним из узких мест в системе, но вероятно не самым тормозящим.