org.jetlinks.rule.engine.defaults.codec.EnumCodec Maven / Gradle / Ivy

Go to download
package org.jetlinks.rule.engine.defaults.codec;

import io.netty.buffer.Unpooled;
import lombok.AllArgsConstructor;
import org.jetlinks.rule.engine.api.codec.Codec;
import org.jetlinks.rule.engine.api.Payload;

import javax.annotation.Nonnull;
import java.util.Arrays;

@AllArgsConstructor(staticName = "of")
public class EnumCodec> implements Codec {

    private final T[] values;

    @Override
    public T decode(@Nonnull Payload payload) {
        byte[] bytes = payload.bodyAsBytes();

        if (bytes.length > 0 && bytes[0] <= values.length) {
            return values[bytes[0] & 0xFF];
        }
        throw new IllegalArgumentException("can not decode payload " + Arrays.toString(bytes) + " to enums " + Arrays.toString(values));
    }

    @Override
    public Payload encode(T body) {
        return () -> Unpooled.wrappedBuffer(new byte[]{(byte) body.ordinal()});
    }


}