com.netflix.kayenta.azure.security.AzureCredentials Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2019 Microsoft Corporation.
 *
 * 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.kayenta.azure.security;

import com.microsoft.azure.storage.*;
import com.microsoft.azure.storage.blob.*;
import lombok.Getter;
import lombok.ToString;
import lombok.extern.slf4j.Slf4j;

@ToString
@Slf4j
public class AzureCredentials {

  @Getter private String storageAccountName;

  @Getter private String accountAccessKey;

  @Getter private String endpointSuffix;

  public AzureCredentials(
      String storageAccountName, String accountAccessKey, String endpointSuffix) {
    this.storageAccountName = storageAccountName;
    this.accountAccessKey = accountAccessKey;
    this.endpointSuffix = endpointSuffix;
  }

  public CloudBlobContainer getAzureContainer(String containerName) throws Exception {
    final String storageConnectionString =
        "DefaultEndpointsProtocol=https;"
            + "AccountName="
            + this.storageAccountName
            + ";"
            + "AccountKey="
            + this.accountAccessKey
            + ";"
            + "EndpointSuffix="
            + this.endpointSuffix;
    // Retrieve storage account from connection-string.
    CloudStorageAccount storageAccount = CloudStorageAccount.parse(storageConnectionString);

    // Create the blob client.
    CloudBlobClient blobClient = storageAccount.createCloudBlobClient();

    // Get a reference to a container.
    // The container name must be lower case

    CloudBlobContainer container = blobClient.getContainerReference(containerName);

    // Create the container if it does not exist.
    container.createIfNotExists();

    return container;
  }
}