Changeset 36565

Timestamp:

02/17/2016 10:51:01 PM (8 years ago)

Author:

azaozz

Message:

Improve the performance of wp_upload_dir():

• Cache the output in non-persistent cache.
• Cache the result from wp_mkdir_p() in persistent cache (when present).
• Introduce wp_get_upload_dir() for use when not uploading files. It is equivalent to wp_upload_dir() but does not check for the existence or create the upload directory.
• Change tests to use the non-cached _wp_upload_dir(). They change options on the fly (should never be used in production) to simulate different environments.
• Introduce _upload_dir_no_subdir() and _upload_dir_https() to facilitate testing. These use the proper upload_dir filter to simulate different environments.

Props kovshenin, azaozz.
See #34359.

Location:

trunk

Files:

• src/wp-includes/functions.php (modified) (3 diffs)
• tests/phpunit/includes/functions.php (modified) (1 diff)
• tests/phpunit/tests/media.php (modified) (2 diffs)
• tests/phpunit/tests/post/attachments.php (modified) (2 diffs)
• tests/phpunit/tests/upload.php (modified) (4 diffs)

trunk/src/wp-includes/functions.php

@@ -1801 +1801 @@
 
 /**
+ * Get uploads directory information.
+ *
+ * Same as wp_upload_dir() but "light weight" as it doesn't attempt to create the uploads directory.
+ * Intended for use in themes, when only 'basedir' and 'baseurl' are needed, generally in all cases when not uploading files.
+ *
+ * @since 4.5.0
+ *
+ * @return array See wp_upload_dir() for description.
+ */
+function wp_get_upload_dir() {
+    return wp_upload_dir( null, false );
+}
+
+/**
  * Get an array containing the current upload directory's path and url.
  *
@@ -1825 +1839 @@
  * 'basedir' - path without subdir.
  * 'baseurl' - URL path without subdir.
- * 'error' - set to false.
+ * 'error' - false or error message.
  *
  * @since 2.0.0
+ * @uses _wp_upload_dir()
  *
  * @param string $time Optional. Time formatted in 'yyyy/mm'. Default null.
+ * @param bool   $create_dir Optional. Whether to check and create the uploads directory. Default true (backwards compatible).
+ * @param bool   $refresh_cache Optional. Whether to refresh the cache. Default false.
  * @return array See above for description.
  */
-function wp_upload_dir( $time = null ) {
+function wp_upload_dir( $time = null, $create_dir = true, $refresh_cache = false ) {
+    static $cache = array();
+
+    $key = sprintf( '%d-%s', get_current_blog_id(), (string) $time );
+
+    if ( $refresh_cache || empty( $cache[ $key ] ) ) {
+        $cache[ $key ] = _wp_upload_dir( $time );
+    }
+
+    /**
+     * Filter the uploads directory data.
+     *
+     * @since 2.0.0
+     *
+     * @param array $uploads Array of upload directory data with keys of 'path',
+     *                       'url', 'subdir, 'basedir', and 'error'.
+     */
+    $uploads = apply_filters( 'upload_dir', $cache[ $key ] );
+
+    if ( $create_dir ) {
+        $path = $uploads['path'];
+        $tested_paths = wp_cache_get( 'upload_dir_tested_paths' );
+
+        if ( ! is_array( $tested_paths ) ) {
+            $tested_paths = array();
+        }
+
+        if ( array_key_exists( $path, $tested_paths ) ) {
+            $uploads['error'] = $tested_paths[ $path ];
+        } else {
+            if ( ! wp_mkdir_p( $path ) ) {
+                if ( 0 === strpos( $uploads['basedir'], ABSPATH ) ) {
+                    $error_path = str_replace( ABSPATH, '', $uploads['basedir'] ) . $uploads['subdir'];
+                } else {
+                    $error_path = basename( $uploads['basedir'] ) . $uploads['subdir'];
+                }
+
+                $uploads['error'] = sprintf( __( 'Unable to create directory %s. Is its parent directory writable by the server?' ), $error_path );
+            }
+
+            $tested_paths[ $path ] = $uploads['error'];
+            wp_cache_set( 'upload_dir_tested_paths', $tested_paths );
+        }
+    }
+
+    return $uploads;
+}
+
+/**
+ * A non-filtered, non-cached version of wp_upload_dir() that doesn't check the path.
+ *
+ * @access private
+ *
+ * @param string $time Optional. Time formatted in 'yyyy/mm'. Default null.
+ * @return array See wp_upload_dir()
+ */
+function _wp_upload_dir( $time = null ) {
     $siteurl = get_option( 'siteurl' );
     $upload_path = trim( get_option( 'upload_path' ) );
@@ -1921 +1994 @@
     $url .= $subdir;
 
-    /**
-     * Filter the uploads directory data.
-     *
-     * @since 2.0.0
-     *
-     * @param array $uploads Array of upload directory data with keys of 'path',
-     *                       'url', 'subdir, 'basedir', and 'error'.
-     */
-    $uploads = apply_filters( 'upload_dir',
-        array(
-            'path'    => $dir,
-            'url'     => $url,
-            'subdir'  => $subdir,
-            'basedir' => $basedir,
-            'baseurl' => $baseurl,
-            'error'   => false,
-        ) );
-
-    // Make sure we have an uploads directory.
-    if ( ! wp_mkdir_p( $uploads['path'] ) ) {
-        if ( 0 === strpos( $uploads['basedir'], ABSPATH ) )
-            $error_path = str_replace( ABSPATH, '', $uploads['basedir'] ) . $uploads['subdir'];
-        else
-            $error_path = basename( $uploads['basedir'] ) . $uploads['subdir'];
-
-        $message = sprintf( __( 'Unable to create directory %s. Is its parent directory writable by the server?' ), $error_path );
-        $uploads['error'] = $message;
-    }
-
-    return $uploads;
+    return array(
+        'path'    => $dir,
+        'url'     => $url,
+        'subdir'  => $subdir,
+        'basedir' => $basedir,
+        'baseurl' => $baseurl,
+        'error'   => false,
+    );
 }
 

trunk/tests/phpunit/includes/functions.php

@@ -116 +116 @@
     update_option( 'permalink_structure', '/%year%/%monthnum%/%day%/%postname%/' );
 }
+
+/**
+ * Helper used with the `upload_dir` filter to remove the /year/month sub directories from the uploads path and URL.
+ */
+function _upload_dir_no_subdir( $uploads ) {
+    $subdir = $uploads['subdir'];
+
+    $uploads['subdir'] = '';
+    $uploads['path'] = str_replace( $subdir, '', $uploads['path'] );
+    $uploads['url'] = str_replace( $subdir, '', $uploads['url'] );
+
+    return $uploads;
+}
+
+/**
+ * Helper used with the `upload_dir` filter to set https upload URL.
+ */ 
+function _upload_dir_https( $uploads ) {
+    $uploads['url'] = str_replace( 'http://', 'https://', $uploads['url'] );
+    $uploads['baseurl'] = str_replace( 'http://', 'https://', $uploads['baseurl'] );
+
+    return $uploads;
+}

trunk/tests/phpunit/tests/media.php

@@ -896 +896 @@
         global $_wp_additional_image_sizes;
 
-        // Save the current setting for uploads folders
-        $uploads_use_yearmonth_folders = get_option( 'uploads_use_yearmonth_folders' );
-
         // Disable date organized uploads
-        update_option( 'uploads_use_yearmonth_folders', 0 );
+        add_filter( 'upload_dir', '_upload_dir_no_subdir' );
 
         // Make an image.
@@ -938 +935 @@
         // Remove the attachment
         wp_delete_attachment( $id );
-
-        // Leave the uploads option the way you found it.
-        update_option( 'uploads_use_yearmonth_folders', $uploads_use_yearmonth_folders );
+        remove_filter( 'upload_dir', '_upload_dir_no_subdir' );
     }
 

trunk/tests/phpunit/tests/post/attachments.php

@@ -442 +442 @@
      */
     public function test_wp_get_attachment_url_should_force_https_when_administering_over_https_and_siteurl_is_https() {
-        // Must set the upload_url_path to fake out `wp_upload_dir()`.
-        $siteurl = get_option( 'siteurl' );
-        update_option( 'upload_url_path', set_url_scheme( $siteurl, 'https' ) . '/uploads' );
+        // Set https upload URL 
+        add_filter( 'upload_dir', '_upload_dir_https' );
 
         $filename = ( DIR_TESTDATA . '/images/test-image.jpg' );
@@ -464 +463 @@
         $_SERVER['HTTPS'] = $is_ssl ? 'on' : 'off';
         set_current_screen( 'front' );
+        remove_filter( 'upload_dir', '_upload_dir_https' );
 
         $this->assertSame( set_url_scheme( $url, 'https' ), $url );

trunk/tests/phpunit/tests/upload.php

@@ -27 +27 @@
         // wp_upload_dir() with default parameters
         $info = wp_upload_dir();
-        $this->assertEquals( get_option( 'siteurl' ) . '/wp-content/uploads/' . gmstrftime('%Y/%m'), $info['url'] );
-        $this->assertEquals( ABSPATH . 'wp-content/uploads/' . gmstrftime('%Y/%m'), $info['path'] );
-        $this->assertEquals( gmstrftime('/%Y/%m'), $info['subdir'] );
-        $this->assertEquals( '', $info['error'] );
+        $subdir = gmstrftime('/%Y/%m');
+
+        $this->assertEquals( get_option( 'siteurl' ) . '/wp-content/uploads' . $subdir, $info['url'] );
+        $this->assertEquals( ABSPATH . 'wp-content/uploads' . $subdir, $info['path'] );
+        $this->assertEquals( $subdir, $info['subdir'] );
+        $this->assertEquals( false, $info['error'] );
     }
 
@@ -36 +38 @@
         // wp_upload_dir() with a relative upload path that is not 'wp-content/uploads'
         update_option( 'upload_path', 'foo/bar' );
-        $info = wp_upload_dir();
-        $this->delete_folders( ABSPATH . 'foo' );
+        $info = _wp_upload_dir();
+        $subdir = gmstrftime('/%Y/%m');
 
-        $this->assertEquals( get_option( 'siteurl' ) . '/foo/bar/' . gmstrftime('%Y/%m'), $info['url'] );
-        $this->assertEquals( ABSPATH . 'foo/bar/' . gmstrftime('%Y/%m'), $info['path'] );
-        $this->assertEquals( gmstrftime('/%Y/%m'), $info['subdir'] );
-        $this->assertEquals( '', $info['error'] );
+        $this->assertEquals( get_option( 'siteurl' ) . '/foo/bar' . $subdir, $info['url'] );
+        $this->assertEquals( ABSPATH . 'foo/bar' . $subdir, $info['path'] );
+        $this->assertEquals( $subdir, $info['subdir'] );
+        $this->assertEquals( false, $info['error'] );
     }
 
@@ -50 +52 @@
     function test_upload_dir_absolute() {
         $path = '/tmp/wp-unit-test';
+
         // wp_upload_dir() with an absolute upload path
         update_option( 'upload_path', $path );
+
         // doesn't make sense to use an absolute file path without setting the url path
         update_option( 'upload_url_path', '/baz' );
-        $info = wp_upload_dir();
-        $this->delete_folders( $path );
 
-        $this->assertEquals( '/baz/' . gmstrftime('%Y/%m'), $info['url'] );
-        $this->assertEquals( "$path/" . gmstrftime('%Y/%m'), $info['path'] );
-        $this->assertEquals( gmstrftime('/%Y/%m'), $info['subdir'] );
-        $this->assertEquals( '', $info['error'] );
+        // Use `_wp_upload_dir()` directly to bypass caching and work with the changed options.
+        // It doesn't create the /year/month directories.
+        $info = _wp_upload_dir();
+        $subdir = gmstrftime('/%Y/%m');
+
+        $this->assertEquals( '/baz' . $subdir, $info['url'] );
+        $this->assertEquals( $path . $subdir, $info['path'] );
+        $this->assertEquals( $subdir, $info['subdir'] );
+        $this->assertEquals( false, $info['error'] );
     }
 
     function test_upload_dir_no_yearnum() {
         update_option( 'uploads_use_yearmonth_folders', 0 );
-        $info = wp_upload_dir();
+
+        // Use `_wp_upload_dir()` directly to bypass caching and work with the changed options.
+        $info = _wp_upload_dir();
+
         $this->assertEquals( get_option( 'siteurl' ) . '/wp-content/uploads', $info['url'] );
         $this->assertEquals( ABSPATH . 'wp-content/uploads', $info['path'] );
         $this->assertEquals( '', $info['subdir'] );
-        $this->assertEquals( '', $info['error'] );
+        $this->assertEquals( false, $info['error'] );
     }
 
     function test_upload_path_absolute() {
         update_option( 'upload_url_path', 'http://example.org/asdf' );
-        $info = wp_upload_dir();
-        $this->assertEquals( 'http://example.org/asdf/' . gmstrftime('%Y/%m'), $info['url'] );
-        $this->assertEquals( ABSPATH . 'wp-content/uploads/' . gmstrftime('%Y/%m'), $info['path'] );
-        $this->assertEquals( gmstrftime('/%Y/%m'), $info['subdir'] );
-        $this->assertEquals( '', $info['error'] );
+
+        // Use `_wp_upload_dir()` directly to bypass caching and work with the changed options.
+        // It doesn't create the /year/month directories.
+        $info = _wp_upload_dir();
+        $subdir = gmstrftime('/%Y/%m');
+
+        $this->assertEquals( 'http://example.org/asdf' . $subdir, $info['url'] );
+        $this->assertEquals( ABSPATH . 'wp-content/uploads' . $subdir, $info['path'] );
+        $this->assertEquals( $subdir, $info['subdir'] );
+        $this->assertEquals( false, $info['error'] );
     }
 
@@ -84 +99 @@
         // upload path setting is empty - it should default to 'wp-content/uploads'
         update_option('upload_path', '');
-        $info = wp_upload_dir();
-        $this->assertEquals( get_option( 'siteurl' ) . '/wp-content/uploads/' . gmstrftime('%Y/%m'), $info['url'] );
-        $this->assertEquals( ABSPATH . 'wp-content/uploads/' . gmstrftime('%Y/%m'), $info['path'] );
-        $this->assertEquals( gmstrftime('/%Y/%m'), $info['subdir'] );
-        $this->assertEquals( '', $info['error'] );
+
+        // Use `_wp_upload_dir()` directly to bypass caching and work with the changed options.
+        // It doesn't create the /year/month directories.
+        $info = _wp_upload_dir();
+        $subdir = gmstrftime('/%Y/%m');
+
+        $this->assertEquals( get_option( 'siteurl' ) . '/wp-content/uploads' . $subdir, $info['url'] );
+        $this->assertEquals( ABSPATH . 'wp-content/uploads' . $subdir, $info['path'] );
+        $this->assertEquals( $subdir, $info['subdir'] );
+        $this->assertEquals( false, $info['error'] );
     }