Search in sources :

Example 1 with OHttpRequestWrapper

use of com.orientechnologies.orient.server.network.protocol.http.OHttpRequestWrapper in project orientdb by orientechnologies.

the class OServerCommandAbstractLogic method execute.

@Override
public boolean execute(final OHttpRequest iRequest, final OHttpResponse iResponse) throws Exception {
    final String[] parts = init(iRequest, iResponse);
    ODatabaseDocument db = null;
    try {
        db = getProfiledDatabaseInstance(iRequest);
        final OFunction f = db.getMetadata().getFunctionLibrary().getFunction(parts[2]);
        if (f == null)
            throw new IllegalArgumentException("Function '" + parts[2] + "' is not configured");
        if (iRequest.httpMethod.equalsIgnoreCase("GET") && !f.isIdempotent()) {
            iResponse.send(OHttpUtils.STATUS_BADREQ_CODE, OHttpUtils.STATUS_BADREQ_DESCRIPTION, OHttpUtils.CONTENT_TEXT_PLAIN, "GET method is not allowed to execute function '" + parts[2] + "' because has been declared as non idempotent. Use POST instead.", null);
            return false;
        }
        Object[] args = new String[parts.length - 3];
        for (int i = 3; i < parts.length; ++i) args[i - 3] = parts[i];
        // BIND CONTEXT VARIABLES
        final OBasicCommandContext context = new OBasicCommandContext();
        context.setVariable("session", OHttpSessionManager.getInstance().getSession(iRequest.sessionId));
        context.setVariable("request", new OHttpRequestWrapper(iRequest, (String[]) args));
        context.setVariable("response", new OHttpResponseWrapper(iResponse));
        final Object functionResult;
        if (args.length == 0 && iRequest.content != null && !iRequest.content.isEmpty()) {
            // PARSE PARAMETERS FROM CONTENT PAYLOAD
            try {
                final ODocument params = new ODocument().fromJSON(iRequest.content);
                functionResult = f.executeInContext(context, params.toMap());
            } catch (Exception e) {
                throw OException.wrapException(new OCommandScriptException("Error on parsing parameters from request body"), e);
            }
        } else
            functionResult = f.executeInContext(context, args);
        handleResult(iRequest, iResponse, functionResult);
    } catch (OCommandScriptException e) {
        // EXCEPTION
        final StringBuilder msg = new StringBuilder(256);
        for (Exception currentException = e; currentException != null; currentException = (Exception) currentException.getCause()) {
            if (msg.length() > 0)
                msg.append("\n");
            msg.append(currentException.getMessage());
        }
        if (isJsonResponse(iResponse)) {
            sendJsonError(iResponse, OHttpUtils.STATUS_BADREQ_CODE, OHttpUtils.STATUS_BADREQ_DESCRIPTION, OHttpUtils.CONTENT_TEXT_PLAIN, msg.toString(), null);
        } else {
            iResponse.send(OHttpUtils.STATUS_BADREQ_CODE, OHttpUtils.STATUS_BADREQ_DESCRIPTION, OHttpUtils.CONTENT_TEXT_PLAIN, msg.toString(), null);
        }
    } finally {
        if (db != null)
            db.close();
    }
    return false;
}
Also used : OCommandScriptException(com.orientechnologies.orient.core.command.script.OCommandScriptException) OBasicCommandContext(com.orientechnologies.orient.core.command.OBasicCommandContext) OHttpResponseWrapper(com.orientechnologies.orient.server.network.protocol.http.OHttpResponseWrapper) OException(com.orientechnologies.common.exception.OException) IOException(java.io.IOException) OCommandScriptException(com.orientechnologies.orient.core.command.script.OCommandScriptException) OFunction(com.orientechnologies.orient.core.metadata.function.OFunction) OHttpRequestWrapper(com.orientechnologies.orient.server.network.protocol.http.OHttpRequestWrapper) ODatabaseDocument(com.orientechnologies.orient.core.db.document.ODatabaseDocument) ODocument(com.orientechnologies.orient.core.record.impl.ODocument)

Aggregations

OException (com.orientechnologies.common.exception.OException)1 OBasicCommandContext (com.orientechnologies.orient.core.command.OBasicCommandContext)1 OCommandScriptException (com.orientechnologies.orient.core.command.script.OCommandScriptException)1 ODatabaseDocument (com.orientechnologies.orient.core.db.document.ODatabaseDocument)1 OFunction (com.orientechnologies.orient.core.metadata.function.OFunction)1 ODocument (com.orientechnologies.orient.core.record.impl.ODocument)1 OHttpRequestWrapper (com.orientechnologies.orient.server.network.protocol.http.OHttpRequestWrapper)1 OHttpResponseWrapper (com.orientechnologies.orient.server.network.protocol.http.OHttpResponseWrapper)1 IOException (java.io.IOException)1