nl.topicus.jdbc.RunningOperationsStore Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spanner-jdbc Show documentation
JDBC Driver for Google Cloud Spanner
There is a newer version: 1.1.6
package nl.topicus.jdbc;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;

import nl.topicus.jdbc.shaded.com.google.cloud.Timestamp;
import nl.topicus.jdbc.shaded.com.google.cloud.spanner.Operation;
import nl.topicus.jdbc.shaded.com.google.cloud.spanner.ResultSets;
import nl.topicus.jdbc.shaded.com.google.cloud.spanner.Struct;
import nl.topicus.jdbc.shaded.com.google.cloud.spanner.Type;
import nl.topicus.jdbc.shaded.com.google.cloud.spanner.Type.StructField;
import nl.topicus.jdbc.shaded.com.google.cloud.spanner.Value;
import nl.topicus.jdbc.shaded.com.google.rpc.Code;
import nl.topicus.jdbc.shaded.com.google.spanner.admin.database.v1.UpdateDatabaseDdlMetadata;

import nl.topicus.jdbc.exception.CloudSpannerSQLException;
import nl.topicus.jdbc.resultset.CloudSpannerResultSet;

/**
 * This class maintains a list of all long running (DDL-)operations of a
 * connection.
 * 
 * @author loite
 *
 */
class RunningOperationsStore
{
	private static final class DdlOperation
	{
		private final Timestamp timeStarted;

		private final List sql;

		private Operation operation;

		private DdlOperation(Timestamp timeStarted, List sql,
				Operation operation)
		{
			this.timeStarted = timeStarted;
			this.sql = sql;
			this.operation = operation;
		}

	}

	private List operations = new ArrayList<>();

	RunningOperationsStore()
	{
	}

	void addOperation(List sql, Operation operation)
	{
		operations.add(new DdlOperation(Timestamp.now(), sql, operation));
	}

	int clearFinishedOperations()
	{
		int count = 0;
		int index = 0;
		while (index < operations.size())
		{
			DdlOperation op = operations.get(index);
			op.operation = op.operation.reload();
			if (op.operation.isDone())
			{
				operations.remove(index);
				count++;
			}
			else
			{
				index++;
			}
		}
		return count;
	}

	void waitForOperations() throws SQLException
	{
		boolean finished = false;
		while (!finished)
		{
			finished = true;
			for (DdlOperation op : operations)
			{
				op.operation = op.operation.reload();
				if (!op.operation.isDone())
				{
					finished = false;
					break;
				}
			}
			try
			{
				Thread.sleep(1000l);
			}
			catch (InterruptedException e)
			{
				Thread.currentThread().interrupt();
				throw new CloudSpannerSQLException("Wait for DDL-operations interrupted", Code.CANCELLED, e);
			}
		}
	}

	/**
	 * 
	 * @return A result set of all DDL operations that have been issued on this
	 *         connection since the last clear operation.
	 */
	ResultSet getOperations(Statement statement)
	{
		List rows = new ArrayList<>(operations.size());
		for (DdlOperation op : operations)
		{
			op.operation = op.operation.reload();
			String exception = null;
			try
			{
				op.operation.getResult();
			}
			catch (Exception e)
			{
				exception = e.getMessage();
			}
			for (String ddl : op.sql)
			{
				rows.add(Struct.newBuilder().add("NAME", Value.string(op.operation.getName()))
						.add("TIME_STARTED", Value.timestamp(op.timeStarted)).add("STATEMENT", Value.string(ddl))
						.add("DONE", Value.bool(op.operation.isDone())).add("EXCEPTION", Value.string(exception))
						.build());
			}
		}
		nl.topicus.jdbc.shaded.com.google.cloud.spanner.ResultSet rs = ResultSets.forRows(Type.struct(StructField.of("NAME", Type.string()),
				StructField.of("TIME_STARTED", Type.timestamp()), StructField.of("STATEMENT", Type.string()),
				StructField.of("DONE", Type.bool()), StructField.of("EXCEPTION", Type.string())), rows);
		return new CloudSpannerResultSet(statement, rs);
	}

}