package com.atlassian.messagequeue.internal.core.messagevalidators;
   
   import com.atlassian.messagequeue.MessageValidatorKey;
   import com.atlassian.messagequeue.registry.MessageValidator;
   import com.atlassian.messagequeue.TenantDataIdSupplier;
   import com.atlassian.messagequeue.internal.core.NestedMessageConstants;
   import com.atlassian.messagequeue.registry.MessageContext;
   
   import java.util.Map;
  
  /**
   * Test validator that uses the data id attribute
   */
  public class MessageTenantDataIdValidator implements MessageValidator {
      private TenantDataIdSupplier tenantDataIdSupplier;
      public static final MessageValidatorKey KEY = MessageValidatorKey.of(MessageTenantDataIdValidator.class.toString());
  
      public MessageTenantDataIdValidator(TenantDataIdSupplier tenantDataIdSupplier) {
          this.tenantDataIdSupplier = tenantDataIdSupplier;
      }
  
      @Override
      public boolean isValid(MessageContext messageContext, Map<String, String> messageAttributes) {
          final String tenantDataId = messageAttributes.get(NestedMessageConstants.TENANT_DATA_ID_ATTRIBUTE_NAME);
          return tenantDataId == null || tenantDataIdSupplier.get().equals(tenantDataId);
      }
  
      @Override
      public void handleInvalidMessage(MessageContext messageContext, Map<String, String> messageAttributes) {
          messageContext.acknowledge();
      }
  }