Changeset 49212

Timestamp:

10/19/2020 09:49:58 PM (4 years ago)

Author:

helen

Message:

Multisite: More specific caching for get_dirsize.

Instead of one cache entry for all upload folders for a site on multisite, this now caches for each folder and invalidates that cache based on context. In multisite, this should speed up get_dirsize calls since older directories that are much less likely to change will no longer have the size recalculated.

Props janthiel, A5hleyRich, batmoo.
Fixes #19879.

Location:

trunk

Files:

• src/wp-admin/includes/file.php (modified) (1 diff)
• src/wp-includes/functions.php (modified) (5 diffs)
• src/wp-includes/post.php (modified) (1 diff)
• tests/phpunit/tests/multisite/dirsizeCache.php (added)

trunk/src/wp-admin/includes/file.php

@@ -929 +929 @@
 
     if ( is_multisite() ) {
-        delete_transient( 'dirsize_cache' );
+         );
     }
 

trunk/src/wp-includes/functions.php

@@ -2740 +2740 @@
     // Compute the URL.
     $url = $upload['url'] . "/$filename";
+
+
+
+
 
     /** This filter is documented in wp-admin/includes/file.php */
@@ -7561 +7565 @@
  */
 function get_dirsize( $directory, $max_execution_time = null ) {
-    $dirsize = get_transient( 'dirsize_cache' );
-
-    if ( is_array( $dirsize ) && isset( $dirsize[ $directory ]['size'] ) ) {
-        return $dirsize[ $directory ]['size'];
-    }
-
-    if ( ! is_array( $dirsize ) ) {
-        $dirsize = array();
-    }
 
     // Exclude individual site directories from the total when checking the main site of a network,
     // as they are subdirectories and should not be counted.
     if ( is_multisite() && is_main_site() ) {
-        $dirsize[ $directory ]['size'] = recurse_dirsize( $directory, $directory . '/sites', $max_execution_time );
+        $ = recurse_dirsize( $directory, $directory . '/sites', $max_execution_time );
     } else {
-        $dirsize[ $directory ]['size'] = recurse_dirsize( $directory, null, $max_execution_time );
-    }
-
-    set_transient( 'dirsize_cache', $dirsize, HOUR_IN_SECONDS );
-    return $dirsize[ $directory ]['size'];
+        $size = recurse_dirsize( $directory, null, $max_execution_time );
+    }
+
+    return $size;
 }
 
@@ -7592 +7586 @@
  * @since 4.3.0 $exclude parameter added.
  * @since 5.2.0 $max_execution_time parameter added.
+
  *
  * @param string       $directory          Full path of a directory.
@@ -7598 +7593 @@
  * @param int          $max_execution_time Maximum time to run before giving up. In seconds. The timeout is global
  *                                         and is measured from the moment WordPress started to load.
+
+
  * @return int|false|null Size in bytes if a valid directory. False if not. Null if timeout.
  */
-function recurse_dirsize( $directory, $exclude = null, $max_execution_time = null ) {
+function recurse_dirsize( $directory, $exclude = null, $max_execution_time = null ) {
     $size = 0;
 
     $directory = untrailingslashit( $directory );
+
+
+
+
+
+
+
+
+
+
+
 
     if ( ! file_exists( $directory ) || ! is_dir( $directory ) || ! is_readable( $directory ) ) {
@@ -7631 +7639 @@
     }
 
-    $handle = opendir( $directory );
-    if ( $handle ) {
-        while ( ( $file = readdir( $handle ) ) !== false ) {
-            $path = $directory . '/' . $file;
-            if ( '.' !== $file && '..' !== $file ) {
-                if ( is_file( $path ) ) {
-                    $size += filesize( $path );
-                } elseif ( is_dir( $path ) ) {
-                    $handlesize = recurse_dirsize( $path, $exclude, $max_execution_time );
-                    if ( $handlesize > 0 ) {
-                        $size += $handlesize;
+    /**
+    * Filters the amount of storage space used by one directory and all it's children, in megabytes.
+    * Return the actual used space to shortcircuit the recursive PHP file size calculation and use something else
+    * like a CDN API or native operating system tools for better performance
+    *
+    * @since 5.6.0
+    *
+    * @param int|false $space_used The amount of used space, in bytes. Default 0.
+    */
+    $size = apply_filters( 'calculate_current_dirsize', $size, $directory, $exclude, $max_execution_time, $directory_cache );
+
+    if ( 0 === $size ) {
+        $handle = opendir( $directory );
+        if ( $handle ) {
+            while ( ( $file = readdir( $handle ) ) !== false ) {
+                $path = $directory . '/' . $file;
+                if ( '.' !== $file && '..' !== $file ) {
+                    if ( is_file( $path ) ) {
+                        $size += filesize( $path );
+                    } elseif ( is_dir( $path ) ) {
+                        $handlesize = recurse_dirsize( $path, $exclude, $max_execution_time, $directory_cache );
+                        if ( $handlesize > 0 ) {
+                            $size += $handlesize;
+                        }
+                    }
+
+                    if ( $max_execution_time > 0 && microtime( true ) - WP_START_TIMESTAMP > $max_execution_time ) {
+                        // Time exceeded. Give up instead of risking a fatal timeout.
+                        $size = null;
+                        break;
                     }
                 }
-
-                if ( $max_execution_time > 0 && microtime( true ) - WP_START_TIMESTAMP > $max_execution_time ) {
-                    // Time exceeded. Give up instead of risking a fatal timeout.
-                    $size = null;
-                    break;
-                }
             }
-        }
-        closedir( $handle );
-    }
+            closedir( $handle );
+        }
+    }
+    $directory_cache[ $cache_path ] = $size;
+
+    // Only write the transient on the top level call and not on recursive calls
+    if ( $save_cache ) {
+        set_transient( 'dirsize_cache', $directory_cache );
+    }
+
     return $size;
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
 }
 

trunk/src/wp-includes/post.php

@@ -5915 +5915 @@
 
     if ( is_multisite() ) {
-        delete_transient( 'dirsize_cache' );
+         );
     }