网络编程(15)——服务器如何主动退出

十五、day15

服务器主动退出一直是服务器设计必须考虑的一个方向,旨在能通过捕获信号使服务器安全退出。我们可以通过asio提供的信号机制绑定回调函数即可实现优雅退出。

之前服务器的主函数如下

cpp 复制代码
#include "CSession.h"
#include "CServer.h"

int main()
{
    try {
        boost::asio::io_context ioc;
        CServer s(ioc, 10086);
        ioc.run();
    }
    catch (std::exception& e) {
        std::cerr << "Exception: " << e.what() << '\n';
    }
    boost::asio::io_context io_context;
}

有两种修改方式,第一种基于condition_variable和lock_quit;第二种基于asio库

1)第一种实现方式

cpp 复制代码
#include "CSession.h"
#include "CServer.h"
#include <csignal>
#include <thread>
#include <mutex>

bool bstop = false;
std::condition_variable cond_quit;
std::mutex mutex_quit;

void sig_handler(int sig) {
    if (sig == SIGINT or sig == SIGTERM) {
        std::unique_lock<std::mutex> lock_quit(mutex_quit);
        bstop = true;
        cond_quit.notify_one();
    }
}

int main()
{
    try {
        boost::asio::io_context ioc;
        std::thread net_work_thread([&ioc] { // 使用子线程运行网络服务
            CServer s(ioc, 10086);
            ioc.run();
            });
        // 主线程执行以下操作
        // 捕获信号,并执行信号对应的回调函数
        signal(SIGINT, sig_handler);
        signal(SIGTERM, sig_handler);
        // 循环会占用资源,为了防止占用资源我们选择挂起
        while (!bstop) {
            std::unique_lock<std::mutex> lock_quit(mutex_quit);
            cond_quit.wait(lock_quit);
        }

        ioc.stop();
        net_work_thread.join();
    }
    catch (std::exception& e) {
        std::cerr << "Exception: " << e.what() << '\n';
    }
    boost::asio::io_context io_context;
}
  • 条件变量、关服标志、锁必须都是全局变量

  • 自定义关服回调函数sig_handler一旦捕获到SIGINT(Ctrl+C触发)或SIGTERM(终止进程信号),系统调用sig_handler函数。在回调函数内首先进行加锁,以免重复操作,然后将关服标志位置为true,并将主线程唤醒,'}'结束后自动解锁

  • 启动子线程运行网络服务

  • 主线程种设置信号处理函数并将主线程挂起,等待退出信号(循环会占用cpu资源,这里通过挂起避免循环,节省资源)

  • 当主线程被唤醒后,退出循环,执行ioc.stop() 来停止io_context,并将子线程中的ioc.run()返回

  • net_work_thread.join()确保子线程在主线程退出之前先结束运行,防止资源泄露

注:信号处理是异步的,即使主线程在等待条件变量,也可以被信号打断,执行信号处理函数

2)第二种实现方式

cpp 复制代码
int main()
{
    try {
        boost::asio::io_context ioc;
        boost::asio::signal_set signals(ioc, SIGINT, SIGTERM);
        // 必须异步等待,否则建立线程进行处理
        signals.async_wait([&ioc](auto, auto) {
            ioc.stop();
            });

        CServer s(ioc, 10086);
        ioc.run();
    }
    catch (std::exception& e) {
        std::cerr << "Exception: " << e.what() << '\n';
    }
    boost::asio::io_context io_context;
}

第二种方式是我推荐的,最优雅最简单。

主要通过asio库的signal_set 函数执行信号处理操作,允许用户在指定的io_context 上异步等待这些信号的到来,并在信号到达时执行相应的回调

  • 异步处理: 通过将信号处理与 io_context 结合,程序能够继续处理其他异步任务,而不需要在信号到达时阻塞主线程。

  • 多线程支持: 如果程序使用多线程,信号的处理仍然可以有效地协调各个线程的操作。

3)测试

两种方式均能得到以下测试结果

相关推荐
珹洺1 小时前
C++从入门到实战(十)类和对象(最终部分)static成员,内部类,匿名对象与对象拷贝时的编译器优化详解
java·数据结构·c++·redis·后端·算法·链表
写bug的小屁孩1 小时前
移动零+复写零+快乐数+盛最多水的容器+有效三角形的个数
c++·算法·双指针
DARLING Zero two♡1 小时前
C++底层学习精进:模板进阶
开发语言·c++·模板
勘察加熊人2 小时前
c++生成html文件helloworld
开发语言·c++·html
羑悻的小杀马特2 小时前
【狂热算法篇】探寻图论幽径:Bellman - Ford 算法的浪漫征程(通俗易懂版)
c++·算法·图论·bellman_ford算法
basketball6166 小时前
C++ STL常用算法之常用排序算法
c++·算法·排序算法
moz与京6 小时前
[附C++,JS,Python题解] Leetcode 面试150题(10)——轮转数组
c++·python·leetcode
三金C_C8 小时前
单例模式解析
单例模式·设计模式·线程锁
晚雾也有归处8 小时前
链表(C++)
数据结构·c++·链表
勘察加熊人8 小时前
c++实现录音系统
开发语言·c++