WebSocketServlet.h

/* Copyright (C) 2026 by Marc Boris Duerner
   SPDX-License-Identifier: LGPL-2.1-or-later WITH mif-exception
*/
 
#ifndef PT_HTTP_WEBSOCKETSERVLET_H
#define PT_HTTP_WEBSOCKETSERVLET_H
 
#include <Pt/Http/Api.h>
#include <Pt/Connectable.h>
#include <Pt/NonCopyable.h>
#include <vector>
#include <cstddef>
 
namespace Pt {
 
namespace Http {
 
class Stream;
class WebSocketService;
class WebSocketSession;

class PT_HTTP_API WebSocketServlet : public Connectable
                                   , private NonCopyable
{
    friend class WebSocketService;
    friend class WebSocketSession;
 
    public:
        explicit WebSocketServlet(WebSocketService& service);

        ~WebSocketServlet();

        WebSocketService& service()
        { return *_service; }

        const WebSocketService& service() const
        { return *_service; }

        std::size_t size() const
        { return _sessions.size(); }
 
    private:
        void onUpgrade(Stream& stream);
 
        void onSessionClosed(WebSocketSession& session);
 
        void releaseSession(WebSocketSession* session);
 
        WebSocketService* _service;
        std::vector<WebSocketSession*> _sessions;
};
 
} // namespace Http
 
} // namespace Pt
 
#endif