org.apache.directory.server.ldap.handlers.extended.StoredProcedureExtendedOperationHandler.java Source code

Introduction

Here is the source code for org.apache.directory.server.ldap.handlers.extended.StoredProcedureExtendedOperationHandler.java

Source

/*
 *  Licensed to the Apache Software Foundation (ASF) under one
 *  or more contributor license agreements.  See the NOTICE file
 *  distributed with this work for additional information
 *  regarding copyright ownership.  The ASF 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.apache.directory.server.ldap.handlers.extended;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import org.apache.commons.lang.SerializationUtils;
import org.apache.directory.api.ldap.codec.api.LdapApiServiceFactory;
import org.apache.directory.api.ldap.extras.extended.storedProcedure.StoredProcedureRequest;
import org.apache.directory.api.ldap.extras.extended.storedProcedure.StoredProcedureResponse;
import org.apache.directory.api.ldap.model.entry.Entry;
import org.apache.directory.api.ldap.model.name.Dn;
import org.apache.directory.api.ldap.sp.LdapContextParameter;
import org.apache.directory.server.core.api.sp.StoredProcEngine;
import org.apache.directory.server.core.api.sp.StoredProcEngineConfig;
import org.apache.directory.server.core.api.sp.StoredProcExecutionManager;
import org.apache.directory.server.core.api.sp.java.JavaStoredProcEngineConfig;
import org.apache.directory.server.ldap.ExtendedOperationHandler;
import org.apache.directory.server.ldap.LdapServer;
import org.apache.directory.server.ldap.LdapSession;

/**
 * A Handler for the StoredProcedure extended operation
 * 
 * @author <a href="mailto:dev@directory.apache.org">Apache Directory Project</a>
 */
public class StoredProcedureExtendedOperationHandler
        implements ExtendedOperationHandler<StoredProcedureRequest, StoredProcedureResponse> {
    private StoredProcExecutionManager manager;
    private static final Object[] EMPTY_CLASS_ARRAY = new Object[0];

    public StoredProcedureExtendedOperationHandler() {
        super();
        //StoredProcEngineConfig javaxScriptSPEngineConfig = new JavaxStoredProcEngineConfig();
        StoredProcEngineConfig javaSPEngineConfig = new JavaStoredProcEngineConfig();
        List<StoredProcEngineConfig> spEngineConfigs = new ArrayList<>();
        //spEngineConfigs.add( javaxScriptSPEngineConfig );
        spEngineConfigs.add(javaSPEngineConfig);
        String spContainer = "ou=Stored Procedures,ou=system";
        this.manager = new StoredProcExecutionManager(spContainer, spEngineConfigs);
    }

    public void handleExtendedOperation(LdapSession session, StoredProcedureRequest req) throws Exception {
        String procedure = req.getProcedureSpecification();
        Entry spUnit = manager.findStoredProcUnit(session.getCoreSession(), procedure);
        StoredProcEngine engine = manager.getStoredProcEngineInstance(spUnit);

        List<Object> valueList = new ArrayList<>(req.size());

        for (int ii = 0; ii < req.size(); ii++) {
            byte[] serializedValue = (byte[]) req.getParameterValue(ii);
            Object value = SerializationUtils.deserialize(serializedValue);

            if (value.getClass().equals(LdapContextParameter.class)) {
                String paramCtx = ((LdapContextParameter) value).getValue();
                value = session.getCoreSession().lookup(new Dn(paramCtx));
            }

            valueList.add(value);
        }

        Object[] values = valueList.toArray(EMPTY_CLASS_ARRAY);
        Object response = engine.invokeProcedure(session.getCoreSession(), procedure, values);
        byte[] serializedResponse = SerializationUtils.serialize((Serializable) response);
        StoredProcedureResponse resp = LdapApiServiceFactory.getSingleton()
                .newExtendedResponse(req.getRequestName(), req.getMessageId(), serializedResponse);
        session.getIoSession().write(resp);
    }

    /**
     * {@inheritDoc}
     */
    public String getOid() {
        return StoredProcedureRequest.EXTENSION_OID;
    }

    private static final Set<String> EXTENSION_OIDS;

    static {
        Set<String> s = new HashSet<>();
        s.add(StoredProcedureRequest.EXTENSION_OID);
        s.add(StoredProcedureResponse.EXTENSION_OID);
        EXTENSION_OIDS = Collections.unmodifiableSet(s);
    }

    /**
     * {@inheritDoc}
     */
    public Set<String> getExtensionOids() {
        return EXTENSION_OIDS;
    }

    /**
     * {@inheritDoc}
     */
    public void setLdapServer(LdapServer ldapServer) {
    }
}