class-wp-metadata-lazyloader.php 5.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171
  1. <?php
  2. /**
  3. * Meta API: WP_Metadata_Lazyloader class
  4. *
  5. * @package WordPress
  6. * @subpackage Meta
  7. * @since 4.5.0
  8. */
  9. /**
  10. * Core class used for lazy-loading object metadata.
  11. *
  12. * When loading many objects of a given type, such as posts in a WP_Query loop, it often makes
  13. * sense to prime various metadata caches at the beginning of the loop. This means fetching all
  14. * relevant metadata with a single database query, a technique that has the potential to improve
  15. * performance dramatically in some cases.
  16. *
  17. * In cases where the given metadata may not even be used in the loop, we can improve performance
  18. * even more by only priming the metadata cache for affected items the first time a piece of metadata
  19. * is requested - ie, by lazy-loading it. So, for example, comment meta may not be loaded into the
  20. * cache in the comments section of a post until the first time get_comment_meta() is called in the
  21. * context of the comment loop.
  22. *
  23. * WP uses the WP_Metadata_Lazyloader class to queue objects for metadata cache priming. The class
  24. * then detects the relevant get_*_meta() function call, and queries the metadata of all queued objects.
  25. *
  26. * Do not access this class directly. Use the wp_metadata_lazyloader() function.
  27. *
  28. * @since 4.5.0
  29. */
  30. #[AllowDynamicProperties]
  31. class WP_Metadata_Lazyloader {
  32. /**
  33. * Pending objects queue.
  34. *
  35. * @since 4.5.0
  36. * @var array
  37. */
  38. protected $pending_objects;
  39. /**
  40. * Settings for supported object types.
  41. *
  42. * @since 4.5.0
  43. * @var array
  44. */
  45. protected $settings = array();
  46. /**
  47. * Constructor.
  48. *
  49. * @since 4.5.0
  50. */
  51. public function __construct() {
  52. $this->settings = array(
  53. 'term' => array(
  54. 'filter' => 'get_term_metadata',
  55. 'callback' => array( $this, 'lazyload_term_meta' ),
  56. ),
  57. 'comment' => array(
  58. 'filter' => 'get_comment_metadata',
  59. 'callback' => array( $this, 'lazyload_comment_meta' ),
  60. ),
  61. );
  62. }
  63. /**
  64. * Adds objects to the metadata lazy-load queue.
  65. *
  66. * @since 4.5.0
  67. *
  68. * @param string $object_type Type of object whose meta is to be lazy-loaded. Accepts 'term' or 'comment'.
  69. * @param array $object_ids Array of object IDs.
  70. * @return void|WP_Error WP_Error on failure.
  71. */
  72. public function queue_objects( $object_type, $object_ids ) {
  73. if ( ! isset( $this->settings[ $object_type ] ) ) {
  74. return new WP_Error( 'invalid_object_type', __( 'Invalid object type.' ) );
  75. }
  76. $type_settings = $this->settings[ $object_type ];
  77. if ( ! isset( $this->pending_objects[ $object_type ] ) ) {
  78. $this->pending_objects[ $object_type ] = array();
  79. }
  80. foreach ( $object_ids as $object_id ) {
  81. // Keyed by ID for faster lookup.
  82. if ( ! isset( $this->pending_objects[ $object_type ][ $object_id ] ) ) {
  83. $this->pending_objects[ $object_type ][ $object_id ] = 1;
  84. }
  85. }
  86. add_filter( $type_settings['filter'], $type_settings['callback'] );
  87. /**
  88. * Fires after objects are added to the metadata lazy-load queue.
  89. *
  90. * @since 4.5.0
  91. *
  92. * @param array $object_ids Array of object IDs.
  93. * @param string $object_type Type of object being queued.
  94. * @param WP_Metadata_Lazyloader $lazyloader The lazy-loader object.
  95. */
  96. do_action( 'metadata_lazyloader_queued_objects', $object_ids, $object_type, $this );
  97. }
  98. /**
  99. * Resets lazy-load queue for a given object type.
  100. *
  101. * @since 4.5.0
  102. *
  103. * @param string $object_type Object type. Accepts 'comment' or 'term'.
  104. * @return void|WP_Error WP_Error on failure.
  105. */
  106. public function reset_queue( $object_type ) {
  107. if ( ! isset( $this->settings[ $object_type ] ) ) {
  108. return new WP_Error( 'invalid_object_type', __( 'Invalid object type.' ) );
  109. }
  110. $type_settings = $this->settings[ $object_type ];
  111. $this->pending_objects[ $object_type ] = array();
  112. remove_filter( $type_settings['filter'], $type_settings['callback'] );
  113. }
  114. /**
  115. * Lazy-loads term meta for queued terms.
  116. *
  117. * This method is public so that it can be used as a filter callback. As a rule, there
  118. * is no need to invoke it directly.
  119. *
  120. * @since 4.5.0
  121. *
  122. * @param mixed $check The `$check` param passed from the 'get_term_metadata' hook.
  123. * @return mixed In order not to short-circuit `get_metadata()`. Generally, this is `null`, but it could be
  124. * another value if filtered by a plugin.
  125. */
  126. public function lazyload_term_meta( $check ) {
  127. if ( ! empty( $this->pending_objects['term'] ) ) {
  128. update_termmeta_cache( array_keys( $this->pending_objects['term'] ) );
  129. // No need to run again for this set of terms.
  130. $this->reset_queue( 'term' );
  131. }
  132. return $check;
  133. }
  134. /**
  135. * Lazy-loads comment meta for queued comments.
  136. *
  137. * This method is public so that it can be used as a filter callback. As a rule, there is no need to invoke it
  138. * directly, from either inside or outside the `WP_Query` object.
  139. *
  140. * @since 4.5.0
  141. *
  142. * @param mixed $check The `$check` param passed from the {@see 'get_comment_metadata'} hook.
  143. * @return mixed The original value of `$check`, so as not to short-circuit `get_comment_metadata()`.
  144. */
  145. public function lazyload_comment_meta( $check ) {
  146. if ( ! empty( $this->pending_objects['comment'] ) ) {
  147. update_meta_cache( 'comment', array_keys( $this->pending_objects['comment'] ) );
  148. // No need to run again for this set of comments.
  149. $this->reset_queue( 'comment' );
  150. }
  151. return $check;
  152. }
  153. }