How to use

io.prestosql.sql.tree.Isolation

constructor

@Override
public Node visitRepeatableRead(SqlBaseParser.RepeatableReadContext context)
{
  return new Isolation(getLocation(context), Isolation.Level.REPEATABLE_READ);
}

@Override
public Node visitReadCommitted(SqlBaseParser.ReadCommittedContext context)
{
  return new Isolation(getLocation(context), Isolation.Level.READ_COMMITTED);
}

@Override
public Node visitSerializable(SqlBaseParser.SerializableContext context)
{
  return new Isolation(getLocation(context), Isolation.Level.SERIALIZABLE);
}

@Override
public Node visitReadCommitted(SqlBaseParser.ReadCommittedContext context)
{
  return new Isolation(getLocation(context), Isolation.Level.READ_COMMITTED);
}

@Override
public Node visitSerializable(SqlBaseParser.SerializableContext context)
{
  return new Isolation(getLocation(context), Isolation.Level.SERIALIZABLE);
}

@Override
public Node visitReadUncommitted(SqlBaseParser.ReadUncommittedContext context)
{
  return new Isolation(getLocation(context), Isolation.Level.READ_UNCOMMITTED);
}

@Override
public Node visitRepeatableRead(SqlBaseParser.RepeatableReadContext context)
{
  return new Isolation(getLocation(context), Isolation.Level.REPEATABLE_READ);
}

@Override
public Node visitReadUncommitted(SqlBaseParser.ReadUncommittedContext context)
{
  return new Isolation(getLocation(context), Isolation.Level.READ_UNCOMMITTED);
}

assertStatement("START TRANSACTION ISOLATION LEVEL READ UNCOMMITTED",
    new StartTransaction(ImmutableList.of(
        new Isolation(Isolation.Level.READ_UNCOMMITTED))));
assertStatement("START TRANSACTION ISOLATION LEVEL READ COMMITTED",
    new StartTransaction(ImmutableList.of(
        new Isolation(Isolation.Level.READ_COMMITTED))));
assertStatement("START TRANSACTION ISOLATION LEVEL REPEATABLE READ",
    new StartTransaction(ImmutableList.of(
        new Isolation(Isolation.Level.REPEATABLE_READ))));
assertStatement("START TRANSACTION ISOLATION LEVEL SERIALIZABLE",
    new StartTransaction(ImmutableList.of(
        new Isolation(Isolation.Level.SERIALIZABLE))));
assertStatement("START TRANSACTION READ ONLY",
    new StartTransaction(ImmutableList.of(
assertStatement("START TRANSACTION ISOLATION LEVEL READ COMMITTED, READ ONLY",
    new StartTransaction(ImmutableList.of(
        new Isolation(Isolation.Level.READ_COMMITTED),
        new TransactionAccessMode(true))));
assertStatement("START TRANSACTION READ ONLY, ISOLATION LEVEL READ COMMITTED",
    new StartTransaction(ImmutableList.of(
        new TransactionAccessMode(true),
        new Isolation(Isolation.Level.READ_COMMITTED))));
assertStatement("START TRANSACTION READ WRITE, ISOLATION LEVEL SERIALIZABLE",
    new StartTransaction(ImmutableList.of(
        new TransactionAccessMode(false),
        new Isolation(Isolation.Level.SERIALIZABLE))));

@Test
public void testStartTransactionTooManyIsolationLevels()
{
  Session session = sessionBuilder()
      .setClientTransactionSupport()
      .build();
  TransactionManager transactionManager = createTestTransactionManager();
  QueryStateMachine stateMachine = createQueryStateMachine("START TRANSACTION", session, transactionManager);
  assertFalse(stateMachine.getSession().getTransactionId().isPresent());
  try {
    getFutureValue(new StartTransactionTask().execute(
        new StartTransaction(ImmutableList.of(new Isolation(Isolation.Level.READ_COMMITTED), new Isolation(Isolation.Level.READ_COMMITTED))),
        transactionManager,
        metadata,
        new AllowAllAccessControl(),
        stateMachine,
        emptyList()));
    fail();
  }
  catch (SemanticException e) {
    assertEquals(e.getCode(), INVALID_TRANSACTION_MODE);
  }
  assertTrue(transactionManager.getAllTransactionInfos().isEmpty());
  assertFalse(stateMachine.getQueryInfo(Optional.empty()).isClearTransactionId());
  assertFalse(stateMachine.getQueryInfo(Optional.empty()).getStartedTransactionId().isPresent());
}

@Test
public void testStartTransactionTooManyIsolationLevels()
{
  Session session = sessionBuilder()
      .setClientTransactionSupport()
      .build();
  TransactionManager transactionManager = createTestTransactionManager();
  QueryStateMachine stateMachine = createQueryStateMachine("START TRANSACTION", session, transactionManager);
  assertFalse(stateMachine.getSession().getTransactionId().isPresent());
  try {
    getFutureValue(new StartTransactionTask().execute(
        new StartTransaction(ImmutableList.of(new Isolation(Isolation.Level.READ_COMMITTED), new Isolation(Isolation.Level.READ_COMMITTED))),
        transactionManager,
        metadata,
        new AllowAllAccessControl(),
        stateMachine,
        emptyList()));
    fail();
  }
  catch (SemanticException e) {
    assertEquals(e.getCode(), INVALID_TRANSACTION_MODE);
  }
  assertTrue(transactionManager.getAllTransactionInfos().isEmpty());
  assertFalse(stateMachine.getQueryInfo(Optional.empty()).isClearTransactionId());
  assertFalse(stateMachine.getQueryInfo(Optional.empty()).getStartedTransactionId().isPresent());
}

@Test
public void testStartTransactionExplicitModes()
{
  Session session = sessionBuilder()
      .setClientTransactionSupport()
      .build();
  TransactionManager transactionManager = createTestTransactionManager();
  QueryStateMachine stateMachine = createQueryStateMachine("START TRANSACTION", session, transactionManager);
  assertFalse(stateMachine.getSession().getTransactionId().isPresent());
  getFutureValue(new StartTransactionTask().execute(
      new StartTransaction(ImmutableList.of(new Isolation(Isolation.Level.SERIALIZABLE), new TransactionAccessMode(true))),
      transactionManager,
      metadata,
      new AllowAllAccessControl(),
      stateMachine,
      emptyList()));
  assertFalse(stateMachine.getQueryInfo(Optional.empty()).isClearTransactionId());
  assertTrue(stateMachine.getQueryInfo(Optional.empty()).getStartedTransactionId().isPresent());
  assertEquals(transactionManager.getAllTransactionInfos().size(), 1);
  TransactionInfo transactionInfo = transactionManager.getTransactionInfo(stateMachine.getQueryInfo(Optional.empty()).getStartedTransactionId().get());
  assertEquals(transactionInfo.getIsolationLevel(), IsolationLevel.SERIALIZABLE);
  assertTrue(transactionInfo.isReadOnly());
  assertFalse(transactionInfo.isAutoCommitContext());
}

@Test
public void testStartTransactionExplicitModes()
{
  Session session = sessionBuilder()
      .setClientTransactionSupport()
      .build();
  TransactionManager transactionManager = createTestTransactionManager();
  QueryStateMachine stateMachine = createQueryStateMachine("START TRANSACTION", session, transactionManager);
  assertFalse(stateMachine.getSession().getTransactionId().isPresent());
  getFutureValue(new StartTransactionTask().execute(
      new StartTransaction(ImmutableList.of(new Isolation(Isolation.Level.SERIALIZABLE), new TransactionAccessMode(true))),
      transactionManager,
      metadata,
      new AllowAllAccessControl(),
      stateMachine,
      emptyList()));
  assertFalse(stateMachine.getQueryInfo(Optional.empty()).isClearTransactionId());
  assertTrue(stateMachine.getQueryInfo(Optional.empty()).getStartedTransactionId().isPresent());
  assertEquals(transactionManager.getAllTransactionInfos().size(), 1);
  TransactionInfo transactionInfo = transactionManager.getTransactionInfo(stateMachine.getQueryInfo(Optional.empty()).getStartedTransactionId().get());
  assertEquals(transactionInfo.getIsolationLevel(), IsolationLevel.SERIALIZABLE);
  assertTrue(transactionInfo.isReadOnly());
  assertFalse(transactionInfo.isAutoCommitContext());
}