SystemBalanceLoader.h

Go to the documentation of this file.

// Copyright 2024 libsese
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
 
 
#pragma once
 
#include "sese/event/Event.h"
#include "sese/net/ReusableSocket.h"
#include "sese/thread/Thread.h"
 
#include <atomic>
#include <functional>
#include <mutex>
 
namespace sese::service {
 
class SystemBalanceLoader final {
public:
    ~SystemBalanceLoader() noexcept;
 
    void setThreads(size_t th) noexcept;
 
    void setAddress(const net::IPAddress::Ptr &addr) noexcept { SystemBalanceLoader::address = addr; }
 
    void setAcceptTimeout(uint32_t to) noexcept { SystemBalanceLoader::timeout = to; }
 
    void setDispatchTimeout(uint32_t to) noexcept { SystemBalanceLoader::timeout = to; }
 
    [[nodiscard]] bool isStarted() const { return _isStart; }
 
    template<class SERVICE>
    bool init() noexcept;
 
    template<class SERVICE>
    bool init(std::function<SERVICE *()> creator) noexcept;
 
    void start() noexcept;
 
    void stop() noexcept;
 
protected:
    std::atomic_bool _isStart{false};
    std::atomic_bool _isStop{false};
 
    uint32_t timeout = 100;
    size_t threads{2};
    std::vector<socket_t> socketVector;
    std::vector<event::EventLoop *> eventLoopVector;
    std::vector<sese::Thread::Ptr> threadVector;
    sese::net::IPAddress::Ptr address = sese::net::IPv4Address::localhost(8080);
};
} // namespace sese::service
 
template<class SERVICE>
bool sese::service::SystemBalanceLoader::init() noexcept {
    return sese::service::SystemBalanceLoader::init<SERVICE>([]() -> SERVICE * { return new SERVICE; });
}
 
// GCOVR_EXCL_START
 
template<class SERVICE>
bool sese::service::SystemBalanceLoader::init(std::function<SERVICE *()> creator) noexcept {
    if (address == nullptr) return false;
 
    sese::net::ReusableSocket reusable_socket(address);
    for (size_t i = 0; i < threads; ++i) {
        auto sub_socket = reusable_socket.makeRawSocket();
        if (sub_socket == -1) {
            goto freeSocket;
        }
        if (0 != sese::net::Socket::setNonblocking(sub_socket)) {
            goto freeSocket;
        }
        if (0 != sese::net::Socket::listen(sub_socket, 32)) {
            goto freeSocket;
        }
        socketVector.emplace_back(sub_socket);
    }
 
    for (size_t i = 0; i < threads; ++i) {
        auto event = creator();
        if (event == nullptr) {
            goto freeEvent;
        }
        event->setListenFd(static_cast<int>(socketVector[i]));
        if (!event->init()) {
            delete event;
            goto freeEvent;
        } else {
            eventLoopVector.emplace_back(event);
        }
    }
 
    return true;
 
freeEvent:
    for (decltype(auto) event_loop: eventLoopVector) {
        delete event_loop;
    }
    eventLoopVector.clear();
 
freeSocket:
    for (decltype(auto) sub_socket: socketVector) {
        sese::net::Socket::close(sub_socket);
    }
    socketVector.clear();
    return false;
}
 
// GCOVR_EXCL_STOP