org.mariadb.r2dbc.codec.Codecs Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of r2dbc-mariadb Show documentation
R2DBC driver for MariaDB
The newest version!
// SPDX-License-Identifier: Apache-2.0
// Copyright (c) 2020-2024 MariaDB Corporation Ab

package org.mariadb.r2dbc.codec;

import io.r2dbc.spi.Blob;
import io.r2dbc.spi.Clob;
import io.r2dbc.spi.Parameter;
import io.r2dbc.spi.R2dbcType;
import java.io.InputStream;
import java.math.BigDecimal;
import java.math.BigInteger;
import java.nio.ByteBuffer;
import java.time.Duration;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.LocalTime;
import java.util.BitSet;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
import java.util.stream.Stream;
import org.mariadb.r2dbc.codec.list.*;
import org.mariadb.r2dbc.util.Assert;
import org.mariadb.r2dbc.util.BindValue;

public final class Codecs {

  public static final Map> typeMapper = typeToCodec();
  public static final Codec[] LIST =
      new Codec[] {
        BigDecimalCodec.INSTANCE,
        BigIntegerCodec.INSTANCE,
        BooleanCodec.INSTANCE,
        ByteBufferCodec.INSTANCE,
        BlobCodec.INSTANCE,
        ByteArrayCodec.INSTANCE,
        ByteCodec.INSTANCE,
        BitSetCodec.INSTANCE,
        ClobCodec.INSTANCE,
        DoubleCodec.INSTANCE,
        LongCodec.INSTANCE,
        FloatCodec.INSTANCE,
        IntCodec.INSTANCE,
        LocalDateCodec.INSTANCE,
        LocalDateTimeCodec.INSTANCE,
        LocalTimeCodec.INSTANCE,
        DurationCodec.INSTANCE,
        ShortCodec.INSTANCE,
        StreamCodec.INSTANCE,
        StringCodec.INSTANCE,
        UuidCodec.INSTANCE
      };
  private static final Map> r2dbcTypeMapper = r2dbcTypeToDataTypeMap();
  private static final Map, Codec> codecMapper = classToCodecMap();

  public static BindValue encodeNull(Class type, int index) {
    if (type == null) return new BindValue(StringCodec.INSTANCE, null);
    return new BindValue(codecFromClass(type, index), null);
  }

  public static BindValue encode(Object value, int index) {

    Codec codec = StringCodec.INSTANCE;
    Object parameterValue = value;

    if (value instanceof Parameter) {
      Parameter parameter = (Parameter) value;
      parameterValue = parameter.getValue();

      if (parameterValue == null) {
        if (parameter.getType() instanceof R2dbcType) {
          codec = codecFromR2dbcType((R2dbcType) parameter.getType());
        }
        return new BindValue(codec, null);
      }
    }

    if (parameterValue == null) {
      return new BindValue(codec, null);
    }
    codec = codecFromClass(parameterValue.getClass(), index);
    return new BindValue(codec, parameterValue);
  }

  public static Codec codecFromClass(Class javaType, int index) {
    Codec codec;
    if ((codec = codecMapper.get(javaType)) != null) return codec;
    if ((codec = codecByClass(javaType, index)) != null) return codec;
    return StringCodec.INSTANCE;
  }

  public static Codec codecFromR2dbcType(R2dbcType type) {
    Assert.requireNonNull(type, "type must not be null");
    Codec codec = r2dbcTypeMapper.get(type);
    if (codec != null) return codec;
    return StringCodec.INSTANCE;
  }

  private static Map> typeToCodec() {
    Map> myMap = new HashMap<>();
    myMap.put(BigDecimal.class, BigDecimalCodec.INSTANCE);
    myMap.put(BigInteger.class, BigIntegerCodec.INSTANCE);
    myMap.put(BitSet.class, BitSetCodec.INSTANCE);
    myMap.put(Blob.class, BlobCodec.INSTANCE);
    myMap.put(Boolean.class, BooleanCodec.INSTANCE);
    myMap.put(byte[].class, ByteArrayCodec.INSTANCE);
    myMap.put(ByteBuffer.class, ByteBufferCodec.INSTANCE);
    myMap.put(Byte.class, ByteCodec.INSTANCE);
    myMap.put(Clob.class, ClobCodec.INSTANCE);
    myMap.put(Double.class, DoubleCodec.INSTANCE);
    myMap.put(Duration.class, DurationCodec.INSTANCE);
    myMap.put(Float.class, FloatCodec.INSTANCE);
    myMap.put(Integer.class, IntCodec.INSTANCE);
    myMap.put(LocalDate.class, LocalDateCodec.INSTANCE);
    myMap.put(LocalDateTime.class, LocalDateTimeCodec.INSTANCE);
    myMap.put(LocalTime.class, LocalTimeCodec.INSTANCE);
    myMap.put(Long.class, LongCodec.INSTANCE);
    myMap.put(Short.class, ShortCodec.INSTANCE);
    myMap.put(Stream.class, StreamCodec.INSTANCE);
    myMap.put(String.class, StringCodec.INSTANCE);
    myMap.put(UUID.class, UuidCodec.INSTANCE);
    return myMap;
  }

  private static Map> r2dbcTypeToDataTypeMap() {
    Map> myMap = new HashMap<>();
    myMap.put(R2dbcType.NCHAR, StringCodec.INSTANCE);
    myMap.put(R2dbcType.CHAR, StringCodec.INSTANCE);
    myMap.put(R2dbcType.NVARCHAR, StringCodec.INSTANCE);
    myMap.put(R2dbcType.VARCHAR, StringCodec.INSTANCE);
    myMap.put(R2dbcType.CLOB, ClobCodec.INSTANCE);
    myMap.put(R2dbcType.NCLOB, ClobCodec.INSTANCE);
    myMap.put(R2dbcType.BOOLEAN, BooleanCodec.INSTANCE);
    myMap.put(R2dbcType.TINYINT, ByteCodec.INSTANCE);
    myMap.put(R2dbcType.BINARY, ByteArrayCodec.INSTANCE);
    myMap.put(R2dbcType.VARBINARY, ByteArrayCodec.INSTANCE);
    myMap.put(R2dbcType.BLOB, BlobCodec.INSTANCE);
    myMap.put(R2dbcType.INTEGER, IntCodec.INSTANCE);
    myMap.put(R2dbcType.SMALLINT, ShortCodec.INSTANCE);
    myMap.put(R2dbcType.BIGINT, BigIntegerCodec.INSTANCE);
    myMap.put(R2dbcType.NUMERIC, BigDecimalCodec.INSTANCE);
    myMap.put(R2dbcType.DECIMAL, BigDecimalCodec.INSTANCE);
    myMap.put(R2dbcType.FLOAT, FloatCodec.INSTANCE);
    myMap.put(R2dbcType.REAL, FloatCodec.INSTANCE);
    myMap.put(R2dbcType.DOUBLE, DoubleCodec.INSTANCE);
    myMap.put(R2dbcType.DATE, LocalDateCodec.INSTANCE);
    myMap.put(R2dbcType.TIME, LocalTimeCodec.INSTANCE);
    myMap.put(R2dbcType.TIME_WITH_TIME_ZONE, LocalTimeCodec.INSTANCE);
    myMap.put(R2dbcType.TIMESTAMP, LocalDateTimeCodec.INSTANCE);
    myMap.put(R2dbcType.TIMESTAMP_WITH_TIME_ZONE, LocalDateTimeCodec.INSTANCE);
    return myMap;
  }

  private static Map, Codec> classToCodecMap() {
    Map, Codec> myMap = new HashMap<>();
    myMap.put(String.class, StringCodec.INSTANCE);
    myMap.put(Clob.class, ClobCodec.INSTANCE);
    myMap.put(InputStream.class, StreamCodec.INSTANCE);
    myMap.put(Boolean.class, BooleanCodec.INSTANCE);
    myMap.put(byte[].class, ByteArrayCodec.INSTANCE);
    myMap.put(Blob.class, BlobCodec.INSTANCE);
    myMap.put(ByteBuffer.class, ByteBufferCodec.INSTANCE);
    myMap.put(BitSet.class, BitSetCodec.INSTANCE);
    myMap.put(Integer.class, IntCodec.INSTANCE);
    myMap.put(Short.class, ShortCodec.INSTANCE);
    myMap.put(BigInteger.class, BigIntegerCodec.INSTANCE);
    myMap.put(Long.class, LongCodec.INSTANCE);
    myMap.put(BigDecimal.class, BigDecimalCodec.INSTANCE);
    myMap.put(Float.class, FloatCodec.INSTANCE);
    myMap.put(Double.class, DoubleCodec.INSTANCE);
    myMap.put(LocalDate.class, LocalDateCodec.INSTANCE);
    myMap.put(LocalTime.class, LocalTimeCodec.INSTANCE);
    myMap.put(Duration.class, DurationCodec.INSTANCE);
    myMap.put(LocalDateTime.class, LocalDateTimeCodec.INSTANCE);
    myMap.put(UUID.class, UuidCodec.INSTANCE);
    return myMap;
  }

  public static Codec codecByClass(Class value, int index) {
    for (Codec codec : Codecs.LIST) {
      if (codec.canEncode(value)) {
        return codec;
      }
    }
    throw new IllegalArgumentException(
        String.format("No encoder for class %s (parameter at index %s) ", value.getName(), index));
  }
}