package server import ( v1 "eino-project/api/customer/v1" "eino-project/internal/conf" "eino-project/internal/service" "github.com/go-kratos/kratos/v2/log" "github.com/go-kratos/kratos/v2/middleware/recovery" "github.com/go-kratos/kratos/v2/transport/http" ) // NewHTTPServer new an HTTP server. func NewHTTPServer(c *conf.Server, customerService *service.CustomerService, logger log.Logger) *http.Server { var opts = []http.ServerOption{ http.Middleware( recovery.Recovery(), ), } if c.Http.Network != "" { opts = append(opts, http.Network(c.Http.Network)) } if c.Http.Addr != "" { opts = append(opts, http.Address(c.Http.Addr)) } if c.Http.Timeout != nil { opts = append(opts, http.Timeout(c.Http.Timeout.AsDuration())) } srv := http.NewServer(opts...) // 注册标准的gRPC-Gateway路由 v1.RegisterCustomerServiceHTTPServer(srv, customerService) // 添加SSE流式聊天的自定义路由 srv.HandleFunc("/api/chat/stream", customerService.HandleStreamChat) return srv }