网络编程(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)测试

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

相关推荐
wayuncn19 分钟前
月付物理服务器租用平台-青蛙云
运维·服务器·服务器租用·服务器托管·物理机租用
望获linux21 分钟前
【实时Linux实战系列】CPU 隔离与屏蔽技术
java·linux·运维·服务器·操作系统·开源软件·嵌入式软件
0wioiw036 分钟前
C#基础(项目结构和编译运行)
linux·运维·服务器
秋风&萧瑟1 小时前
【C++】C++中的友元函数和友元类
c++
2401_873587822 小时前
Linux常见指令以及权限理解
linux·运维·服务器
梁诚斌2 小时前
使用OpenSSL接口读取pem编码格式文件中的证书
开发语言·c++
晨曦丿3 小时前
双11服务器
linux·服务器·网络
wanhengidc4 小时前
UDP服务器主要是指什么意思?
服务器·网络协议·udp
李迟4 小时前
在Linux服务器上使用kvm创建虚拟机
java·linux·服务器
从后端到QT4 小时前
SRS流媒体服务器(8)源码分析之rtc/rtmp互相转码详解
运维·服务器·实时音视频