package com.tinkerpop.gremlin.server.op.session;
import com.tinkerpop.gremlin.driver.Tokens;
import com.tinkerpop.gremlin.driver.message.RequestMessage;
import com.tinkerpop.gremlin.driver.message.ResponseMessage;
import com.tinkerpop.gremlin.driver.message.ResponseStatusCode;
import com.tinkerpop.gremlin.server.Context;
import com.tinkerpop.gremlin.server.Settings;
import com.tinkerpop.gremlin.server.op.AbstractEvalOpProcessor;
import com.tinkerpop.gremlin.server.op.OpProcessorException;
import com.tinkerpop.gremlin.util.function.ThrowingConsumer;
import java.util.HashMap;
import java.util.Optional;
/**
* Simple {@link com.tinkerpop.gremlin.server.OpProcessor} implementation that handles {@code ScriptEngine}
* script evaluation in the context of a session.
*
* @author Stephen Mallette (http://stephen.genoprime.com)
*/
public class SessionOpProcessor extends AbstractEvalOpProcessor {
public static final String OP_PROCESSOR_NAME = "session";
/**
* Configuration setting for how long a session will be available before it timesout.
*/
public static final String CONFIG_SESSION_TIMEOUT = "sessionTimeout";
/**
* Default timeout for a session is eight hours.
*/
public static final long DEFAULT_SESSION_TIMEOUT = 28800000l;
static final Settings.ProcessorSettings DEFAULT_SETTINGS = new Settings.ProcessorSettings();
static {
DEFAULT_SETTINGS.className = SessionOpProcessor.class.getCanonicalName();
DEFAULT_SETTINGS.config = new HashMap<String,Object>() {{
put(CONFIG_SESSION_TIMEOUT, DEFAULT_SESSION_TIMEOUT);
}};
}
@Override
public String getName() {
return OP_PROCESSOR_NAME;
}
@Override
public ThrowingConsumer<Context> getEvalOp() {
return SessionOps::evalOp;
}
@Override
protected Optional<ThrowingConsumer<Context>> validateEvalMessage(final RequestMessage message) throws OpProcessorException {
super.validateEvalMessage(message);
if (!message.optionalArgs(Tokens.ARGS_SESSION).isPresent()) {
final String msg = String.format("A message with an [%s] op code requires a [%s] argument.", Tokens.OPS_EVAL, Tokens.ARGS_SESSION);
throw new OpProcessorException(msg, ResponseMessage.build(message).code(ResponseStatusCode.REQUEST_ERROR_INVALID_REQUEST_ARGUMENTS).result(msg).create());
}
return Optional.empty();
}
}