PHP libevent实现聊天室

点击(此处)折叠或打开

1. <?php
   
2. //守护进程化
   
3. daemonize();
   
4. 
   
5. $socket = stream_socket_server ('tcp://127.0.0.1:9998', $errno, $errstr);
   
6. stream_set_blocking($socket, 0);
   
7. $base = event_base_new();
   
8. $event = event_new();
   
9. event_set($event, $socket, EV_READ | EV_PERSIST, 'ev_accept', $base);
   
10. event_base_set($event, $base);
    
11. event_add($event);
    
12. event_base_loop($base);
    
13. 
    
14. $GLOBALS['connections'] = array();
    
15. $GLOBALS['buffers'] = array();
    
16. 
    
17. function ev_accept($socket, $flag, $base) {
    
18.     static $id = 0;
    
19. 
    
20.     $connection = stream_socket_accept($socket);
    
21.     stream_set_blocking($connection, 0);
    
22. 
    
23.     $id += 1;
    
24. 
    
25.     $buffer = event_buffer_new($connection, 'ev_read', NULL, 'ev_error', $id);
    
26.     event_buffer_base_set($buffer, $base);
    
27.     event_buffer_timeout_set($buffer, 30, 30);
    
28.     event_buffer_watermark_set($buffer, EV_READ, 0, 0xffffff);
    
29.     event_buffer_priority_set($buffer, 10);
    
30.     event_buffer_enable($buffer, EV_READ | EV_PERSIST);
    
31. 
    
32.     // we need to save both buffer and connection outside
    
33.     $GLOBALS['connections'][$id] = $connection;
    
34.     $GLOBALS['buffers'][$id] = $buffer;
    
35. }
    
36. 
    
37. function ev_error($buffer, $error, $id) {
    
38.     event_buffer_disable($GLOBALS['buffers'][$id], EV_READ | EV_WRITE);
    
39.     event_buffer_free($GLOBALS['buffers'][$id]);
    
40.     fclose($GLOBALS['connections'][$id]);
    
41.     unset($GLOBALS['buffers'][$id], $GLOBALS['connections'][$id]);
    
42. }
    
43. 
    
44. function ev_read($buffer, $id) {
    
45.     while ($read = event_buffer_read($buffer, 256)) {
    
46.         foreach( $GLOBALS['connections'] as $ids => $conn ) {
    
47.                 if( $ids != $id ) {
    
48.                         fwrite($conn,$read);
    
49.                 }
    
50.         }
    
51.     }
    
52. }
    
53. 
    
54. 
    
55. function daemonize()
    
56. {
    
57.         $pid = pcntl_fork();
    
58.         if ($pid == -1)
    
59.         {
    
60.                 die("fork(1) failed!\n");
    
61.         }
    
62.         elseif ($pid > 0)
    
63.         {
    
64.                 //让由用户启动的进程退出
    
65.                 exit(0);
    
66.         }
    
67. 
    
68.         //建立一个有别于终端的新session以脱离终端
    
69.         posix_setsid();
    
70. 
    
71.         $pid = pcntl_fork();
    
72.         if ($pid == -1)
    
73.         {
    
74.                 die("fork(2) failed!\n");
    
75.         }
    
76.         elseif ($pid > 0)
    
77.         {
    
78.                 //父进程退出, 剩下子进程成为最终的独立进程
    
79.                 exit(0);
    
80.         }
    
81. }
    
82. 
    
83. ?>