org.zrgs.spring.statemachine.AbstractStateMachineCommands.java Source code

Introduction

Here is the source code for org.zrgs.spring.statemachine.AbstractStateMachineCommands.java

Source

/*
 * Copyright 2015 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
 *
 * 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.zrgs.spring.statemachine;

import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.shell.core.CommandMarker;
import org.springframework.shell.core.annotation.CliCommand;
import org.springframework.statemachine.StateMachine;
import org.springframework.statemachine.state.State;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

import java.util.Iterator;
import java.util.Map.Entry;
import java.util.Set;

import javax.inject.Inject;

@SuppressWarnings("SpringAutowiredFieldsWarningInspection")
@Component
public class AbstractStateMachineCommands<S, E> implements CommandMarker {

    @SuppressWarnings("SpringJavaAutowiringInspection")
    @Inject
    private StateMachine<S, E> stateMachine;

    @Inject
    @Qualifier("stateChartModel")
    private String stateChartModel;

    protected StateMachine<S, E> getStateMachine() {
        return stateMachine;
    }

    @SuppressWarnings("unused")
    @CliCommand(value = "sm state", help = "Prints current state")
    public String state() {
        State<S, E> state = stateMachine.getState();
        if (state != null) {
            return StringUtils.collectionToCommaDelimitedString(state.getIds());
        } else {
            return "No state";
        }
    }

    @SuppressWarnings("unused")
    @CliCommand(value = "sm start", help = "Start a state machine")
    public String start() {
        stateMachine.start();
        return "State machine started";
    }

    @SuppressWarnings("unused")
    @CliCommand(value = "sm stop", help = "Stop a state machine")
    public String stop() {
        stateMachine.stop();
        return "State machine stopped";
    }

    @SuppressWarnings("unused")
    @CliCommand(value = "sm print", help = "Print state machine")
    public String print() {
        return stateChartModel;
    }

    @SuppressWarnings("unused")
    @CliCommand(value = "sm variables", help = "Prints extended state variables")
    public String variables() {
        StringBuilder buf = new StringBuilder();
        Set<Entry<Object, Object>> entrySet = stateMachine.getExtendedState().getVariables().entrySet();
        Iterator<Entry<Object, Object>> iterator = entrySet.iterator();
        if (entrySet.isEmpty()) {
            while (iterator.hasNext()) {
                Entry<Object, Object> e = iterator.next();
                buf.append(e.getKey()).append("=").append(e.getValue());
                if (iterator.hasNext()) {
                    buf.append("\n");
                }
            }
        } else {
            buf.append("No variables");
        }
        return buf.toString();
    }
}