io.nosqlbench.adapter.diag.optasks.DiagTask_erroroncycle Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2022 nosqlbench
 *
 * 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 io.nosqlbench.adapter.diag.optasks;

import io.nosqlbench.nb.annotations.Service;
import io.nosqlbench.api.config.standard.ConfigModel;
import io.nosqlbench.api.config.standard.NBConfigModel;
import io.nosqlbench.api.config.standard.NBConfiguration;
import io.nosqlbench.api.config.standard.Param;

import java.util.Map;

/**
 * Cause a blocking call to delay the initialization
 * of this owning operation for a number of milliseconds.
 */
@Service(value = DiagTask.class, selector = "erroroncycle")
public class DiagTask_erroroncycle implements DiagTask {

    private String name;
    private long error_on_cycle;

    @Override
    public void applyConfig(NBConfiguration cfg) {
        this.name = cfg.get("name", String.class);
        error_on_cycle = cfg.get("erroroncycle", long.class);
    }

    @Override
    public NBConfigModel getConfigModel() {
        return ConfigModel.of(DiagTask_erroroncycle.class)
                .add(Param.required("name", String.class))
                .add(Param.defaultTo("erroroncycle", 1L))
                .asReadOnly();
    }

    @Override
    public Map apply(Long aLong, Map stringObjectMap) {
        if (error_on_cycle == aLong) {
            throw new RuntimeException("Diag was requested to stop on cycle " + error_on_cycle);
        }
        return Map.of();
    }

    @Override
    public String getName() {
        return name;
    }
}