package com.atlassian.messagequeue.internal.core.messagevalidators;
   
   import com.atlassian.messagequeue.registry.MessageValidator;
   import com.atlassian.messagequeue.internal.core.NestedMessageConstants;
   import com.atlassian.messagequeue.registry.MessageContext;
   import org.slf4j.Logger;
   import org.slf4j.LoggerFactory;
   
   import java.util.Map;
  
  /**
   * Ensures that every message has a message runner key.
   */
  public class MessageRunnerKeyValidator implements MessageValidator {
      private static final Logger log = LoggerFactory.getLogger(MessageRunnerKeyValidator.class);
  
      @Override
      public boolean isValid(MessageContext messageContext, Map<String, String> messageAttributes) {
          return messageAttributes.get(NestedMessageConstants.MESSAGE_RUNNER_KEY_ATTRIBUTE_NAME) != null;
      }
  
      @Override
      public void handleInvalidMessage(MessageContext messageContext, Map<String, String> messageAttributes) {
          final String messageId = messageContext.getMessageId().orElse("Not present");
          log.error("Message received that does not contain required '{}' attribute. Message will be acknowledged (messageID: {})", NestedMessageConstants.MESSAGE_RUNNER_KEY_ATTRIBUTE_NAME, messageId);
          messageContext.acknowledge();
      }
  }