SocketServer.h

Go to the documentation of this file.

/*
 * SNode.C - A Slim Toolkit for Network Communication
 * Copyright (C) Volker Christian <me@vchrist.at>
 *               2020, 2021, 2022, 2023, 2024, 2025, 2026
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published
 * by the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
 
/*
 * MIT License
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
 
#ifndef CORE_SOCKET_STREAM_SOCKETSERVERNEW_H
#define CORE_SOCKET_STREAM_SOCKETSERVERNEW_H
 
#include "core/EventReceiver.h"
#include "core/SNodeC.h"
#include "core/socket/Socket.h"                    // IWYU pragma: export
#include "core/socket/State.h"                     // IWYU pragma: export
#include "core/socket/stream/AutoConnectControl.h" // IWYU pragma: export
#include "core/timer/Timer.h"
 
#ifndef DOXYGEN_SHOULD_SKIP_THIS
 
#include "log/Logger.h"
#include "utils/Random.h"
 
#include <algorithm>
#include <functional>  // IWYU pragma: export
#include <type_traits> // IWYU pragma: export
 
#endif /* DOXYGEN_SHOULD_SKIP_THIS */
 
namespace core::socket::stream {
 
    template <typename SocketAcceptorT, typename SocketContextFactoryT, typename... Args>
        requires std::is_base_of_v<core::eventreceiver::AcceptEventReceiver, SocketAcceptorT> &&
                 std::is_base_of_v<core::socket::stream::SocketContextFactory, SocketContextFactoryT>
    class SocketServer : public core::socket::Socket<typename SocketAcceptorT::Config> {
    private:
        using SocketAcceptor = SocketAcceptorT;
        using SocketContextFactory = SocketContextFactoryT;
 
        using Super = core::socket::Socket<typename SocketAcceptor::Config>;
 
    public:
        using SocketConnection = typename SocketAcceptor::SocketConnection;
        using SocketAddress = typename SocketAcceptor::SocketAddress;
        using Config = typename SocketAcceptor::Config;
 
    private:
        struct Context {
            Context(const std::shared_ptr<SocketContextFactory>& socketContextFactory,
                    const std::function<void(SocketConnection*)>& onConnect,
                    const std::function<void(SocketConnection*)>& onConnected,
                    const std::function<void(SocketConnection*)>& onDisconnect)
                : socketContextFactory(socketContextFactory)
                , onConnect(onConnect)
                , onConnected(onConnected)
                , onDisconnect(onDisconnect)
                , onInitState([]([[maybe_unused]] core::eventreceiver::AcceptEventReceiver* descriptorEventReceiver) {
                })
                , onAutoConnectControl([]([[maybe_unused]] const std::shared_ptr<AutoConnectControl>& onAutoConnectControl) {
                }) {
            }
 
            std::shared_ptr<SocketContextFactory> socketContextFactory;
 
            std::function<void(SocketConnection*)> onConnect;
            std::function<void(SocketConnection*)> onConnected;
            std::function<void(SocketConnection*)> onDisconnect;
            std::function<void(core::eventreceiver::AcceptEventReceiver*)> onInitState;
 
            std::function<void(const std::shared_ptr<AutoConnectControl>&)> onAutoConnectControl;
            std::shared_ptr<AutoConnectControl> autoConnectControl;
        };
 
        SocketServer(const std::shared_ptr<Config>& config,
                     const std::shared_ptr<SocketContextFactory>& socketContextFactory,
                     const std::function<void(SocketConnection*)> onConnect,
                     const std::function<void(SocketConnection*)> onConnected,
                     const std::function<void(SocketConnection*)> onDisconnect)
            : Super(config)
            , sharedContext(std::make_shared<Context>(socketContextFactory, onConnect, onConnected, onDisconnect)) {
        }
 
        SocketServer(const std::shared_ptr<Config>& config, const std::shared_ptr<Context>& sharedContext)
            : Super(config)
            , sharedContext(sharedContext) {
        }
 
    public:
        SocketServer(const std::string& name,
                     const std::function<void(SocketConnection*)>& onConnect,
                     const std::function<void(SocketConnection*)>& onConnected,
                     const std::function<void(SocketConnection*)>& onDisconnect,
                     Args&&... args)
            : Super(name)
            , sharedContext(std::make_shared<Context>(
                  std::make_shared<SocketContextFactory>(std::forward<Args>(args)...),
                  [onConnect](SocketConnection* socketConnection) { // onConnect
                      LOG(DEBUG) << socketConnection->getConnectionName() << ": OnConnect";
 
                      LOG(DEBUG) << "  Local: " << socketConnection->getLocalAddress().toString();
                      LOG(DEBUG) << "   Peer: " << socketConnection->getRemoteAddress().toString();
 
                      if (onConnect) {
                          onConnect(socketConnection);
                      }
                  },
                  [onConnected](SocketConnection* socketConnection) { // onConnected
                      LOG(DEBUG) << socketConnection->getConnectionName() << ": OnConnected";
 
                      LOG(DEBUG) << "  Local: " << socketConnection->getLocalAddress().toString();
                      LOG(DEBUG) << "   Peer: " << socketConnection->getRemoteAddress().toString();
 
                      if (onConnected) {
                          onConnected(socketConnection);
                      }
                  },
                  [onDisconnect](SocketConnection* socketConnection) { // onDisconnect
                      LOG(DEBUG) << socketConnection->getConnectionName() << ": OnDisconnect";
 
                      LOG(DEBUG) << "            Local: " << socketConnection->getLocalAddress().toString();
                      LOG(DEBUG) << "             Peer: " << socketConnection->getRemoteAddress().toString();
 
                      LOG(DEBUG) << "     Online Since: " << socketConnection->getOnlineSince();
                      LOG(DEBUG) << "  Online Duration: " << socketConnection->getOnlineDuration();
 
                      LOG(DEBUG) << "     Total Queued: " << socketConnection->getTotalQueued();
                      LOG(DEBUG) << "       Total Sent: " << socketConnection->getTotalSent();
                      LOG(DEBUG) << "      Write Delta: " << socketConnection->getTotalQueued() - socketConnection->getTotalSent();
                      LOG(DEBUG) << "       Total Read: " << socketConnection->getTotalRead();
                      LOG(DEBUG) << "  Total Processed: " << socketConnection->getTotalProcessed();
                      LOG(DEBUG) << "       Read Delta: " << socketConnection->getTotalRead() - socketConnection->getTotalProcessed();
 
                      if (onDisconnect) {
                          onDisconnect(socketConnection);
                      }
                  })) {
        }
 
        SocketServer(const std::function<void(SocketConnection*)>& onConnect,
                     const std::function<void(SocketConnection*)>& onConnected,
                     const std::function<void(SocketConnection*)>& onDisconnect,
                     Args&&... args)
            : SocketServer("", onConnect, onConnected, onDisconnect, std::forward<Args>(args)...) {
        }
 
        SocketServer(const std::string& name, Args&&... args)
            : SocketServer(name, {}, {}, {}, std::forward<Args>(args)...) {
        }
 
        explicit SocketServer(Args&&... args)
            : SocketServer("", std::forward<Args>(args)...) {
        }
 
    private:
        const SocketServer& realListen(const std::function<void(const SocketAddress&, core::socket::State)>& onStatus,
                                       unsigned int tries,
                                       double retryTimeoutScale) const {
            core::EventReceiver::atNextTick(
                [config = this->config, sharedContext = this->sharedContext, onStatus, tries, retryTimeoutScale] {
                    LOG(DEBUG) << config->getInstanceName() << ": Initiating listen";
 
                    if (core::SNodeC::state() == core::State::RUNNING || core::SNodeC::state() == core::State::INITIALIZED) {
                        auto autoConnectControl = sharedContext->autoConnectControl;
                        if (!autoConnectControl) {
                            autoConnectControl = std::make_shared<AutoConnectControl>();
                            sharedContext->autoConnectControl = autoConnectControl;
                            if (sharedContext->onAutoConnectControl) {
                                sharedContext->onAutoConnectControl(autoConnectControl);
                            }
                        }
 
                        new SocketAcceptor(
                            sharedContext->socketContextFactory,
                            sharedContext->onConnect,
                            sharedContext->onConnected,
                            sharedContext->onDisconnect,
                            sharedContext->onInitState,
                            [config, sharedContext, autoConnectControl, onStatus, tries, retryTimeoutScale](
                                const SocketAddress& socketAddress, core::socket::State state) {
                                const bool retryFlag = (state & core::socket::State::NO_RETRY) == 0;
                                state &= ~core::socket::State::NO_RETRY;
                                onStatus(socketAddress, state);
 
                                if (retryFlag && config->getRetry() // Shall we potentially retry? In case are the ...
                                    && autoConnectControl->isRetryEnabled() &&
                                    (config->getRetryTries() == 0 ||
                                     tries < config->getRetryTries()) // ... limits not reached and has an ...
                                    && (state == core::socket::State::ERROR ||
                                        (state == core::socket::State::FATAL && config->getRetryOnFatal()))) { // error occurred?
                                    double relativeRetryTimeout =
                                        config->getRetryLimit() > 0
                                            ? std::min<double>(config->getRetryTimeout() * retryTimeoutScale, config->getRetryLimit())
                                            : config->getRetryTimeout() * retryTimeoutScale;
                                    relativeRetryTimeout -= utils::Random::getInRange(-config->getRetryJitter(), config->getRetryJitter()) *
                                                            relativeRetryTimeout / 100.;
 
                                    LOG(INFO) << config->getInstanceName() << ": Retry listen in " << relativeRetryTimeout << " seconds";
 
                                    autoConnectControl->armRetryTimer(
                                        relativeRetryTimeout,
                                        [config, sharedContext, autoConnectControl, /*generation,*/ onStatus, tries, retryTimeoutScale]() {
                                            if (!autoConnectControl->isRetryEnabled()) {
                                                return;
                                            }
                                            if (config->getRetry()) {
                                                SocketServer(config, sharedContext)
                                                    .realListen(onStatus, tries + 1, retryTimeoutScale * config->getRetryBase());
                                            } else {
                                                LOG(INFO) << config->getInstanceName() << ": Retry listen disabled during wait";
                                            }
                                        });
                                }
                            },
                            config);
                    }
                });
 
            return *this;
        }
 
    public:
        const SocketServer& listen(const std::function<void(const SocketAddress&, core::socket::State)>& onStatus) const {
            return realListen(onStatus, 0, 1);
        }
 
        const SocketServer& listen(const SocketAddress& localAddress,
                                   const std::function<void(const SocketAddress&, core::socket::State)>& onStatus) const {
            Super::config->Local::setSocketAddress(localAddress);
 
            return listen(onStatus);
        }
 
        const SocketServer& listen(const SocketAddress& localAddress,
                                   int backlog,
                                   const std::function<void(const SocketAddress&, core::socket::State)>& onStatus) const {
            Super::config->Local::setBacklog(backlog);
 
            return listen(localAddress, onStatus);
        }
 
        std::function<void(SocketConnection*)>& getOnConnect() const {
            return sharedContext->onConnect;
        }
 
        const SocketServer& setOnConnect(const std::function<void(SocketConnection*)>& onConnect, bool initialize = false) const {
            sharedContext->onConnect =
                initialize ? onConnect : [oldOnConnect = sharedContext->onConnect, onConnect](SocketConnection* socketConnection) {
                    oldOnConnect(socketConnection);
                    onConnect(socketConnection);
                };
 
            return *this;
        }
 
        std::function<void(SocketConnection*)>& getOnConnected() const {
            return sharedContext->onConnected;
        }
 
        const SocketServer& setOnConnected(const std::function<void(SocketConnection*)>& onConnected, bool initialize = false) const {
            sharedContext->onConnected =
                initialize ? onConnected : [oldOnConnected = sharedContext->onConnected, onConnected](SocketConnection* socketConnection) {
                    oldOnConnected(socketConnection);
                    onConnected(socketConnection);
                };
 
            return *this;
        }
 
        std::function<void(SocketConnection*)>& getOnDisconnect() const {
            return sharedContext->onDisconnect;
        }
 
        SocketServer& setOnDisconnect(const std::function<void(SocketConnection*)>& onDisconnect, bool initialize = false) {
            sharedContext->onDisconnect =
                initialize ? onDisconnect
                           : [oldOnDisconnect = sharedContext->onDisconnect, onDisconnect](SocketConnection* socketConnection) {
                                 oldOnDisconnect(socketConnection);
                                 onDisconnect(socketConnection);
                             };
 
            return *this;
        }
 
        std::function<void(const std::shared_ptr<AutoConnectControl>&)>& getOnAutoConnectControl() {
            return sharedContext->onAutoConnectControl;
        }
 
        const SocketServer&
        setOnAutoConnectControl(const std::function<void(const std::shared_ptr<AutoConnectControl>&)>& onAutoConnectControl,
                                bool initialize = false) const {
            sharedContext->onAutoConnectControl = initialize ? onAutoConnectControl
                                                             : [oldOnAutoConnectControl = sharedContext->onAutoConnectControl,
                                                                onAutoConnectControl](const std::shared_ptr<AutoConnectControl>& control) {
                                                                   oldOnAutoConnectControl(control);
                                                                   onAutoConnectControl(control);
                                                               };
 
            if (sharedContext->autoConnectControl && onAutoConnectControl) {
                core::EventReceiver::atNextTick([sharedContext = this->sharedContext, onAutoConnectControl] {
                    onAutoConnectControl(sharedContext->autoConnectControl);
                });
            }
 
            return *this;
        }
 
        std::function<void(core::eventreceiver::AcceptEventReceiver*)>& getOnInitState() const {
            return sharedContext->onInitState;
        }
 
        const SocketServer& setOnInitState(const std::function<void(core::eventreceiver::AcceptEventReceiver*)>& onInitState,
                                           bool initialize = false) const {
            sharedContext->onInitState = initialize ? onInitState
                                                    : [oldOnInitState = sharedContext->onInitState,
                                                       onInitState](core::eventreceiver::AcceptEventReceiver* descriptorEventReceiver) {
                                                          oldOnInitState(descriptorEventReceiver);
                                                          onInitState(descriptorEventReceiver);
                                                      };
 
            return *this;
        }
 
        std::shared_ptr<SocketContextFactory> getSocketContextFactory() const {
            return sharedContext->socketContextFactory;
        }
 
    private:
        std::shared_ptr<Context> sharedContext;
    };
 
    template <typename SocketServer, typename... Args>
    SocketServer Server(const std::string& instanceName,
                        const std::function<void(typename SocketServer::Config&)>& configurator,
                        Args&&... socketContextFactoryArgs) {
        const SocketServer socketServer(instanceName, std::forward<Args>(socketContextFactoryArgs)...);
 
        configurator(socketServer.getConfig());
 
        return socketServer;
    }
 
    template <typename SocketServer, typename... Args>
    SocketServer Server(const std::string& instanceName, Args&&... socketContextFactoryArgs) {
        return SocketServer(instanceName, std::forward<Args>(socketContextFactoryArgs)...);
    }
 
} // namespace core::socket::stream
 
#endif // CORE_SOCKET_STREAM_SOCKETSERVERNEW_H