redis源码阅读之ae_kqueue

点击( 此处 )折叠或打开

1. typedef struct aeApiState {
2.      int kqfd ;
3.     struct kevent * events ;
4. } aeApiState ;

点击( 此处 )折叠或打开

1. static int aeApiCreate ( aeEventLoop * eventLoop ) {
2.     aeApiState * state = zmalloc ( sizeof ( aeApiState ) ) ;
4.      if ( ! state ) return - 1 ;
5.     state - > events = zmalloc ( sizeof ( struct kevent ) * eventLoop - > setsize ) ;
6.      if ( ! state - > events ) {
7.         zfree ( state ) ;
8.         return - 1 ;
9.      }
10.     state - > kqfd = kqueue ( ) ;
11.      if ( state - > kqfd = = - 1 ) {
12.         zfree ( state - > events ) ;
13.         zfree ( state ) ;
14.         return - 1 ;
15.      }
16.     eventLoop - > apidata = state ;
17.     return 0 ;
18. }
20. static int aeApiResize ( aeEventLoop * eventLoop , int setsize ) {
21.     aeApiState * state = eventLoop - > apidata ;
23.     state - > events = zrealloc ( state - > events , sizeof ( struct kevent ) * setsize ) ;
24.     return 0 ;
25. }
27. static void aeApiFree ( aeEventLoop * eventLoop ) {
28.     aeApiState * state = eventLoop - > apidata ;
30.     close ( state - > kqfd ) ;
31.     zfree ( state - > events ) ;
32.     zfree ( state ) ;
33. }
35. static int aeApiAddEvent ( aeEventLoop * eventLoop , int fd , int mask ) {
36.     aeApiState * state = eventLoop - > apidata ;
37.     struct kevent ke ;
39.      if ( mask & AE_READABLE ) {
40.         EV_SET ( & ke , fd , EVFILT_READ , EV_ADD , 0 , 0 , NULL ) ;
41.          if ( kevent ( state - > kqfd , & ke , 1 , NULL , 0 , NULL ) = = - 1 ) return - 1 ;
42.      }
43.      if ( mask & AE_WRITABLE ) {
44.         EV_SET ( & ke , fd , EVFILT_WRITE , EV_ADD , 0 , 0 , NULL ) ;
45.          if ( kevent ( state - > kqfd , & ke , 1 , NULL , 0 , NULL ) = = - 1 ) return - 1 ;
46.      }
47.     return 0 ;
48. }
50. static void aeApiDelEvent ( aeEventLoop * eventLoop , int fd , int mask ) {
51.     aeApiState * state = eventLoop - > apidata ;
52.     struct kevent ke ;
54.      if ( mask & AE_READABLE ) {
55.         EV_SET ( & ke , fd , EVFILT_READ , EV_DELETE , 0 , 0 , NULL ) ;
56.         kevent ( state - > kqfd , & ke , 1 , NULL , 0 , NULL ) ;
57.      }
58.      if ( mask & AE_WRITABLE ) {
59.         EV_SET ( & ke , fd , EVFILT_WRITE , EV_DELETE , 0 , 0 , NULL ) ;
60.         kevent ( state - > kqfd , & ke , 1 , NULL , 0 , NULL ) ;
61.      }
62. }
63. static char * aeApiName ( void ) {
64.     return "kqueue" ;
65. }

点击( 此处 )折叠或打开

1. static int aeApiPoll ( aeEventLoop * eventLoop , struct timeval * tvp ) {
2.     aeApiState * state = eventLoop - > apidata ;
3.      int retval , numevents = 0 ;
5.      if ( tvp ! = NULL ) {
6.         struct timespec timeout ;
7.         timeout . tv_sec = tvp - > tv_sec ;
8.         timeout . tv_nsec = tvp - > tv_usec * 1000 ;
9.         retval = kevent ( state - > kqfd , NULL , 0 , state - > events , eventLoop - > setsize ,
10.                          & timeout ) ;
11.      } else {
12.         retval = kevent ( state - > kqfd , NULL , 0 , state - > events , eventLoop - > setsize ,
13.                          NULL ) ;
14.      }
16.      if ( retval > 0 ) {
17.          int j ;
19.         numevents = retval ;
20.          for ( j = 0 ; j < numevents ; j + + ) {
21.              int mask = 0 ;
22.             struct kevent * e = state - > events + j ;
24.              if ( e - > filter = = EVFILT_READ ) mask | = AE_READABLE ;
25.              if ( e - > filter = = EVFILT_WRITE ) mask | = AE_WRITABLE ;
26.             eventLoop - > fired [ j ] . fd = e - > ident ;
27.             eventLoop - > fired [ j ] . mask = mask ;
28.          }
29.      }
30.     return numevents ;
31. }