wret instead of ret var

[brisk.git] / web / spush / brisk-spush.php

diff --git a/web/spush/brisk-spush.php b/web/spush/brisk-spush.php
index 28088f3..788de8c 100755 (executable)
--- a/web/spush/brisk-spush.php
+++ b/web/spush/brisk-spush.php
@@ -243,10 +243,14 @@ function main()
                                 $header_out = array();
                                 if (!isset($cookie['sess'])
                                     || (($user = $room->get_user($cookie['sess'], $idx)) == FALSE)) {
-                                    $body = index_rd_ifra_fini(TRUE);
-                                    fwrite($new_socket, headers_render($header_out).$body);
-                                    fflush($new_socket);
-                                    fclose($new_socket);
+                                    $content = index_rd_ifra_fini(TRUE);
+                                    
+                                    $pgflush = new PageFlush($new_socket, $curtime, 20, $header_out, $content);
+
+                                    if ($pgflush->try_flush($curtime) == FALSE) {
+                                        // Add $pgflush to the pgflush array
+                                        array_push($pages_flush, $pgflush);
+                                    }
                                     break;
                                 }
                                 // close a previous opened index_read_ifra socket, if exists
@@ -258,9 +262,18 @@ function main()
                                     $user->rd_socket_set(NULL);
                                 }
 
-                                $body = "";
-                                index_rd_ifra_init($room, $user, $header_out, $body, $get, $post, $cookie);
-                                fwrite($new_socket, headers_render($header_out).$body);
+                                $content = "";
+                                index_rd_ifra_init($room, $user, $header_out, $content, $get, $post, $cookie);
+                                $content_l = mb_strlen($content, "ASCII");
+
+                                $wret = @fwrite($new_socket, headers_render($header_out).$content);
+                                if ($wret < $content_l) {
+                                    printf("TROUBLES WITH FWRITE: %d\n", $wret);
+                                    $user->rd_cache_set(mb_substr($content, $wret, $content_l - $wret, "ASCII"));
+                                }
+                                else {
+                                    $user->rd_cache_set("");
+                                }
                                 fflush($new_socket);
 
                                 $s2u[intval($new_socket)] = $idx;
@@ -331,21 +344,21 @@ function main()
         foreach ($socks as $k => $sock) {
             if (isset($s2u[intval($sock)])) {
                 $user = $room->user[$s2u[intval($sock)]];
-                $body = $user->rd_cache_get();
-                if ($body == "")
-                    index_rd_ifra_main($room, $user, $body);
+                $content = $user->rd_cache_get();
+                if ($content == "")
+                    index_rd_ifra_main($room, $user, $content);
 
-                if ($body == "" && $user->rd_kalive_is_expired($curtime)) {
-                    $body = index_rd_ifra_keepalive($user);
+                if ($content == "" && $user->rd_kalive_is_expired($curtime)) {
+                    $content = index_rd_ifra_keepalive($user);
                 }
 
-                if ($body != "") {
-                    echo "SPIA: [".substr($body, 0, 60)."...]\n";
-                    $body_l = mb_strlen($body, "ASCII");
-                    $ret = @fwrite($sock, $body);
-                    if ($ret < $body_l) {
-                        printf("TROUBLE WITH FWRITE: %d\n", $ret);
-                        $user->rd_cache_set(mb_substr($body, $ret, $body_l - $ret, "ASCII"));
+                if ($content != "") {
+                    echo "SPIA: [".substr($content, 0, 60)."...]\n";
+                    $content_l = mb_strlen($content, "ASCII");
+                    $wret = @fwrite($sock, $content);
+                    if ($wret < $content_l) {
+                        printf("TROUBLE WITH FWRITE: %d\n", $wret);
+                        $user->rd_cache_set(mb_substr($content, $wret, $content_l - $wret, "ASCII"));
                     }
                     else {
                         $user->rd_cache_set("");