net.udidb.server.driver.ServerModule.java Source code

Introduction

Here is the source code for net.udidb.server.driver.ServerModule.java

Source

/*
 * Copyright (c) 2011-2015, Dan McNulty
 * All rights reserved.
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */

package net.udidb.server.driver;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.module.SimpleModule;
import com.google.inject.AbstractModule;
import com.google.inject.name.Names;

import io.vertx.core.Vertx;
import io.vertx.core.impl.VertxFactoryImpl;
import net.libudi.api.UdiProcessManager;
import net.libudi.api.jni.impl.UdiProcessManagerImpl;
import net.sourcecrumbs.api.files.BinaryReader;
import net.sourcecrumbs.refimpl.CrossPlatformBinaryReader;
import net.udidb.engine.context.DebuggeeContextManager;
import net.udidb.engine.context.DebuggeeContextManagerImpl;
import net.udidb.engine.events.EventPump;
import net.udidb.engine.events.EventSink;
import net.udidb.engine.expr.ExpressionCompilerDelegate;
import net.udidb.engine.ops.impls.help.HelpMessageProvider;
import net.udidb.engine.ops.results.DeferredResult;
import net.udidb.engine.ops.results.OperationResultVisitor;
import net.udidb.engine.ops.results.TableResult;
import net.udidb.engine.ops.results.TableRow;
import net.udidb.engine.ops.results.ValueResult;
import net.udidb.engine.ops.results.VoidResult;
import net.udidb.engine.source.InMemorySourceLineRowFactory;
import net.udidb.engine.source.SourceLineRowFactory;
import net.udidb.expr.ExpressionCompiler;
import net.udidb.expr.values.ExpressionValue;
import net.udidb.server.api.results.ExpressionValueSerializer;
import net.udidb.server.api.results.DeferredResultMixIn;
import net.udidb.server.api.results.TableResultMixIn;
import net.udidb.server.api.results.TableRowMixIn;
import net.udidb.server.api.results.ValueResultMixIn;
import net.udidb.server.api.results.VoidResultMixIn;
import net.udidb.server.engine.OperationEngine;
import net.udidb.server.engine.ServerEngine;
import net.udidb.server.engine.ServerEngineImpl;
import net.udidb.server.engine.ServerEventDispatcher;
import net.udidb.server.wamp.InMemoryConnectorProvider;
import ws.wamp.jawampa.ApplicationError;
import ws.wamp.jawampa.WampClient;
import ws.wamp.jawampa.WampClientBuilder;
import ws.wamp.jawampa.WampRouter;
import ws.wamp.jawampa.WampRouterBuilder;

/**
 * A Guice module defining dependencies for running the debugger within a server
 *
 * @author mcnulty
 */
public class ServerModule extends AbstractModule {
    private static final String WAMP_REALM = "udidb";
    private static final String INTERNAL_CLIENT_URI = "udidb://wamp";

    @Override
    protected void configure() {
        // JSON configuration
        SimpleModule simpleModule = new SimpleModule();
        simpleModule.addSerializer(ExpressionValue.class, new ExpressionValueSerializer());

        ObjectMapper objectMapper = new ObjectMapper();
        objectMapper.addMixInAnnotations(VoidResult.class, VoidResultMixIn.class);
        objectMapper.addMixInAnnotations(DeferredResult.class, DeferredResultMixIn.class);
        objectMapper.addMixInAnnotations(TableResult.class, TableResultMixIn.class);
        objectMapper.addMixInAnnotations(ValueResult.class, ValueResultMixIn.class);
        objectMapper.addMixInAnnotations(TableRow.class, TableRowMixIn.class);
        objectMapper.registerModule(simpleModule);
        bind(ObjectMapper.class).toInstance(objectMapper);

        // REST API configuration
        Vertx vertx = new VertxFactoryImpl().vertx();
        bind(Vertx.class).toInstance(vertx);

        // Engine configuration
        bind(String[].class).annotatedWith(Names.named("OP_PACKAGES"))
                .toInstance(new String[] { "net.udidb.engine.ops.impls" });

        bind(DebuggeeContextManager.class).to(DebuggeeContextManagerImpl.class);

        bind(HelpMessageProvider.class).asEagerSingleton();

        bind(UdiProcessManager.class).toInstance(new UdiProcessManagerImpl());

        bind(BinaryReader.class).toInstance(new CrossPlatformBinaryReader());

        bind(ExpressionCompiler.class).toInstance(new ExpressionCompilerDelegate());

        bind(SourceLineRowFactory.class).toInstance(new InMemorySourceLineRowFactory());

        bind(ServerEngine.class).to(ServerEngineImpl.class);

        bind(OperationResultVisitor.class).to(OperationEngine.class);

        bind(ServerEventDispatcher.class).asEagerSingleton();

        bind(EventPump.class).asEagerSingleton();

        bind(EventSink.class).to(ServerEventDispatcher.class);

        WampRouter wampRouter = configureWampRouter();
        bind(WampRouter.class).toInstance(wampRouter);

        bind(WampClient.class).toInstance(configureWampClient(wampRouter));

    }

    private WampRouter configureWampRouter() {
        try {
            return new WampRouterBuilder().addRealm(WAMP_REALM).build();
        } catch (ApplicationError e) {
            // TODO there might be a better way to communicate this error
            throw new RuntimeException(e);
        }
    }

    private WampClient configureWampClient(WampRouter wampRouter) {
        try {
            WampClient wampClient = new WampClientBuilder().withRealm(WAMP_REALM).withUri(INTERNAL_CLIENT_URI)
                    .withConnectorProvider(new InMemoryConnectorProvider(wampRouter)).build();
            wampClient.open();
            return wampClient;
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }
}