io.zulia.util.CursorHelper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of zulia-common Show documentation

zulia-common

There is a newer version: 1.6.4

package io.zulia.util;

import java.nio.charset.Charset;
import java.util.Base64;

import static io.zulia.message.ZuliaQuery.LastIndexResult;
import static io.zulia.message.ZuliaQuery.LastResult;
import static io.zulia.message.ZuliaQuery.ScoredResult;

/**
 * Created by Payam Meyer on 8/7/17.
 * @author pmeyer
 */
public class CursorHelper {

	private static Charset utf8Charset = Charset.forName("utf-8");

	public static String getStaticIndexCursor(LastResult lastResult) {
		return new String(Base64.getEncoder().encode(lastResult.toByteArray()), utf8Charset);
	}

	public static String getUniqueSortedCursor(LastResult lastResult) {

		LastResult.Builder lastResultBuilder = LastResult.newBuilder();
		for (LastIndexResult lastIndexResult : lastResult.getLastIndexResultList()) {
			LastIndexResult.Builder lastIndexResultBuilder = LastIndexResult.newBuilder();
			lastIndexResultBuilder.setIndexName(lastIndexResult.getIndexName());
			for (ScoredResult scoredResult : lastIndexResult.getLastForShardList()) {
				ScoredResult.Builder scoredResultBuilder = ScoredResult.newBuilder(scoredResult).clearScore().clearResultDocument();
				lastIndexResultBuilder.addLastForShard(scoredResultBuilder);
			}
			lastResultBuilder.addLastIndexResult(lastIndexResultBuilder);
		}

		return new String(Base64.getEncoder().encode(lastResultBuilder.build().toByteArray()), utf8Charset);
	}

	public static LastResult getLastResultFromCursor(String cursor) {
		try {
			return LastResult.parseFrom(Base64.getDecoder().decode(cursor.getBytes(utf8Charset)));
		}
		catch (Exception e) {
			throw new RuntimeException("Invalid cursor");
		}
	}
}