From 98581635943ca28633294d8849014bd2fc852d36 Mon Sep 17 00:00:00 2001
From: lafer-fz <42789804+lafer-fz@users.noreply.github.com>
Date: Tue, 16 Apr 2024 15:44:58 +0800
Subject: [PATCH] fix: import alias index "

---
 cmd/hz/generator/custom_files.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cmd/hz/generator/custom_files.go b/cmd/hz/generator/custom_files.go
index 1315948f1..d3a905ba7 100644
--- a/cmd/hz/generator/custom_files.go
+++ b/cmd/hz/generator/custom_files.go
@@ -259,7 +259,7 @@ func getInsertImportContent(tplInfo *Template, renderInfo interface{}, fileConte
 				}
 				imptSlice = append(imptSlice, [2]string{"", impt[1 : len(impt)-1]})
 			} else { // 3. alias "import"
-				idx := strings.Index(impt, "\n")
+				idx := strings.Index(impt, "\"")
 				if idx == -1 {
 					return nil, fmt.Errorf("error import format for file: %s", tplInfo.Path)
 				}