All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.glassfish.jersey.media.sse.EventSource Maven / Gradle / Ivy

The newest version!
/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright (c) 2012-2015 Oracle and/or its affiliates. All rights reserved.
 *
 * The contents of this file are subject to the terms of either the GNU
 * General Public License Version 2 only ("GPL") or the Common Development
 * and Distribution License("CDDL") (collectively, the "License").  You
 * may not use this file except in compliance with the License.  You can
 * obtain a copy of the License at
 * http://glassfish.java.net/public/CDDL+GPL_1_1.html
 * or packager/legal/LICENSE.txt.  See the License for the specific
 * language governing permissions and limitations under the License.
 *
 * When distributing the software, include this License Header Notice in each
 * file and include the License file at packager/legal/LICENSE.txt.
 *
 * GPL Classpath Exception:
 * Oracle designates this particular file as subject to the "Classpath"
 * exception as provided by Oracle in the GPL Version 2 section of the License
 * file that accompanied this code.
 *
 * Modifications:
 * If applicable, add the following below the License Header, with the fields
 * enclosed by brackets [] replaced by your own identifying information:
 * "Portions Copyright [year] [name of copyright owner]"
 *
 * Contributor(s):
 * If you wish your version of this file to be governed by only the CDDL or
 * only the GPL Version 2, indicate your decision by adding "[Contributor]
 * elects to include this software in this distribution under the [CDDL or GPL
 * Version 2] license."  If you don't indicate a single choice of license, a
 * recipient has the option to distribute your version of this file under
 * either the CDDL, the GPL Version 2 or to extend the choice of license to
 * its licensees as provided above.  However, if you add GPL Version 2 code
 * and therefore, elected the GPL Version 2 license, then the option applies
 * only if the new code is made subject to such option by the copyright
 * holder.
 */
package org.glassfish.jersey.media.sse;

import java.util.Collection;
import java.util.Collections;
import java.util.Date;
import java.util.List;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicReference;
import java.util.logging.Level;
import java.util.logging.Logger;

import javax.ws.rs.ServiceUnavailableException;
import javax.ws.rs.client.Invocation;
import javax.ws.rs.client.WebTarget;

import org.glassfish.jersey.internal.util.ExtendedLogger;

import jersey.repackaged.com.google.common.util.concurrent.ThreadFactoryBuilder;

/**
 * Client for reading and processing {@link InboundEvent incoming Server-Sent Events}.
 * 

* Instances of this class are thread safe. To build a new instance, you can use one of the * available public {@code EventSource} constructors that produce pre-configured event * source instances. Alternatively, you can create a new {@link EventSource.Builder} instance * using {@link #target(javax.ws.rs.client.WebTarget) EventSource.target(endpoint)} factory method. * Compared to {@code EventSource} constructors, an event source builder provides greater flexibility * when custom-configuring a new event source builder. *

*

* Once an {@link EventSource} is created, it {@link #open opens a connection} to the associated {@link WebTarget web target} * and starts processing any incoming inbound events. * Whenever a new event is received, an {@link EventSource#onEvent(InboundEvent)} method is called as well as any * registered {@link EventListener event listeners} are notified (see {@link EventSource#register(EventListener)} * and {@link EventSource#register(EventListener, String, String...)}. *

*

Reconnect support

*

* The {@code EventSource} supports automated recuperation from a connection loss, including * negotiation of delivery of any missed events based on the last received SSE event {@code id} field value, provided * this field is set by the server and the negotiation facility is supported by the server. In case of a connection loss, * the last received SSE event {@code id} field value is send in the {@value SseFeature#LAST_EVENT_ID_HEADER} HTTP * request header as part of a new connection request sent to the SSE endpoint. Upon a receipt of such reconnect request, the SSE * endpoint that supports this negotiation facility is expected to replay all missed events. Note however, that this is a * best-effort mechanism which does not provide any guaranty that all events would be delivered without a loss. You should * therefore not rely on receiving every single event and design your client application code accordingly. *

*

* By default, when a connection the the SSE endpoint is lost, the event source will wait {@value #RECONNECT_DEFAULT} ms * before attempting to reconnect to the SSE endpoint. The SSE endpoint can however control the client-side retry delay * by including a special {@code retry} field value in the any send event. Jersey {@code EventSource} implementation * tracks any received SSE event {@code retry} field values set by the endpoint and adjusts the reconnect delay accordingly, * using the last received {@code retry} field value as the reconnect delay. *

*

* In addition to handling the standard connection losses, Jersey {@code EventSource} automatically deals with any * {@code HTTP 503 Service Unavailable} responses from SSE endpoint, that contain a * {@value javax.ws.rs.core.HttpHeaders#RETRY_AFTER} HTTP header with a valid value. The * HTTP 503 + {@value javax.ws.rs.core.HttpHeaders#RETRY_AFTER} technique is often used by HTTP endpoints * as a means of connection and traffic throttling. In case a * HTTP 503 + {@value javax.ws.rs.core.HttpHeaders#RETRY_AFTER} response is received in return to a connection * request, Jersey {@code EventSource} will automatically schedule a new reconnect attempt and use the received * {@value javax.ws.rs.core.HttpHeaders#RETRY_AFTER} HTTP header value as a one-time override of the reconnect delay. *

*

Using HTTP persistent connections

*

* The experience has shown that persistent HTTP connection management in the {@link java.net.HttpURLConnection}, * that is used as a default Jersey client {@link org.glassfish.jersey.client.spi.Connector connector}, is fragile. * It is unfortunately quite possible, that under heavy load the client and server connections may get out of sync, * causing Jersey {@code EventSource} hang on a connection that has already been closed by a server, but has not been * properly cleaned up by the {@link java.net.HttpURLConnection} management code, and has been reused for a re-connect * request instead. To avoid this issue, Jersey {@code EventSource} implementation by default disables * persistent HTTP connections when connecting * (or reconnecting) to the SSE endpoint. *

*

* In case you are using Jersey event source with a Jersey client * {@link org.glassfish.jersey.client.ClientConfig#connectorProvider(org.glassfish.jersey.client.spi.ConnectorProvider)} * connector provider configured to use some other client {@code ConnectorProvider} implementation able to reliably * manage persistent HTTP connections (such as {@code org.glassfish.jersey.grizzly.connector.GrizzlyConnectorProvider} or * {@code org.glassfish.jersey.apache.connector.ApacheConnectorProvider}), or in case you simply need to use persistent * HTTP connections, you may do so by invoking the {@link Builder#usePersistentConnections() usePersistentConnections()} method * on an event source builder prior to creating a new event source instance. *

* * @author Pavel Bucek (pavel.bucek at oracle.com) * @author Marek Potociar (marek.potociar at oracle.com) */ public class EventSource implements EventListener { /** * Default SSE {@link EventSource} reconnect delay value in milliseconds. * * @since 2.3 */ public static final long RECONNECT_DEFAULT = 500; private enum State { READY, OPEN, CLOSED } private static final Level CONNECTION_ERROR_LEVEL = Level.FINE; private static final ExtendedLogger LOGGER = new ExtendedLogger(Logger.getLogger(EventSource.class.getName()), Level.FINEST); /** * SSE streaming resource target. */ private final WebTarget target; /** * Default reconnect delay. */ private final long reconnectDelay; /** * Flag indicating if the persistent HTTP connections should be disabled. */ private final boolean disableKeepAlive; /** * Incoming SSE event processing task executor. */ private final ScheduledExecutorService executor; /** * Event source internal state. */ private final AtomicReference state = new AtomicReference<>(State.READY); /** * List of all listeners not bound to receive only events of a particular name. */ private final List unboundListeners = new CopyOnWriteArrayList<>(); /** * A map of listeners bound to receive only events of a particular name. */ private final ConcurrentMap> boundListeners = new ConcurrentHashMap<>(); /** * Jersey {@link EventSource} builder class. * * Event source builder provides methods that let you conveniently configure and subsequently build * a new {@code EventSource} instance. You can obtain a new event source builder instance using * a static {@link EventSource#target(javax.ws.rs.client.WebTarget) EventSource.target(endpoint)} factory method. *

* For example: *

     * EventSource es = EventSource.target(endpoint).named("my source")
     *                             .reconnectingEvery(5, SECONDS)
     *                             .open();
     * 
*

* * @since 2.3 */ public static class Builder { private final WebTarget endpoint; private long reconnect = EventSource.RECONNECT_DEFAULT; private String name = null; private boolean disableKeepAlive = true; private Builder(final WebTarget endpoint) { this.endpoint = endpoint; } /** * Set a custom name for the event source. *

* At present, custom event source name is mainly useful to be able to distinguish different event source * event processing threads from one another. If not set, a default name will be generated using the * SSE endpoint URI. *

* * @param name custom event source name. * @return updated event source builder instance. */ public Builder named(String name) { this.name = name; return this; } /** * Instruct event source to use * persistent HTTP connections when connecting * (or reconnecting) to the SSE endpoint, provided the mechanism is supported by the underlying client * {@link org.glassfish.jersey.client.spi.Connector}. *

* By default, the persistent HTTP connections are disabled for the reasons discussed in the {@link EventSource} * javadoc. *

* * @return updated event source builder instance. */ @SuppressWarnings("unused") public Builder usePersistentConnections() { disableKeepAlive = false; return this; } /** * Set the initial reconnect delay to be used by the event source. *

* Note that this value may be later overridden by the SSE endpoint using either a {@code retry} SSE event field * or HTTP 503 + {@value javax.ws.rs.core.HttpHeaders#RETRY_AFTER} mechanism as described * in the {@link EventSource} javadoc. *

* * @param delay the default time to wait before attempting to recover from a connection loss. * @param unit time unit of the reconnect delay parameter. * @return updated event source builder instance. */ @SuppressWarnings("unused") public Builder reconnectingEvery(final long delay, TimeUnit unit) { reconnect = unit.toMillis(delay); return this; } /** * Build new SSE event source pointing at a SSE streaming {@link WebTarget web target}. *

* The returned event source is ready, but not {@link EventSource#open() connected} to the SSE endpoint. * It is expected that you will manually invoke its {@link #open()} method once you are ready to start * receiving SSE events. In case you want to build an event source instance that is already connected * to the SSE endpoint, use the event source builder {@link #open()} method instead. *

*

* Once the event source is open, the incoming events are processed by the event source in an * asynchronous task that runs in an internal single-threaded {@link ScheduledExecutorService * scheduled executor service}. *

* * @return new event source instance, ready to be connected to the SSE endpoint. * @see #open() */ public EventSource build() { return new EventSource(endpoint, name, reconnect, disableKeepAlive, false); } /** * Build new SSE event source pointing at a SSE streaming {@link WebTarget web target}. *

* The returned event source is already {@link EventSource#open() connected} to the SSE endpoint * and is processing any new incoming events. In case you want to build an event source instance * that is already ready, but not automatically connected to the SSE endpoint, use the event source * builder {@link #build()} method instead. *

*

* The incoming events are processed by the event source in an asynchronous task that runs in an * internal single-threaded {@link ScheduledExecutorService scheduled executor service}. *

* * @return new event source instance, already connected to the SSE endpoint. * @see #build() */ public EventSource open() { // opening directly in the constructor is just plain ugly... final EventSource source = new EventSource(endpoint, name, reconnect, disableKeepAlive, false); source.open(); return source; } } /** * Create a new {@link EventSource.Builder event source builder} that provides convenient way how to * configure and fine-tune various aspects of a newly prepared event source instance. * * @param endpoint SSE streaming endpoint. Must not be {@code null}. * @return a builder of a new event source instance pointing at the specified SSE streaming endpoint. * @throws NullPointerException in case the supplied web target is {@code null}. * @since 2.3 */ public static Builder target(WebTarget endpoint) { return new Builder(endpoint); } /** * Create new SSE event source and open a connection it to the supplied SSE streaming {@link WebTarget web target}. * * This constructor is performs the same series of actions as a call to: *
EventSource.target(endpoint).open()
*

* The created event source instance automatically {@link #open opens a connection} to the supplied SSE streaming * web target and starts processing incoming {@link InboundEvent events}. *

*

* The incoming events are processed by the event source in an asynchronous task that runs in an * internal single-threaded {@link ScheduledExecutorService scheduled executor service}. *

* * @param endpoint SSE streaming endpoint. Must not be {@code null}. * @throws NullPointerException in case the supplied web target is {@code null}. */ public EventSource(final WebTarget endpoint) { this(endpoint, true); } /** * Create new SSE event source pointing at a SSE streaming {@link WebTarget web target}. * * This constructor is performs the same series of actions as a call to: *
     * if (open) {
     *     EventSource.target(endpoint).open();
     * } else {
     *     EventSource.target(endpoint).build();
     * }
*

* If the supplied {@code open} flag is {@code true}, the created event source instance automatically * {@link #open opens a connection} to the supplied SSE streaming web target and starts processing incoming * {@link InboundEvent events}. * Otherwise, if the {@code open} flag is set to {@code false}, the created event source instance * is not automatically connected to the web target. In this case it is expected that the user who * created the event source will manually invoke its {@link #open()} method. *

*

* Once the event source is open, the incoming events are processed by the event source in an * asynchronous task that runs in an internal single-threaded {@link ScheduledExecutorService * scheduled executor service}. *

* * @param endpoint SSE streaming endpoint. Must not be {@code null}. * @param open if {@code true}, the event source will immediately connect to the SSE endpoint, * if {@code false}, the connection will not be established until {@link #open()} method is * called explicitly on the event stream. * @throws NullPointerException in case the supplied web target is {@code null}. */ public EventSource(final WebTarget endpoint, final boolean open) { this(endpoint, null, RECONNECT_DEFAULT, true, open); } private EventSource(final WebTarget target, final String name, final long reconnectDelay, final boolean disableKeepAlive, final boolean open) { if (target == null) { throw new NullPointerException("Web target is 'null'."); } this.target = SseFeature.register(target); this.reconnectDelay = reconnectDelay; this.disableKeepAlive = disableKeepAlive; final String esName = (name == null) ? createDefaultName(target) : name; this.executor = Executors.newSingleThreadScheduledExecutor( new ThreadFactoryBuilder().setNameFormat(esName + "-%d").setDaemon(true).build()); if (open) { open(); } } private static String createDefaultName(WebTarget target) { return String.format("jersey-sse-event-source-[%s]", target.getUri().toASCIIString()); } /** * Open the connection to the supplied SSE underlying {@link WebTarget web target} and start processing incoming * {@link InboundEvent events}. * * @throws IllegalStateException in case the event source has already been opened earlier. */ public void open() { if (!state.compareAndSet(State.READY, State.OPEN)) { switch (state.get()) { case OPEN: throw new IllegalStateException(LocalizationMessages.EVENT_SOURCE_ALREADY_CONNECTED()); case CLOSED: throw new IllegalStateException(LocalizationMessages.EVENT_SOURCE_ALREADY_CLOSED()); } } EventProcessor processor = new EventProcessor(reconnectDelay, null); executor.submit(processor); // return only after the first request to the SSE endpoint has been made processor.awaitFirstContact(); } /** * Check if this event source instance has already been {@link #open() opened}. * * @return {@code true} if this event source is open, {@code false} otherwise. */ public boolean isOpen() { return state.get() == State.OPEN; } /** * Register new {@link EventListener event listener} to receive all streamed {@link InboundEvent SSE events}. * * @param listener event listener to be registered with the event source. * @see #register(EventListener, String, String...) */ public void register(final EventListener listener) { register(listener, null); } /** * Add name-bound {@link EventListener event listener} which will be called only for incoming SSE * {@link InboundEvent events} whose {@link InboundEvent#getName() name} is equal to the specified * name(s). * * @param listener event listener to register with this event source. * @param eventName inbound event name. * @param eventNames additional event names. * @see #register(EventListener) */ public void register(final EventListener listener, final String eventName, final String... eventNames) { if (eventName == null) { unboundListeners.add(listener); } else { addBoundListener(eventName, listener); if (eventNames != null) { for (String name : eventNames) { addBoundListener(name, listener); } } } } private void addBoundListener(final String name, final EventListener listener) { List listeners = boundListeners.putIfAbsent(name, new CopyOnWriteArrayList<>(Collections.singleton(listener))); if (listeners != null) { // alas, new listener collection registration conflict: // need to add the new listener to the existing listener collection listeners.add(listener); } } /** * {@inheritDoc} *

* The default {@code EventSource} implementation is empty, users can override this method to handle * incoming {@link InboundEvent}s. *

*

* Note that overriding this method may be necessary to make sure no {@code InboundEvent incoming events} * are lost in case the event source is constructed using {@link #EventSource(javax.ws.rs.client.WebTarget)} * constructor or in case a {@code true} flag is passed to the {@link #EventSource(javax.ws.rs.client.WebTarget, boolean)} * constructor, since the connection is opened as as part of the constructor call and the event processing starts * immediately. Therefore any {@link EventListener}s registered later after the event source has been constructed * may miss the notifications about the one or more events that arrive immediately after the connection to the * event source is established. *

* * @param inboundEvent received inbound event. */ @Override public void onEvent(final InboundEvent inboundEvent) { // do nothing } /** * Close this event source. * * The method will wait up to 5 seconds for the internal event processing task to complete. */ public void close() { close(5, TimeUnit.SECONDS); } /** * Close this event source and wait for the internal event processing task to complete * for up to the specified amount of wait time. *

* The method blocks until the event processing task has completed execution after a shutdown * request, or until the timeout occurs, or the current thread is interrupted, whichever happens * first. *

*

* In case the waiting for the event processing task has been interrupted, this method restores * the {@link Thread#interrupted() interrupt} flag on the thread before returning {@code false}. *

* * @param timeout the maximum time to wait. * @param unit the time unit of the timeout argument. * @return {@code true} if this executor terminated and {@code false} if the timeout elapsed * before termination or the termination was interrupted. */ public boolean close(final long timeout, final TimeUnit unit) { shutdown(); try { if (!executor.awaitTermination(timeout, unit)) { LOGGER.log(CONNECTION_ERROR_LEVEL, LocalizationMessages.EVENT_SOURCE_SHUTDOWN_TIMEOUT(target.getUri().toString())); return false; } } catch (InterruptedException e) { LOGGER.log(CONNECTION_ERROR_LEVEL, LocalizationMessages.EVENT_SOURCE_SHUTDOWN_INTERRUPTED(target.getUri().toString())); Thread.currentThread().interrupt(); return false; } return true; } private void shutdown() { if (state.getAndSet(State.CLOSED) != State.CLOSED) { // shut down only if has not been shut down before LOGGER.debugLog("Shutting down event processing."); executor.shutdownNow(); } } /** * Private event processor task responsible for connecting to the SSE stream and processing * incoming SSE events as well as handling any connection issues. */ private class EventProcessor implements Runnable, EventListener { /** * Open connection response arrival synchronization latch. */ private final CountDownLatch firstContactSignal; /** * Last received event id. */ private String lastEventId; /** * Re-connect delay. */ private long reconnectDelay; public EventProcessor(final long reconnectDelay, final String lastEventId) { /** * Synchronization barrier used to signal that the initial contact with SSE endpoint * has been made. */ this.firstContactSignal = new CountDownLatch(1); this.reconnectDelay = reconnectDelay; this.lastEventId = lastEventId; } private EventProcessor(final EventProcessor that) { this.firstContactSignal = null; this.reconnectDelay = that.reconnectDelay; this.lastEventId = that.lastEventId; } @Override public void run() { LOGGER.debugLog("Listener task started."); EventInput eventInput = null; try { try { final Invocation.Builder request = prepareHandshakeRequest(); if (state.get() == State.OPEN) { // attempt to connect only if even source is open LOGGER.debugLog("Connecting..."); eventInput = request.get(EventInput.class); LOGGER.debugLog("Connected!"); } } finally { if (firstContactSignal != null) { // release the signal regardless of event source state or connection request outcome firstContactSignal.countDown(); } } final Thread execThread = Thread.currentThread(); while (state.get() == State.OPEN && !execThread.isInterrupted()) { if (eventInput == null || eventInput.isClosed()) { LOGGER.debugLog("Connection lost - scheduling reconnect in {0} ms", reconnectDelay); scheduleReconnect(reconnectDelay); break; } else { this.onEvent(eventInput.read()); } } } catch (ServiceUnavailableException ex) { LOGGER.debugLog("Received HTTP 503"); long delay = reconnectDelay; if (ex.hasRetryAfter()) { LOGGER.debugLog("Recovering from HTTP 503 using HTTP Retry-After header value as a reconnect delay"); final Date requestTime = new Date(); delay = ex.getRetryTime(requestTime).getTime() - requestTime.getTime(); delay = (delay > 0) ? delay : 0; } LOGGER.debugLog("Recovering from HTTP 503 - scheduling to reconnect in {0} ms", delay); scheduleReconnect(delay); } catch (Exception ex) { if (LOGGER.isLoggable(CONNECTION_ERROR_LEVEL)) { LOGGER.log(CONNECTION_ERROR_LEVEL, String.format("Unable to connect - closing the event source to %s.", target.getUri().toASCIIString()), ex); } // if we're here, an unrecoverable error has occurred - just turn off the lights... EventSource.this.shutdown(); } finally { if (eventInput != null && !eventInput.isClosed()) { eventInput.close(); } LOGGER.debugLog("Listener task finished."); } } /** * Called by the event source when an inbound event is received. * * This listener aggregator method is responsible for invoking {@link EventSource#onEvent(InboundEvent)} * method on the owning event source as well as for notifying all registered {@link EventListener event listeners}. * * @param event incoming {@link InboundEvent inbound event}. */ @Override public void onEvent(final InboundEvent event) { if (event == null) { return; } LOGGER.debugLog("New event received."); if (event.getId() != null) { lastEventId = event.getId(); } if (event.isReconnectDelaySet()) { reconnectDelay = event.getReconnectDelay(); } notify(EventSource.this, event); notify(unboundListeners, event); final String eventName = event.getName(); if (eventName != null) { final List eventListeners = boundListeners.get(eventName); if (eventListeners != null) { notify(eventListeners, event); } } } private void notify(final Collection listeners, final InboundEvent event) { for (EventListener listener : listeners) { notify(listener, event); } } private void notify(final EventListener listener, final InboundEvent event) { try { listener.onEvent(event); } catch (Exception ex) { if (LOGGER.isLoggable(Level.FINE)) { LOGGER.log(Level.FINE, String.format("Event notification in a listener of %s class failed.", listener.getClass().getName()), ex); } } } /** * Schedule a new event processor task to reconnect after the specified {@code delay} [milliseconds]. * * If the {@code delay} is zero or negative, the new reconnect task will be scheduled immediately. * The {@code reconnectDelay} and {@code lastEventId} field values are propagated into the newly * scheduled task. *

* The method will silently abort in case the event source is not {@link EventSource#isOpen() open}. *

* * @param delay specifies the amount of time [milliseconds] to wait before attempting a reconnect. * If zero or negative, the new reconnect task will be scheduled immediately. */ private void scheduleReconnect(final long delay) { final State s = state.get(); if (s != State.OPEN) { LOGGER.debugLog("Aborting reconnect of event source in {0} state", state); return; } // propagate the current reconnectDelay, but schedule based on the delay parameter final EventProcessor processor = new EventProcessor(this); if (delay > 0) { executor.schedule(processor, delay, TimeUnit.MILLISECONDS); } else { executor.submit(processor); } } private Invocation.Builder prepareHandshakeRequest() { final Invocation.Builder request = target.request(SseFeature.SERVER_SENT_EVENTS_TYPE); if (lastEventId != null && !lastEventId.isEmpty()) { request.header(SseFeature.LAST_EVENT_ID_HEADER, lastEventId); } if (disableKeepAlive) { request.header("Connection", "close"); } return request; } /** * Await the initial contact with the SSE endpoint. */ public void awaitFirstContact() { LOGGER.debugLog("Awaiting first contact signal."); try { if (firstContactSignal == null) { return; } try { firstContactSignal.await(); } catch (InterruptedException ex) { LOGGER.log(CONNECTION_ERROR_LEVEL, LocalizationMessages.EVENT_SOURCE_OPEN_CONNECTION_INTERRUPTED(), ex); Thread.currentThread().interrupt(); } } finally { LOGGER.debugLog("First contact signal released."); } } } }




© 2015 - 2024 Weber Informatics LLC | Privacy Policy