I/O 多路复用之 select

#include <sys/select.h>

# 清除掉该描述符
 void FD_CLR(fd, fd_set *fdset);

# 拷贝描述符集合
 void FD_COPY(fd_set *fdset_orig, fd_set *fdset_copy);

# 常用 判断该描述符是不是准备好了。Linux返回0或1。Mac表现不一样，返回位的十进制大小，比如fd=3在则返回8
 int FD_ISSET(fd, fd_set *fdset);

# 常用 把关注的描述符放到集合里
 void FD_SET(fd, fd_set *fdset);

# 清空描述符集合
 void FD_ZERO(fd_set *fdset);

int select(int nfds, fd_set *restrict readfds, fd_set *restrict writefds, fd_set *restrict errorfds, struct timeval *restrict timeout);

// 数组元素是一个long int类型的
typedef long int __fd_mask;

// 一个long int8字节，NFDBITS=8*8=64bit
#define __NFDBITS (8 * (int) sizeof (__fd_mask))

// 有的平台把它当32
# define NFDBITS __NFDBITS

// 忽略了一些兼容问题
typedef struct {
  __fd_mask __fds_bits[__FD_SETSIZE / __NFDBITS];
# define __FDS_BITS(set) ((set)->__fds_bits)
} fd_set;

typedef struct {
    long fds_bits[1024/64];
} fd_set;

#include <stdio.h>
#include <stdlib.h>
#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>
#include <arpa/inet.h>

int main(void) {
    fd_set rfds;
    struct timeval tv;
    int retval;

    /* 查看常量值 FD_SETSIZE  NFDBITS */
    printf("FD_SETSIZE=%d，NFDBITS=%lu， FD_SETSIZE / NFDBITS=%lu\n", FD_SETSIZE, NFDBITS, FD_SETSIZE / NFDBITS);

    /* 下面是连接服务端代码 */

    // 套接字文件描述符
    int clientSocket;
    // 描述服务器的socket
    struct sockaddr_in serverAddr;
    char recvbuf[200];
    int dataLen;

    clientSocket = socket(AF_INET, SOCK_STREAM, 0);
    // 客户端打开的网络socket描述符= 3 （0 1 2是标准输入输出错误，进程下文件描述符从小到大）
    printf("clientSocket: %d\n", clientSocket);

    serverAddr.sin_family = AF_INET;
    serverAddr.sin_port = htons(8090);
    serverAddr.sin_addr.s_addr = inet_addr("127.0.0.1");
    // connect服务端
    connect(clientSocket, (struct sockaddr *) &serverAddr, sizeof(serverAddr));

    /* 连接服务端结束 */

    /* 监视标准输入描述符，看什么时候有输入进来*/
    FD_ZERO(&rfds);// 清空描述符集合
    FD_SET(0, &rfds);// 把标准输入放进去
    FD_SET(clientSocket, &rfds);//

    /* 等待5s */
    tv.tv_sec = 5;
    tv.tv_usec = 0;

    // select 调用
    retval = select(clientSocket + 1, &rfds, NULL, NULL, &tv);

    /* Mac系统下运行tv不受改变；Linux下运行改变了time参数，tv.tv_sec是0 */
    printf("time %ld\n", tv.tv_sec);

    if (retval == -1){
        perror("select()");
    } else if (retval){
        // 标准输入准备就绪
        if(FD_ISSET(0, &rfds)){
            printf("Std Input is available now. retval=%d\n", retval);
        }
        // 网络socket准备就绪
        if(FD_ISSET(clientSocket, &rfds)){
            recvbuf[0] = '\0';
            dataLen = recv(clientSocket, recvbuf, 200, 0);
            printf("dataLen: %d, recvbuf: %s\n", dataLen, recvbuf);
            close(clientSocket);
        }
    } else {
        printf("No data within five seconds.\n");
    }
    exit(EXIT_SUCCESS);
}

package main

import (
  "time"
  "net"
)

func main(){
  listen, _ := net.Listen("tcp", "localhost:8090")
  for{
    conn, _ := listen.Accept()
    go handlefunc(conn)
  }
}

/**
 间隔两秒后，发一条消息给客户端
 忽略错误
 */
func handlefunc(conn net.Conn){
  defer conn.Close()
  time.Sleep(2*time.Second)
  str := "send a string to client " + conn.RemoteAddr().String()
  conn.Write([]byte(str))
}