Java tutorial
/* * Licensed 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 com.facebook.presto.example; import com.facebook.presto.spi.RecordCursor; import com.facebook.presto.spi.RecordSet; import com.facebook.presto.spi.type.Type; import com.google.common.base.Throwables; import com.google.common.collect.ImmutableList; import com.google.common.io.ByteSource; import com.google.common.io.Resources; import java.net.MalformedURLException; import java.util.List; import org.apache.lucene.queryparser.classic.ParseException; import static java.util.Objects.requireNonNull; public class LuceneRecordSet implements RecordSet { private final List<LuceneColumnHandle> columnHandles; private final List<Type> columnTypes; private final ByteSource byteSource; public LuceneRecordSet(LuceneSplit split, List<LuceneColumnHandle> columnHandles) { requireNonNull(split, "split is null"); this.columnHandles = requireNonNull(columnHandles, "column handles is null"); ImmutableList.Builder<Type> types = ImmutableList.builder(); for (LuceneColumnHandle column : columnHandles) { types.add(column.getColumnType()); } this.columnTypes = types.build(); try { byteSource = Resources.asByteSource(split.getUri().toURL()); } catch (MalformedURLException e) { throw Throwables.propagate(e); } } @Override public List<Type> getColumnTypes() { return columnTypes; } @Override public RecordCursor cursor() { LuceneRecordCursor lrc = null; try { lrc = new LuceneRecordCursor(columnHandles); } catch (ParseException e) { e.printStackTrace(); } return lrc; } }