use of org.apache.ignite.internal.processors.rest.request.GridRestBaselineRequest in project ignite by apache.
the class GridJettyRestHandler method createRequest.
/**
* Creates REST request.
*
* @param cmd Command.
* @param params Parameters.
* @param req Servlet request.
* @return REST request.
* @throws IgniteCheckedException If creation failed.
*/
@Nullable
private GridRestRequest createRequest(GridRestCommand cmd, Map<String, String> params, HttpServletRequest req) throws IgniteCheckedException {
GridRestRequest restReq;
switch(cmd) {
case GET_OR_CREATE_CACHE:
{
GridRestCacheRequest restReq0 = new GridRestCacheRequest();
restReq0.cacheName(params.get(CACHE_NAME_PARAM));
String templateName = params.get(TEMPLATE_NAME_PARAM);
if (!F.isEmpty(templateName))
restReq0.templateName(templateName);
String backups = params.get(BACKUPS_PARAM);
CacheConfigurationOverride cfg = new CacheConfigurationOverride();
// Set cache backups.
if (!F.isEmpty(backups)) {
try {
cfg.backups(Integer.parseInt(backups));
} catch (NumberFormatException e) {
throw new IgniteCheckedException("Failed to parse number of cache backups: " + backups, e);
}
}
// Set cache group name.
String cacheGrp = params.get(CACHE_GROUP_PARAM);
if (!F.isEmpty(cacheGrp))
cfg.cacheGroup(cacheGrp);
// Set cache data region name.
String dataRegion = params.get(DATA_REGION_PARAM);
if (!F.isEmpty(dataRegion))
cfg.dataRegion(dataRegion);
// Set cache write mode.
String wrtSyncMode = params.get(WRITE_SYNCHRONIZATION_MODE_PARAM);
if (!F.isEmpty(wrtSyncMode)) {
try {
cfg.writeSynchronizationMode(CacheWriteSynchronizationMode.valueOf(wrtSyncMode));
} catch (IllegalArgumentException e) {
throw new IgniteCheckedException("Failed to parse cache write synchronization mode: " + wrtSyncMode, e);
}
}
if (!cfg.isEmpty())
restReq0.configuration(cfg);
restReq = restReq0;
break;
}
case DESTROY_CACHE:
{
GridRestCacheRequest restReq0 = new GridRestCacheRequest();
restReq0.cacheName(params.get(CACHE_NAME_PARAM));
restReq = restReq0;
break;
}
case ATOMIC_DECREMENT:
case ATOMIC_INCREMENT:
{
DataStructuresRequest restReq0 = new DataStructuresRequest();
restReq0.key(params.get("key"));
restReq0.initial(longValue("init", params, null));
restReq0.delta(longValue("delta", params, null));
restReq = restReq0;
break;
}
case CACHE_CONTAINS_KEY:
case CACHE_CONTAINS_KEYS:
case CACHE_GET:
case CACHE_GET_ALL:
case CACHE_GET_AND_PUT:
case CACHE_GET_AND_REPLACE:
case CACHE_PUT_IF_ABSENT:
case CACHE_GET_AND_PUT_IF_ABSENT:
case CACHE_PUT:
case CACHE_PUT_ALL:
case CACHE_REMOVE:
case CACHE_REMOVE_VALUE:
case CACHE_REPLACE_VALUE:
case CACHE_GET_AND_REMOVE:
case CACHE_REMOVE_ALL:
case CACHE_CLEAR:
case CACHE_ADD:
case CACHE_CAS:
case CACHE_METRICS:
case CACHE_SIZE:
case CACHE_UPDATE_TLL:
case CACHE_METADATA:
case CACHE_REPLACE:
case CACHE_APPEND:
case CACHE_PREPEND:
{
GridRestCacheRequest restReq0 = new GridRestCacheRequest();
String cacheName = params.get(CACHE_NAME_PARAM);
restReq0.cacheName(F.isEmpty(cacheName) ? null : cacheName);
String keyType = params.get("keyType");
String valType = params.get("valueType");
Converter converter = new Converter(cacheName);
restReq0.key(converter.convert(keyType, params.get("key")));
restReq0.value(converter.convert(valType, params.get("val")));
restReq0.value2(converter.convert(valType, params.get("val2")));
Object val1 = converter.convert(valType, params.get("val1"));
if (val1 != null)
restReq0.value(val1);
// Cache operations via REST will use binary objects.
restReq0.cacheFlags(intValue("cacheFlags", params, KEEP_BINARIES_MASK));
restReq0.ttl(longValue("exp", params, null));
if (cmd == CACHE_GET_ALL || cmd == CACHE_PUT_ALL || cmd == CACHE_REMOVE_ALL || cmd == CACHE_CONTAINS_KEYS) {
List<Object> keys = converter.values(keyType, "k", params);
List<Object> vals = converter.values(valType, "v", params);
if (keys.size() < vals.size())
throw new IgniteCheckedException("Number of keys must be greater or equals to number of values.");
Map<Object, Object> map = U.newHashMap(keys.size());
Iterator<Object> keyIt = keys.iterator();
Iterator<Object> valIt = vals.iterator();
while (keyIt.hasNext()) map.put(keyIt.next(), valIt.hasNext() ? valIt.next() : null);
restReq0.values(map);
}
restReq = restReq0;
break;
}
case TOPOLOGY:
case NODE:
{
GridRestTopologyRequest restReq0 = new GridRestTopologyRequest();
restReq0.includeMetrics(Boolean.parseBoolean(params.get("mtr")));
restReq0.includeAttributes(Boolean.parseBoolean(params.get("attr")));
String caches = params.get("caches");
restReq0.includeCaches(caches == null || Boolean.parseBoolean(caches));
restReq0.nodeIp(params.get("ip"));
restReq0.nodeId(uuidValue("id", params));
restReq = restReq0;
break;
}
case EXE:
case RESULT:
case NOOP:
{
GridRestTaskRequest restReq0 = new GridRestTaskRequest();
restReq0.taskId(params.get("id"));
restReq0.taskName(params.get("name"));
restReq0.params(new Converter().values(null, "p", params));
restReq0.async(Boolean.parseBoolean(params.get("async")));
restReq0.timeout(longValue("timeout", params, 0L));
restReq = restReq0;
break;
}
case LOG:
{
GridRestLogRequest restReq0 = new GridRestLogRequest();
restReq0.path(params.get("path"));
restReq0.from(intValue("from", params, -1));
restReq0.to(intValue("to", params, -1));
restReq = restReq0;
break;
}
case DATA_REGION_METRICS:
case DATA_STORAGE_METRICS:
case NAME:
case VERSION:
case PROBE:
{
restReq = new GridRestRequest();
break;
}
case CLUSTER_ACTIVE:
case CLUSTER_INACTIVE:
case CLUSTER_ACTIVATE:
case CLUSTER_DEACTIVATE:
case CLUSTER_CURRENT_STATE:
{
GridRestChangeStateRequest restReq0 = new GridRestChangeStateRequest();
if (cmd == CLUSTER_CURRENT_STATE)
restReq0.reqCurrentState();
else if (cmd == CLUSTER_ACTIVE || cmd == CLUSTER_ACTIVATE)
restReq0.active(true);
else
restReq0.active(false);
restReq0.forceDeactivation(booleanValue(GridRestClusterStateRequest.ARG_FORCE, params, false));
restReq = restReq0;
break;
}
case CLUSTER_STATE:
case CLUSTER_SET_STATE:
{
GridRestClusterStateRequest restReq0 = new GridRestClusterStateRequest();
if (cmd == CLUSTER_STATE)
restReq0.reqCurrentMode();
else {
ClusterState newState = enumValue("state", params, ClusterState.class);
restReq0.state(newState);
restReq0.forceDeactivation(booleanValue(GridRestClusterStateRequest.ARG_FORCE, params, false));
}
restReq = restReq0;
break;
}
case CLUSTER_NAME:
{
restReq = new GridRestClusterNameRequest();
break;
}
case BASELINE_CURRENT_STATE:
case BASELINE_SET:
case BASELINE_ADD:
case BASELINE_REMOVE:
{
GridRestBaselineRequest restReq0 = new GridRestBaselineRequest();
restReq0.topologyVersion(longValue("topVer", params, null));
restReq0.consistentIds(new Converter().values(null, "consistentId", params));
restReq = restReq0;
break;
}
case AUTHENTICATE:
{
restReq = new GridRestRequest();
break;
}
case ADD_USER:
case REMOVE_USER:
case UPDATE_USER:
{
RestUserActionRequest restReq0 = new RestUserActionRequest();
restReq0.user(params.get("user"));
restReq0.password(params.get("password"));
restReq = restReq0;
break;
}
case EXECUTE_SQL_QUERY:
case EXECUTE_SQL_FIELDS_QUERY:
{
RestQueryRequest restReq0 = new RestQueryRequest();
String cacheName = params.get(CACHE_NAME_PARAM);
restReq0.sqlQuery(params.get("qry"));
restReq0.arguments(new Converter(cacheName).values(null, "arg", params).toArray());
restReq0.typeName(params.get("type"));
String pageSize = params.get("pageSize");
if (pageSize != null)
restReq0.pageSize(Integer.parseInt(pageSize));
String keepBinary = params.get("keepBinary");
if (keepBinary != null)
restReq0.keepBinary(Boolean.parseBoolean(keepBinary));
String distributedJoins = params.get("distributedJoins");
if (distributedJoins != null)
restReq0.distributedJoins(Boolean.parseBoolean(distributedJoins));
restReq0.cacheName(cacheName);
if (cmd == EXECUTE_SQL_QUERY)
restReq0.queryType(RestQueryRequest.QueryType.SQL);
else
restReq0.queryType(RestQueryRequest.QueryType.SQL_FIELDS);
restReq = restReq0;
break;
}
case EXECUTE_SCAN_QUERY:
{
RestQueryRequest restReq0 = new RestQueryRequest();
restReq0.sqlQuery(params.get("qry"));
String pageSize = params.get("pageSize");
if (pageSize != null)
restReq0.pageSize(Integer.parseInt(pageSize));
restReq0.cacheName(params.get(CACHE_NAME_PARAM));
restReq0.className(params.get("className"));
restReq0.queryType(RestQueryRequest.QueryType.SCAN);
restReq = restReq0;
break;
}
case FETCH_SQL_QUERY:
{
RestQueryRequest restReq0 = new RestQueryRequest();
String qryId = params.get("qryId");
if (qryId != null)
restReq0.queryId(Long.parseLong(qryId));
String pageSize = params.get("pageSize");
if (pageSize != null)
restReq0.pageSize(Integer.parseInt(pageSize));
restReq0.cacheName(params.get(CACHE_NAME_PARAM));
restReq = restReq0;
break;
}
case CLOSE_SQL_QUERY:
{
RestQueryRequest restReq0 = new RestQueryRequest();
String qryId = params.get("qryId");
if (qryId != null)
restReq0.queryId(Long.parseLong(qryId));
restReq0.cacheName(params.get(CACHE_NAME_PARAM));
restReq = restReq0;
break;
}
case NODE_STATE_BEFORE_START:
{
restReq = new GridRestNodeStateBeforeStartRequest();
break;
}
case WARM_UP:
{
GridRestWarmUpRequest restReq0 = new GridRestWarmUpRequest();
restReq0.stopWarmUp(Boolean.parseBoolean(String.valueOf(params.get("stopWarmUp"))));
restReq = restReq0;
break;
}
default:
throw new IgniteCheckedException("Invalid command: " + cmd);
}
restReq.address(new InetSocketAddress(req.getRemoteAddr(), req.getRemotePort()));
restReq.command(cmd);
Object certs = req.getAttribute("javax.servlet.request.X509Certificate");
if (certs instanceof X509Certificate[])
restReq.certificates((X509Certificate[]) certs);
// TODO: In IGNITE 3.0 we should check credentials only for AUTHENTICATE command.
if (!credentials(params, IGNITE_LOGIN, IGNITE_PASSWORD, restReq))
credentials(params, USER_PARAM, PWD_PARAM, restReq);
String clientId = params.get("clientId");
try {
if (clientId != null)
restReq.clientId(UUID.fromString(clientId));
} catch (Exception ignored) {
// Ignore invalid client id. Rest handler will process this logic.
}
String destId = params.get("destId");
try {
if (destId != null)
restReq.destinationId(UUID.fromString(destId));
} catch (IllegalArgumentException ignored) {
// Don't fail - try to execute locally.
}
String sesTokStr = params.get("sessionToken");
try {
if (sesTokStr != null) {
// Token is a UUID encoded as 16 bytes as HEX.
byte[] bytes = U.hexString2ByteArray(sesTokStr);
if (bytes.length == 16)
restReq.sessionToken(bytes);
}
} catch (IllegalArgumentException ignored) {
// Ignore invalid session token.
}
return restReq;
}
use of org.apache.ignite.internal.processors.rest.request.GridRestBaselineRequest in project ignite by apache.
the class GridBaselineCommandHandler method handleAsync.
/**
* {@inheritDoc}
*/
@Override
public IgniteInternalFuture<GridRestResponse> handleAsync(GridRestRequest req) {
assert req != null;
assert SUPPORTED_COMMANDS.contains(req.command());
assert req instanceof GridRestBaselineRequest : "Invalid type of baseline request.";
if (log.isDebugEnabled())
log.debug("Handling baseline REST request: " + req);
GridRestBaselineRequest req0 = (GridRestBaselineRequest) req;
try {
IgniteClusterEx cluster = ctx.grid().cluster();
List<Object> consistentIds = req0.consistentIds();
switch(req0.command()) {
case BASELINE_CURRENT_STATE:
{
break;
}
case BASELINE_SET:
{
Long topVer = req0.topologyVersion();
if (topVer == null && consistentIds == null)
throw new IgniteCheckedException("Failed to handle request (either topVer or consistentIds should be specified).");
if (topVer != null)
cluster.setBaselineTopology(topVer);
else
cluster.setBaselineTopology(filterServerNodesByConsId(consistentIds));
break;
}
case BASELINE_ADD:
{
if (consistentIds == null)
throw new IgniteCheckedException(missingParameter("consistentIds"));
Set<BaselineNode> baselineTop = new HashSet<>(currentBaseLine());
baselineTop.addAll(filterServerNodesByConsId(consistentIds));
cluster.setBaselineTopology(baselineTop);
break;
}
case BASELINE_REMOVE:
{
if (consistentIds == null)
throw new IgniteCheckedException(missingParameter("consistentIds"));
Collection<BaselineNode> baseline = currentBaseLine();
Set<BaselineNode> baselineTop = new HashSet<>(baseline);
baselineTop.removeAll(filterNodesByConsId(baseline, consistentIds));
cluster.setBaselineTopology(baselineTop);
break;
}
default:
assert false : "Invalid command for baseline handler: " + req;
}
return new GridFinishedFuture<>(new GridRestResponse(currentState()));
} catch (IgniteCheckedException e) {
return new GridFinishedFuture<>(e);
} finally {
if (log.isDebugEnabled())
log.debug("Handled baseline REST request: " + req);
}
}
Aggregations