Add a Hub to the new grid, emulating the grid3 hub

java/server/src/org/openqa/selenium/grid/commands/Hub.java

@@ -0,0 +1,98 @@
+// Licensed to the Software Freedom Conservancy (SFC) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The SFC licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+package org.openqa.selenium.grid.commands;
+
+import com.google.auto.service.AutoService;
+
+import com.beust.jcommander.JCommander;
+import com.beust.jcommander.ParameterException;
+
+import org.openqa.selenium.cli.CliCommand;
+import org.openqa.selenium.grid.config.AnnotatedConfig;
+import org.openqa.selenium.grid.config.CompoundConfig;
+import org.openqa.selenium.grid.config.ConcatenatingConfig;
+import org.openqa.selenium.grid.config.Config;
+import org.openqa.selenium.grid.config.EnvConfig;
+import org.openqa.selenium.grid.distributor.Distributor;
+import org.openqa.selenium.grid.distributor.local.LocalDistributor;
+import org.openqa.selenium.grid.node.local.NodeFlags;
+import org.openqa.selenium.grid.router.Router;
+import org.openqa.selenium.grid.server.BaseServer;
+import org.openqa.selenium.grid.server.BaseServerFlags;
+import org.openqa.selenium.grid.server.BaseServerOptions;
+import org.openqa.selenium.grid.server.HelpFlags;
+import org.openqa.selenium.grid.server.Server;
+import org.openqa.selenium.grid.server.W3CCommandHandler;
+import org.openqa.selenium.grid.sessionmap.SessionMap;
+import org.openqa.selenium.grid.sessionmap.local.LocalSessionMap;
+
+@AutoService(CliCommand.class)
+public class Hub implements CliCommand {
+
+  @Override
+  public String getName() {
+    return "hub";
+  }
+
+  @Override
+  public String getDescription() {
+    return "A grid hub, composed of sessions, distributor, and router";
+  }
+
+  @Override
+  public Executable configure(String... args) {
+    HelpFlags help = new HelpFlags();
+    BaseServerFlags baseFlags = new BaseServerFlags(4444);
+    NodeFlags nodeFlags = new NodeFlags();
+
+    JCommander commander = JCommander.newBuilder()
+        .programName("standalone")
+        .addObject(baseFlags)
+        .addObject(help)
+        .addObject(nodeFlags)
+        .build();
+
+    return () -> {
+      try {
+        commander.parse(args);
+      } catch (ParameterException e) {
+        System.err.println(e.getMessage());
+        commander.usage();
+        return;
+      }
+
+      if (help.displayHelp(commander, System.out)) {
+        return;
+      }
+
+      Config config = new CompoundConfig(
+          new AnnotatedConfig(help),
+          new AnnotatedConfig(baseFlags),
+          new EnvConfig(),
+          new ConcatenatingConfig("selenium", '.', System.getProperties()));
+
+      SessionMap sessions = new LocalSessionMap();
+      Distributor distributor = new LocalDistributor();
+      Router router = new Router(sessions, distributor);
+
+      Server<?> server = new BaseServer<>(new BaseServerOptions(config));
+      server.addHandler(router, (inj, req) -> new W3CCommandHandler(router));
+      server.start();
+    };
+  }
+}