package com.chinaztt.mes.technology.state.routing;

import com.chinaztt.mes.technology.entity.Routing;
import com.chinaztt.mes.technology.state.routing.constant.RoutingEvents;
import com.chinaztt.mes.technology.state.routing.constant.RoutingStates;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.statemachine.StateMachineContext;
import org.springframework.statemachine.StateMachinePersist;
import org.springframework.statemachine.config.EnableStateMachineFactory;
import org.springframework.statemachine.config.EnumStateMachineConfigurerAdapter;
import org.springframework.statemachine.config.builders.*;
import org.springframework.statemachine.persist.DefaultStateMachinePersister;
import org.springframework.statemachine.persist.StateMachinePersister;
import org.springframework.statemachine.support.DefaultStateMachineContext;

import java.util.EnumSet;

/**
 * @Author: zhangxy
 * @Date: 2020-08-24 17:21
 */
@Configuration
@EnableStateMachineFactory(name = "routingStateMachineFactory")
public class RoutingStateMachineConfig extends EnumStateMachineConfigurerAdapter<RoutingStates, RoutingEvents> {

	public final static String MACHINE_ID = "routingStateMachine";

	@Override
	public void configure(StateMachineStateConfigurer<RoutingStates, RoutingEvents> states) throws Exception {
		states.withStates()
				.initial(RoutingStates.DRAFT)
				.states(EnumSet.allOf(RoutingStates.class));
	}

	@Override
	public void configure(StateMachineTransitionConfigurer<RoutingStates, RoutingEvents> transitions) throws Exception {
		transitions
				.withExternal()
				.source(RoutingStates.DRAFT).target(RoutingStates.ACCEPTED)
				.event(RoutingEvents.ACCEPT)
				.and()
				.withExternal()
				.source(RoutingStates.ACCEPTED).target(RoutingStates.DRAFT)
				.event(RoutingEvents.REVOKE)
				.and()
				.withExternal()
				.source(RoutingStates.ACCEPTED).target(RoutingStates.CANCELLED)
				.event(RoutingEvents.CANCEL);
	}

	@Override
	public void configure(StateMachineConfigurationConfigurer<RoutingStates, RoutingEvents> config)
			throws Exception {
		config.withConfiguration().machineId(MACHINE_ID);
	}

	/**
	 * 持久化配置
	 *
	 * @return
	 */
	@Bean(name ="routingPersister")
	public StateMachinePersister<RoutingStates, RoutingEvents, Routing> persister() {
		return new DefaultStateMachinePersister<>(new StateMachinePersist<RoutingStates, RoutingEvents, Routing>() {
			@Override
			public void write(StateMachineContext<RoutingStates, RoutingEvents> stateMachineContext, Routing routing) throws Exception {

			}

			@Override
			public StateMachineContext<RoutingStates, RoutingEvents> read(Routing routing) throws Exception {
				StateMachineContext<RoutingStates, RoutingEvents> result = new DefaultStateMachineContext<RoutingStates, RoutingEvents>(RoutingStates.getEnum(routing.getState()),
						null, null, null, null, MACHINE_ID);
				return result;
			}
		});
	}

}