Added generic stopwatch collector and tests.

src/Record/Hook_Record_Collector.php

@@ -4,34 +4,28 @@ namespace Rarst\Laps\Record;
 
 use Rarst\Laps\Event\Hook_Event_Config_Interface;
 use Symfony\Component\Stopwatch\Stopwatch;
-use Symfony\Component\Stopwatch\StopwatchEvent;
 
 /**
  * Processes events based on hooked starts and stops.
  */
-class Hook_Record_Collector implements Record_Collector_Interface {
+class Hook_Record_Collector extends Stopwatch_Record_Collector {
 
-	/** @var Hook_Event_Config_Interface[] $event_configs  */
+	/** @var Hook_Event_Config_Interface[] $event_configs */
 	protected $event_configs = [];
 
-	/** @var Stopwatch $stopwatch */
-	protected $stopwatch;
-
 	/** @var array $events */
 	protected $events = [];
 
 	/**
-	 * @param array     $event_configs Starts and stops configuration.
 	 * @param Stopwatch $stopwatch     Stopwatch instance.
+	 * @param array     $event_configs Starts and stops configuration.
 	 */
-	public function __construct( array $event_configs, Stopwatch $stopwatch ) {
+	public function __construct( Stopwatch $stopwatch, array $event_configs ) {
 
-		$this->stopwatch = $stopwatch;
-
-		$this->stopwatch->start( 'Plugins Load', 'plugin' );
+		parent::__construct( $stopwatch );
 
+		$this->start( 'Plugins Load', 'plugin' );
 		$this->add_events( $event_configs['core']->get_events() );
-
 		unset( $event_configs['core'] );
 		$this->event_configs = $event_configs;
 
@@ -48,9 +42,7 @@ class Hook_Record_Collector implements Record_Collector_Interface {
 		$this->events = array_merge( $this->events, $stops );
 
 		foreach ( $stops as $hook_name => $data ) {
-
 			foreach ( array_keys( $data ) as $priority ) {
-
 				add_action( $hook_name, [ $this, 'tick' ], $priority );
 			}
 		}
@@ -91,12 +83,12 @@ class Hook_Record_Collector implements Record_Collector_Interface {
 			'category' => null,
 		] );
 
-		if ( 'stop' === $event['action'] && ! $this->stopwatch->isStarted( $event['event'] ) ) {
-			return $input;
+		if ( 'start' === $event['action'] ) {
+			$this->start( $event['event'], $event['category'] );
+		} else {
+			$this->stop( $event['event'] );
 		}
 
-		$this->stopwatch->{$event['action']}( $event['event'], $event['category'] );
-
 		return $input;
 	}
 
@@ -105,29 +97,8 @@ class Hook_Record_Collector implements Record_Collector_Interface {
 	 */
 	public function get_records() {
 
-		if ( $this->stopwatch->isStarted( 'Toolbar' ) ) {
-			$this->stopwatch->stop( 'Toolbar' );
-		}
+		$this->stopwatch->stop( 'Toolbar' );
 
-		$events = $this->stopwatch->getSectionEvents( '__root__' );
-
-		foreach ( array_keys( $events ) as $name ) {
-			if ( $this->stopwatch->isStarted( $name ) ) {
-				unset( $events[ $name ] );
-			}
-		}
-
-		return array_map( [ $this, 'transform' ], array_keys( $events ), $events );
-	}
-
-	/**
-	 * @param string         $name  Event name.
-	 * @param StopwatchEvent $event Stopwatch event instance.
-	 *
-	 * @return Stopwatch_Record
-	 */
-	protected function transform( $name, StopwatchEvent $event ) {
-
-		return new Stopwatch_Record( $name, $event );
+		return parent::get_records();
 	}
 }

src/Record/Http_Record_Collector.php

@@ -3,22 +3,18 @@
 namespace Rarst\Laps\Record;
 
 use Symfony\Component\Stopwatch\Stopwatch;
-use Symfony\Component\Stopwatch\StopwatchEvent;
 
 /**
  * Captures time of network requests made with HTTP API.
  */
-class Http_Record_Collector implements Record_Collector_Interface {
-
-	/** @var Stopwatch $stopwatch */
-	protected $stopwatch;
+class Http_Record_Collector extends Stopwatch_Record_Collector {
 
 	/**
 	 * @param Stopwatch $stopwatch Stopwatch instance.
 	 */
 	public function __construct( Stopwatch $stopwatch ) {
 
-		$this->stopwatch = $stopwatch;
+		parent::__construct( $stopwatch );
 
 		add_action( 'pre_http_request', [ $this, 'pre_http_request' ], 10, 3 );
 		add_action( 'http_api_debug', [ $this, 'http_api_debug' ], 10, 5 );
@@ -35,7 +31,7 @@ class Http_Record_Collector implements Record_Collector_Interface {
 	 */
 	public function pre_http_request( $false, $args, $url ) {
 
-		$this->stopwatch->start( $url, 'http' );
+		$this->start( $url, 'http' );
 
 		return $false;
 	}
@@ -55,29 +51,8 @@ class Http_Record_Collector implements Record_Collector_Interface {
 	 */
 	public function http_api_debug( $response, $type, $class, $args, $url ) {
 
-		$this->stopwatch->stop( $url );
+		$this->stop( $url );
 
 		return $response;
 	}
-
-	/**
-	 * @return Stopwatch_Record[]
-	 */
-	public function get_records() {
-
-		$events = $this->stopwatch->getSectionEvents( '__root__' );
-
-		return array_map( [ $this, 'transform' ], array_keys( $events ), $events );
-	}
-
-	/**
-	 * @param string         $name  Event name.
-	 * @param StopwatchEvent $event Stopwatch event instance.
-	 *
-	 * @return Stopwatch_Record
-	 */
-	protected function transform( $name, StopwatchEvent $event ) {
-
-		return new Stopwatch_Record( $name, $event );
-	}
 }

src/Record/Stopwatch_Record_Collector.php

@@ -0,0 +1,73 @@
+<?php
+
+namespace Rarst\Laps\Record;
+
+use Symfony\Component\Stopwatch\Stopwatch;
+use Symfony\Component\Stopwatch\StopwatchEvent;
+
+/**
+ * Generic Stopwatch–based collector. Can be used by itself or extended.
+ */
+class Stopwatch_Record_Collector implements Record_Collector_Interface {
+
+	/** @var Stopwatch $stopwatch */
+	protected $stopwatch;
+
+	/**
+	 * @param Stopwatch $stopwatch Stopwatch instance.
+	 */
+	public function __construct( Stopwatch $stopwatch ) {
+		$this->stopwatch = $stopwatch;
+	}
+
+	/**
+	 * @param string $name     The event name.
+	 * @param string $category The event category.
+	 *
+	 * @return StopwatchEvent
+	 */
+	public function start( $name, $category = null ) {
+		return $this->stopwatch->start( $name, $category );
+	}
+
+	/**
+	 * @param string $name The event name.
+	 *
+	 * @return bool|StopwatchEvent
+	 */
+	public function stop( $name ) {
+
+		if ( $this->stopwatch->isStarted( $name ) ) {
+			return $this->stopwatch->stop( $name );
+		}
+
+		return false;
+	}
+
+	/**
+	 * @return Stopwatch_Record[]
+	 */
+	public function get_records() {
+
+		$events = $this->stopwatch->getSectionEvents( '__root__' );
+
+		foreach ( array_keys( $events ) as $name ) {
+			if ( $this->stopwatch->isStarted( $name ) ) {
+				unset( $events[ $name ] );
+			}
+		}
+
+		return array_map( [ $this, 'transform' ], array_keys( $events ), $events );
+	}
+
+	/**
+	 * @param string         $name  Event name.
+	 * @param StopwatchEvent $event Stopwatch event instance.
+	 *
+	 * @return Stopwatch_Record
+	 */
+	protected function transform( $name, StopwatchEvent $event ) {
+
+		return new Stopwatch_Record( $name, $event );
+	}
+}