feat: support for octane (#106)

Author: bhushan

src/DataProviders/LaravelResponseDataProvider.php

@@ -10,12 +10,14 @@
 use Treblle\Php\Contract\ErrorDataProvider;
 use Treblle\Php\DataTransferObject\Response;
 use Treblle\Php\Contract\ResponseDataProvider;
+use Symfony\Component\HttpFoundation\Request as SymfonyRequest;
 use Symfony\Component\HttpFoundation\Response as SymfonyResponse;
 
 final class LaravelResponseDataProvider implements ResponseDataProvider
 {
     public function __construct(
         private readonly FieldMasker     $fieldMasker,
+        private \Illuminate\Http\Request|SymfonyRequest $request,
         private readonly JsonResponse|\Illuminate\Http\Response|SymfonyResponse $response,
         private ErrorDataProvider                                      &$errorDataProvider,
     ) {
@@ -58,6 +60,12 @@ private function getLoadTimeInMilliseconds(): float
         $currentTimeInMilliseconds = microtime(true) * 1000;
         $requestTimeInMilliseconds = microtime(true) * 1000;
 
+        if ($this->request->attributes->has('treblle_request_started_at')) {
+            $requestTimeInMilliseconds = $this->request->attributes->get('treblle_request_started_at') * 1000;
+
+            return $currentTimeInMilliseconds - $requestTimeInMilliseconds;
+        }
+
         if (isset($_SERVER['REQUEST_TIME_FLOAT'])) {
             $requestTimeInMilliseconds = (float)$_SERVER['REQUEST_TIME_FLOAT'] * 1000;
         } elseif (defined('LARAVEL_START')) {

src/Middlewares/TreblleMiddleware.php

@@ -48,7 +48,7 @@ public function terminate(Request $request, JsonResponse|Response|SymfonyRespons
         $fieldMasker = new FieldMasker($maskedFields);
         $errorProvider = new InMemoryErrorDataProvider();
         $requestProvider = new LaravelRequestDataProvider($fieldMasker, $request);
-        $responseProvider = new LaravelResponseDataProvider($fieldMasker, $response, $errorProvider);
+        $responseProvider = new LaravelResponseDataProvider($fieldMasker, $request, $response, $errorProvider);
 
         if (! empty($response->exception)) {
             $errorProvider->addError(new Error(

src/TreblleServiceProvider.php

@@ -7,6 +7,7 @@
 use function config;
 use Illuminate\Routing\Router;
 use Illuminate\Support\ServiceProvider;
+use Illuminate\Contracts\Events\Dispatcher;
 use Illuminate\Foundation\Console\AboutCommand;
 use Treblle\Laravel\Middlewares\TreblleMiddleware;
 use Illuminate\Contracts\Container\BindingResolutionException;
@@ -34,6 +35,13 @@ public function boot(): void
             $router->aliasMiddleware('treblle', TreblleMiddleware::class);
         }
 
+        /** @var Dispatcher $events */
+        $events = $this->app->make(Dispatcher::class);
+
+        $events->listen('Laravel\Octane\Events\RequestReceived', function ($event): void {
+            $event->request->attributes->set('treblle_request_started_at', microtime(true));
+        });
+
         AboutCommand::add(
             section: 'Treblle',
             data: static fn (): array => [