#919 - io.ebean package initial

src/test/java/org/tests/batchinsert/TestBatchInsertFlush.java

@@ -0,0 +1,117 @@
+package org.tests.batchinsert;
+
+import io.ebean.BaseTestCase;
+import io.ebean.Ebean;
+import io.ebean.EbeanServer;
+import io.ebean.Transaction;
+import io.ebean.annotation.Transactional;
+import io.ebean.PersistBatch;
+import org.tests.model.basic.EBasicVer;
+import org.junit.Test;
+
+import java.sql.Timestamp;
+
+import static org.junit.Assert.assertNotNull;
+
+public class TestBatchInsertFlush extends BaseTestCase {
+
+  @Test
+  @Transactional(batch = PersistBatch.ALL)
+  public void transactional_flushOnGetId() {
+
+    EbeanServer server = Ebean.getDefaultServer();
+
+    EBasicVer b1 = new EBasicVer("b1");
+    server.save(b1);
+
+    EBasicVer b2 = new EBasicVer("b2");
+    server.save(b2);
+
+    Integer id = b1.getId();
+    assertNotNull(id);
+    EBasicVer b3 = new EBasicVer("b3");
+    server.save(b3);
+  }
+
+  @Test
+  public void testFlushOnGetId() {
+
+    EbeanServer server = Ebean.getDefaultServer();
+    Transaction txn = server.beginTransaction();
+    try {
+      txn.setBatch(PersistBatch.ALL);
+
+      EBasicVer b1 = new EBasicVer("b1");
+      server.save(b1, txn);
+
+      EBasicVer b2 = new EBasicVer("b2");
+      server.save(b2, txn);
+
+      Integer id = b1.getId();
+      assertNotNull(id);
+
+      EBasicVer b3 = new EBasicVer("b3");
+      server.save(b3, txn);
+
+      txn.commit();
+
+    } finally {
+      txn.end();
+    }
+  }
+
+  @Test
+  public void testFlushOnGetProperty() {
+
+    EbeanServer server = Ebean.getDefaultServer();
+    Transaction txn = server.beginTransaction();
+    try {
+      txn.setBatch(PersistBatch.ALL);
+
+      EBasicVer b1 = new EBasicVer("b1");
+      server.save(b1, txn);
+
+      EBasicVer b2 = new EBasicVer("b2");
+      server.save(b2, txn);
+
+      // flush here
+      Timestamp lastUpdate = b1.getLastUpdate();
+      assertNotNull(lastUpdate);
+
+      EBasicVer b3 = new EBasicVer("b3");
+      server.save(b3, txn);
+
+      txn.commit();
+
+    } finally {
+      txn.end();
+    }
+  }
+
+  @Test
+  public void testFlushOnSetProperty() {
+
+    EbeanServer server = Ebean.getDefaultServer();
+    Transaction txn = server.beginTransaction();
+    try {
+      txn.setBatch(PersistBatch.ALL);
+
+      EBasicVer b1 = new EBasicVer("b1");
+      server.save(b1, txn);
+
+      EBasicVer b2 = new EBasicVer("b2");
+      server.save(b2, txn);
+
+      // flush here
+      b1.setDescription("modify");
+
+      EBasicVer b3 = new EBasicVer("b3");
+      server.save(b3, txn);
+
+      txn.commit();
+
+    } finally {
+      txn.end();
+    }
+  }
+}

src/test/java/org/tests/batchinsert/TestBatchInsertSimple.java

@@ -0,0 +1,213 @@
+package org.tests.batchinsert;
+
+import io.ebean.BaseTestCase;
+import io.ebean.Ebean;
+import io.ebean.Transaction;
+import io.ebean.annotation.Transactional;
+import io.ebean.PersistBatch;
+import org.tests.model.basic.UTDetail;
+import org.tests.model.basic.UTMaster;
+import org.junit.Test;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Random;
+
+//import static org.assertj.core.api.Assertions.assertThat;
+//import static org.junit.Assert.assertNull;
+
+public class TestBatchInsertSimple extends BaseTestCase {
+
+  Random random = new Random();
+
+  @Test
+  public void testJdbcBatchPerRequestWithMasterAndDetails() {
+
+    int numOfMasters = 4;// 2 + random.nextInt(8);
+
+    Transaction transaction = Ebean.beginTransaction();
+    try {
+      transaction.setBatch(PersistBatch.NONE);
+      transaction.setBatchOnCascade(PersistBatch.INSERT);
+      transaction.setBatchSize(30);
+      // setBatchGetGeneratedKeys MUST be turned off for MS SQL Server because :(
+      transaction.setBatchGetGeneratedKeys(false);
+
+      for (int i = 0; i < numOfMasters; i++) {
+        UTMaster master = createMasterAndDetails(i, 20);
+        master.save();
+      }
+
+      transaction.commit();
+
+    } finally {
+      transaction.end();
+    }
+  }
+
+  @Test
+  public void testTransactional() {
+
+    if (isMsSqlServer()) return;
+
+    saveWithFullBatchMode();
+  }
+
+  @Transactional(batchSize = 50)
+  public void saveWithFullBatchMode() {
+
+    int numOfMasters = 4;
+
+    for (int i = 0; i < numOfMasters; i++) {
+      UTMaster master = createMasterAndDetails(i, 5);
+      // the save is 'batched' and does not execute immediately
+      // ... it now acts more like 'merge/persist'
+      master.save();
+    }
+  }
+
+//  @Test
+//  public void testTransactional_skipGeneratedBeans() {
+//
+//    if (isMsSqlServer()) return;
+//
+//    List<UTMaster> beans = saveWithFullBatchMode_skipGeneratedKeys();
+//    for (UTMaster bean : beans) {
+//      assertNull(bean.getId());
+//    }
+//  }
+//
+//  @Transactional(batch=PersistBatch.ALL, batchSize=50, getGeneratedKeys = false)
+//  public List<UTMaster> saveWithFullBatchMode_skipGeneratedKeys() {
+//
+//    Transaction transaction = server().currentTransaction();
+//    SpiTransaction spiTxn = (SpiTransaction)transaction;
+//    Boolean generatedKeys = spiTxn.getBatchGetGeneratedKeys();
+//
+//    assertThat(generatedKeys).isFalse();
+//
+//    List<UTMaster> beans = new ArrayList<UTMaster>();
+//    for (int i = 0; i < 4; i++) {
+//     beans.add(createMaster(i));
+//    }
+//
+//    server().saveAll(beans);
+//    return beans;
+//  }
+
+  @Test
+  public void testJdbcBatchPerRequestWithMasterOnly() {
+
+    int numOfMasters = 4;
+
+    Transaction transaction = Ebean.beginTransaction();
+    try {
+      transaction.setBatch(PersistBatch.NONE);
+      transaction.setBatchOnCascade(PersistBatch.INSERT);
+      transaction.setBatchSize(30);
+      // setBatchGetGeneratedKeys MUST be turned off for MS SQL Server because :(
+      transaction.setBatchGetGeneratedKeys(false);
+
+      for (int i = 0; i < numOfMasters; i++) {
+        UTMaster master = createMaster(i);
+        Ebean.save(master);
+      }
+
+      transaction.commit();
+
+    } finally {
+      transaction.end();
+    }
+  }
+
+  @Test
+  public void testJdbcBatchOnCollection() {
+
+    // MS SQL Server doesn't like batch inserts when we need getGeneratedKeys
+    if (isMsSqlServer()) return;
+
+    int numOfMasters = 3;
+
+    List<UTMaster> masters = new ArrayList<>();
+    for (int i = 0; i < numOfMasters; i++) {
+      masters.add(createMasterAndDetails(i, 7));
+    }
+
+    Transaction transaction = Ebean.beginTransaction();
+    try {
+      transaction.setBatch(PersistBatch.NONE);
+      transaction.setBatchOnCascade(PersistBatch.ALL);
+      transaction.setBatchSize(20);
+
+      // escalate based on batchOnCascade value
+      Ebean.saveAll(masters);
+
+      transaction.commit();
+
+    } finally {
+      transaction.end();
+    }
+  }
+
+  @Test
+  public void testJdbcBatchOnCollectionNoTransaction() {
+
+    int numOfMasters = 3;
+
+    List<UTMaster> masters = new ArrayList<>();
+    for (int i = 0; i < numOfMasters; i++) {
+      masters.add(createMasterAndDetails(i, 5));
+    }
+
+    // escalate based on batchOnCascade value
+    Ebean.saveAll(masters);
+
+    for (int i = 0; i < masters.size(); i++) {
+      UTMaster utMaster = masters.get(i);
+      utMaster.setName(utMaster.getName() + "-Mod");
+      if (i % 2 == 0) {
+        // make the updates a little bit different
+        utMaster.setDescription("Blah");
+      }
+    }
+
+    Ebean.saveAll(masters);
+  }
+
+  private UTMaster createMasterAndDetails(int masterPos, int size) {
+
+    UTMaster master = createMaster(masterPos);
+    List<UTDetail> details = new ArrayList<>();
+
+    int count = 2 + random.nextInt(size);
+
+    for (int i = 0; i < count; i++) {
+
+      int qty = 1 + random.nextInt(99);
+      double amount = random.nextDouble();
+
+      details.add(createDetail(masterPos + "-" + i, qty, amount));
+    }
+    master.setDetails(details);
+    return master;
+  }
+
+  private UTMaster createMaster(int position) {
+    UTMaster m = new UTMaster();
+    m.setName("batchInsert-master" + position);
+    return m;
+  }
+
+  private UTDetail createDetail(String position, int qty, double amount) {
+
+    UTDetail detail = new UTDetail();
+    detail.setName("batchInsert-detail-" + position);
+    detail.setQty(qty);
+    detail.setAmount(amount);
+
+    // System.out.println("-- "+detail);
+
+    return detail;
+  }
+
+}

src/test/java/org/tests/batchinsert/TestBatchInsertWithInitialisedCollection.java

@@ -0,0 +1,53 @@
+package org.tests.batchinsert;
+
+import io.ebean.BaseTestCase;
+import io.ebean.Ebean;
+import io.ebean.Transaction;
+import io.ebean.PersistBatch;
+import org.tests.model.basic.OCachedBean;
+import org.avaje.ebeantest.LoggedSqlCollector;
+import org.junit.Test;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+public class TestBatchInsertWithInitialisedCollection extends BaseTestCase {
+
+  @Test
+  public void test() {
+
+    if (isMsSqlServer()) return;
+
+    List<OCachedBean> list = new ArrayList();
+
+    for (int i = 0; i < 3; i++) {
+      OCachedBean bean = new OCachedBean();
+      bean.setName("name " + i);
+      list.add(bean);
+    }
+
+    LoggedSqlCollector.start();
+
+    Transaction txn = Ebean.beginTransaction();
+    try {
+      txn.setBatch(PersistBatch.ALL);
+
+      Ebean.saveAll(list);
+      txn.commit();
+
+    } finally {
+      txn.end();
+    }
+
+    List<String> loggedSql = LoggedSqlCollector.stop();
+    assertThat(loggedSql).hasSize(3);
+
+    for (String sql : loggedSql) {
+      assertThat(sql).contains("insert into o_cached_bean (");
+      assertThat(sql).contains("name) values (?");
+    }
+
+  }
+}

src/test/java/org/tests/batchinsert/TestBatchOnCascadeExceptionHandling.java

@@ -0,0 +1,110 @@
+package org.tests.batchinsert;
+
+import io.ebean.BaseTestCase;
+import io.ebean.EbeanServer;
+import io.ebean.Transaction;
+import io.ebean.PersistBatch;
+import io.ebeaninternal.api.SpiTransaction;
+import io.ebeaninternal.server.persist.BatchControl;
+import org.tests.model.basic.EBasicWithUniqueCon;
+import org.tests.model.basic.EOptOneB;
+import org.tests.model.basic.EOptOneC;
+import org.assertj.core.api.Assertions;
+import org.junit.Test;
+
+import javax.persistence.PersistenceException;
+import java.sql.SQLException;
+import java.sql.Savepoint;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+public class TestBatchOnCascadeExceptionHandling extends BaseTestCase {
+
+  @Test
+  public void testBatchScenarioWithSavepoint() throws SQLException {
+    server().save(createEntityWithName("conflict", "before"));
+
+    Transaction txn = server().beginTransaction();
+    try {
+      EBasicWithUniqueCon v2 = createEntityWithName("conflict", "after");
+      txn.flushBatch();
+      Savepoint sp = txn.getConnection().setSavepoint();
+      try {
+        server().save(v2); // unique key violation
+      } catch (PersistenceException e) {
+        txn.getConnection().rollback(sp);
+
+        EBasicWithUniqueCon conflicting = server().find(EBasicWithUniqueCon.class).where().eq("name", "conflict").findUnique();
+        assertThat(conflicting).isNotNull();
+        server().delete(conflicting);
+        server().save(v2); // try again
+      }
+      txn.commit();
+    } finally {
+      txn.end();
+    }
+
+    EBasicWithUniqueCon winner = server().find(EBasicWithUniqueCon.class).where().eq("name", "conflict").findUnique();
+    assertThat(winner).isNotNull();
+    assertThat(winner.getDescription()).isEqualTo("after");
+  }
+
+  @Test
+  public void testBatchedInsertFailure() {
+    server().save(createEntityWithName("foo"));
+    testBatchOnCascadeIsExceptionSafe(server(), () -> {
+      server().save(createEntityWithName("foo")); // duplicate name on insert
+    });
+  }
+
+  @Test
+  public void testBatchedUpdateFailure() {
+    server().save(createEntityWithName("bla"));
+    final EBasicWithUniqueCon bar = createEntityWithName("bar");
+    server().save(bar);
+    testBatchOnCascadeIsExceptionSafe(server(), () -> {
+      bar.setName("bla");
+      server().save(bar); // duplicate name on update
+    });
+  }
+
+  @Test
+  public void testBatchedDeleteFailure() {
+    final EOptOneC c = new EOptOneC();
+    server().save(c);
+    final EOptOneB b = new EOptOneB();
+    b.setC(c);
+    server().save(b);
+    testBatchOnCascadeIsExceptionSafe(server(), () -> {
+      server().delete(c); // foreign key violation
+    });
+  }
+
+  protected void testBatchOnCascadeIsExceptionSafe(EbeanServer server, Runnable failingOperation) {
+    Transaction txn = server.beginTransaction();
+    assertThat(txn.getBatch()).isSameAs(PersistBatch.NONE);
+    assertThat(txn.getBatchOnCascade()).isSameAs(PersistBatch.ALL);
+
+    try {
+      failingOperation.run();
+      Assertions.fail("PersistenceException expected");
+    } catch (PersistenceException e) {
+      assertThat(txn.getBatch()).as("batch mode").isSameAs(PersistBatch.NONE); // should not have changed
+      BatchControl bc = ((SpiTransaction) txn).getBatchControl();
+      assertThat(bc == null || bc.isEmpty()).as("batch emtpy").isTrue();
+    } finally {
+      txn.end();
+    }
+  }
+
+  protected EBasicWithUniqueCon createEntityWithName(String name) {
+    return createEntityWithName(name, null);
+  }
+
+  protected EBasicWithUniqueCon createEntityWithName(String name, String description) {
+    EBasicWithUniqueCon it = new EBasicWithUniqueCon();
+    it.setName(name);
+    it.setDescription(description);
+    return it;
+  }
+}

src/test/java/org/tests/batchinsert/TestBatchSaveWithGetBeanId.java

@@ -0,0 +1,34 @@
+package org.tests.batchinsert;
+
+import io.ebean.BaseTestCase;
+import io.ebean.Ebean;
+import io.ebean.EbeanServer;
+import io.ebean.annotation.Transactional;
+import org.tests.model.basic.Customer;
+import org.junit.Test;
+
+import static org.junit.Assert.assertNotNull;
+
+public class TestBatchSaveWithGetBeanId extends BaseTestCase {
+
+  /**
+   * Making this transaction with batchSize means that the insert
+   * below does not occur immediately ... and the getBeanId()
+   * should invoke the flush (and hence trigger the insert).
+   */
+  @Transactional(batchSize = 10)
+  @Test
+  public void test() {
+
+    EbeanServer server = Ebean.getDefaultServer();
+    Customer model = new Customer();
+    model.setName("foo");
+
+    server.insert(model);
+
+    // should invoke a flush which then means the
+    // insert occurs and the bean has an Id value
+    Object beanId = server.getBeanId(model);
+    assertNotNull(beanId);
+  }
+}