/**
* 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.hadoop.hive.serde2.objectinspector;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;
/**
* DefaultListObjectInspector works on list data that is stored as a Java List
* or Java Array object.
*
* Always use the ObjectInspectorFactory to create new ObjectInspector objects,
* instead of directly creating an instance of this class.
*/
public class StandardListObjectInspector implements SettableListObjectInspector {
private ObjectInspector listElementObjectInspector;
protected StandardListObjectInspector() {
super();
}
/**
* Call ObjectInspectorFactory.getStandardListObjectInspector instead.
*/
protected StandardListObjectInspector(
ObjectInspector listElementObjectInspector) {
this.listElementObjectInspector = listElementObjectInspector;
}
public final Category getCategory() {
return Category.LIST;
}
// without data
public ObjectInspector getListElementObjectInspector() {
return listElementObjectInspector;
}
// with data
@SuppressWarnings({ "rawtypes", "unchecked" })
public Object getListElement(Object data, int index) {
if (data == null) {
return null;
}
// We support List<Object>, Set<Object> and Object[]
// so we have to do differently.
if (! (data instanceof List)) {
if (! (data instanceof Set)) {
Object[] list = (Object[]) data;
if (index < 0 || index >= list.length) {
return null;
}
return list[index];
} else {
data = new ArrayList((Set<?>) data);
}
}
List<?> list = (List<?>) data;
if (index < 0 || index >= list.size()) {
return null;
}
return list.get(index);
}
public int getListLength(Object data) {
if (data == null) {
return -1;
}
// We support List<Object>, Set<Object> and Object[]
// so we have to do differently.
if (! (data instanceof List)) {
if (! (data instanceof Set)) {
Object[] list = (Object[]) data;
return list.length;
} else {
Set<?> set = (Set<?>) data;
return set.size();
}
} else {
List<?> list = (List<?>) data;
return list.size();
}
}
@SuppressWarnings({ "rawtypes", "unchecked" })
public List<?> getList(Object data) {
if (data == null) {
return null;
}
// We support List<Object>, Set<Object> and Object[]
// so we have to do differently.
if (! (data instanceof List)) {
if (! (data instanceof Set)) {
data = java.util.Arrays.asList((Object[]) data);
} else {
data = new ArrayList((Set<?>) data);
}
}
List<?> list = (List<?>) data;
return list;
}
public String getTypeName() {
return org.apache.hadoop.hive.serde.serdeConstants.LIST_TYPE_NAME + "<"
+ listElementObjectInspector.getTypeName() + ">";
}
// /////////////////////////////
// SettableListObjectInspector
@Override
public Object create(int size) {
List<Object> a = new ArrayList<Object>(size);
for (int i = 0; i < size; i++) {
a.add(null);
}
return a;
}
@Override
public Object resize(Object list, int newSize) {
List<Object> a = (List<Object>) list;
while (a.size() < newSize) {
a.add(null);
}
while (a.size() > newSize) {
a.remove(a.size() - 1);
}
return a;
}
@Override
public Object set(Object list, int index, Object element) {
List<Object> a = (List<Object>) list;
a.set(index, element);
return a;
}
}