org.apache.tika.server.client.TikaClient Maven / Gradle / Ivy
/*
* 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.tika.server.client;
import java.io.IOException;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.List;
import java.util.Random;
import org.apache.tika.config.TikaConfig;
import org.apache.tika.exception.TikaException;
import org.apache.tika.metadata.serialization.JsonFetchEmitTuple;
import org.apache.tika.metadata.serialization.JsonFetchEmitTupleList;
import org.apache.tika.pipes.FetchEmitTuple;
public class TikaClient {
private final Random random = new Random();
private final List clients;
private TikaClient(List clients) {
this.clients = clients;
}
public static TikaClient get(TikaConfig tikaConfig, List tikaServers)
throws TikaClientConfigException {
List clients = new ArrayList<>();
for (String url : tikaServers) {
clients.add(TikaHttpClient.get(url));
}
return new TikaClient(clients);
}
/*public List parse(InputStream is, Metadata metadata)
throws IOException, TikaException {
}*/
public TikaEmitterResult parse(FetchEmitTuple fetchEmit) throws IOException, TikaException {
TikaHttpClient client = getHttpClient();
StringWriter writer = new StringWriter();
JsonFetchEmitTuple.toJson(fetchEmit, writer);
return client.postJson(writer.toString());
}
public TikaEmitterResult parseAsync(List tuples)
throws IOException, TikaException {
StringWriter writer = new StringWriter();
JsonFetchEmitTupleList.toJson(tuples, writer);
TikaHttpClient client = getHttpClient();
return client.postJsonAsync(writer.toString());
}
private TikaHttpClient getHttpClient() {
if (clients.size() == 1) {
return clients.get(0);
}
int index = random.nextInt(clients.size());
return clients.get(index);
}
}