Apache
/
activemq-artemis
mirror of https://github.com/apache/activemq-artemis.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

This closes #1065

This commit is contained in:
Justin Bertram 2017-03-06 14:20:54 -06:00
parent 3566077b6e 3882b00735
commit 973c519217
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
artemis-server/src/main/java/org/apache/activemq/artemis/spi/core/protocol/MessagePersister.java
View File

@ -18,11 +18,14 @@
package org.apache.activemq.artemis.spi.core.protocol; package org.apache.activemq.artemis.spi.core.protocol;
import java.util.Map; import java.util.Map;
import java.util.ServiceLoader;
import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.ConcurrentHashMap;
import org.apache.activemq.artemis.api.core.ActiveMQBuffer; import org.apache.activemq.artemis.api.core.ActiveMQBuffer;
import org.apache.activemq.artemis.api.core.Message; import org.apache.activemq.artemis.api.core.Message;
import org.apache.activemq.artemis.core.message.impl.CoreMessagePersister;
import org.apache.activemq.artemis.core.persistence.Persister; import org.apache.activemq.artemis.core.persistence.Persister;
import org.apache.activemq.artemis.core.protocol.core.impl.CoreProtocolManagerFactory;
import org.jboss.logging.Logger; import org.jboss.logging.Logger;
public class MessagePersister implements Persister<Message> { public class MessagePersister implements Persister<Message> {
@ -34,6 +37,14 @@ public class MessagePersister implements Persister<Message> {
/** This will be used for reading messages */ /** This will be used for reading messages */
private static Map<Byte, Persister<Message>> protocols = new ConcurrentHashMap<>(); private static Map<Byte, Persister<Message>> protocols = new ConcurrentHashMap<>();
static {
MessagePersister.registerPersister(CoreProtocolManagerFactory.ID, CoreMessagePersister.getInstance());
Iterable<ProtocolManagerFactory> protocols = ServiceLoader.load(ProtocolManagerFactory.class, MessagePersister.class.getClassLoader());
for (ProtocolManagerFactory next : protocols) {
MessagePersister.registerPersister(next.getStoreID(), next.getPersister());
}
}
public static void registerProtocol(ProtocolManagerFactory manager) { public static void registerProtocol(ProtocolManagerFactory manager) {
Persister<Message> messagePersister = manager.getPersister(); Persister<Message> messagePersister = manager.getPersister();
@ -49,7 +60,9 @@ public class MessagePersister implements Persister<Message> {
} }
public static void registerPersister(byte recordType, Persister<Message> persister) { public static void registerPersister(byte recordType, Persister<Message> persister) {
protocols.put(recordType, persister); if (persister != null) {
protocols.put(recordType, persister);
}
} }
public static MessagePersister getInstance() { public static MessagePersister getInstance() {