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

Merge pull request #90 from runcommand/23-all-stages

Browse source 
Use --all flag to profile all stages
This commit is contained in:
Daniel Bachhuber 2016-10-07 15:50:35 -07:00 • committed by GitHub
commit 9dbc52727c
2 changed files with 89 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

52
features/profile-stage.feature
View file

@ -30,3 +30,55 @@ Feature: Profile the template render stage
| wp_footer |
| |
| total |
Scenario: Use --all flag to profile all stages
Given a WP install
When I run `wp profile stage --all --fields=hook`
Then STDOUT should be a table containing rows:
| hook |
| |
| muplugins_loaded |
| |
| plugins_loaded |
| |
| setup_theme |
| |
| after_setup_theme |
| |
| init |
| |
| wp_loaded |
| |
| parse_request |
| |
| send_headers |
| |
| pre_get_posts |
| |
| the_posts |
| |
| wp |
| |
| template_redirect |
| |
| template_include |
| |
| wp_head |
| |
| loop_start |
| |
| loop_end |
| |
| wp_footer |
| |
| total |
Scenario: Invalid stage specified
Given a WP install
When I try `wp profile stage foo_bar`
Then STDERR should be:
"""
Error: Invalid stage. Must be one of bootstrap, main_query, template, or use --all.
"""

60
inc/class-command.php
View file

@ -26,6 +26,9 @@ class Command {
* [<stage>]
* : Drill down into a specific stage.
*
* [--all]
* : Expand upon all stages.
*
* [--url=<url>]
* : Execute a request against a specified URL. Defaults to the home URL.
*
@ -48,13 +51,11 @@ class Command {
public function stage( $args, $assoc_args ) {
global $wpdb;
if ( isset( $args[0] ) ) {
$this->focus_stage = $args[0];
}
$this->focus_stage = Utils\get_flag_value( $assoc_args, 'all', isset( $args[0] ) ? $args[0] : null );
$valid_stages = array( 'bootstrap', 'main_query', 'template' );
if ( $this->focus_stage && ! in_array( $this->focus_stage, $valid_stages, true ) ) {
WP_CLI::error( 'Invalid stage. Must be one of: ' . implode( ', ', $valid_stages ) );
if ( $this->focus_stage && ( true !== $this->focus_stage && ! in_array( $this->focus_stage, $valid_stages, true ) ) ) {
WP_CLI::error( 'Invalid stage. Must be one of ' . implode( ', ', $valid_stages ) . ', or use --all.' );
}
$this->run_profiler();
@ -424,15 +425,41 @@ class Command {
private function load_wordpress_with_template() {
global $wp_query;
if ( 'bootstrap' === $this->focus_stage ) {
$this->set_stage_hooks( array(
$stage_hooks = array(
'bootstrap' => array(
'muplugins_loaded',
'plugins_loaded',
'setup_theme',
'after_setup_theme',
'init',
'wp_loaded',
) );
),
'main_query' => array(
'parse_request',
'send_headers',
'pre_get_posts',
'the_posts',
'wp',
),
'template' => array(
'template_redirect',
'template_include',
'wp_head',
'loop_start',
'loop_end',
'wp_footer',
),
);
if ( true === $this->focus_stage ) {
$hooks = array();
foreach( $stage_hooks as $stage_hook ) {
$hooks = array_merge( $hooks, $stage_hook );
}
$this->set_stage_hooks( $hooks );
}
if ( 'bootstrap' === $this->focus_stage ) {
$this->set_stage_hooks( $stage_hooks['bootstrap'] );
} else if ( ! $this->focus_stage && ! $this->focus_hook ) {
$logger = new Logger( array( 'stage' => 'bootstrap' ) );
$logger->start();
@ -448,13 +475,7 @@ class Command {
// Set up main_query main WordPress query.
if ( 'main_query' === $this->focus_stage ) {
$this->set_stage_hooks( array(
'parse_request',
'send_headers',
'pre_get_posts',
'the_posts',
'wp',
) );
$this->set_stage_hooks( $stage_hooks['main_query'] );
} else if ( ! $this->focus_stage && ! $this->focus_hook ) {
$logger = new Logger( array( 'stage' => 'main_query' ) );
$logger->start();
@ -477,14 +498,7 @@ class Command {
// Load the theme template.
if ( 'template' === $this->focus_stage ) {
$this->set_stage_hooks( array(
'template_redirect',
'template_include',
'wp_head',
'loop_start',
'loop_end',
'wp_footer',
) );
$this->set_stage_hooks( $stage_hooks['template'] );
} else if ( ! $this->focus_stage && ! $this->focus_hook ) {
$logger = new Logger( array( 'stage' => 'template' ) );
$logger->start();