com.mastfrog.email.server.EmailServiceModule Maven / Gradle / Ivy

Go to download
package com.mastfrog.email.server;

import com.google.inject.AbstractModule;
import com.google.inject.Inject;
import com.google.inject.TypeLiteral;
import com.google.inject.multibindings.Multibinder;
import com.google.inject.name.Named;
import com.google.inject.name.Names;
import com.mastfrog.giulius.Dependencies;
import com.mastfrog.settings.Settings;
import com.mastfrog.shutdown.hooks.ShutdownHookRegistry;
import com.mastfrog.util.preconditions.Exceptions;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;

/**
 * Configures mail sending
 *
 * @author Tim Boudreau
 */
public final class EmailServiceModule extends AbstractModule {

    private final Settings settings;
    public static final String SETTINGS_KEY_EMAIL_SEND_THREADS = "email.send.threads";

    public EmailServiceModule(Settings settings) {
        this.settings = settings;
    }

    @Override
    protected void configure() {
        if (Dependencies.isProductionMode(settings)) {
            // fail fast
            System.out.println("Production mode - using real mail server");
            bind(EmailServerService.class).to(RealMailServer.class).asEagerSingleton();
//            bind(HtmlTemplateProvider.class).asEagerSingleton();
        } else {
            System.out.println("Development mode - will not send real emails");
        }
        bind(ShutdownMailqueue.class).asEagerSingleton();
        bind(ExecutorService.class).annotatedWith(Names.named("mailqueue")).toInstance(Executors.newFixedThreadPool(
                settings.getInt(SETTINGS_KEY_EMAIL_SEND_THREADS, 2)));
        // Guice requires this even if none get bound elsewhere
        Multibinder> placeholder
                = Multibinder.newSetBinder(binder(), ENUM_EMAIL_TEMPLATE_PROVIDER_LITERAL);
    }
    public static TypeLiteral> ENUM_EMAIL_TEMPLATE_PROVIDER_LITERAL = new TL();

    private static class TL extends TypeLiteral> {

    }

    static class ShutdownMailqueue implements Runnable {

        private final ExecutorService svc;

        @Inject
        ShutdownMailqueue(@Named("mailqueue") ExecutorService svc, ShutdownHookRegistry reg) {
            this.svc = svc;
            reg.add(this);
        }

        @Override
        public void run() {
            svc.shutdownNow();
            System.out.println("Waiting for mail queue to empty");
            try {
                svc.awaitTermination(10, TimeUnit.SECONDS);
            } catch (InterruptedException ex) {
                Exceptions.printStackTrace(ex);
            }
        }
    }
}