Fix import issue

Author: lwj5

modules/openapi-generator/src/main/resources/go-server/routers.mustache

@@ -13,6 +13,7 @@ import (
 	{{/mux}}
 	{{#chi}}
 	"github.com/go-chi/chi/v5"
+	"github.com/go-chi/chi/v5/middleware"
 	{{#featureCORS}}
 	"github.com/go-chi/cors"
 	{{/featureCORS}}
@@ -45,14 +46,25 @@ type Router interface {
 const errMsgRequiredMissing = "required parameter is missing"
 
 // NewRouter creates a new router for any number of api routers
+func NewRouter(routers ...Router) {{#routers}}{{#mux}}*mux.Router{{/mux}}{{#chi}}chi.Router{{/chi}}{{/routers}} {
 {{#routers}}
 	{{#mux}}
-func NewRouter(routers ...Router) *mux.Router {
 	router := mux.NewRouter().StrictSlash(true)
+	{{/mux}}
+	{{#chi}}
+	router := chi.NewRouter()
+	router.Use(middleware.Logger)
+	{{#featureCORS}}
+	router.Use(cors.Handler(cors.Options{}))
+	{{/featureCORS}}
+	{{/chi}}
+{{/routers}}
 	for _, api := range routers {
 		for _, route := range api.Routes() {
 			var handler http.Handler
 			handler = route.HandlerFunc
+{{#routers}}
+	{{#mux}}
 			handler = Logger(handler, route.Name)
 			{{#featureCORS}}
 			handler = handlers.CORS()(handler)
@@ -63,26 +75,13 @@ func NewRouter(routers ...Router) *mux.Router {
 				Path(route.Pattern).
 				Name(route.Name).
 				Handler(handler)
-		}
-	}
 	{{/mux}}
 	{{#chi}}
-func NewRouter(routers ...Router) chi.Router {
-	router := chi.NewRouter()
-	{{#featureCORS}}
-	r.Use(cors.Handler(cors.Options{}))
-	{{/featureCORS}}
-	for _, api := range routers {
-		for _, route := range api.Routes() {
-			var handler http.Handler
-			handler = route.HandlerFunc
-			handler = Logger(handler, route.Name)
-
 			router.Method(route.Method, route.Pattern, handler)
-		}
-	}
 	{{/chi}}
 {{/routers}}
+		}
+	}
 
 	return router
 }