com.linkedin.pinot.server.starter.helix.DefaultHelixStarterServerConfig.java Source code

Introduction

Here is the source code for com.linkedin.pinot.server.starter.helix.DefaultHelixStarterServerConfig.java

Source

/**
 * Copyright (C) 2014-2016 LinkedIn Corp. (pinot-core@linkedin.com)
 *
 * 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.
 */
package com.linkedin.pinot.server.starter.helix;

import com.linkedin.pinot.common.utils.CommonConstants;
import com.linkedin.pinot.server.conf.ServerConf;
import java.util.Iterator;
import org.apache.commons.configuration.Configuration;
import org.apache.commons.configuration.PropertiesConfiguration;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class DefaultHelixStarterServerConfig {
    private static final Logger LOGGER = LoggerFactory.getLogger(DefaultHelixStarterServerConfig.class);

    public static ServerConf getDefaultHelixServerConfig(Configuration externalConfigs) {
        Configuration defaultConfigs = loadDefaultServerConf();
        @SuppressWarnings("unchecked")
        Iterator<String> iterable = externalConfigs.getKeys();
        while (iterable.hasNext()) {
            String key = iterable.next();
            defaultConfigs.setProperty(key, externalConfigs.getProperty(key));
            LOGGER.info("External config key: {}, value: {}", key, externalConfigs.getProperty(key));
        }
        return new ServerConf(defaultConfigs);
    }

    public static Configuration loadDefaultServerConf() {
        Configuration serverConf = new PropertiesConfiguration();
        serverConf.addProperty(CommonConstants.Server.CONFIG_OF_INSTANCE_DATA_DIR,
                CommonConstants.Server.DEFAULT_INSTANCE_DATA_DIR);
        serverConf.addProperty(CommonConstants.Server.CONFIG_OF_INSTANCE_SEGMENT_TAR_DIR,
                CommonConstants.Server.DEFAULT_INSTANCE_SEGMENT_TAR_DIR);

        serverConf.addProperty(CommonConstants.Server.CONFIG_OF_INSTANCE_READ_MODE,
                CommonConstants.Server.DEFAULT_READ_MODE);
        serverConf.addProperty(CommonConstants.Server.CONFIG_OF_INSTANCE_DATA_MANAGER_CLASS,
                CommonConstants.Server.DEFAULT_DATA_MANAGER_CLASS);
        serverConf.addProperty(CommonConstants.Server.CONFIG_OF_INSTANCE_SEGMENT_METADATA_LOADER_CLASS,
                CommonConstants.Server.DEFAULT_SEGMENT_METADATA_LOADER_CLASS);
        serverConf.addProperty(CommonConstants.Server.CONFIG_OF_SEGMENT_FORMAT_VERSION,
                CommonConstants.Server.DEFAULT_SEGMENT_FORMAT_VERSION);

        // query executor parameters
        serverConf.addProperty(CommonConstants.Server.CONFIG_OF_QUERY_EXECUTOR_PRUNER_CLASS,
                " DataSchemaSegmentPruner,TimeSegmentPruner,ValidSegmentPruner");
        serverConf.addProperty("pinot.server.query.executor.pruner.DataSchemaSegmentPruner.id", "0");
        serverConf.addProperty("pinot.server.query.executor.pruner.TimeSegmentPruner.id", "1");
        serverConf.addProperty("pinot.server.query.executor.pruner.ValidSegmentPruner.id", "2");
        serverConf.addProperty(CommonConstants.Server.CONFIG_OF_QUERY_EXECUTOR_TIMEOUT,
                CommonConstants.Server.DEFAULT_QUERY_EXECUTOR_TIMEOUT);
        serverConf.addProperty(CommonConstants.Server.CONFIG_OF_QUERY_EXECUTOR_CLASS,
                CommonConstants.Server.DEFAULT_QUERY_EXECUTOR_CLASS);

        serverConf.addProperty(CommonConstants.Server.CONFIG_OF_HELIX_FLAPPING_TIMEWINDOW_MS,
                CommonConstants.Server.DEFAULT_HELIX_FLAPPING_TIMEWINDOW_MS);

        // request handler factory parameters
        serverConf.addProperty(CommonConstants.Server.CONFIG_OF_REQUEST_HANDLER_FACTORY_CLASS,
                CommonConstants.Server.DEFAULT_REQUEST_HANDLER_FACTORY_CLASS);

        // netty port
        serverConf.addProperty(CommonConstants.Server.CONFIG_OF_NETTY_PORT,
                CommonConstants.Helix.DEFAULT_SERVER_NETTY_PORT);

        return serverConf;
    }

}