wp-cli/cache-command
2
0
Fork 0
mirror of https://hk.gh-proxy.com/https://github.com/wp-cli/profile-command.git synced 2025-08-18 06:11:48 +08:00
Code Issues Projects Packages Activity Actions

Use more precise hooks for profiling intermediate stages

Browse source
This commit is contained in:
Daniel Bachhuber 2016-10-09 15:43:02 -07:00
parent e6a05be838
commit 0a365862d3
2 changed files with 54 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

9
features/profile-hook.feature
View file

@ -21,6 +21,15 @@ Feature: Profile a specific hook
| smilies_init() | 2 | 0 |
| feed_links() | 8 | 0 |
Scenario: Profile an intermediate stage hook
Given a WP install
When I run `wp profile hook wp_head:before --fields=callback,cache_hits,cache_misses`
Then STDOUT should be a table containing rows:
| callback | cache_hits | cache_misses |
| locate_template() | 0 | 0 |
| load_template() | 0 | 0 |
Scenario: Profile a hook before the template is loaded
Given a WP install

75
inc/class-profiler.php
View file

@ -74,10 +74,50 @@ class Profiler {
define( 'SAVEQUERIES', true );
}
});
WP_CLI::add_wp_hook( 'all', array( $this, 'wp_hook_begin' ) );
if ( 'hook' === $this->type
&& ':before' === substr( $this->focus, -7, 7 ) ) {
$stage_hooks = array();
foreach( $this->stage_hooks as $hooks ) {
$stage_hooks = array_merge( $stage_hooks, $hooks );
}
$end_hook = substr( $this->focus, 0, -7 );
$key = array_search( $end_hook, $stage_hooks );
$start_hook = $stage_hooks[ $key - 1 ];
WP_CLI::add_wp_hook( $start_hook, array( $this, 'wp_tick_profile_begin' ), 9999 );
WP_CLI::add_wp_hook( $end_hook, array( $this, 'wp_tick_profile_end' ), -9999 );
} else {
WP_CLI::add_wp_hook( 'all', array( $this, 'wp_hook_begin' ) );
}
WP_CLI::add_wp_hook( 'pre_http_request', array( $this, 'wp_request_begin' ) );
WP_CLI::add_wp_hook( 'http_api_debug', array( $this, 'wp_request_end' ) );
$this->load_wordpress_with_template();
if ( 'hook' === $this->type
&& ':before' === substr( $this->focus, -7, 7 ) ) {
foreach( $this->loggers as $hash => $logger ) {
list( $name, $location ) = self::get_name_location_from_callback( $logger['callback'] );
$logger['callback'] = $name;
$logger['location'] = $location;
$this->loggers[ $hash ] = new Logger( $logger );
}
}
}
/**
* Start profiling function calls on the end of this filter
*/
public function wp_tick_profile_begin( $value = null ) {
register_tick_function( array( $this, 'handle_function_tick' ) );
declare( ticks = 1 );
return $value;
}
/**
* Stop profiling function calls at the beginning of this filter
*/
public function wp_tick_profile_end( $value = null ) {
unregister_tick_function( array( $this, 'handle_function_tick' ) );
$this->tick_callback = null;
return $value;
}
/**
@ -113,19 +153,10 @@ class Profiler {
$this->previous_filter_callbacks = null;
}
if ( 'hook' === $this->type && 0 === $this->filter_depth ) {
if ( $current_filter === $this->focus || true === $this->focus ) {
$this->wrap_current_filter_callbacks( $current_filter );
} else if ( ':before' === substr( $this->focus, -7, 7 ) && $current_filter === substr( $this->focus, 0, -7 ) ) {
unregister_tick_function( array( $this, 'handle_function_tick' ) );
$this->tick_callback = null;
foreach( $this->loggers as $hash => $logger ) {
list( $name, $location ) = self::get_name_location_from_callback( $logger['callback'] );
$logger['callback'] = $name;
$logger['location'] = $location;
$this->loggers[ $hash ] = new Logger( $logger );
}
}
if ( 'hook' === $this->type
&& 0 === $this->filter_depth
&& ( $current_filter === $this->focus || true === $this->focus ) ) {
$this->wrap_current_filter_callbacks( $current_filter );
}
$this->filter_depth++;
@ -197,22 +228,6 @@ class Profiler {
$this->filter_depth--;
$stage_hooks = array();
foreach( $this->stage_hooks as $hooks ) {
$stage_hooks = array_merge( $stage_hooks, $hooks );
}
if ( 'hook' === $this->type
&& 0 === $this->filter_depth
&& ':before' === substr( $this->focus, -7, 7 ) ) {
$key = array_search( $current_filter, $stage_hooks );
if ( false !== $key
&& isset( $stage_hooks[ $key + 1 ] )
&& $stage_hooks[ $key + 1 ] === substr( $this->focus, 0, -7 ) ) {
register_tick_function( array( $this, 'handle_function_tick' ) );
declare( ticks = 1 );
}
}
return $filter_value;
}