Subversion Repositories SmartDukaan

//  Copyright (c) 2007-2008 Facebook

//

//  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.

//

// See accompanying file LICENSE or visit the Scribe site at:

// http://developers.facebook.com/scribe/

//

// @author Bobby Johnson

// @author Jason Sobel

#ifndef SCRIBE_CONF_H

#define SCRIBE_CONF_H

#include <string>

#include <vector>

#include <queue>

#include <iostream>

#include <fstream>

#include "src/gen-cpp/scribe.h"

/*

 * This class reads and parses a configuration

 * describing a hierarchy of store objects.

 *

 * It reads a conf file with a proprietary format, although it could

 * be changed to xml (or anything else that supports hierarchy) by only

 * changing the code in this class.

 */

class StoreConf;

typedef boost::shared_ptr<StoreConf> pStoreConf;

typedef std::map<std::string, std::string> string_map_t;

typedef std::map<std::string, pStoreConf> store_conf_map_t;

class StoreConf {

 public:

  StoreConf();

  virtual ~StoreConf();

  // Return value is true if the key exists, and false if it doesn't.

  // This doesn't check for garbage ints or empty strings.

  // The return parameter is untouched if the key isn't found.

  void getAllStores(std::vector<pStoreConf>& _return);

  bool getStore(const std::string& storeName, pStoreConf& _return);

  bool getInt(const std::string& intName, long int& _return);

  bool getUnsigned(const std::string& intName, unsigned long int& _return);

  bool getString(const std::string& stringName, std::string& _return);

  void setString(const std::string& stringName, const std::string& value);

  void setUnsigned(const std::string& intName, unsigned long value);

  // Reads configuration from a file and throws an exception if it fails.

  void parseConfig(const std::string& filename);

 private:

  string_map_t values;

  store_conf_map_t stores;

  static bool parseStore(/*in,out*/ std::queue<std::string>& raw_config,

                         /*out*/ StoreConf* parsed_config);

  static std::string trimString(const std::string& str);

  bool readConfFile(const std::string& filename, std::queue<std::string>& _return);

};

#endif //!defined SCRIBE_CONF_H