feat: support sse in api files (#5074)

Signed-off-by: Kevin Wan <wanjunfeng@gmail.com>

tools/goctl/api/gogen/genroutes_test.go

@@ -3,6 +3,8 @@ package gogen
 import (
 	"testing"
 	"time"
+
+	"github.com/zeromicro/go-zero/tools/goctl/api/spec"
 )
 
 func Test_formatDuration(t *testing.T) {
@@ -25,3 +27,173 @@ func Test_formatDuration(t *testing.T) {
 		}
 	}
 }
+
+func TestSSESupport(t *testing.T) {
+	// Test API spec with SSE enabled
+	apiSpec := &spec.ApiSpec{
+		Service: spec.Service{
+			Groups: []spec.Group{
+				{
+					Annotation: spec.Annotation{
+						Properties: map[string]string{
+							"sse":    "true",
+							"prefix": "/api/v1",
+						},
+					},
+					Routes: []spec.Route{
+						{
+							Method:  "get",
+							Path:    "/events",
+							Handler: "StreamEvents",
+						},
+					},
+				},
+			},
+		},
+	}
+
+	groups, err := getRoutes(apiSpec)
+	if err != nil {
+		t.Fatalf("getRoutes failed: %v", err)
+	}
+
+	if len(groups) != 1 {
+		t.Fatalf("Expected 1 group, got %d", len(groups))
+	}
+
+	group := groups[0]
+	if !group.sseEnabled {
+		t.Error("Expected SSE to be enabled")
+	}
+
+	if group.prefix != "/api/v1" {
+		t.Errorf("Expected prefix '/api/v1', got '%s'", group.prefix)
+	}
+
+	if len(group.routes) != 1 {
+		t.Fatalf("Expected 1 route, got %d", len(group.routes))
+	}
+
+	route := group.routes[0]
+	if route.method != "http.MethodGet" {
+		t.Errorf("Expected method 'http.MethodGet', got '%s'", route.method)
+	}
+
+	if route.path != "/events" {
+		t.Errorf("Expected path '/events', got '%s'", route.path)
+	}
+}
+
+func TestSSEWithOtherFeatures(t *testing.T) {
+	// Test API spec with SSE and other features
+	apiSpec := &spec.ApiSpec{
+		Service: spec.Service{
+			Groups: []spec.Group{
+				{
+					Annotation: spec.Annotation{
+						Properties: map[string]string{
+							"sse":        "true",
+							"jwt":        "Auth",
+							"signature":  "true",
+							"prefix":     "/api/v1",
+							"timeout":    "30s",
+							"middleware": "AuthMiddleware,LogMiddleware",
+						},
+					},
+					Routes: []spec.Route{
+						{
+							Method:  "get",
+							Path:    "/events",
+							Handler: "StreamEvents",
+						},
+					},
+				},
+			},
+		},
+	}
+
+	groups, err := getRoutes(apiSpec)
+	if err != nil {
+		t.Fatalf("getRoutes failed: %v", err)
+	}
+
+	if len(groups) != 1 {
+		t.Fatalf("Expected 1 group, got %d", len(groups))
+	}
+
+	group := groups[0]
+
+	// Verify all features are enabled
+	if !group.sseEnabled {
+		t.Error("Expected SSE to be enabled")
+	}
+
+	if !group.jwtEnabled {
+		t.Error("Expected JWT to be enabled")
+	}
+
+	if !group.signatureEnabled {
+		t.Error("Expected signature to be enabled")
+	}
+
+	if group.authName != "Auth" {
+		t.Errorf("Expected authName 'Auth', got '%s'", group.authName)
+	}
+
+	if group.prefix != "/api/v1" {
+		t.Errorf("Expected prefix '/api/v1', got '%s'", group.prefix)
+	}
+
+	if group.timeout != "30s" {
+		t.Errorf("Expected timeout '30s', got '%s'", group.timeout)
+	}
+
+	expectedMiddlewares := []string{"AuthMiddleware", "LogMiddleware"}
+	if len(group.middlewares) != len(expectedMiddlewares) {
+		t.Errorf("Expected %d middlewares, got %d", len(expectedMiddlewares), len(group.middlewares))
+	}
+
+	for i, expected := range expectedMiddlewares {
+		if group.middlewares[i] != expected {
+			t.Errorf("Expected middleware[%d] '%s', got '%s'", i, expected, group.middlewares[i])
+		}
+	}
+}
+
+func TestSSEDisabled(t *testing.T) {
+	// Test API spec without SSE
+	apiSpec := &spec.ApiSpec{
+		Service: spec.Service{
+			Groups: []spec.Group{
+				{
+					Annotation: spec.Annotation{
+						Properties: map[string]string{
+							"prefix": "/api/v1",
+						},
+					},
+					Routes: []spec.Route{
+						{
+							Method:  "get",
+							Path:    "/status",
+							Handler: "GetStatus",
+						},
+					},
+				},
+			},
+		},
+	}
+
+	groups, err := getRoutes(apiSpec)
+	if err != nil {
+		t.Fatalf("getRoutes failed: %v", err)
+	}
+
+	if len(groups) != 1 {
+		t.Fatalf("Expected 1 group, got %d", len(groups))
+	}
+
+	group := groups[0]
+	if group.sseEnabled {
+		t.Error("Expected SSE to be disabled")
+	}
+}