de.codecentric.spring.boot.chaos.monkey.endpoints.BaseChaosMonkeyEndpoint Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2022 the original author or 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
 *
 *     https://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 de.codecentric.spring.boot.chaos.monkey.endpoints;

import de.codecentric.spring.boot.chaos.monkey.configuration.ChaosMonkeyProperties;
import de.codecentric.spring.boot.chaos.monkey.configuration.ChaosMonkeySettings;
import de.codecentric.spring.boot.chaos.monkey.configuration.WatcherProperties;
import de.codecentric.spring.boot.chaos.monkey.endpoints.dto.AssaultPropertiesUpdate;
import de.codecentric.spring.boot.chaos.monkey.endpoints.dto.ChaosMonkeyStatusResponseDto;
import java.time.Duration;

public class BaseChaosMonkeyEndpoint {

    protected final ChaosMonkeySettings chaosMonkeySettings;

    public BaseChaosMonkeyEndpoint(ChaosMonkeySettings chaosMonkeySettings) {
        this.chaosMonkeySettings = chaosMonkeySettings;
    }

    public AssaultPropertiesUpdate getAssaultProperties() {
        return this.chaosMonkeySettings.getAssaultProperties().toDto();
    }

    public ChaosMonkeyStatusResponseDto enableChaosMonkey() {
        return setChaosMonkeyEnabled(true);
    }

    public ChaosMonkeyStatusResponseDto disableChaosMonkey() {
        return setChaosMonkeyEnabled(false);
    }

    private ChaosMonkeyStatusResponseDto setChaosMonkeyEnabled(boolean enabled) {
        ChaosMonkeyProperties chaosMonkeyProperties = this.chaosMonkeySettings.getChaosMonkeyProperties();
        Long lastTimestamp = chaosMonkeyProperties.getLastEnabledToggleTimestamp();
        boolean wasEnabled = chaosMonkeyProperties.isEnabled();

        chaosMonkeyProperties.setEnabled(enabled);

        Duration enabledFor = wasEnabled && lastTimestamp != null ? Duration.ofMillis(System.currentTimeMillis() - lastTimestamp) : null;
        Long timestamp = chaosMonkeyProperties.getLastEnabledToggleTimestamp();

        return new ChaosMonkeyStatusResponseDto(enabled, timestamp, enabledFor);
    }

    public ChaosMonkeyStatusResponseDto getStatus() {
        ChaosMonkeyProperties chaosMonkeyProperties = this.chaosMonkeySettings.getChaosMonkeyProperties();
        Long lastEnabledToggleTime = chaosMonkeyProperties.getLastEnabledToggleTimestamp();
        boolean enabled = chaosMonkeyProperties.isEnabled();
        Duration enabledFor = enabled && lastEnabledToggleTime != null ? Duration.ofMillis(System.currentTimeMillis() - lastEnabledToggleTime) : null;
        return new ChaosMonkeyStatusResponseDto(enabled, lastEnabledToggleTime, enabledFor);
    }

    public WatcherProperties getWatcherProperties() {
        return this.chaosMonkeySettings.getWatcherProperties();
    }
}