Make waiting for ExtensionAPIServer configurable (#694)

pkg/server/server.go

@@ -64,6 +64,7 @@ type Server struct {
 	cacheFactory *factory.CacheFactory
 
 	extensionAPIServer            ExtensionAPIServer
+	SkipWaitForExtensionAPIServer bool
 
 	authMiddleware      auth.Middleware
 	controllers         *Controllers
@@ -100,6 +101,9 @@ type Options struct {
 	// In most cases, you'll want to use [github.com/rancher/steve/pkg/ext.NewExtensionAPIServer]
 	// to create an ExtensionAPIServer.
 	ExtensionAPIServer ExtensionAPIServer
+
+	// SkipWaitForExtensionAPIServer allows serving requests despite the ExtensionAPIServer may not have been registered yet.
+	SkipWaitForExtensionAPIServer bool
 }
 
 func New(ctx context.Context, restConfig *rest.Config, opts *Options) (*Server, error) {
@@ -132,6 +136,7 @@ func New(ctx context.Context, restConfig *rest.Config, opts *Options) (*Server,
 		SQLCache:                      opts.SQLCache,
 		cacheFactory:                  cacheFactory,
 		extensionAPIServer:            opts.ExtensionAPIServer,
+		SkipWaitForExtensionAPIServer: opts.SkipWaitForExtensionAPIServer,
 	}
 
 	if err := setup(ctx, server); err != nil {
@@ -256,11 +261,16 @@ func setup(ctx context.Context, server *Server) error {
 		sf)
 
 	next := http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
+		if server.extensionAPIServer == nil || server.SkipWaitForExtensionAPIServer {
+			server.next.ServeHTTP(rw, req)
+			return
+		}
+
 		select {
 		case <-server.extensionAPIServer.Registered():
 			server.next.ServeHTTP(rw, req)
 		default:
-			http.NotFoundHandler().ServeHTTP(rw, req)
+			http.Error(rw, "API Aggregation not ready", http.StatusServiceUnavailable)
 		}
 	})