hnhzServer/fixed/memdb-server@0.5.7/app/config.js

// Copyright 2015 The MemDB Authors.
//
// 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 the AUTHORS file
// for names of contributors.

'use strict';

var P = require('bluebird');
var path = require('path');
var fs = require('fs-extra');
var mkdirp = require('mkdirp');
var memdbLogger = require('memdb-logger');
var logger = memdbLogger.getLogger('memdb', __filename);
var utils = require('./utils');

var _config = null;

exports.init = function(confPath, shardId){
    var searchPaths = [];
    var homePath = process.env.HOME || process.env.USERPROFILE || process.env.HOMEPATH;

    var localDataPath = path.join(homePath, '.memdb');
    mkdirp(localDataPath);

    searchPaths = confPath ? [confPath] : [path.join(homePath, '.memdb/memdb.conf.js'), '/etc/memdb.conf.js'];

    var conf = null;
    for(var i=0; i<searchPaths.length; i++){
        if(fs.existsSync(searchPaths[i])){
            confPath = path.resolve(searchPaths[i]);
            conf = require(confPath);
            exports.path = confPath;
            break;
        }
    }
    if(!conf){
        if(confPath){
            throw new Error('config file not found - ' + searchPaths);
        }

        // copy and load default config
        var confTemplatePath = path.join(__dirname, '../memdb.conf.js');
        var defaultConfPath = path.join(localDataPath, 'memdb.conf.js');
        fs.copySync(confTemplatePath, defaultConfPath);

        conf = require(defaultConfPath);
    }

    // Configure promise
    if(conf.promise && conf.promise.longStackTraces){
        P.longStackTraces();
    }

    // Configure log
    var logConf = conf.log || {};

    var logPath = logConf.path || path.join(localDataPath, 'log');
    mkdirp(logPath);

    console.log('log path: %s', logPath);
    memdbLogger.configure(path.join(__dirname, 'log4js.json'), {shardId : shardId || '$', base : logPath});

    var level = logConf.level || 'INFO';
    memdbLogger.setGlobalLogLevel(memdbLogger.levels[level]);

    // heapdump
    if(conf.heapdump){
        require('heapdump');
    }

    _config = conf;
};

exports.getShardIds = function(){
    if(!_config){
        throw new Error('please config.init first');
    }
    return Object.keys(_config.shards);
};

exports.shardConfig = function(shardId){
    if(!_config){
        throw new Error('please config.init first');
    }

    var conf = utils.clone(_config);

    var shardConf = conf.shards && conf.shards[shardId];
    if(!shardConf){
        throw new Error('shard ' + shardId + ' not configured');
    }
    // Override shard specific config
    for(var key in shardConf){
        conf[key] = shardConf[key];
    }

    conf.shardId = shardId;
    return conf;
};

exports.clusterConfig = function(){
    return _config;
};