io.zulia.server.index.federator.GetFieldNamesRequestFederator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of zulia-server Show documentation
$project.description
There is a newer version: 1.6.4
package io.zulia.server.index.federator;

import io.zulia.message.ZuliaBase;
import io.zulia.message.ZuliaBase.Node;
import io.zulia.message.ZuliaServiceOuterClass.GetFieldNamesRequest;
import io.zulia.message.ZuliaServiceOuterClass.GetFieldNamesResponse;
import io.zulia.message.ZuliaServiceOuterClass.IndexRouting;
import io.zulia.message.ZuliaServiceOuterClass.InternalGetFieldNamesRequest;
import io.zulia.server.connection.client.InternalClient;
import io.zulia.server.index.ZuliaIndex;

import java.io.IOException;
import java.util.Collection;
import java.util.List;
import java.util.Set;
import java.util.TreeSet;
import java.util.concurrent.ExecutorService;

public class GetFieldNamesRequestFederator extends MasterSlaveNodeRequestFederator {
	private final InternalClient internalClient;
	private final ZuliaIndex index;

	public GetFieldNamesRequestFederator(Node thisNode, Collection otherNodesActive, ZuliaBase.MasterSlaveSettings masterSlaveSettings, ZuliaIndex index,
			ExecutorService pool, InternalClient internalClient) throws IOException {
		super(thisNode, otherNodesActive, masterSlaveSettings, index, pool);
		this.internalClient = internalClient;
		this.index = index;
	}

	@Override
	protected GetFieldNamesResponse processExternal(Node node, GetFieldNamesRequest request) throws Exception {
		IndexRouting indexRouting = getIndexRouting(node).get(0);
		InternalGetFieldNamesRequest internalRequest = InternalGetFieldNamesRequest.newBuilder().setIndexRouting(indexRouting).setGetFieldNamesRequest(request)
				.build();
		return internalClient.getFieldNames(node, internalRequest);
	}

	@Override
	protected GetFieldNamesResponse processInternal(Node node, GetFieldNamesRequest request) throws Exception {
		IndexRouting indexRouting = getIndexRouting(node).get(0);
		InternalGetFieldNamesRequest internalRequest = InternalGetFieldNamesRequest.newBuilder().setIndexRouting(indexRouting).setGetFieldNamesRequest(request)
				.build();
		return internalGetFieldNames(index, internalRequest);
	}

	public static GetFieldNamesResponse internalGetFieldNames(ZuliaIndex index, InternalGetFieldNamesRequest request) throws Exception {
		return index.getFieldNames(request);
	}

	public GetFieldNamesResponse getResponse(GetFieldNamesRequest request) throws Exception {

		Set fieldNames = new TreeSet<>();
		List responses = send(request);
		for (GetFieldNamesResponse response : responses) {
			fieldNames.addAll(response.getFieldNameList());
		}

		GetFieldNamesResponse.Builder responseBuilder = GetFieldNamesResponse.newBuilder();
		responseBuilder.addAllFieldName(fieldNames);
		return responseBuilder.build();
	}
}