network/util/ConfigProvider.java

0001 /*
0002  * Licensed to the Apache Software Foundation (ASF) under one or more
0003  * contributor license agreements.  See the NOTICE file distributed with
0004  * this work for additional information regarding copyright ownership.
0005  * The ASF licenses this file to You under the Apache License, Version 2.0
0006  * (the "License"); you may not use this file except in compliance with
0007  * the License.  You may obtain a copy of the License at
0008  *
0009  *    http://www.apache.org/licenses/LICENSE-2.0
0010  *
0011  * Unless required by applicable law or agreed to in writing, software
0012  * distributed under the License is distributed on an "AS IS" BASIS,
0013  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
0014  * See the License for the specific language governing permissions and
0015  * limitations under the License.
0016  */
0017
0018 package org.apache.spark.network.util;
0019
0020 import java.util.Map;
0021 import java.util.NoSuchElementException;
0022
0023 /**
0024  * Provides a mechanism for constructing a {@link TransportConf} using some sort of configuration.
0025  */
0026 public abstract class ConfigProvider {
0027   /** Obtains the value of the given config, throws NoSuchElementException if it doesn't exist. */
0028   public abstract String get(String name);
0029
0030   /** Returns all the config values in the provider. */
0031   public abstract Iterable<Map.Entry<String, String>> getAll();
0032
0033   public String get(String name, String defaultValue) {
0034     try {
0035       return get(name);
0036     } catch (NoSuchElementException e) {
0037       return defaultValue;
0038     }
0039   }
0040
0041   public int getInt(String name, int defaultValue) {
0042     return Integer.parseInt(get(name, Integer.toString(defaultValue)));
0043   }
0044
0045   public long getLong(String name, long defaultValue) {
0046     return Long.parseLong(get(name, Long.toString(defaultValue)));
0047   }
0048
0049   public double getDouble(String name, double defaultValue) {
0050     return Double.parseDouble(get(name, Double.toString(defaultValue)));
0051   }
0052
0053   public boolean getBoolean(String name, boolean defaultValue) {
0054     return Boolean.parseBoolean(get(name, Boolean.toString(defaultValue)));
0055   }
0056
0057 }