org.zodiac.fastorm.rdb.codec.EnumValueCodec Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of zodiac-fastorm-rdb Show documentation
Zodiac FastORM RDB
The newest version!
package org.zodiac.fastorm.rdb.codec;

import java.lang.reflect.Array;
import java.util.Arrays;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collector;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import org.zodiac.fastorm.core.ValueCodec;

public class EnumValueCodec implements ValueCodec {

    private static final Collector COLLECTOR = Collectors.joining(",");

    private static final Function SPLITTER = str -> str.split("[,]");

    @SuppressWarnings("all")
    private final Class type;
    @SuppressWarnings("all")
    private final Object[] values;

    private boolean isArray;

    private boolean toMask;

    public EnumValueCodec(Class type) {
        if (type.isArray()) {
            this.type = type.getComponentType();
            this.isArray = true;
        } else {
            this.type = type;
        }
        values = this.type.getEnumConstants();
        if (values == null) {
            throw new IllegalArgumentException(type + " must be enum");
        }
    }

    public EnumValueCodec(Class type, boolean toMask) {
        this(type);
        this.toMask = toMask;
    }

    public boolean isArray() {
        return isArray;
    }

    public boolean isToMask() {
        return toMask;
    }

    @Override
    @SuppressWarnings("all")
    public Object encode(Object value) {
        if (value instanceof String && toMask) {
            String name = String.valueOf(value);
            value = Enum.valueOf(type, name);
        }

        if (value instanceof Enum) {
            if (!toMask) {
                return ((Enum) value).name();
            } else {
                return enumToMask(((Enum) value));
            }
        }

        if (value instanceof Enum[]) {
            if (!toMask) {
                return Stream.of(((Enum[]) value))
                             .map(Enum::name)
                             .collect(COLLECTOR);
            } else {
                return enumToMask(((Enum[]) value));
            }
        }
        return value;
    }

    @Override
    public Object decode(Object data) {
        if (data instanceof String) {
            if (!isArray) {
                for (Object value : values) {
                    if (((Enum) value).name().equalsIgnoreCase(String.valueOf(data))) {
                        return value;
                    }
                }
                return null;
            } else {
                List arr = Arrays.asList(SPLITTER.apply(((String) data)));
                return Stream
                        .of(type.getEnumConstants())
                        .map(Enum.class::cast)
                        .filter(e -> arr.contains(e.name()))
                        .toArray(l -> (Enum[]) Array.newInstance(type, l));
            }
        }
        if (data instanceof Number) {
            long val = ((Number) data).longValue();

            Stream stream = Stream
                    .of(type.getEnumConstants())
                    .map(Enum.class::cast)
                    .filter(e -> toMask ? enumInMask(val, e) : e.ordinal() == val);

            if (isArray) {
                return stream.toArray(l -> (Enum[]) Array.newInstance(type, l));
            } else {
                return stream.findFirst().orElse(null);
            }

        }
        return data;
    }

    private boolean enumInMask(long mask, Enum e) {
        return (mask & (1L << e.ordinal())) != 0;
    }

    private long enumToMask(Enum... enums) {
        if (enums == null) {
            return 0L;
        }
        long value = 0L;
        for (Enum e : enums) {
            value |= (1L << e.ordinal());
        }
        return value;
    }

}