PushyServiceImpl.java

/**
 * 
 */
package jframe.pushy.impl;

import java.io.File;
import java.util.Date;

import jframe.core.plugin.annotation.InjectPlugin;
import jframe.core.plugin.annotation.Injector;
import jframe.core.plugin.annotation.Start;
import jframe.core.plugin.annotation.Stop;
import jframe.pushy.PushyConf;
import jframe.pushy.PushyPlugin;
import jframe.pushy.PushyService;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.relayrides.pushy.apns.ApnsEnvironment;
import com.relayrides.pushy.apns.PushManager;
import com.relayrides.pushy.apns.PushManagerConfiguration;
import com.relayrides.pushy.apns.util.SSLContextUtil;
import com.relayrides.pushy.apns.util.SimpleApnsPushNotification;
import com.relayrides.pushy.apns.util.TokenUtil;

/**
 * @author dzh
 * @date Sep 30, 2014 9:04:30 AM
 * @since 1.0
 */
@Injector
public class PushyServiceImpl implements PushyService {

	static Logger LOG = LoggerFactory.getLogger(PushyServiceImpl.class);

	@InjectPlugin
	static PushyPlugin plugin;

	static String FILE_CONF = "file.pushy";

	PushManager<SimpleApnsPushNotification> pushManager;

	// ExecutorService exeSvc;

	// NioEventLoopGroup eventGroup;

	@Start
	void start() {
		String conf = plugin.getConfig(FILE_CONF);
		if (!new File(conf).exists()) {
			LOG.error("Not found pushy config file {}", conf);
			return;
		}

		try {
			// exeSvc = new ThreadPoolExecutor(1, Runtime.getRuntime()
			// .availableProcessors() + 1, 60L, TimeUnit.SECONDS,
			// new LinkedBlockingQueue<Runnable>());
			// eventGroup = new NioEventLoopGroup(1);

			PushyConf.init(conf);

			PushManagerConfiguration pushConf = new PushManagerConfiguration();
			pushConf.setConcurrentConnectionCount(PushyConf.PUSH_CONN_COUNT);
			pushManager = new PushManager<SimpleApnsPushNotification>(
					getEnvironment(PushyConf.HOST, PushyConf.HOST_PORT,
							PushyConf.FEEDBACK, PushyConf.FEEDBACK_PORT),
					SSLContextUtil.createDefaultSSLContext(
							plugin.getConfig("app.conf") + "/"
									+ PushyConf.IOS_AUTH,
							PushyConf.IOS_PASSWORD), null, null, null,
					pushConf, "PushManager");

			pushManager.start();
		} catch (Exception e) {
			LOG.error(e.getMessage());
			return;
		}
		LOG.info("PushyService start successfully!");

		// pushManager.registerRejectedNotificationListener(listener);
		// pushManager.registerFailedConnectionListener(listener)
		// class MyRejectedNotificationListener implements
		// RejectedNotificationListener<SimpleApnsPushNotification> {
		//
		// @Override
		// public void handleRejectedNotification(
		// final PushManager<? extends SimpleApnsPushNotification> pushManager,
		// final SimpleApnsPushNotification notification,
		// final RejectedNotificationReason reason) {

		// System.out.format("%s was rejected with rejection reason %s\n",
		// notification, reason);
		// }
		// }
		// pushManager
		// .registerRejectedNotificationListener(new
		// MyRejectedNotificationListener());

		// class MyFailedConnectionListener implements
		// FailedConnectionListener<SimpleApnsPushNotification> {
		// @Override
		// public void handleFailedConnection(
		// final PushManager<? extends SimpleApnsPushNotification> pushManager,
		// final Throwable cause) {
		// System.out.println(cause.getMessage());
		// }
		// }
		// pushManager
		// .registerFailedConnectionListener(new MyFailedConnectionListener());
	}

	public void setPushManager(
			PushManager<SimpleApnsPushNotification> pushManager) {
		this.pushManager = pushManager;
	}

	public static ApnsEnvironment getEnvironment(String host, String port,
			String feedback, String fdPort) {
		return new ApnsEnvironment(host, Integer.parseInt(port), feedback,
				Integer.parseInt(fdPort));
	}

	@Stop
	public void stop() {
		if (pushManager != null) {
			try {
				pushManager.shutdown(10 * 1000);
			} catch (InterruptedException e) {
				LOG.error(e.getMessage());
			}
		}

		// if (eventGroup != null) {
		// eventGroup.shutdownGracefully();
		// }
		// if (exeSvc != null) {
		// try {
		// exeSvc.shutdown();
		// exeSvc.awaitTermination(6, TimeUnit.SECONDS);
		// } catch (InterruptedException e) {
		// LOG.error(e.getMessage());
		// }
		// }
		plugin = null;
	}

	@Override
	public void sendMessage(String token, String payload) throws Exception {
		sendMessage(token, payload, null);
	}

	// public List<String> getExpiredTokens(long timeout, TimeUnit timeoutUnit)
	// {
	// List<ExpiredToken> list = null;
	// try {
	// list = pushManager.getExpiredTokens(10, TimeUnit.SECONDS);
	// } catch (InterruptedException | FeedbackConnectionException e) {
	// LOG.error(e.getMessage());
	// }
	// if (list == null) {
	// list = Collections.emptyList();
	// }
	//
	// List<String> tokens = new ArrayList<String>(list.size());
	// for (ExpiredToken t : list) {
	// tokens.add(TokenUtil.tokenBytesToString(t.getToken()));
	// }
	// return tokens;
	// }

	@Override
	public void sendMessage(String token, String payload, Date expirationDate)
			throws Exception {
		try {
			pushManager.getQueue().put(
					new SimpleApnsPushNotification(TokenUtil
							.tokenStringToByteArray(token), payload,
							expirationDate));
		} catch (InterruptedException e) {
			LOG.warn(e.getMessage());
		}
	}

}