From 14f2726694274aa6dadd2993c01b197d145b9076 Mon Sep 17 00:00:00 2001
From: tangwei <tw2066@163.com>
Date: Fri, 22 Nov 2024 20:45:58 +0800
Subject: [PATCH] optimize code

---
 src/Listener/AfterDtoStartListener.php | 7 +++++--
 src/Swagger/SwaggerOpenApi.php         | 4 ++--
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/Listener/AfterDtoStartListener.php b/src/Listener/AfterDtoStartListener.php
index ba5a2cc..6bc09c4 100644
--- a/src/Listener/AfterDtoStartListener.php
+++ b/src/Listener/AfterDtoStartListener.php
@@ -26,7 +26,8 @@ public function __construct(
         private SwaggerComponents $swaggerComponents,
         private SwaggerConfig $swaggerConfig,
         private DtoConfig $dtoConfig,
-    ) {}
+    ) {
+    }
 
     public function listen(): array
     {
@@ -49,11 +50,13 @@ public function process(object $event): void
         if (! $this->swaggerConfig->getOutputDir()) {
             return;
         }
+
+        $this->swaggerOpenApi->init($server['name']);
+
         if ($this->dtoConfig->isScanCacheable()) {
             return;
         }
 
-        $this->swaggerOpenApi->init();
         /** @var SwaggerPaths $swagger */
         $swagger = make(SwaggerPaths::class, [$server['name']]);
         foreach ($router->getData() ?? [] as $routeData) {
diff --git a/src/Swagger/SwaggerOpenApi.php b/src/Swagger/SwaggerOpenApi.php
index 8f9bfa7..5065b4a 100644
--- a/src/Swagger/SwaggerOpenApi.php
+++ b/src/Swagger/SwaggerOpenApi.php
@@ -30,7 +30,7 @@ public function __construct(
     ) {
     }
 
-    public function init(): void
+    public function init(string $serverName): void
     {
         $this->openApi = new OpenApi();
         $this->openApi->paths = [];
@@ -45,6 +45,7 @@ public function init(): void
         $this->setComponentsSecuritySchemes();
         $this->setSecurity();
         $this->setExternalDocs();
+        $this->serverNameAll[] = $serverName;
     }
 
     public function clean(): void
@@ -139,7 +140,6 @@ public function save(string $serverName): void
         }
         $outputFile = $outputDir . '/' . $serverName . '.' . $this->swaggerConfig->getFormat();
         $this->openApi->saveAs($outputFile);
-        $this->serverNameAll[] = $serverName;
     }
 
     protected function setInfo(): void