commit push server

src/main/java/com/notnoop/apns/internal/QueuedApnsService.java

@@ -0,0 +1,126 @@
+/*
+ *  Copyright 2009, Mahmood Ali.
+ *  All rights reserved.
+ *
+ *  Redistribution and use in source and binary forms, with or without
+ *  modification, are permitted provided that the following conditions are
+ *  met:
+ *
+ *    * Redistributions of source code must retain the above copyright
+ *      notice, this list of conditions and the following disclaimer.
+ *    * Redistributions in binary form must reproduce the above
+ *      copyright notice, this list of conditions and the following disclaimer
+ *      in the documentation and/or other materials provided with the
+ *      distribution.
+ *    * Neither the name of Mahmood Ali. nor the names of its
+ *      contributors may be used to endorse or promote products derived from
+ *      this software without specific prior written permission.
+ *
+ *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ *  A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ *  OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ *  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ *  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ *  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ *  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+package com.notnoop.apns.internal;
+
+import java.util.Date;
+import java.util.Map;
+import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.concurrent.BlockingQueue;
+import java.util.concurrent.Executors;
+import java.util.concurrent.LinkedBlockingQueue;
+import java.util.concurrent.ThreadFactory;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.notnoop.apns.ApnsNotification;
+import com.notnoop.apns.ApnsService;
+import com.notnoop.exceptions.NetworkIOException;
+
+public class QueuedApnsService extends AbstractApnsService {
+
+	private static final Logger logger = LoggerFactory.getLogger(QueuedApnsService.class);
+	
+    private ApnsService service;
+    private BlockingQueue<ApnsNotification> queue;
+    private AtomicBoolean started = new AtomicBoolean(false);
+
+    public QueuedApnsService(ApnsService service) {
+        this(service, null);
+    }
+
+    public QueuedApnsService(ApnsService service, final ThreadFactory tf) {
+        super(null);
+        this.service = service;
+        this.queue = new LinkedBlockingQueue<ApnsNotification>();
+        this.threadFactory = tf == null ? Executors.defaultThreadFactory() : tf;
+        this.thread = null;
+    }
+
+    @Override
+    public void push(ApnsNotification msg) {
+        if (!started.get()) {
+            throw new IllegalStateException("service hasn't be started or was closed");
+        }
+        queue.add(msg);
+    }
+
+    private final ThreadFactory threadFactory;
+    private Thread thread;
+    private volatile boolean shouldContinue;
+
+    public void start() {
+        if (started.getAndSet(true)) {
+            // I prefer if we throw a runtime IllegalStateException here,
+            // but I want to maintain semantic backward compatibility.
+            // So it is returning immediately here
+            return;
+        }
+
+        service.start();
+        shouldContinue = true;
+        thread = threadFactory.newThread(new Runnable() {
+            public void run() {
+                while (shouldContinue) {
+                    try {
+                        ApnsNotification msg = queue.take();
+                        service.push(msg);
+                    } catch (InterruptedException e) {
+                    	// ignore
+                    } catch (NetworkIOException e) {
+                    	// ignore: failed connect...
+                    } catch (Exception e) {
+                    	// weird if we reached here - something wrong is happening, but we shouldn't stop the service anyway!
+                    	logger.warn("Unexpected message caught... Shouldn't be here", e);
+                    }
+                }
+            }
+        });
+        thread.start();
+    }
+
+    public void stop() {
+        started.set(false);
+        shouldContinue = false;
+        thread.interrupt();
+        service.stop();
+    }
+
+    @Override
+    public Map<String, Date> getInactiveDevices() throws NetworkIOException {
+        return service.getInactiveDevices();
+    }
+
+    public void testConnection() throws NetworkIOException {
+        service.testConnection();
+    }
+
+}