package com.atlassian.messagequeue.internal.sqs;
   
   import com.atlassian.messagequeue.MessageRunnerKey;
   
   import static java.util.Objects.requireNonNull;
   
   /**
    * Default implementation of {@link SQSMessageRunnerKeyToProducerMapper}
    * Return {@link SQSProducerQueueConfig} for provided {@link MessageRunnerKey} from the map
   * or default one if {@link MessageRunnerKey} is not in the map
   */
  public class DefaultSQSMessageRunnerKeyToProducerMapper implements SQSMessageRunnerKeyToProducerMapper {
  
      private final SQSConfig sqsConfig;
  
      public DefaultSQSMessageRunnerKeyToProducerMapper(SQSConfig sqsConfig) {
          this.sqsConfig = requireNonNull(sqsConfig);
      }
  
      @Override
      public SQSProducerQueueConfig getQueueConfigForMessageRunner(MessageRunnerKey messageRunnerKey) {
          return sqsConfig.getOutboundQueueNameMappings().getOrDefault(messageRunnerKey, sqsConfig.getDefaultQueue());
      }
  }