Update engine.go

engine.go

@@ -43,29 +43,42 @@
 }
 
 type OpenAPIConfig struct {
+	MiddlewareDisplayLimit int
+}
+
+func DefaultOpenAPIConfig() OpenAPIConfig {
+    return OpenAPIConfig{
+        MiddlewareDisplayLimit: 5,
+    }
 	// Local path to save the OpenAPI JSON spec
 	JSONFilePath string
 	// If true, the server will not serve nor generate any OpenAPI resources
 	Disabled bool
 	// If true, the engine will not print messages
 	DisableMessages bool
 	// If true, the engine will not save the OpenAPI JSON spec locally
 	DisableLocalSave bool
 	// Pretty prints the OpenAPI spec with proper JSON indentation
 	PrettyFormatJSON bool
 }
 
 var defaultOpenAPIConfig = OpenAPIConfig{
 	JSONFilePath: "doc/openapi.json",
 }
+//implemnt action
+func WithOpenAPIConfig(config OpenAPIConfig) func(*Engine) {
+    return func(e *Engine) {
+        e.openAPIConfig = config
+    }
+}
 
 // WithRequestContentType sets the accepted content types for the engine.
 // By default, the accepted content types is */*.
 func WithRequestContentType(consumes ...string) func(*Engine) {
 	return func(e *Engine) { e.requestContentTypes = consumes }
 }
 
 func WithOpenAPIConfig(config OpenAPIConfig) func(*Engine) {
 	return func(e *Engine) {
 		if config.JSONFilePath != "" {
 			e.OpenAPIConfig.JSONFilePath = config.JSONFilePath
@@ -77,6 +90,14 @@
 	}
 }
 
+func (e *Engine) DisplayMiddlewares() []Middleware {
+    limit := e.openAPIConfig.MiddlewareDisplayLimit
+    if len(e.middlewares) < limit {
+        return e.middlewares
+    }
+    return e.middlewares[:limit]
+}
+
 // WithErrorHandler sets a customer error handler for the server
 func WithErrorHandler(errorHandler func(err error) error) func(*Engine) {
 	return func(e *Engine) {
@@ -150,6 +171,28 @@
 	return json.Marshal(s.OpenAPI.Description())
 }
 
+func TestDefaultMiddlewareLimit(t *testing.T) {
+    engine := NewEngine(DefaultOpenAPIConfig())
+    engine.middlewares = []Middleware{"A", "B", "C", "D", "E", "F"}
+
+    displayed := engine.DisplayMiddlewares()
+    if len(displayed) != 5 {
+        t.Errorf("Expected 5 middlewares, got %d", len(displayed))
+    }
+}
+
+func TestCustomMiddlewareLimit(t *testing.T) {
+    config := DefaultOpenAPIConfig()
+    config.MiddlewareDisplayLimit = 3
+    engine := NewEngine(config)
+    engine.middlewares = []Middleware{"A", "B", "C", "D", "E"}
+
+    displayed := engine.DisplayMiddlewares()
+    if len(displayed) != 3 {
+        t.Errorf("Expected 3 middlewares, got %d", len(displayed))
+    }
+}
+
 func (e *Engine) printOpenAPIMessage(msg string) {
 	if !e.OpenAPIConfig.DisableMessages {
 		slog.Info(msg)