org.openqa.grid.internal.utils.configuration.GridHubConfiguration Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of selenium-server Show documentation
Show all versions of selenium-server Show documentation
Selenium automates browsers. That's it! What you do with that power is entirely up to you.
// 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.grid.internal.utils.configuration;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonObject;
import com.google.gson.TypeAdapter;
import com.google.gson.annotations.Expose;
import com.google.gson.stream.JsonReader;
import com.google.gson.stream.JsonWriter;
import com.beust.jcommander.Parameter;
import org.openqa.grid.common.JSONConfigurationUtils;
import org.openqa.grid.common.exception.GridConfigurationException;
import org.openqa.grid.internal.listeners.Prioritizer;
import org.openqa.grid.internal.utils.CapabilityMatcher;
import org.openqa.grid.internal.utils.DefaultCapabilityMatcher;
import org.openqa.grid.internal.utils.configuration.converters.StringToClassConverter;
import org.openqa.grid.internal.utils.configuration.validators.FileExistsValueValidator;
import java.io.IOException;
public class GridHubConfiguration extends GridConfiguration {
public static final String DEFUALT_HUB_CONFIG_FILE = "defaults/DefaultHub.json";
private static final GridHubConfiguration DEFAULT_CONFIG = loadFromJSON(DEFUALT_HUB_CONFIG_FILE);
/*
* config parameters which do not serialize or de-serialize
*/
@Parameter(
names = "-hubConfig",
description = " filename: a JSON file (following grid2 format), which defines the hub properties",
validateValueWith = FileExistsValueValidator.class
)
public String hubConfig;
/*
* config parameters which serialize and deserialize to/from json
*/
@Expose
@Parameter(
names = { "-matcher", "-capabilityMatcher" },
description = " class name : a class implementing the CapabilityMatcher interface. Specifies the logic the hub will follow to define whether a request can be assigned to a node. For example, if you want to have the matching process use regular expressions instead of exact match when specifying browser version. ALL nodes of a grid ecosystem would then use the same capabilityMatcher, as defined here. Default is org.openqa.grid.internal.utils.DefaultCapabilityMatcher",
converter = StringToClassConverter.CapabilityMatcherStringConverter.class
)
public CapabilityMatcher capabilityMatcher = new DefaultCapabilityMatcher();
@Expose
@Parameter(
names = "-newSessionWaitTimeout",
description = " in ms : The time after which a new test waiting for a node to become available will time out. When that happens, the test will throw an exception before attempting to start a browser. Defaults to no timeout ( -1 )"
)
public Integer newSessionWaitTimeout = -1;
@Expose
@Parameter(
names = "-prioritizer",
description = " class name : a class implementing the Prioritizer interface. Specify a custom Prioritizer if you want to sort the order in which new session requests are processed when there is a queue. Default to null ( no priority = FIFO )",
converter = StringToClassConverter.PrioritizerStringConverter.class
)
public Prioritizer prioritizer = null;
@Expose
@Parameter(
names = "-throwOnCapabilityNotPresent",
description = " true or false : If true, the hub will reject all test requests if no compatible proxy is currently registered. If set to false, the request will queue until a node supporting the capability is registered with the grid. Default is true"
)
public Boolean throwOnCapabilityNotPresent = true;
/**
* Init with built-in defaults
*/
public GridHubConfiguration() {
role = "hub";
if (DEFAULT_CONFIG != null) {
merge(DEFAULT_CONFIG);
}
}
/**
* @param filePath hub config json file to load configuration from
*/
public static GridHubConfiguration loadFromJSON(String filePath) {
return loadFromJSON(JSONConfigurationUtils.loadJSON(filePath));
}
/**
* @param json JsonObject to load configuration from
*/
public static GridHubConfiguration loadFromJSON(JsonObject json) {
try {
GsonBuilder builder = new GsonBuilder();
GridHubConfiguration.staticAddJsonTypeAdapter(builder);
return builder.excludeFieldsWithoutExposeAnnotation().create()
.fromJson(json, GridHubConfiguration.class);
} catch (Throwable e) {
throw new GridConfigurationException("Error with the JSON of the config : " + e.getMessage(),
e);
}
}
public void merge(GridNodeConfiguration other) {
super.merge(other);
}
public void merge(GridHubConfiguration other) {
super.merge(other);
if (isMergeAble(other.capabilityMatcher, capabilityMatcher)) {
capabilityMatcher = other.capabilityMatcher;
}
if (isMergeAble(other.newSessionWaitTimeout, newSessionWaitTimeout)) {
newSessionWaitTimeout = other.newSessionWaitTimeout;
}
if (isMergeAble(other.prioritizer, prioritizer)) {
prioritizer = other.prioritizer;
}
if (isMergeAble(other.throwOnCapabilityNotPresent, throwOnCapabilityNotPresent)) {
throwOnCapabilityNotPresent = other.throwOnCapabilityNotPresent;
}
}
@Override
public String toString(String format) {
StringBuilder sb = new StringBuilder();
sb.append(super.toString(format));
sb.append(toString(format, "hubConfig", hubConfig));
sb.append(toString(format, "capabilityMatcher", capabilityMatcher.getClass().getCanonicalName()));
sb.append(toString(format, "newSessionWaitTimeout", newSessionWaitTimeout));
sb.append(toString(format, "prioritizer", prioritizer != null? prioritizer.getClass().getCanonicalName(): null));
sb.append(toString(format, "throwOnCapabilityNotPresent", throwOnCapabilityNotPresent));
return sb.toString();
}
@Override
protected void addJsonTypeAdapter(GsonBuilder builder) {
super.addJsonTypeAdapter(builder);
GridHubConfiguration.staticAddJsonTypeAdapter(builder);
}
protected static void staticAddJsonTypeAdapter(GsonBuilder builder) {
builder.registerTypeAdapter(CapabilityMatcher.class, new CapabilityMatcherAdapter().nullSafe());
builder.registerTypeAdapter(Prioritizer.class, new PrioritizerAdapter().nullSafe());
}
protected static class SimpleClassNameAdapter extends TypeAdapter {
@Override
public void write(JsonWriter out, T value) throws IOException {
out.value(value.getClass().getCanonicalName());
}
@Override
public T read(JsonReader in) throws IOException {
String value = in.nextString();
try {
return (T) Class.forName(value).newInstance();
} catch (Exception e) {
throw new RuntimeException(String.format("String %s could not be coerced to class: %s", value, Class.class.getName()), e);
}
}
}
protected static class CapabilityMatcherAdapter extends SimpleClassNameAdapter {
}
protected static class PrioritizerAdapter extends SimpleClassNameAdapter {
}
}