benchmarks/src/main/java/org/apache/omid/benchmarks/tso/TSOServerBenchmarkConfig.java

/*
 * 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.omid.benchmarks.tso;

import com.google.inject.AbstractModule;
import org.apache.omid.YAMLUtils;
import org.apache.omid.benchmarks.utils.IntegerGenerator;
import org.apache.omid.metrics.MetricsRegistry;
import org.apache.omid.tools.hbase.SecureHBaseConfig;
import org.apache.omid.tso.client.OmidClientConfiguration;

public class TSOServerBenchmarkConfig extends SecureHBaseConfig {

    private static final String CONFIG_FILE_NAME = "tso-server-benchmark-config.yml";
    private static final String DEFAULT_CONFIG_FILE_NAME = "default-tso-server-benchmark-config.yml";

    private long benchmarkRunLengthInMins;

    private int txRunners;
    private int txRateInRequestPerSecond;
    private long warmUpPeriodInSecs;
    private IntegerGenerator cellIdGenerator;
    private int writesetSize;
    private boolean fixedWritesetSize;
    private int percentageOfReadOnlyTxs;
    private long commitDelayInMs;

    private OmidClientConfiguration omidClientConfiguration;
    private AbstractModule commitTableStoreModule;

    private MetricsRegistry metrics;

    // ----------------------------------------------------------------------------------------------------------------
    // Instantiation
    // ----------------------------------------------------------------------------------------------------------------

    TSOServerBenchmarkConfig() {
        this(CONFIG_FILE_NAME);
    }

    TSOServerBenchmarkConfig(String configFileName) {
        new YAMLUtils().loadSettings(configFileName, DEFAULT_CONFIG_FILE_NAME, this);
    }

    // ----------------------------------------------------------------------------------------------------------------
    // Getters and setters for config params
    // ----------------------------------------------------------------------------------------------------------------

    public long getBenchmarkRunLengthInMins() {
        return benchmarkRunLengthInMins;
    }

    public void setBenchmarkRunLengthInMins(long benchmarkRunLengthInMins) {
        this.benchmarkRunLengthInMins = benchmarkRunLengthInMins;
    }

    public int getTxRunners() {
        return txRunners;
    }

    public void setTxRunners(int txRunners) {
        this.txRunners = txRunners;
    }

    public int getTxRateInRequestPerSecond() {
        return txRateInRequestPerSecond;
    }

    public void setTxRateInRequestPerSecond(int txRateInRequestPerSecond) {
        this.txRateInRequestPerSecond = txRateInRequestPerSecond;
    }

    public long getWarmUpPeriodInSecs() {
        return warmUpPeriodInSecs;
    }

    public void setWarmUpPeriodInSecs(long warmUpPeriodInSecs) {
        this.warmUpPeriodInSecs = warmUpPeriodInSecs;
    }

    public IntegerGenerator getCellIdGenerator() {
        return cellIdGenerator;
    }

    public void setCellIdGenerator(IntegerGenerator cellIdGenerator) {
        this.cellIdGenerator = cellIdGenerator;
    }

    public int getWritesetSize() {
        return writesetSize;
    }

    public void setWritesetSize(int writesetSize) {
        this.writesetSize = writesetSize;
    }

    public boolean isFixedWritesetSize() {
        return fixedWritesetSize;
    }

    public void setFixedWritesetSize(boolean fixedWritesetSize) {
        this.fixedWritesetSize = fixedWritesetSize;
    }

    public int getPercentageOfReadOnlyTxs() {
        return percentageOfReadOnlyTxs;
    }

    public void setPercentageOfReadOnlyTxs(int percentageOfReadOnlyTxs) {
        this.percentageOfReadOnlyTxs = percentageOfReadOnlyTxs;
    }

    public long getCommitDelayInMs() {
        return commitDelayInMs;
    }

    public void setCommitDelayInMs(long commitDelayInMs) {
        this.commitDelayInMs = commitDelayInMs;
    }

    public OmidClientConfiguration getOmidClientConfiguration() {
        return omidClientConfiguration;
    }

    public void setOmidClientConfiguration(OmidClientConfiguration omidClientConfiguration) {
        this.omidClientConfiguration = omidClientConfiguration;
    }

    public AbstractModule getCommitTableStoreModule() {
        return commitTableStoreModule;
    }

    public void setCommitTableStoreModule(AbstractModule commitTableStoreModule) {
        this.commitTableStoreModule = commitTableStoreModule;
    }

    public MetricsRegistry getMetrics() {
        return metrics;
    }

    public void setMetrics(MetricsRegistry metrics) {
        this.metrics = metrics;
    }

}