mul-agent-virtual-simulation-system/train-server/src/main/java/top/haijunit/train/domain/constant/SwitchStateConverter.java

package top.haijunit.train.domain.constant;

import jakarta.persistence.AttributeConverter;
import jakarta.persistence.Converter;

import java.util.Optional;

/**
 * @author zhanghaijun
 * @date 2023/11/27 16:50
 * @description [一句话描述该类的功能]
 */
@Converter(autoApply = true)
public class SwitchStateConverter implements AttributeConverter<SwitchStateEnum, Integer> {
    @Override
    public Integer convertToDatabaseColumn(SwitchStateEnum state) {
        return Optional.of(state).map(SwitchStateEnum::getCode).orElse(SwitchStateEnum.Forward.getCode());
    }

    @Override
    public SwitchStateEnum convertToEntityAttribute(Integer code) {
        return SwitchStateEnum.getByCode(code);
    }
}