Refresh query duration handling (#541)

* refresh query duration handling * adjust naming * ensure only called when methods exist
laravel · Jun 23, 2022 · 4cc7c1a · 4cc7c1a
1 parent edde78b
commit 4cc7c1a
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/src/Concerns/ProvidesDefaultConfigurationOptions.php b/src/Concerns/ProvidesDefaultConfigurationOptions.php
@@ -47,6 +47,7 @@ public static function prepareApplicationForNextOperation(): array
             \Laravel\Octane\Listeners\GiveNewApplicationInstanceToViewFactory::class,
             \Laravel\Octane\Listeners\FlushDatabaseRecordModificationState::class,
             \Laravel\Octane\Listeners\FlushDatabaseQueryLog::class,
+            \Laravel\Octane\Listeners\RefreshQueryDurationHandling::class,
             \Laravel\Octane\Listeners\FlushLogContext::class,
             \Laravel\Octane\Listeners\FlushArrayCache::class,
             \Laravel\Octane\Listeners\FlushMonologState::class,

diff --git a/src/Listeners/RefreshQueryDurationHandling.php b/src/Listeners/RefreshQueryDurationHandling.php
@@ -0,0 +1,29 @@
+<?php
+
+namespace Laravel\Octane\Listeners;
+
+class RefreshQueryDurationHandling
+{
+    /**
+     * Handle the event.
+     *
+     * @param  mixed  $event
+     * @return void
+     */
+    public function handle($event): void
+    {
+        if (! $event->sandbox->resolved('db')) {
+            return;
+        }
+
+        foreach ($event->sandbox->make('db')->getConnections() as $connection) {
+            if (
+                method_exists($connection, 'resetTotalQueryDuration')
+                && method_exists($connection, 'allowQueryDurationHandlersToRunAgain')
+            ) {
+                $connection->resetTotalQueryDuration();
+                $connection->allowQueryDurationHandlersToRunAgain();
+            }
+        }
+    }
+}