/*
* Copyright © 2014 Cask Data, Inc.
*
* 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.
*/
package co.cask.cdap.app.guice;
import co.cask.cdap.app.runtime.ProgramRunner;
import co.cask.cdap.app.runtime.ProgramRuntimeService;
import co.cask.cdap.common.conf.CConfiguration;
import co.cask.cdap.common.conf.Constants;
import co.cask.cdap.common.logging.common.LocalLogWriter;
import co.cask.cdap.common.logging.common.LogWriter;
import co.cask.cdap.internal.app.queue.QueueReaderFactory;
import co.cask.cdap.internal.app.runtime.ProgramRunnerFactory;
import co.cask.cdap.internal.app.runtime.batch.MapReduceProgramRunner;
import co.cask.cdap.internal.app.runtime.flow.FlowProgramRunner;
import co.cask.cdap.internal.app.runtime.flow.FlowletProgramRunner;
import co.cask.cdap.internal.app.runtime.procedure.ProcedureProgramRunner;
import co.cask.cdap.internal.app.runtime.service.InMemoryProgramRuntimeService;
import co.cask.cdap.internal.app.runtime.service.InMemoryRunnableRunner;
import co.cask.cdap.internal.app.runtime.service.InMemoryServiceRunner;
import co.cask.cdap.internal.app.runtime.spark.SparkProgramRunner;
import co.cask.cdap.internal.app.runtime.webapp.IntactJarHttpHandler;
import co.cask.cdap.internal.app.runtime.webapp.JarHttpHandler;
import co.cask.cdap.internal.app.runtime.webapp.WebappHttpHandlerFactory;
import co.cask.cdap.internal.app.runtime.webapp.WebappProgramRunner;
import co.cask.cdap.internal.app.runtime.workflow.WorkflowProgramRunner;
import com.google.common.base.Preconditions;
import com.google.inject.Inject;
import com.google.inject.PrivateModule;
import com.google.inject.Provider;
import com.google.inject.Provides;
import com.google.inject.Scopes;
import com.google.inject.Singleton;
import com.google.inject.assistedinject.FactoryModuleBuilder;
import com.google.inject.multibindings.MapBinder;
import com.google.inject.name.Named;
import org.apache.twill.api.ServiceAnnouncer;
import org.apache.twill.common.Cancellable;
import org.apache.twill.discovery.Discoverable;
import org.apache.twill.discovery.DiscoveryService;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.util.Map;
/**
*
*/
final class InMemoryProgramRunnerModule extends PrivateModule {
/**
* Configures a {@link com.google.inject.Binder} via the exposed methods.
*/
@Override
protected void configure() {
// Bind and expose LogWriter (a bit hacky, but needed by MapReduce for now)
bind(LogWriter.class).to(LocalLogWriter.class);
expose(LogWriter.class);
// Bind ServiceAnnouncer for procedure.
bind(ServiceAnnouncer.class).to(DiscoveryServiceAnnouncer.class);
// For Binding queue stuff
bind(QueueReaderFactory.class).in(Scopes.SINGLETON);
// Bind ProgramRunner
MapBinder<ProgramRunnerFactory.Type, ProgramRunner> runnerFactoryBinder =
MapBinder.newMapBinder(binder(), ProgramRunnerFactory.Type.class, ProgramRunner.class);
runnerFactoryBinder.addBinding(ProgramRunnerFactory.Type.FLOW).to(FlowProgramRunner.class);
runnerFactoryBinder.addBinding(ProgramRunnerFactory.Type.FLOWLET).to(FlowletProgramRunner.class);
runnerFactoryBinder.addBinding(ProgramRunnerFactory.Type.PROCEDURE).to(ProcedureProgramRunner.class);
runnerFactoryBinder.addBinding(ProgramRunnerFactory.Type.MAPREDUCE).to(MapReduceProgramRunner.class);
runnerFactoryBinder.addBinding(ProgramRunnerFactory.Type.SPARK).to(SparkProgramRunner.class);
runnerFactoryBinder.addBinding(ProgramRunnerFactory.Type.WORKFLOW).to(WorkflowProgramRunner.class);
runnerFactoryBinder.addBinding(ProgramRunnerFactory.Type.WEBAPP).to(WebappProgramRunner.class);
//twill support in standalone
runnerFactoryBinder.addBinding(ProgramRunnerFactory.Type.SERVICE).to(InMemoryServiceRunner.class);
runnerFactoryBinder.addBinding(ProgramRunnerFactory.Type.RUNNABLE).to(InMemoryRunnableRunner.class);
bind(ProgramRunnerFactory.class).to(InMemoryFlowProgramRunnerFactory.class).in(Scopes.SINGLETON);
// Note: Expose for test cases. Need to refactor test cases.
expose(ProgramRunnerFactory.class);
// Bind and expose runtime service
bind(ProgramRuntimeService.class).to(InMemoryProgramRuntimeService.class).in(Scopes.SINGLETON);
expose(ProgramRuntimeService.class);
// For binding DataSet transaction stuff
install(new DataFabricFacadeModule());
// Create webapp http handler factory.
install(new FactoryModuleBuilder().implement(JarHttpHandler.class, IntactJarHttpHandler.class)
.build(WebappHttpHandlerFactory.class));
}
@Singleton
@Provides
private LocalLogWriter providesLogWriter(CConfiguration configuration) {
return new LocalLogWriter(configuration);
}
@Singleton
private static final class InMemoryFlowProgramRunnerFactory implements ProgramRunnerFactory {
private final Map<ProgramRunnerFactory.Type, Provider<ProgramRunner>> providers;
@Inject
private InMemoryFlowProgramRunnerFactory(Map<ProgramRunnerFactory.Type, Provider<ProgramRunner>> providers) {
this.providers = providers;
}
@Override
public ProgramRunner create(ProgramRunnerFactory.Type programType) {
Provider<ProgramRunner> provider = providers.get(programType);
Preconditions.checkNotNull(provider, "Unsupported program type: " + programType);
return provider.get();
}
}
@Singleton
private static final class DiscoveryServiceAnnouncer implements ServiceAnnouncer {
private final DiscoveryService discoveryService;
private final InetAddress hostname;
@Inject
private DiscoveryServiceAnnouncer(DiscoveryService discoveryService,
@Named(Constants.AppFabric.SERVER_ADDRESS) InetAddress hostname) {
this.discoveryService = discoveryService;
this.hostname = hostname;
}
@Override
public Cancellable announce(final String serviceName, final int port) {
return discoveryService.register(new Discoverable() {
@Override
public String getName() {
return serviceName;
}
@Override
public InetSocketAddress getSocketAddress() {
return new InetSocketAddress(hostname, port);
}
});
}
}
}