package com.atlassian.messagequeue.internal.core;
   
   import com.atlassian.messagequeue.MessageValidatorKey;
   import com.atlassian.messagequeue.registry.MessageValidator;
   import com.atlassian.messagequeue.registry.MessageValidatorRegistryService;
   
   import javax.annotation.concurrent.ThreadSafe;
   import java.util.Collection;
   import java.util.concurrent.ConcurrentHashMap;
  import java.util.concurrent.ConcurrentMap;
  
  /**
   * A default message runner registry that stores registrations in memory in a concurrent data structure.
   *
   * It's important to realise that any validators registered here will only validate/filter messages received by the current node.
   *
   */
  @ThreadSafe
  public class DefaultMessageValidatorRegistryService implements MessageValidatorRegistryService, MessageValidatorRegistryHelper {
      private final ConcurrentMap<MessageValidatorKey, MessageValidator> registry;
  
      public DefaultMessageValidatorRegistryService() {
          this.registry = new ConcurrentHashMap<>();
      }
  
      @Override
      public void registerMessageValidator(MessageValidatorKey messageRunnerKey, MessageValidator messageRunner) {
          registry.put(messageRunnerKey, messageRunner);
      }
  
      @Override
      public void unregisterMessageValidator(MessageValidatorKey messageRunnerKey) {
          registry.remove(messageRunnerKey);
      }
  
      @Override
      public Collection<MessageValidator> getAllValidators() {
          return registry.values();
      }
  }