org.apache.cassandra.locator.SnitchProperties Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of cassandra-all Show documentation
Show all versions of cassandra-all Show documentation
The Apache Cassandra Project develops a highly scalable second-generation distributed database, bringing together Dynamo's fully distributed design and Bigtable's ColumnFamily-based data model.
/*
* 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.cassandra.locator;
import java.io.InputStream;
import java.net.URL;
import java.util.Properties;
import com.google.common.annotations.VisibleForTesting;
import org.apache.cassandra.io.util.FileUtils;
import org.apache.cassandra.utils.Pair;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import static org.apache.cassandra.config.CassandraRelevantProperties.CASSANDRA_RACKDC_PROPERTIES;
public class SnitchProperties
{
private static final Logger logger = LoggerFactory.getLogger(SnitchProperties.class);
public static final String RACKDC_PROPERTY_FILENAME = CASSANDRA_RACKDC_PROPERTIES.getKey();
private final Properties properties;
public SnitchProperties()
{
properties = new Properties();
InputStream stream = null;
String configURL = CASSANDRA_RACKDC_PROPERTIES.getString();
try
{
URL url;
if (configURL == null)
url = SnitchProperties.class.getClassLoader().getResource(RACKDC_PROPERTY_FILENAME);
else
url = new URL(configURL);
stream = url.openStream(); // catch block handles potential NPE
properties.load(stream);
}
catch (Exception e)
{
// do not throw exception here, just consider this an incomplete or an empty property file.
logger.warn("Unable to read {}", ((configURL != null) ? configURL : RACKDC_PROPERTY_FILENAME));
}
finally
{
FileUtils.closeQuietly(stream);
}
}
@VisibleForTesting
public SnitchProperties(Properties properties)
{
this.properties = properties;
}
@SafeVarargs
public SnitchProperties(Pair... pairs)
{
properties = new Properties();
for (Pair pair : pairs)
properties.setProperty(pair.left, pair.right);
}
/**
* Get a snitch property value or return defaultValue if not defined.
*/
public String get(String propertyName, String defaultValue)
{
return properties.getProperty(propertyName, defaultValue);
}
public SnitchProperties add(String key, String value)
{
properties.put(key, value);
return this;
}
/**
* Returns this instance of snitch properties if key is present
* otherwise create new instance of properties and put key with a give value into it
*
* @param key key to add
* @param value value to add
* @return same properties if key is present or new object with added key and value if not
*/
public SnitchProperties putIfAbsent(String key, String value)
{
if (contains(key))
return this;
Properties p = new Properties();
p.putAll(this.properties);
p.put(key, value);
return new SnitchProperties(p);
}
public boolean contains(String propertyName)
{
return properties.containsKey(propertyName);
}
public String getDcSuffix()
{
return properties.getProperty("dc_suffix", "");
}
@Override
public String toString()
{
return "SnitchProperties{" +
"properties=" + (properties != null ? properties.toString() : "null") +
'}';
}
}