nl.topicus.jdbc.statement.Mutations Maven / Gradle / Ivy

package nl.topicus.jdbc.statement;

import java.util.Arrays;
import java.util.List;

import nl.topicus.jdbc.shaded.com.google.cloud.spanner.Mutation;

class Mutations
{
	private final List mutations;

	private final AbstractTablePartWorker worker;

	/**
	 * Single mutation
	 * 
	 * @param mutation
	 */
	Mutations(Mutation mutation)
	{
		this.mutations = Arrays.asList(mutation);
		this.worker = null;
	}

	Mutations(List mutations)
	{
		this.mutations = mutations;
		this.worker = null;
	}

	Mutations(AbstractTablePartWorker worker)
	{
		this.mutations = null;
		this.worker = worker;
	}

	List getMutations()
	{
		if (isWorker())
			throw new IllegalStateException(
					"Cannot call getMutations() on a Mutations-object that returns its results as a worker");
		return mutations;
	}

	AbstractTablePartWorker getWorker()
	{
		return worker;
	}

	boolean isWorker()
	{
		return worker != null;
	}

	long getNumberOfResults()
	{
		if (isWorker())
			return worker.getRecordCount();
		return mutations.size();
	}

}