| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you 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 org.apache.jackrabbit.oak.blob.cloud.azure.blobstorage; |
| |
| import com.google.common.base.Strings; |
| import com.microsoft.azure.storage.CloudStorageAccount; |
| import com.microsoft.azure.storage.OperationContext; |
| import com.microsoft.azure.storage.RetryExponentialRetry; |
| import com.microsoft.azure.storage.RetryNoRetry; |
| import com.microsoft.azure.storage.RetryPolicy; |
| import com.microsoft.azure.storage.StorageException; |
| import com.microsoft.azure.storage.blob.CloudBlobClient; |
| import com.microsoft.azure.storage.blob.CloudBlobContainer; |
| import org.apache.jackrabbit.core.data.DataStoreException; |
| import org.apache.jackrabbit.oak.commons.PropertiesUtil; |
| |
| import java.io.File; |
| import java.io.FileInputStream; |
| import java.io.IOException; |
| import java.io.InputStream; |
| import java.net.InetSocketAddress; |
| import java.net.Proxy; |
| import java.net.SocketAddress; |
| import java.net.URISyntaxException; |
| import java.security.InvalidKeyException; |
| import java.util.Properties; |
| |
| public final class Utils { |
| |
| public static final String DEFAULT_CONFIG_FILE = "azure.properties"; |
| |
| public static final String DASH = "-"; |
| |
| /** |
| * private constructor so that class cannot initialized from outside. |
| */ |
| private Utils() { |
| } |
| |
| /** |
| * Create CloudBlobClient from properties. |
| * |
| * @param connectionString connectionString to configure @link {@link CloudBlobClient} |
| * @return {@link CloudBlobClient} |
| */ |
| public static CloudBlobClient getBlobClient(final String connectionString) throws URISyntaxException, InvalidKeyException { |
| CloudStorageAccount account = CloudStorageAccount.parse(connectionString); |
| CloudBlobClient client = account.createCloudBlobClient(); |
| return client; |
| } |
| |
| public static CloudBlobContainer getBlobContainer(final String connectionString, final String containerName) throws DataStoreException { |
| try { |
| CloudBlobClient client = Utils.getBlobClient(connectionString); |
| return client.getContainerReference(containerName); |
| } catch (InvalidKeyException | URISyntaxException | StorageException e) { |
| throw new DataStoreException(e); |
| } |
| } |
| |
| public static void setProxyIfNeeded(final Properties properties) { |
| String proxyHost = properties.getProperty(AzureConstants.PROXY_HOST); |
| String proxyPort = properties.getProperty(AzureConstants.PROXY_PORT); |
| |
| if (!Strings.isNullOrEmpty(proxyHost) && |
| Strings.isNullOrEmpty(proxyPort)) { |
| int port = Integer.parseInt(proxyPort); |
| SocketAddress proxyAddr = new InetSocketAddress(proxyHost, port); |
| Proxy proxy = new Proxy(Proxy.Type.HTTP, proxyAddr); |
| OperationContext.setDefaultProxy(proxy); |
| } |
| } |
| |
| public static RetryPolicy getRetryPolicy(final String maxRequestRetry) { |
| int retries = PropertiesUtil.toInteger(maxRequestRetry, -1); |
| if (retries < 0) { |
| return null; |
| } |
| if (retries == 0) { |
| return new RetryNoRetry(); |
| } |
| return new RetryExponentialRetry(RetryPolicy.DEFAULT_CLIENT_BACKOFF, retries); |
| } |
| |
| |
| public static String getConnectionStringFromProperties(Properties properties) { |
| |
| String sasUri = properties.getProperty(AzureConstants.AZURE_SAS, ""); |
| String blobEndpoint = properties.getProperty(AzureConstants.AZURE_BLOB_ENDPOINT, ""); |
| String connectionString = properties.getProperty(AzureConstants.AZURE_CONNECTION_STRING, ""); |
| |
| if (!connectionString.isEmpty()) { |
| return connectionString; |
| } |
| |
| if (!sasUri.isEmpty()) { |
| return getConnectionStringForSas(sasUri, blobEndpoint); |
| } |
| |
| return getConnectionString( |
| properties.getProperty(AzureConstants.AZURE_STORAGE_ACCOUNT_NAME, ""), |
| properties.getProperty(AzureConstants.AZURE_STORAGE_ACCOUNT_KEY, "")); |
| } |
| |
| private static String getConnectionStringForSas(String sasUri, String blobEndpoint) { |
| return String.format("BlobEndpoint=%s;SharedAccessSignature=%s", blobEndpoint, sasUri); |
| } |
| |
| public static String getConnectionString(final String accountName, final String accountKey) { |
| return String.format( |
| "DefaultEndpointsProtocol=https;AccountName=%s;AccountKey=%s", |
| accountName, |
| accountKey |
| ); |
| } |
| |
| /** |
| * Read a configuration properties file. If the file name ends with ";burn", |
| * the file is deleted after reading. |
| * |
| * @param fileName the properties file name |
| * @return the properties |
| * @throws java.io.IOException if the file doesn't exist |
| */ |
| public static Properties readConfig(String fileName) throws IOException { |
| if (!new File(fileName).exists()) { |
| throw new IOException("Config file not found. fileName=" + fileName); |
| } |
| Properties prop = new Properties(); |
| InputStream in = null; |
| try { |
| in = new FileInputStream(fileName); |
| prop.load(in); |
| } finally { |
| if (in != null) { |
| in.close(); |
| } |
| } |
| return prop; |
| } |
| } |