org.drools.grid.remote.GridNodeRemoteClient Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2010 salaboy.
 *
 * Licensed 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.
 * under the License.
 */

package org.drools.grid.remote;

import java.net.InetSocketAddress;
import java.util.Arrays;
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.ConcurrentHashMap;
import org.drools.KnowledgeBaseFactoryService;
import org.drools.builder.KnowledgeBuilderFactoryService;
import org.drools.grid.Grid;
import org.drools.grid.GridNode;
import org.drools.grid.GridServiceDescription;
import org.drools.grid.io.ConversationManager;
import org.drools.grid.io.impl.CommandImpl;
import org.drools.grid.remote.mina.MinaConnector;
import org.drools.grid.service.directory.Address;
import org.drools.runtime.StatefulKnowledgeSession;
import org.drools.util.ServiceRegistry;
import org.drools.util.ServiceRegistryImpl;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class GridNodeRemoteClient
        implements
        GridNode {

    private static Logger logger = LoggerFactory.getLogger(GridNodeRemoteClient.class);

    private GridServiceDescription    gsd;
    private Grid                      grid;
    private final Map localContext    = new ConcurrentHashMap();
    private final ServiceRegistry     serviceRegistry = ServiceRegistryImpl.getInstance();
    private MinaConnector             connector       = new MinaConnector();

    public GridNodeRemoteClient(Grid grid,
                                GridServiceDescription gsd) {
        this.gsd = gsd;
        this.grid = grid;
        init( this.localContext );
    }

    public  T get(String identifier,
                     Class cls) {
        //@TODO: this was done just as a hack to have the information ready, requires review and refactoring
        if(cls.isAssignableFrom(String.class)){
            CommandImpl cmd = new CommandImpl( "lookupKsessionId",
                    Arrays.asList( new Object[]{identifier} ) );

            ConversationManager connm = this.grid.get( ConversationManager.class );
            Object result = ConversationUtil.sendMessage( connm,
                    (InetSocketAddress) ((Map)this.gsd.getAddresses()).get( "socket" ).getObject(),
                    this.gsd.getId(),
                    cmd );
            return (T)result;
        }
        if(cls.isAssignableFrom(StatefulKnowledgeSession.class) ){
            if ( logger.isDebugEnabled() ) {
                logger.debug( "(" + Thread.currentThread().getId() + ")"+Thread.currentThread().getName() +"GNRC I'm now trying to locate a happy SKS..." + identifier );
            }
            CommandImpl cmd = new CommandImpl( "lookupKsession",
                    Arrays.asList( new Object[]{identifier} ) );
            if ( logger.isDebugEnabled() ) {
                logger.debug( "(" + Thread.currentThread().getId() + ")"+Thread.currentThread().getName() +"GNRC I have the command ready" );
            }

            ConversationManager connm = this.grid.get( ConversationManager.class );
            if ( logger.isDebugEnabled() ) {
                logger.debug( "(" + Thread.currentThread().getId() + ")"+Thread.currentThread().getName() +"GNRC I have the convo manager ready" );
                logger.debug( "(" + Thread.currentThread().getId() + ")"+Thread.currentThread().getName() +"GNRC Sending to " + ((Map)this.gsd.getAddresses()).get( "socket" ).getObject() );
                logger.debug( "(" + Thread.currentThread().getId() + ")"+Thread.currentThread().getName() +"GNRC The gsd, whatever it is " + this.gsd.getId() );
            }

            Object result = ConversationUtil.sendMessage( connm,
                    (InetSocketAddress) ((Map)this.gsd.getAddresses()).get( "socket" ).getObject(),
                    this.gsd.getId(),
                    cmd );

            if ( logger.isDebugEnabled() ) {
                logger.debug( "(" + Thread.currentThread().getId() + ")"+Thread.currentThread().getName() +"GNRC Ready to send it back" );
            }

            return (T)new StatefulKnowledgeSessionRemoteClient((String)result, gsd, connm);
        }
        T service = (T) localContext.get( identifier );
        if ( service == null ) {
            service = this.serviceRegistry.get( cls );
        }

        return service;
    }

    public  T get(Class serviceClass) {
        return get( serviceClass.getName(),
                serviceClass );
    }

    public void set(String identifier,
                    Object object) {
        //We need a way to do it more generic, so we can set whatever we want.

        if(object instanceof StatefulKnowledgeSessionRemoteClient){
            String localId = UUID.randomUUID().toString();

            CommandImpl cmd = new CommandImpl( "registerKsession",
                    Arrays.asList( new Object[]{identifier, ((StatefulKnowledgeSessionRemoteClient)object).getInstanceId()} ) );

            ConversationManager connm = this.grid.get( ConversationManager.class );
            ConversationUtil.sendMessage( connm,
                    (InetSocketAddress) ((Map)this.gsd.getAddresses()).get( "socket" ).getObject(),
                    this.gsd.getId(),
                    cmd );
        } else{

            throw new UnsupportedOperationException( "Not supported yet." );
        }


    }

    public String getId() {
        return gsd.getId();
    }

    public void init(Object context) {

        this.localContext.put( KnowledgeBuilderFactoryService.class.getName(),
                new KnowledgeBuilderProviderRemoteClient( this.grid,
                        gsd ) );
        this.localContext.put( KnowledgeBaseFactoryService.class.getName(),
                new KnowledgeBaseProviderRemoteClient( this.grid,
                        gsd ) );

    }

    public void dispose() {
        connector.close();
    }

}