#1632 - ENH: Add bind parameter expansion for SqlUpdate

This commit is contained in:
rob bygrave
2019-02-07 23:19:00 +13:00
5 changed files with 282 additions and 10 deletions
@@ -1,8 +1,10 @@
package org.tests.update;
import io.ebean.BaseTestCase;
import io.ebean.DB;
import io.ebean.Ebean;
import io.ebean.SqlUpdate;
import io.ebean.Transaction;
import org.ebeantest.LoggedSqlCollector;
import org.junit.Test;
@@ -34,6 +36,219 @@ public class TestSqlUpdateBindMultipleLists extends BaseTestCase {
assertThat(sql).hasSize(2);
}
@Test
public void positionParamsExpansion() {
SqlUpdate sqlUpdate = Ebean.createSqlUpdate("delete from o_customer where id in (?1)");
sqlUpdate.setParameter(1, asList(9991, 9992, 9993));
sqlUpdate.execute();
assertEquals("delete from o_customer where id in (?,?,?)", sqlUpdate.getGeneratedSql());
sqlUpdate = Ebean.createSqlUpdate("delete from o_customer where id in (?1)");
sqlUpdate.setParameter(1, asList(9991, 9993));
sqlUpdate.execute();
assertEquals("delete from o_customer where id in (?,?)", sqlUpdate.getGeneratedSql());
sqlUpdate = Ebean.createSqlUpdate("delete from o_customer where id in (?1)");
sqlUpdate.setParameter(1, asList(9993));
sqlUpdate.execute();
assertEquals("delete from o_customer where id in (?)", sqlUpdate.getGeneratedSql());
}
@Test
public void positionParamsExpansion_withPrePost() {
SqlUpdate sqlUpdate = Ebean.createSqlUpdate("delete from o_customer where id > ? and id in (?2) and id < ?");
sqlUpdate.setParameter(1, 90);
sqlUpdate.setParameter(2, asList(9991, 9992, 9993));
sqlUpdate.setParameter(3, 91);
sqlUpdate.execute();
assertEquals("delete from o_customer where id > ? and id in (?,?,?) and id < ?", sqlUpdate.getGeneratedSql());
}
@Test
public void positionParamsExpansion_withPrePost_usingParam() {
SqlUpdate sqlUpdate = Ebean.createSqlUpdate("delete from o_customer where id > ? and id in (?2) and id < ?");
sqlUpdate.setParams(90, asList(9991, 9992, 9993), 91);
sqlUpdate.execute();
assertEquals("delete from o_customer where id > ? and id in (?,?,?) and id < ?", sqlUpdate.getGeneratedSql());
}
@Test
public void positionParamsExpansion_withPre() {
SqlUpdate sqlUpdate = Ebean.createSqlUpdate("delete from o_customer where name = ? and id in (?2)");
sqlUpdate.setParameter(1, "Foo");
sqlUpdate.setParameter(2, asList(9991, 9992, 9993));
sqlUpdate.execute();
assertEquals("delete from o_customer where name = ? and id in (?,?,?)", sqlUpdate.getGeneratedSql());
}
@Test
public void positionParamsExpansion_multi() {
SqlUpdate upd = DB.sqlUpdate("delete from o_customer where id in (?1) and name in (?2) and id not in (?3)");
try (Transaction transaction = DB.beginTransaction()) {
transaction.setBatchMode(true);
upd.setParams(asList(9991), asList("Foo"), asList("Bar"));
upd.execute();
upd.setParams(asList(9991, 9992), asList("Foo", "Bar"), asList(1,2,3,4));
upd.execute();
upd.setParams(asList(9991), asList("Foo", "Bar", "Baz"), asList(1,2));
upd.execute();
transaction.commit();
}
}
@Test
public void positionParamsExpansion_withPost() {
SqlUpdate sqlUpdate = Ebean.createSqlUpdate("delete from o_customer where id in (?1) and name = ?");
sqlUpdate.setParameter(1, asList(9991, 9992, 9993));
sqlUpdate.setParameter(2, "Foo");
sqlUpdate.execute();
assertEquals("delete from o_customer where id in (?,?,?) and name = ?", sqlUpdate.getGeneratedSql());
}
@Test
public void positionParamsExpansion_withPost_andBatch_execute() {
SqlUpdate upd = DB.sqlUpdate("delete from o_customer where id in (?1) and name = ?");
try (Transaction transaction = DB.beginTransaction()) {
transaction.setBatchMode(true);
upd.setParams(asList(9991, 9992, 9993), "Foo");
upd.execute();
upd.setParams(asList(9991, 9992), "Bar");
upd.execute();
upd.setParams(asList(9991, 9992, 9999), "Baz");
upd.execute();
transaction.commit();
}
}
@Test
public void positionParamsExpansion_withPost_addBatch_execute() {
SqlUpdate upd = DB.sqlUpdate("delete from o_customer where id in (?1) and name = ?");
try (Transaction transaction = DB.beginTransaction()) {
transaction.setBatchMode(true);
upd.setParams(asList(9991, 9992, 9993), "Foo");
upd.addBatch();
upd.setParams(asList(9991, 9992), "Bar");
upd.addBatch();
upd.setParams(asList(9991, 9992, 9999), "Baz");
upd.addBatch();
transaction.commit();
}
}
@Test
public void test_multipleLists_asPositioned() {
SqlUpdate sqlUpdate = DB.sqlUpdate("delete from o_customer where id in (?1) and name in (?2)");
try (Transaction transaction = DB.beginTransaction()) {
sqlUpdate.setParameter(1, asList(9991, 9992, 9993));
sqlUpdate.setParameter(2, asList("rob", "jim"));
sqlUpdate.execute();
assertEquals("delete from o_customer where id in (?,?,?) and name in (?,?)", sqlUpdate.getGeneratedSql());
sqlUpdate.setParams(asList(9991, 9992), asList("rob", "jim", "sd"));
sqlUpdate.execute();
assertEquals("delete from o_customer where id in (?,?) and name in (?,?,?)", sqlUpdate.getGeneratedSql());
sqlUpdate.setParameter(1, asList(9991, 9992));
sqlUpdate.setParameter(2, asList("rob", "jim"));
sqlUpdate.execute();
assertEquals("delete from o_customer where id in (?,?) and name in (?,?)", sqlUpdate.getGeneratedSql());
sqlUpdate.setParams(asList(9991), asList("rob", "jim"));
sqlUpdate.execute();
assertEquals("delete from o_customer where id in (?) and name in (?,?)", sqlUpdate.getGeneratedSql());
sqlUpdate.setParams(asList(9992), asList("ro3b", "j3im"));
sqlUpdate.execute();
assertEquals("delete from o_customer where id in (?) and name in (?,?)", sqlUpdate.getGeneratedSql());
sqlUpdate.setParameter(1, asList(9992, 4545));
sqlUpdate.setParameter(2, asList("ro3b"));
sqlUpdate.execute();
assertEquals("delete from o_customer where id in (?,?) and name in (?)", sqlUpdate.getGeneratedSql());
transaction.commit();
}
}
@Test
public void test_multipleLists_asPositioned_withBatch() {
SqlUpdate sqlUpdate = DB.sqlUpdate("delete from o_customer where id in (?1) and name in (?2)");
try (Transaction transaction = DB.beginTransaction()) {
transaction.setBatchMode(true);
sqlUpdate.setParameter(1, asList(9991, 9992, 9993));
sqlUpdate.setParameter(2, asList("rob", "jim"));
sqlUpdate.execute();
assertEquals("delete from o_customer where id in (?,?,?) and name in (?,?)", sqlUpdate.getGeneratedSql());
sqlUpdate.setParams(asList(9991, 9992), asList("rob", "jim", "sd"));
sqlUpdate.execute();
assertEquals("delete from o_customer where id in (?,?) and name in (?,?,?)", sqlUpdate.getGeneratedSql());
sqlUpdate.setParameter(1, asList(9991, 9992));
sqlUpdate.setParameter(2, asList("rob", "jim"));
sqlUpdate.execute();
assertEquals("delete from o_customer where id in (?,?) and name in (?,?)", sqlUpdate.getGeneratedSql());
sqlUpdate.setParams(asList(9991), asList("rob", "jim"));
sqlUpdate.execute();
assertEquals("delete from o_customer where id in (?) and name in (?,?)", sqlUpdate.getGeneratedSql());
sqlUpdate.setParams(asList(9992), asList("ro3b", "j3im"));
sqlUpdate.execute();
assertEquals("delete from o_customer where id in (?) and name in (?,?)", sqlUpdate.getGeneratedSql());
sqlUpdate.setParameter(1, asList(9992, 4545));
sqlUpdate.setParameter(2, asList("ro3b"));
sqlUpdate.execute();
assertEquals("delete from o_customer where id in (?,?) and name in (?)", sqlUpdate.getGeneratedSql());
transaction.commit();
}
}
@Test
public void test_multipleLists() {