All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • com.adgear.anoa.read.ThriftFieldWrapper Maven / Gradle / Ivy

    package com.adgear.anoa.read;

import org.apache.thrift.TFieldIdEnum;
import org.apache.thrift.TFieldRequirementType;
import org.apache.thrift.meta_data.EnumMetaData;
import org.apache.thrift.meta_data.FieldMetaData;
import org.apache.thrift.meta_data.FieldValueMetaData;
import org.apache.thrift.meta_data.ListMetaData;
import org.apache.thrift.meta_data.MapMetaData;
import org.apache.thrift.meta_data.SetMetaData;
import org.apache.thrift.meta_data.StructMetaData;
import org.apache.thrift.protocol.TType;

import java.util.function.Supplier;
import java.util.stream.Stream;

class ThriftFieldWrapper implements FieldWrapper {

  final F tFieldIdEnum;
  final boolean isRequired;
  final private Object defaultValue;
  final Supplier defaultValueSupplier;
  final private AbstractReader reader;

  ThriftFieldWrapper(F tFieldIdEnum,
                     FieldMetaData fieldMetaData,
                     Supplier defaultValueSupplier) {
    this.tFieldIdEnum = tFieldIdEnum;
    this.isRequired = (fieldMetaData.requirementType == TFieldRequirementType.REQUIRED);
    this.reader = createReader(fieldMetaData.valueMetaData);
    this.defaultValue = defaultValueSupplier.get();
    this.defaultValueSupplier = defaultValueSupplier;
  }

  @Override
  public Stream getNames() {
    return Stream.of(tFieldIdEnum.getFieldName());
  }

  @Override
  public AbstractReader getReader() {
    return reader;
  }

  @SuppressWarnings("unchecked")
  static private AbstractReader createReader(FieldValueMetaData metaData) {
    switch (metaData.type) {
      case TType.BOOL:
        return new BooleanReader();
      case TType.BYTE:
        return new ByteReader();
      case TType.DOUBLE:
        return new DoubleReader();
      case TType.ENUM:
        return new EnumReader(((EnumMetaData) metaData).enumClass);
      case TType.I16:
        return new ShortReader();
      case TType.I32:
        return new IntegerReader();
      case TType.I64:
        return new LongReader();
      case TType.LIST:
        return new ListReader(createReader(((ListMetaData) metaData).elemMetaData));
      case TType.MAP:
        MapMetaData mapMetaData = (MapMetaData) metaData;
        if (mapMetaData.keyMetaData.type != TType.STRING) {
          throw new RuntimeException("Map key type is not string.");
        }
        return new MapReader(createReader(mapMetaData.valueMetaData));
      case TType.SET:
        return new SetReader(createReader((SetMetaData) metaData));
      case TType.STRUCT:
        return new ThriftReader(((StructMetaData) metaData).structClass);
      case TType.STRING:
        return metaData.isBinary() ? new ByteBufferReader() : new StringReader();
    }
    throw new RuntimeException("Unknown type in metadata " + metaData);
  }

  @Override
  public boolean equalsDefaultValue(Object value) {
    return value == null || value.equals(defaultValue);
  }
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy