Add affordance to restore queries while preparing response

laravel · Jan 11, 2023 · 1a1ed92 · 1a1ed92
1 parent 67c4653
commit 1a1ed92
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 0 deletions.
diff --git a/src/Illuminate/Foundation/Application.php b/src/Illuminate/Foundation/Application.php
@@ -1444,4 +1444,17 @@ public function preventQueriesWhilePreparingResponse($connection = null)
 
         return $this;
     }
+
+    /**
+     * Restores ability for database to query while preparing the request response.
+     *
+     * @param  string|null  $connection
+     * @return $this
+     */
+    public function allowQueriesWhilePreparingResponse($connection = null)
+    {
+        $this['events']->listen(PreparingResponse::class, fn () => $this['db']->connection($connection)->allowQueries());
+
+        return $this;
+    }
 }
diff --git a/tests/Integration/Foundation/ApplicationTest.php b/tests/Integration/Foundation/ApplicationTest.php
@@ -97,6 +97,31 @@ public function toArray($request)
 
         $this->assertCount(1, DB::getQueryLog());
     }
+
+    public function testItcanRestoreValue()
+    {
+        Schema::create('foo', function ($table) {
+            $table->id();
+        });
+        Route::get('test-route', function () {
+            return new class('xxxx') extends JsonResource {
+                public function toArray($request)
+                {
+                    return [
+                        //
+                    ];
+                }
+            };
+        })->middleware(TestMiddleware::class);
+        DB::enableQueryLog();
+        $this->withoutExceptionHandling();
+
+        $this->app->preventQueriesWhilePreparingResponse();
+        $this->app->allowQueriesWhilePreparingResponse();
+        $this->get('test-route')->assertOk();
+
+        $this->assertCount(1, DB::getQueryLog());
+    }
 }
 
 class TestMiddleware