用libevent库实现一个简单的高性能服务器

#include <event2/event.h>
#include <event2/listener.h>
#include <event2/bufferevent.h>

// 回调函数，处理接收到的数据
static void echo_read_cb(struct bufferevent* bev, void* ctx) {
    struct evbuffer* input = bufferevent_get_input(bev);
    struct evbuffer* output = bufferevent_get_output(bev);

    // 将接收到的数据写回客户端
    evbuffer_add_buffer(output, input);
}

// 回调函数，处理连接事件
static void accept_conn_cb(struct evconnlistener* listener, evutil_socket_t fd, struct sockaddr* address, int socklen, void* ctx) {
    struct event_base* base = evconnlistener_get_base(listener);
    struct bufferevent* bev = bufferevent_socket_new(base, fd, BEV_OPT_CLOSE_ON_FREE);

    // 设置回调函数，处理读取事件
    bufferevent_setcb(bev, echo_read_cb, NULL, NULL, NULL);

    // 启用读取事件
    bufferevent_enable(bev, EV_READ | EV_WRITE);
}

// 回调函数，处理错误事件
static void accept_error_cb(struct evconnlistener* listener, void* ctx) {
    struct event_base* base = evconnlistener_get_base(listener);
    int err = EVUTIL_SOCKET_ERROR();
    fprintf(stderr, "Error %d (%s) on the listener. Shutting down.\n", err, evutil_socket_error_to_string(err));

    // 以正常退出码关闭程序
    event_base_loopexit(base, NULL);
}

int main() {
    // 初始化libevent
    struct event_base* base = event_base_new();
    if (!base) {
        fprintf(stderr, "Could not initialize libevent!\n");
        return 1;
    }

    // 创建一个监听器，监听在本地8888端口
    struct sockaddr_in sin;
    memset(&sin, 0, sizeof(sin));
    sin.sin_family = AF_INET;
    sin.sin_port = htons(8888);

    struct evconnlistener* listener = evconnlistener_new_bind(
        base,
        accept_conn_cb,
        NULL,
        LEV_OPT_CLOSE_ON_FREE | LEV_OPT_REUSEABLE,
        -1,
        (struct sockaddr*)&sin,
        sizeof(sin));

    if (!listener) {
        fprintf(stderr, "Could not create listener!\n");
        return 1;
    }

    // 设置监听器的错误回调函数
    evconnlistener_set_error_cb(listener, accept_error_cb);

    // 进入事件循环
    event_base_dispatch(base);

    // 清理资源
    evconnlistener_free(listener);
    event_base_free(base);

    return 0;
}