/* * Copyright (C) 2019 Ryan Murray * * 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 org.apache.arrow.flight.spark; import java.util.Iterator; import org.apache.arrow.flight.Action; import org.apache.arrow.flight.FlightClient; import org.apache.arrow.flight.Location; import org.apache.arrow.flight.Result; import org.apache.arrow.memory.BufferAllocator; import org.apache.arrow.memory.RootAllocator; public class FlightClientFactory implements AutoCloseable { private final BufferAllocator allocator = new RootAllocator(); private final Location defaultLocation; private final String username; private final String password; private final boolean parallel; public FlightClientFactory(Location defaultLocation, String username, String password, boolean parallel) { this.defaultLocation = defaultLocation; this.username = username; this.password = (password == null || password.equals("$NULL$")) ? null : password; this.parallel = parallel; } public FlightClient apply() { FlightClient client = FlightClient.builder(allocator, defaultLocation).build(); client.authenticateBasic(username, password); if (parallel) { Iterator<Result> res = client.doAction(new Action("PARALLEL")); res.forEachRemaining(Object::toString); } return client; } @Override public void close() { allocator.close(); } }