libevent从入门到掌握三

libevent从入门到掌握<三>

案例

利用最简单的socket通信来看一下前面的例子：

代码很简单，不做过多的说明：

cli.cpp:

#include<iostream>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<errno.h>
#include<unistd.h>
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<event.h>
#include<event2/util.h>
 
using namespace std;
 
typedef struct sockaddr cli;
 
void send_msg_cb(int fd, short events, void* arg)
{
    char msg[1024] = {};

	
    int ret = read(fd, msg, sizeof(msg));
    if( ret <= 0 )
    {
        perror("read fail ");
        exit(1);
    }
	//cout<<"msg:"<<msg<<endl;
	
    int sockfd = *((int*)arg);
 
    write(sockfd, msg, ret);
	cout<<"write:"<<msg<<endl;
}
 
 
void read_msg_cb(int fd, short events, void *arg)
{
    char msg1[1024];
 
    int len = read(fd, msg1, sizeof(msg1)-1);
    if( len <= 0 )
    {
        perror("read error!!!");
        exit(1);
    }
    msg1[len] = '\0';
 
    cout<<"read is ok,msg:"<<msg1<<endl;
}
 
 
 

int connect_ser(const char* server_ip, const int port)
{
    int sockfd, status;
    struct sockaddr_in sock;
 
    memset(&sock, 0, sizeof(sock) );
 
    sock.sin_family = AF_INET;
    sock.sin_port = htons(port);
    status = inet_aton(server_ip, &sock.sin_addr);
 
    if( status == 0 )
    {
        errno = EINVAL;
        return -1;
    }
	//创建套接字
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if( -1 == sockfd )
	{
		perror("socket is error");
		exit(-1);
	} 
 
    status = connect(sockfd, (cli*)&sock, sizeof(sock) );
 
    if( status == -1 )
    {
		perror("connect");
        close(sockfd);
        return -1;
    }
 
    evutil_make_socket_nonblocking(sockfd);
 
    return sockfd;
}
 
int main(int argc, char** argv)
{
    if( argc < 3 )
    {
        cout<<"please input two param:"<<endl;
        return -1;
    }

    int sockfd = connect_ser(argv[1], atoi(argv[2]));
    if( sockfd == -1)
    {
		cout<<"connect is error"<<endl;
        return -1;
    }
	//初始化base
    struct event_base* base = event_base_new();
 
    struct event *ev1 = event_new(base, sockfd,
                                        EV_READ | EV_PERSIST,
                                        read_msg_cb, NULL);
    event_add(ev1, NULL);
	//键盘输入
    struct event* ev2 = event_new(base, STDIN_FILENO,
                                      EV_READ | EV_PERSIST, send_msg_cb,
                                      (void*)&sockfd);
 
 
    event_add(ev2, NULL);
 
    event_base_dispatch(base);
 
    cout<<"ending>>>>>"<<endl;
    return 0;
}
 

ser.cpp

#include<stdio.h>
#include<string.h>
#include<errno.h>
#include<iostream>
#include<unistd.h>
#include<event.h>
 
using namespace std;

void accept_cb(int fd, short events, void* arg)
{
    evutil_socket_t sockfd;
 
    struct sockaddr_in client;
    socklen_t len = sizeof(cli);
 
    sockfd = accept(fd, (struct sockaddr*)&cli, &len );
    evutil_make_socket_nonblocking(sockfd);
 
    printf("accept a cli %d\n", sockfd);
 
    struct event_base* base = (event_base*)arg;
 
    //仅仅是为了动态创建一个event结构体
    struct event *ev = event_new(NULL, -1, 0, NULL, NULL);
    //将动态创建的结构体作为event的回调参数
    event_assign(ev, base, sockfd, EV_READ | EV_PERSIST,
                 socket_read_cb, (void*)ev);
 
    event_add(ev, NULL);
}
 
 
void socket_read_cb(int fd, short events, void *arg)
{
    char msg[4096];
    struct event *ev = (struct event*)arg;
    int len = read(fd, msg, sizeof(msg) - 1);
 
 
 
    if( len <= 0 )
    {
        printf("some error happen when read\n");
        event_free(ev);
        close(fd);
        return ;
    }
 
    msg[len] = '\0';
    printf("recv the cli msg: %s", msg);
 
    char reply_msg[4096] = "I have recvieced the msg: ";
    strcat(reply_msg + strlen(reply_msg), msg);
 
    write(fd, reply_msg, strlen(reply_msg) );
}
 
 
 
typedef struct sockaddr SA;
int tcp_server_init(int port, int listen_num)
{
    int errno_save;
    evutil_socket_t listener;
 
    listener = socket(AF_INET, SOCK_STREAM, 0);
    if( listener == -1 )
        return -1;
 
    //允许多次绑定同一个地址。要用在socket和bind之间
    evutil_make_listen_socket_reuseable(listener);
 
    struct sockaddr_in sin;
    sin.sin_family = AF_INET;
    sin.sin_addr.s_addr = 0;
    sin.sin_port = htons(port);
 
    if( ::bind(listener, (SA*)&sin, sizeof(sin)) < 0 )
        goto error;
 
    if( ::listen(listener, listen_num) < 0)
        goto error;
 
 
    //跨平台统一接口，将套接字设置为非阻塞状态
    evutil_make_socket_nonblocking(listener);
 
    return listener;
 
    error:
        errno_save = errno;
        evutil_closesocket(listener);
        errno = errno_save;
 
        return -1;
}

int main(int argc, char** argv)
{
 
    int listener = tcp_server_init(9999, 10);
    if( listener == -1 )
    {
        perror(" tcp_server_init error ");
        return -1;
    }
 
    struct event_base* base = event_base_new();
 
    //添加监听客户端请求连接事件
    struct event* ev_listen = event_new(base, listener, EV_READ | EV_PERSIST,
                                        accept_cb, base);
    event_add(ev_listen, NULL);
 
 
    event_base_dispatch(base);
 
    return 0;
}
 
 

makefile:

all:ser1 cli1

ser1:ser1.cpp
	g++ ser1.cpp -o ser1 -levent

cli1:cli1.cpp
	g++ cli1.cpp -o cli1 -levent

clean:
	rm -f ser1 cli1

这两断代码主要就是使用了前面的event事件进行通信；和之前socket通信是一样的，只是增加的注册函数event，不需要循环检测消息的到来，这些工作被event替代了，是不是感觉非常方便；

之后libevent还封装了很多优秀的接口；后面合并大家一起学习！！！

---

想了解学习更多C++后台服务器方面的知识，请关注：
微信公众号：C++后台服务器开发

---