Redis源碼分析（二十一）--- anet網絡通信的封裝

昨天非常輕松的分析完Redis的事件驅動模型之後，今天我來看看anet的代碼，anet是Redis對于Client/Server的網絡操作的一個小小封裝。代碼中對此檔案的官方解釋為:

/* anet.c -- Basic TCP socket stuff made a bit less boring  

* 基于簡單的基本TCP的socket連接配接  

複制代碼

後面的made a bit less boring這在這裡表示啥意思，就讓我有點費解了，不過前面的是重點，Basic TCP Socket，基于的是TCP 協定的socket連接配接。anet.h的API如下:

int anetTcpConnect(char *err, char *addr, int port); /* TCP的預設連接配接 */  

int anetTcpNonBlockConnect(char *err, char *addr, int port); /* TCP的非阻塞連接配接 */  

int anetUnixConnect(char *err, char *path); /* anet的Unix方式的預設連接配接方式 */  

int anetUnixNonBlockConnect(char *err, char *path); /* anet的Unix方式的非阻塞連接配接方式 */  

int anetRead(int fd, char *buf, int count); /* anet網絡讀取檔案到buffer中操作 */  

int anetResolve(char *err, char *host, char *ipbuf, size_t ipbuf_len); /* 解析所有的東西 */  

int anetResolveIP(char *err, char *host, char *ipbuf, size_t ipbuf_len); /* 單單解析IP的位址 */  

int anetTcpServer(char *err, int port, char *bindaddr, int backlog);  

int anetTcp6Server(char *err, int port, char *bindaddr, int backlog);  

int anetUnixServer(char *err, char *path, mode_t perm, int backlog);  

int anetTcpAccept(char *err, int serversock, char *ip, size_t ip_len, int *port);  

int anetUnixAccept(char *err, int serversock);  

int anetWrite(int fd, char *buf, int count); /* anet通過網絡從buffer中寫入檔案操作 */  

int anetNonBlock(char *err, int fd); /* anet設定非阻塞的方法 */  

int anetEnableTcpNoDelay(char *err, int fd); /* 啟用TCP沒有延遲 */  

int anetDisableTcpNoDelay(char *err, int fd); /* 禁用TCP連接配接沒有延遲 */  

int anetTcpKeepAlive(char *err, int fd); /* 設定TCP保持活躍連接配接狀态。适用于所有系統 */  

int anetPeerToString(int fd, char *ip, size_t ip_len, int *port);  

int anetKeepAlive(char *err, int fd, int interval); /* 設定TCP連接配接一直存活，用來檢測已經死去的結點，interval選項隻适用于Linux下的系統 */  

int anetSockName(int fd, char *ip, size_t ip_len, int *port);  

我們還是能夠看到很多熟悉的方法的，read，write，accept.connect等在任何程式設計語言中都會看到的一些方法。看完這個anet,最直覺的感覺就是作者編寫的這個網絡操作庫就是對于C語言系統網絡庫的又一次簡答的封裝，因為裡面都是直接調用庫方法的函數實作。作者根據自己業務的需要在上面做了小小的封裝。比如說非阻塞的設定；

/* anet設定非阻塞的方法 */  

int anetNonBlock(char *err, int fd)  

{  

    int flags;  

    /* Set the socket non-blocking. 

     * Note that fcntl(2) for F_GETFL and F_SETFL can't be 

     * interrupted by a signal. */  

    if ((flags = fcntl(fd, F_GETFL)) == -1) {  

        anetSetError(err, "fcntl(F_GETFL): %s", strerror(errno));  

        return ANET_ERR;  

    }  

    //調用fcntl方法設定非阻塞方法  

    if (fcntl(fd, F_SETFL, flags | O_NONBLOCK) == -1) {  

        anetSetError(err, "fcntl(F_SETFL,O_NONBLOCK): %s", strerror(errno));  

    return ANET_OK;  

}  

fcntl方法是直接起作用的方法。在整個網絡操作檔案的中，讓我感覺稍有亮點的還是有一些地方的

(1).能設定BLOCK連接配接還是NONE_BLOCKED方式的connect；

/* TCP的預設連接配接 */  

int anetTcpConnect(char *err, char *addr, int port)  

    return anetTcpGenericConnect(err,addr,port,ANET_CONNECT_NONE);  

/* TCP的非阻塞連接配接 */  

int anetTcpNonBlockConnect(char *err, char *addr, int port)  

    return anetTcpGenericConnect(err,addr,port,ANET_CONNECT_NONBLOCK);  

(2).能設定連接配接的Delay的延時與否。:

/* 設定TCP連接配接是否NODelay沒有延遲 */  

static int anetSetTcpNoDelay(char *err, int fd, int val)  

    if (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &val, sizeof(val)) == -1)  

    {  

        anetSetError(err, "setsockopt TCP_NODELAY: %s", strerror(errno));  

/* 啟用TCP沒有延遲 */  

int anetEnableTcpNoDelay(char *err, int fd)  

    return anetSetTcpNoDelay(err, fd, 1);  

/* 禁用TCP連接配接沒有延遲 */  

int anetDisableTcpNoDelay(char *err, int fd)  

    return anetSetTcpNoDelay(err, fd, 0);  

也許在有些情況下對延時要求比較高，就不能有延時。

(3).對ip位址有ipv4和ipv6位址不同的處理方法。這個作者想得還是非常全面的。在對位址做resolve解析的時候就考慮到了這個問題:

/* anetGenericResolve() is called by anetResolve() and anetResolveIP() to 

* do the actual work. It resolves the hostname "host" and set the string 

* representation of the IP address into the buffer pointed by "ipbuf". 

* 

* If flags is set to ANET_IP_ONLY the function only resolves hostnames 

* that are actually already IPv4 or IPv6 addresses. This turns the function 

* into a validating / normalizing function. */  

/* 解析的泛型方法，可以根據條件解析host主機名或IP位址 */  

int anetGenericResolve(char *err, char *host, char *ipbuf, size_t ipbuf_len,  

                       int flags)  

    struct addrinfo hints, *info;  

    int rv;  

    memset(&hints,0,sizeof(hints));  

    if (flags & ANET_IP_ONLY) hints.ai_flags = AI_NUMERICHOST;  

    hints.ai_family = AF_UNSPEC;  

    hints.ai_socktype = SOCK_STREAM;  /* specify socktype to avoid dups */  

    //解析hostName  

    if ((rv = getaddrinfo(host, NULL, &hints, &info)) != 0) {  

        anetSetError(err, "%s", gai_strerror(rv));  

    //根據類型解析ipV4的位址還是ipV6的位址  

    if (info->ai_family == AF_INET) {  

        struct sockaddr_in *sa = (struct sockaddr_in *)info->ai_addr;  

        inet_ntop(AF_INET, &(sa->sin_addr), ipbuf, ipbuf_len);  

    } else {  

        struct sockaddr_in6 *sa = (struct sockaddr_in6 *)info->ai_addr;  

        inet_ntop(AF_INET6, &(sa->sin6_addr), ipbuf, ipbuf_len);  

    freeaddrinfo(info);  

還有一些常見的方法，與我們平時寫代碼時用的手法基本一樣，比如accept()的方法：

/* socket連接配接操作 */  

static int anetGenericAccept(char *err, int s, struct sockaddr *sa, socklen_t *len) {  

    int fd;  

    while(1) {  

        //通過while循環等待連接配接  

        fd = accept(s,sa,len);  

        if (fd == -1) {  

            if (errno == EINTR)  

                continue;  

            else {  

                anetSetError(err, "accept: %s", strerror(errno));  

                return ANET_ERR;  

            }  

        }  

        break;  

    return fd;  

本文作者：HarLock

本文來自雲栖社群合作夥伴rediscn，了解相關資訊可以關注redis.cn網站。