com.netflix.spinnaker.echo.config.AmazonPubsubProperties Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of echo-pubsub-aws Show documentation
Spinnaker Echo
The newest version!
/*
 * Copyright 2017 Netflix, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License")
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.netflix.spinnaker.echo.config;

import java.io.*;
import java.util.List;
import javax.validation.Valid;
import javax.validation.constraints.NotEmpty;
import lombok.Data;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.boot.context.properties.ConfigurationProperties;

@Data
@ConfigurationProperties(prefix = "pubsub.amazon")
public class AmazonPubsubProperties {

  @Valid private List subscriptions;

  @Data
  public static class AmazonPubsubSubscription {

    private static final Logger log = LoggerFactory.getLogger(AmazonPubsubSubscription.class);

    @NotEmpty private String name;

    @NotEmpty private String topicARN;

    @NotEmpty private String queueARN;

    private String templatePath;

    private MessageFormat messageFormat;

    /**
     * Provide an id, present in the message attributes as a string, to use as a unique identifier
     * for processing messages. Fall back to amazon sqs id if alternate Id is not present in the
     * message attributes
     */
    private String alternateIdInMessageAttributes;

    int visibilityTimeout = 30;
    int sqsMessageRetentionPeriodSeconds = 120;
    int waitTimeSeconds = 5;

    // 1 hour default
    private Integer dedupeRetentionSeconds = 3600;

    public AmazonPubsubSubscription() {}

    public AmazonPubsubSubscription(
        String name,
        String topicARN,
        String queueARN,
        String templatePath,
        MessageFormat messageFormat,
        String alternateIdInMessageAttributes,
        Integer dedupeRetentionSeconds) {
      this.name = name;
      this.topicARN = topicARN;
      this.queueARN = queueARN;
      this.templatePath = templatePath;
      this.messageFormat = messageFormat;
      this.alternateIdInMessageAttributes = alternateIdInMessageAttributes;
      if (dedupeRetentionSeconds != null && dedupeRetentionSeconds >= 0) {
        this.dedupeRetentionSeconds = dedupeRetentionSeconds;
      } else {
        if (dedupeRetentionSeconds != null) {
          log.warn("Ignoring dedupeRetentionSeconds invalid value of " + dedupeRetentionSeconds);
        }
      }
    }

    private MessageFormat determineMessageFormat() {
      // Supplying a custom template overrides a MessageFormat choice
      if (!StringUtils.isEmpty(templatePath)) {
        return MessageFormat.CUSTOM;
      } else if (messageFormat == null || messageFormat.equals("")) {
        return MessageFormat.NONE;
      }
      return messageFormat;
    }

    public InputStream readTemplatePath() {
      messageFormat = determineMessageFormat();
      log.info(
          "Using message format: {} to process artifacts for subscription: {}",
          messageFormat,
          name);

      try {
        if (messageFormat == MessageFormat.CUSTOM) {
          try {
            return new FileInputStream(new File(templatePath));
          } catch (FileNotFoundException e) {
            // Check if custom jar path was provided before failing
            return getClass().getResourceAsStream(templatePath);
          }
        } else if (messageFormat.jarPath != null) {
          return getClass().getResourceAsStream(messageFormat.jarPath);
        }
      } catch (Exception e) {
        throw new RuntimeException("Failed to read template in subscription " + name, e);
      }
      return null;
    }
  }

  public static enum MessageFormat {
    S3("/s3.jinja"),
    CUSTOM(),
    NONE();

    private String jarPath;

    MessageFormat(String jarPath) {
      this.jarPath = jarPath;
    }

    MessageFormat() {}
  }
}