content: /*
   * Licensed to the Apache Software Foundation (ASF) under one
   * or more contributor license agreements.  See the NOTICE file
   * distributed with this work for additional information
   * regarding copyright ownership.  The ASF licenses this file
   * to you under the Apache License, Version 2.0 (the
   * "License"); you may not use this file except in compliance
   * with the License.  You may obtain a copy of the License at
   *
   *   http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing,
   * software distributed under the License is distributed on an
   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
   * KIND, either express or implied.  See the License for the
   * specific language governing permissions and limitations
   * under the License.
   */
  package org.apache.asterix.column.values.reader;
  
  import java.io.DataInput;
  import java.io.IOException;
  
  import org.apache.asterix.column.values.IColumnValuesReader;
  import org.apache.asterix.column.values.IColumnValuesReaderFactory;
  import org.apache.asterix.column.values.reader.value.AbstractValueReader;
  import org.apache.asterix.column.values.reader.value.BooleanValueReader;
  import org.apache.asterix.column.values.reader.value.DoubleValueReader;
  import org.apache.asterix.column.values.reader.value.FloatValueReader;
  import org.apache.asterix.column.values.reader.value.LongValueReader;
  import org.apache.asterix.column.values.reader.value.NoOpValueReader;
  import org.apache.asterix.column.values.reader.value.StringValueReader;
  import org.apache.asterix.column.values.reader.value.UUIDValueReader;
  import org.apache.asterix.column.values.reader.value.key.DoubleKeyValueReader;
  import org.apache.asterix.column.values.reader.value.key.FloatKeyValueReader;
  import org.apache.asterix.column.values.reader.value.key.LevelKeyReader;
  import org.apache.asterix.column.values.reader.value.key.LongKeyValueReader;
  import org.apache.asterix.column.values.reader.value.key.StringKeyValueReader;
  import org.apache.asterix.column.values.reader.value.key.UUIDKeyValueReader;
  import org.apache.asterix.om.types.ATypeTag;
  
  public class ColumnValueReaderFactory implements IColumnValuesReaderFactory {
      @Override
      public IColumnValuesReader createValueReader(ATypeTag typeTag, int columnIndex, int maxLevel, boolean primaryKey) {
          return new PrimitiveColumnValuesReader(createReader(typeTag, primaryKey), columnIndex, maxLevel, primaryKey);
      }
  
      @Override
      public IColumnValuesReader createValueReader(ATypeTag typeTag, int columnIndex, int maxLevel, int[] delimiters) {
          return new RepeatedPrimitiveColumnValuesReader(createReader(typeTag, false), columnIndex, maxLevel, delimiters);
      }
  
      @Override
      public IColumnValuesReader createValueReader(DataInput input) throws IOException {
          ATypeTag typeTag = ATypeTag.VALUE_TYPE_MAPPING[input.readByte()];
          int columnIndex = input.readInt();
          int maxLevel = input.readInt();
          boolean primaryKey = input.readBoolean();
          boolean collection = input.readBoolean();
          if (collection) {
              int[] delimiters = new int[input.readInt()];
              for (int i = 0; i < delimiters.length; i++) {
                  delimiters[i] = input.readInt();
              }
              return createValueReader(typeTag, columnIndex, maxLevel, delimiters);
          }
          return createValueReader(typeTag, columnIndex, maxLevel, primaryKey);
      }
  
      private AbstractValueReader createReader(ATypeTag typeTag, boolean primaryKey) {
          switch (typeTag) {
              case MISSING:
              case NULL:
                  return primaryKey ? LevelKeyReader.INSTANCE : NoOpValueReader.INSTANCE;
              case BOOLEAN:
                  return new BooleanValueReader();
              case TINYINT:
              case SMALLINT:
              case INTEGER:
              case BIGINT:
                  return primaryKey ? new LongKeyValueReader(typeTag) : new LongValueReader();
              case FLOAT:
                  return primaryKey ? new FloatKeyValueReader() : new FloatValueReader();
              case DOUBLE:
                  return primaryKey ? new DoubleKeyValueReader() : new DoubleValueReader();
              case STRING:
                  return primaryKey ? new StringKeyValueReader() : new StringValueReader();
              case UUID:
                  return primaryKey ? new UUIDKeyValueReader() : new UUIDValueReader();
              default:
                  throw new UnsupportedOperationException(typeTag + " is not supported");
          }
      }
  }
  