org.wisdom.akka.impl.AkkaBootstrap Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of akka-system Show documentation
The newest version!
/*
 * #%L
 * Wisdom-Framework
 * %%
 * Copyright (C) 2013 - 2014 Wisdom Framework
 * %%
 * 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.
 * #L%
 */
package org.wisdom.akka.impl;

import akka.actor.ActorSystem;
import akka.osgi.OsgiActorSystemFactory;
import com.typesafe.config.ConfigFactory;
import org.apache.felix.ipojo.annotations.*;
import org.osgi.framework.BundleContext;
import org.osgi.framework.ServiceRegistration;
import org.wisdom.akka.AkkaSystemService;
import org.wisdom.api.http.Context;
import org.wisdom.api.http.Result;
import scala.concurrent.ExecutionContext;
import scala.concurrent.Future;

import java.io.InputStream;
import java.util.concurrent.Callable;

/**
 * A component exposing the {@link org.wisdom.akka.AkkaSystemService}.
 */
@Component
@Provides
@Instantiate
public class AkkaBootstrap implements AkkaSystemService {

    private final BundleContext context;
    private ServiceRegistration systemRegistration;

    /**
     * The actor system.
     */
    private ActorSystem system;

    /**
     * Creates the Akka Bootstrap.
     *
     * @param context the bundle context
     */
    public AkkaBootstrap(BundleContext context) {
        this.context = context;
    }

    /**
     * Initializes the Akka System.
     */
    @Validate
    public void start() {
        OsgiActorSystemFactory osgiActorSystemFactory = OsgiActorSystemFactory.apply(context, ConfigFactory.empty());
        system = osgiActorSystemFactory.createActorSystem("wisdom-system");

        systemRegistration = context.registerService(ActorSystem.class, system, null);
    }

    /**
     * Shuts downs the Akka System.
     */
    @Invalidate
    public void stop() {
        unregisterQuietly(systemRegistration);
        systemRegistration = null;
        if (system != null) {
            system.shutdown();
            system = null;
        }
    }

    private void unregisterQuietly(ServiceRegistration registration) {
        if (registration != null) {
            try {
                registration.unregister();
            } catch (Exception e) { //NOSONAR
                // Ignored.
            }
        }
    }

    /**
     * @return the Akka System.
     */
    @Override
    public ActorSystem system() {
        return system;
    }

    /**
     * Executes the given {@link java.util.concurrent.Callable} returning a {@link org.wisdom.api.http.Result},
     * in a separated thread (managed by Akka).
     *
     * @param callable the classloader
     * @param context  the HTTP context used by the given callable
     * @return the future to retrieve the result.
     */
    @Override
    public Future dispatchResultWithContext(Callable callable, Context context) {
        return akka.dispatch.Futures.future(callable,
                new HttpExecutionContext(system.dispatcher(), context, Thread.currentThread().getContextClassLoader()));
    }

    /**
     * Executes the given {@link java.util.concurrent.Callable} returning a {@link org.wisdom.api.http.Result},
     * in a separated thread (managed by Akka). Unlike the
     * {@link #dispatchResultWithContext(java.util.concurrent.Callable, org.wisdom.api.http.Context)} method,
     * this method uses the current HTTP context.
     *
     * @param callable the classloader
     * @return the future to retrieve the result.
     */
    @Override
    public Future dispatchResult(Callable callable) {
        return akka.dispatch.Futures.future(callable,
                new HttpExecutionContext(system.dispatcher(), Context.CONTEXT.get(),
                        Thread.currentThread().getContextClassLoader())
        );
    }

    /**
     * Executes the given {@link java.util.concurrent.Callable} returning an {@link java.io.InputStream},
     * in a separated thread (managed by Akka). Unlike the
     * {@link #dispatchResultWithContext(java.util.concurrent.Callable, org.wisdom.api.http.Context)} method,
     * this method uses the current HTTP context.
     *
     * @param callable the classloader
     * @return the future to retrieve the result.
     */
    @Override
    public Future dispatchInputStream(Callable callable) {
        return akka.dispatch.Futures.future(callable,
                new HttpExecutionContext(system.dispatcher(), Context.CONTEXT.get(),
                        Thread.currentThread().getContextClassLoader())
        );
    }

    /**
     * Executes the given callable in a separated thread (managed by Akka).
     *
     * @param callable the task
     * @param ctx      the execution context in which the callable is going to be executed.
     * @param       the type of result
     * @return the future to retrieve the computed result.
     */
    @Override
    public  Future dispatch(Callable callable, ExecutionContext ctx) {
        return akka.dispatch.Futures.future(callable, ctx);
    }

    /**
     * @return an execution context using the current HTTP context and current thread context classloader.
     */
    public ExecutionContext fromThread() {
        return new HttpExecutionContext(system.dispatcher(), Context.CONTEXT.get(),
                Thread.currentThread().getContextClassLoader());
    }
}