Skip to content

M
mybatis-generator-plugin
Commit ba776224 authored by hewei's avatar hewei
Browse files
Options

LogicalDeletePlugin 测试用例

parent aaad0c6d
Hide whitespace changes
Inline Side-by-side
Showing with 405 additions and 388 deletions
src/test/java/com/itfsw/mybatis/generator/plugins/BatchInsertPluginTest.java
View file @ ba776224
...@@ -100,43 +100,38 @@ public class BatchInsertPluginTest { ...@@ -100,43 +100,38 @@ public class BatchInsertPluginTest {
MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/BatchInsertPlugin/mybatis-generator.xml"); MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/BatchInsertPlugin/mybatis-generator.xml");
tool.generate(new AbstractShellCallback() { tool.generate(new AbstractShellCallback() {
@Override @Override
public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) { public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) throws Exception {
try { // 1. 普通Mapper参数中List泛型为普通Model
// 1. 普通Mapper参数中List泛型为普通Model Class clsTbMapper = loader.loadClass(packagz + ".TbMapper");
Class clsTbMapper = loader.loadClass(packagz + ".TbMapper"); int count = 0;
int count = 0; for (Method method : clsTbMapper.getDeclaredMethods()) {
for (Method method : clsTbMapper.getDeclaredMethods()) { if (method.getName().equals("batchInsert")) {
if (method.getName().equals("batchInsert")) { Assert.assertEquals(Util.getListActualType(method.getGenericParameterTypes()[0]), packagz + ".Tb");
Assert.assertEquals(Util.getListActualType(method.getGenericParameterTypes()[0]), packagz + ".Tb"); count++;
count++;
}
if (method.getName().equals("batchInsertSelective")) {
Assert.assertEquals(Util.getListActualType(method.getGenericParameterTypes()[0]), packagz + ".Tb");
Assert.assertEquals(method.getGenericParameterTypes()[1].getTypeName(), packagz + ".Tb$Column[]");
count++;
}
} }
Assert.assertEquals(count, 2); if (method.getName().equals("batchInsertSelective")) {
Assert.assertEquals(Util.getListActualType(method.getGenericParameterTypes()[0]), packagz + ".Tb");
Assert.assertEquals(method.getGenericParameterTypes()[1].getTypeName(), packagz + ".Tb$Column[]");
count++;
}
}
Assert.assertEquals(count, 2);
// 2. 带有WithBlobs // 2. 带有WithBlobs
Class clsTbBlobsMapper = loader.loadClass(packagz + ".TbBlobsMapper"); Class clsTbBlobsMapper = loader.loadClass(packagz + ".TbBlobsMapper");
count = 0; count = 0;
for (Method method : clsTbBlobsMapper.getDeclaredMethods()) { for (Method method : clsTbBlobsMapper.getDeclaredMethods()) {
if (method.getName().equals("batchInsert")) { if (method.getName().equals("batchInsert")) {
Assert.assertEquals(Util.getListActualType(method.getGenericParameterTypes()[0]), packagz + ".TbBlobsWithBLOBs"); Assert.assertEquals(Util.getListActualType(method.getGenericParameterTypes()[0]), packagz + ".TbBlobsWithBLOBs");
count++; count++;
} }
if (method.getName().equals("batchInsertSelective")) { if (method.getName().equals("batchInsertSelective")) {
Assert.assertEquals(Util.getListActualType(method.getGenericParameterTypes()[0]), packagz + ".TbBlobsWithBLOBs"); Assert.assertEquals(Util.getListActualType(method.getGenericParameterTypes()[0]), packagz + ".TbBlobsWithBLOBs");
Assert.assertEquals(method.getGenericParameterTypes()[1].getTypeName(), packagz + ".TbBlobsWithBLOBs$Column[]"); Assert.assertEquals(method.getGenericParameterTypes()[1].getTypeName(), packagz + ".TbBlobsWithBLOBs$Column[]");
count++; count++;
}
} }
Assert.assertEquals(count, 2);
} catch (Exception e) {
e.printStackTrace();
Assert.assertTrue(false);
} }
Assert.assertEquals(count, 2);
} }
}); });
} }
...@@ -149,31 +144,26 @@ public class BatchInsertPluginTest { ...@@ -149,31 +144,26 @@ public class BatchInsertPluginTest {
MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/BatchInsertPlugin/mybatis-generator.xml"); MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/BatchInsertPlugin/mybatis-generator.xml");
tool.generate(new AbstractShellCallback() { tool.generate(new AbstractShellCallback() {
@Override @Override
public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) { public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) throws Exception {
try { // 1. 测试sql
// 1. 测试sql ObjectUtil tbMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbMapper")));
ObjectUtil tbMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbMapper"))); List<Object> params = new ArrayList<>();
List<Object> params = new ArrayList<>(); params.add(
params.add( new ObjectUtil(loader, packagz + ".Tb")
new ObjectUtil(loader, packagz + ".Tb") .set("field1", "test")
.set("field1", "test") .getObject()
.getObject() );
); params.add(
params.add( new ObjectUtil(loader, packagz + ".Tb")
new ObjectUtil(loader, packagz + ".Tb") .set("field1", "test")
.set("field1", "test") .set("field2", 1)
.set("field2", 1) .getObject()
.getObject() );
); String sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "batchInsert", params);
String sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "batchInsert", params); Assert.assertEquals(sql, "insert into tb (field1, field2) values ('test', null) , ('test', 1)");
Assert.assertEquals(sql, "insert into tb (field1, field2) values ('test', null) , ('test', 1)"); // 2. 执行sql
// 2. 执行sql Object count = tbMapper.invoke("batchInsert", params);
Object count = tbMapper.invoke("batchInsert", params); Assert.assertEquals(count, 2);
Assert.assertEquals(count, 2);
} catch (Exception e) {
e.printStackTrace();
Assert.assertTrue(false);
}
} }
}); });
} }
...@@ -186,38 +176,31 @@ public class BatchInsertPluginTest { ...@@ -186,38 +176,31 @@ public class BatchInsertPluginTest {
MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/BatchInsertPlugin/mybatis-generator.xml"); MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/BatchInsertPlugin/mybatis-generator.xml");
tool.generate(new AbstractShellCallback() { tool.generate(new AbstractShellCallback() {
@Override @Override
public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) { public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) throws Exception {
try { // 1. 测试sql
// 1. 测试sql ObjectUtil tbBlobsMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbBlobsMapper")));
ObjectUtil tbBlobsMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbBlobsMapper"))); List<Object> params = new ArrayList<>();
List<Object> params = new ArrayList<>(); params.add(
params.add( new ObjectUtil(loader, packagz + ".TbBlobsWithBLOBs")
new ObjectUtil(loader, packagz + ".TbBlobsWithBLOBs") .set("field1", "test")
.set("field1", "test") .getObject()
.getObject() );
); params.add(
params.add( new ObjectUtil(loader, packagz + ".TbBlobsWithBLOBs")
new ObjectUtil(loader, packagz + ".TbBlobsWithBLOBs") .set("field1", "test")
.set("field1", "test") .set("field2", "test123")
.set("field2", "test123") .getObject()
.getObject() );
); ObjectUtil columnField2 = new ObjectUtil(loader, packagz + ".TbBlobsWithBLOBs$Column#field2");
ObjectUtil columnField2 = new ObjectUtil(loader, packagz + ".TbBlobsWithBLOBs$Column#field2"); // java 动态参数不能有两个会冲突，最后一个封装成Array!!!必须使用反射创建指定类型数组，不然调用invoke对了可变参数会检查类型！
// java 动态参数不能有两个会冲突，最后一个封装成Array!!!必须使用反射创建指定类型数组，不然调用invoke对了可变参数会检查类型！ Object columns = Array.newInstance(columnField2.getCls(), 1);
Object columns = Array.newInstance(columnField2.getCls(), 1); Array.set(columns, 0, columnField2.getObject());
Array.set(columns, 0, columnField2.getObject());
String sql = SqlHelper.getFormatMapperSql(tbBlobsMapper.getObject(), "batchInsertSelective", params, columns); String sql = SqlHelper.getFormatMapperSql(tbBlobsMapper.getObject(), "batchInsertSelective", params, columns);
Assert.assertEquals(sql, "insert into tb_blobs ( field2 ) values ( 'null' ) , ( 'test123' )"); Assert.assertEquals(sql, "insert into tb_blobs ( field2 ) values ( 'null' ) , ( 'test123' )");
// 2. 执行sql // 2. 执行sql
Object count = tbBlobsMapper.invoke("batchInsertSelective", params, columns); Object count = tbBlobsMapper.invoke("batchInsertSelective", params, columns);
Assert.assertEquals(count, 2); Assert.assertEquals(count, 2);
} catch (Exception e) {
e.printStackTrace();
Assert.assertTrue(false);
} finally {
sqlSession.close();
}
} }
}); });
} }
......
src/test/java/com/itfsw/mybatis/generator/plugins/ExampleEnhancedPluginTest.java
View file @ ba776224
...@@ -56,20 +56,15 @@ public class ExampleEnhancedPluginTest { ...@@ -56,20 +56,15 @@ public class ExampleEnhancedPluginTest {
MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/ExampleEnhancedPlugin/mybatis-generator.xml"); MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/ExampleEnhancedPlugin/mybatis-generator.xml");
tool.generate(new AbstractShellCallback() { tool.generate(new AbstractShellCallback() {
@Override @Override
public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) { public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) throws Exception {
try { ObjectUtil tbMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbMapper")));
ObjectUtil tbMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbMapper")));
ObjectUtil tbExample = new ObjectUtil(loader, packagz + ".TbExample"); ObjectUtil tbExample = new ObjectUtil(loader, packagz + ".TbExample");
ObjectUtil tbExampleCriteria = new ObjectUtil(tbExample.invoke("createCriteria")); ObjectUtil tbExampleCriteria = new ObjectUtil(tbExample.invoke("createCriteria"));
// 调用example方法能正常返回 // 调用example方法能正常返回
String sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectByExample", tbExampleCriteria.invoke("example")); String sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectByExample", tbExampleCriteria.invoke("example"));
Assert.assertEquals(sql, "select id, field1 from tb"); Assert.assertEquals(sql, "select id, field1 from tb");
} catch (Exception e) {
e.printStackTrace();
Assert.assertTrue(false);
}
} }
}); });
} }
...@@ -82,22 +77,17 @@ public class ExampleEnhancedPluginTest { ...@@ -82,22 +77,17 @@ public class ExampleEnhancedPluginTest {
MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/ExampleEnhancedPlugin/mybatis-generator.xml"); MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/ExampleEnhancedPlugin/mybatis-generator.xml");
tool.generate(new AbstractShellCallback() { tool.generate(new AbstractShellCallback() {
@Override @Override
public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) { public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) throws Exception {
try { ObjectUtil tbMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbMapper")));
ObjectUtil tbMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbMapper")));
ObjectUtil tbExample = new ObjectUtil(loader, packagz + ".TbExample"); ObjectUtil tbExample = new ObjectUtil(loader, packagz + ".TbExample");
Object order = Array.newInstance(String.class, 2); Object order = Array.newInstance(String.class, 2);
Array.set(order, 0, "id desc"); Array.set(order, 0, "id desc");
Array.set(order, 1, "field1 asc"); Array.set(order, 1, "field1 asc");
tbExample.invoke("orderBy", order); // 可变参数方法直接设置order by tbExample.invoke("orderBy", order); // 可变参数方法直接设置order by
String sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectByExample", tbExample.getObject()); String sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectByExample", tbExample.getObject());
Assert.assertEquals(sql, "select id, field1 from tb order by id desc , field1 asc"); Assert.assertEquals(sql, "select id, field1 from tb order by id desc , field1 asc");
} catch (Exception e) {
e.printStackTrace();
Assert.assertTrue(false);
}
} }
}); });
} }
...@@ -151,7 +141,7 @@ public class ExampleEnhancedPluginTest { ...@@ -151,7 +141,7 @@ public class ExampleEnhancedPluginTest {
private class TestInvocationHandler implements InvocationHandler { private class TestInvocationHandler implements InvocationHandler {
@Override @Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable { public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
if (method.getName().equals("add")){ if (method.getName().equals("add")) {
ObjectUtil tbExampleCriteria = new ObjectUtil(args[0]); ObjectUtil tbExampleCriteria = new ObjectUtil(args[0]);
tbExampleCriteria.invoke("andIdEqualTo", 5l); tbExampleCriteria.invoke("andIdEqualTo", 5l);
return tbExampleCriteria.getObject(); return tbExampleCriteria.getObject();
......
src/test/java/com/itfsw/mybatis/generator/plugins/IncrementsPluginTest.java
View file @ ba776224
...@@ -64,31 +64,26 @@ public class IncrementsPluginTest { ...@@ -64,31 +64,26 @@ public class IncrementsPluginTest {
MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/IncrementsPlugin/mybatis-generator.xml"); MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/IncrementsPlugin/mybatis-generator.xml");
tool.generate(new AbstractShellCallback() { tool.generate(new AbstractShellCallback() {
@Override @Override
public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) { public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) throws Exception {
try { // 1. 测试生成的方法
// 1. 测试生成的方法 ObjectUtil tbBuilder = new ObjectUtil(loader, packagz + ".Tb$Builder");
ObjectUtil tbBuilder = new ObjectUtil(loader, packagz + ".Tb$Builder"); List<Method> methods = tbBuilder.getMethods("incF1");
List<Method> methods = tbBuilder.getMethods("incF1"); Assert.assertEquals(methods.size(), 2);
Assert.assertEquals(methods.size(), 2); // 自增方法
// 自增方法 Method incMethod = methods.get(0).getParameterTypes().length == 1 ? methods.get(1) : methods.get(0);
Method incMethod = methods.get(0).getParameterTypes().length == 1 ? methods.get(1) : methods.get(0); Assert.assertEquals(incMethod.getParameters()[1].getParameterizedType().getTypeName(), packagz + ".Tb$Builder$Inc");
Assert.assertEquals(incMethod.getParameters()[1].getParameterizedType().getTypeName(), packagz + ".Tb$Builder$Inc");
// 2. 测试有空格
// 2. 测试有空格 ObjectUtil tbKeysBuilder = new ObjectUtil(loader, packagz + ".TbKeys$Builder");
ObjectUtil tbKeysBuilder = new ObjectUtil(loader, packagz + ".TbKeys$Builder"); Assert.assertEquals(tbKeysBuilder.getMethods("incF1").size(), 2);
Assert.assertEquals(tbKeysBuilder.getMethods("incF1").size(), 2); Assert.assertEquals(tbKeysBuilder.getMethods("incF2").size(), 2);
Assert.assertEquals(tbKeysBuilder.getMethods("incF2").size(), 2); Assert.assertEquals(tbKeysBuilder.getMethods("incF3").size(), 2);
Assert.assertEquals(tbKeysBuilder.getMethods("incF3").size(), 2);
// 3. 测试在WithBlobs正确生成
// 3. 测试在WithBlobs正确生成 ObjectUtil tbBlobsWithBLOBs = new ObjectUtil(loader, packagz + ".TbBlobsWithBLOBs$Builder");
ObjectUtil tbBlobsWithBLOBs = new ObjectUtil(loader, packagz + ".TbBlobsWithBLOBs$Builder"); Assert.assertEquals(tbBlobsWithBLOBs.getMethods("incF1").size(), 2);
Assert.assertEquals(tbBlobsWithBLOBs.getMethods("incF1").size(), 2); Assert.assertEquals(tbBlobsWithBLOBs.getMethods("incF2").size(), 1);
Assert.assertEquals(tbBlobsWithBLOBs.getMethods("incF2").size(), 1); Assert.assertEquals(tbBlobsWithBLOBs.getMethods("incF3").size(), 2);
Assert.assertEquals(tbBlobsWithBLOBs.getMethods("incF3").size(), 2);
} catch (Exception e) {
e.printStackTrace();
Assert.assertTrue(false);
}
} }
}); });
} }
...@@ -101,107 +96,102 @@ public class IncrementsPluginTest { ...@@ -101,107 +96,102 @@ public class IncrementsPluginTest {
MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/IncrementsPlugin/mybatis-generator.xml"); MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/IncrementsPlugin/mybatis-generator.xml");
tool.generate(new AbstractShellCallback() { tool.generate(new AbstractShellCallback() {
@Override @Override
public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) { public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) throws Exception {
try {
// 1. 测试updateByExample、updateByExampleSelective // 1. 测试updateByExample、updateByExampleSelective
ObjectUtil tbMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbMapper"))); ObjectUtil tbMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbMapper")));
ObjectUtil tbExample = new ObjectUtil(loader, packagz + ".TbExample"); ObjectUtil tbExample = new ObjectUtil(loader, packagz + ".TbExample");
ObjectUtil criteria = new ObjectUtil(tbExample.invoke("createCriteria")); ObjectUtil criteria = new ObjectUtil(tbExample.invoke("createCriteria"));
criteria.invoke("andIdEqualTo", 3l); criteria.invoke("andIdEqualTo", 3l);
ObjectUtil tbBuilder = new ObjectUtil(loader, packagz + ".Tb$Builder"); ObjectUtil tbBuilder = new ObjectUtil(loader, packagz + ".Tb$Builder");
ObjectUtil tbBuilderInc = new ObjectUtil(loader, packagz + ".Tb$Builder$Inc#INC"); ObjectUtil tbBuilderInc = new ObjectUtil(loader, packagz + ".Tb$Builder$Inc#INC");
tbBuilder.invoke("incF1", 100l, tbBuilderInc.getObject()); tbBuilder.invoke("incF1", 100l, tbBuilderInc.getObject());
// sql // sql
String sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "updateByExample", tbBuilder.invoke("build"), tbExample.getObject()); String sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "updateByExample", tbBuilder.invoke("build"), tbExample.getObject());
Assert.assertEquals(sql, "update tb set id = null, field1 = 'null', inc_f1 = inc_f1 + 100 , inc_f2 = null, inc_f3 = null WHERE ( id = '3' )"); Assert.assertEquals(sql, "update tb set id = null, field1 = 'null', inc_f1 = inc_f1 + 100 , inc_f2 = null, inc_f3 = null WHERE ( id = '3' )");
sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "updateByExampleSelective", tbBuilder.invoke("build"), tbExample.getObject()); sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "updateByExampleSelective", tbBuilder.invoke("build"), tbExample.getObject());
Assert.assertEquals(sql, "update tb SET inc_f1 = inc_f1 + 100 WHERE ( id = '3' )"); Assert.assertEquals(sql, "update tb SET inc_f1 = inc_f1 + 100 WHERE ( id = '3' )");
// 执行 // 执行
// inc_f1 增加100 // inc_f1 增加100
Object result = tbMapper.invoke("updateByExampleSelective", tbBuilder.invoke("build"), tbExample.getObject()); Object result = tbMapper.invoke("updateByExampleSelective", tbBuilder.invoke("build"), tbExample.getObject());
Assert.assertEquals(result, 1); Assert.assertEquals(result, 1);
PreparedStatement preparedStatement = sqlSession.getConnection().prepareStatement("select inc_f1 from tb where id = 3"); PreparedStatement preparedStatement = sqlSession.getConnection().prepareStatement("select inc_f1 from tb where id = 3");
preparedStatement.execute(); preparedStatement.execute();
ResultSet rs = preparedStatement.getResultSet(); ResultSet rs = preparedStatement.getResultSet();
rs.first(); rs.first();
Assert.assertEquals(rs.getInt("inc_f1"), 103); Assert.assertEquals(rs.getInt("inc_f1"), 103);
// inc_f1 再减去50 // inc_f1 再减去50
ObjectUtil tbBuilderDec = new ObjectUtil(loader, packagz + ".Tb$Builder$Inc#DEC"); ObjectUtil tbBuilderDec = new ObjectUtil(loader, packagz + ".Tb$Builder$Inc#DEC");
tbBuilder.invoke("incF1", 50l, tbBuilderDec.getObject()); tbBuilder.invoke("incF1", 50l, tbBuilderDec.getObject());
result = tbMapper.invoke("updateByExampleSelective", tbBuilder.invoke("build"), tbExample.getObject()); result = tbMapper.invoke("updateByExampleSelective", tbBuilder.invoke("build"), tbExample.getObject());
Assert.assertEquals(result, 1); Assert.assertEquals(result, 1);
// 验证执行结果 // 验证执行结果
preparedStatement = sqlSession.getConnection().prepareStatement("select inc_f1 from tb where id = 3"); preparedStatement = sqlSession.getConnection().prepareStatement("select inc_f1 from tb where id = 3");
preparedStatement.execute(); preparedStatement.execute();
rs = preparedStatement.getResultSet(); rs = preparedStatement.getResultSet();
rs.first(); rs.first();
Assert.assertEquals(rs.getInt("inc_f1"), 53); Assert.assertEquals(rs.getInt("inc_f1"), 53);
// 2. 测试updateByPrimaryKey、updateByPrimaryKeySelective // 2. 测试updateByPrimaryKey、updateByPrimaryKeySelective
ObjectUtil tbKeysMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbKeysMapper"))); ObjectUtil tbKeysMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbKeysMapper")));
ObjectUtil tbKeysBuilderInc = new ObjectUtil(loader, packagz + ".TbKeys$Builder$Inc#INC"); ObjectUtil tbKeysBuilderInc = new ObjectUtil(loader, packagz + ".TbKeys$Builder$Inc#INC");
ObjectUtil tbKeysBuilder = new ObjectUtil(loader, packagz + ".TbKeys$Builder"); ObjectUtil tbKeysBuilder = new ObjectUtil(loader, packagz + ".TbKeys$Builder");
tbKeysBuilder.invoke("key1", 1l); tbKeysBuilder.invoke("key1", 1l);
tbKeysBuilder.invoke("key2", "k1"); tbKeysBuilder.invoke("key2", "k1");
tbKeysBuilder.invoke("incF1", 10l, tbKeysBuilderInc.getObject()); tbKeysBuilder.invoke("incF1", 10l, tbKeysBuilderInc.getObject());
tbKeysBuilder.invoke("incF3", 30l, tbKeysBuilderInc.getObject()); tbKeysBuilder.invoke("incF3", 30l, tbKeysBuilderInc.getObject());
// sql // sql
sql = SqlHelper.getFormatMapperSql(tbKeysMapper.getObject(), "updateByPrimaryKey", tbKeysBuilder.invoke("build")); sql = SqlHelper.getFormatMapperSql(tbKeysMapper.getObject(), "updateByPrimaryKey", tbKeysBuilder.invoke("build"));
Assert.assertEquals(sql, "update tb_keys set field1 = 'null', field2 = null, inc_f1 = inc_f1 + 10 , inc_f2 = null , inc_f3 = inc_f3 + 30 where key1 = 1 and key2 = 'k1'"); Assert.assertEquals(sql, "update tb_keys set field1 = 'null', field2 = null, inc_f1 = inc_f1 + 10 , inc_f2 = null , inc_f3 = inc_f3 + 30 where key1 = 1 and key2 = 'k1'");
sql = SqlHelper.getFormatMapperSql(tbKeysMapper.getObject(), "updateByPrimaryKeySelective", tbKeysBuilder.invoke("build")); sql = SqlHelper.getFormatMapperSql(tbKeysMapper.getObject(), "updateByPrimaryKeySelective", tbKeysBuilder.invoke("build"));
Assert.assertEquals(sql, "update tb_keys SET inc_f1 = inc_f1 + 10 , inc_f3 = inc_f3 + 30 where key1 = 1 and key2 = 'k1'"); Assert.assertEquals(sql, "update tb_keys SET inc_f1 = inc_f1 + 10 , inc_f3 = inc_f3 + 30 where key1 = 1 and key2 = 'k1'");
// 执行 // 执行
result = tbKeysMapper.invoke("updateByPrimaryKeySelective", tbKeysBuilder.invoke("build")); result = tbKeysMapper.invoke("updateByPrimaryKeySelective", tbKeysBuilder.invoke("build"));
Assert.assertEquals(result, 1); Assert.assertEquals(result, 1);
// 验证执行结果 // 验证执行结果
preparedStatement = sqlSession.getConnection().prepareStatement("select inc_f1, inc_f3 from tb_keys where key1 = 1 and key2 = 'k1'"); preparedStatement = sqlSession.getConnection().prepareStatement("select inc_f1, inc_f3 from tb_keys where key1 = 1 and key2 = 'k1'");
preparedStatement.execute(); preparedStatement.execute();
rs = preparedStatement.getResultSet(); rs = preparedStatement.getResultSet();
rs.first(); rs.first();
Assert.assertEquals(rs.getInt("inc_f1"), 11); Assert.assertEquals(rs.getInt("inc_f1"), 11);
Assert.assertEquals(rs.getInt("inc_f3"), 33); Assert.assertEquals(rs.getInt("inc_f3"), 33);
// 3. 测试updateByExampleWithBLOBs、updateByPrimaryKeyWithBLOBs // 3. 测试updateByExampleWithBLOBs、updateByPrimaryKeyWithBLOBs
ObjectUtil tbBlobsMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbBlobsMapper"))); ObjectUtil tbBlobsMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbBlobsMapper")));
ObjectUtil tbBlobsExample = new ObjectUtil(loader, packagz + ".TbBlobsExample"); ObjectUtil tbBlobsExample = new ObjectUtil(loader, packagz + ".TbBlobsExample");
ObjectUtil tbBlobsExampleCriteria = new ObjectUtil(tbBlobsExample.invoke("createCriteria")); ObjectUtil tbBlobsExampleCriteria = new ObjectUtil(tbBlobsExample.invoke("createCriteria"));
tbBlobsExampleCriteria.invoke("andIdEqualTo", 3l); tbBlobsExampleCriteria.invoke("andIdEqualTo", 3l);
ObjectUtil tbBlobsWithBLOBsBuilder = new ObjectUtil(loader, packagz + ".TbBlobsWithBLOBs$Builder"); ObjectUtil tbBlobsWithBLOBsBuilder = new ObjectUtil(loader, packagz + ".TbBlobsWithBLOBs$Builder");
ObjectUtil tbBlobsBuilderInc = new ObjectUtil(loader, packagz + ".TbBlobs$Builder$Inc#INC"); ObjectUtil tbBlobsBuilderInc = new ObjectUtil(loader, packagz + ".TbBlobs$Builder$Inc#INC");
tbBlobsWithBLOBsBuilder.invoke("incF1", 100l, tbBlobsBuilderInc.getObject()); tbBlobsWithBLOBsBuilder.invoke("incF1", 100l, tbBlobsBuilderInc.getObject());
tbBlobsWithBLOBsBuilder.invoke("incF2", 50l); tbBlobsWithBLOBsBuilder.invoke("incF2", 50l);
tbBlobsWithBLOBsBuilder.invoke("field3", "blob"); tbBlobsWithBLOBsBuilder.invoke("field3", "blob");
// sql // sql
sql = SqlHelper.getFormatMapperSql(tbBlobsMapper.getObject(), "updateByExampleWithBLOBs", tbBlobsWithBLOBsBuilder.invoke("build"), tbBlobsExample.getObject()); sql = SqlHelper.getFormatMapperSql(tbBlobsMapper.getObject(), "updateByExampleWithBLOBs", tbBlobsWithBLOBsBuilder.invoke("build"), tbBlobsExample.getObject());
Assert.assertEquals(sql, "update tb_blobs set id = null, field1 = 'null', inc_f1 = inc_f1 + 100 , inc_f2 = 50, inc_f3 = null , field2 = 'null', field3 = 'blob' WHERE ( id = '3' )"); Assert.assertEquals(sql, "update tb_blobs set id = null, field1 = 'null', inc_f1 = inc_f1 + 100 , inc_f2 = 50, inc_f3 = null , field2 = 'null', field3 = 'blob' WHERE ( id = '3' )");
tbBlobsWithBLOBsBuilder.invoke("id", 3l); tbBlobsWithBLOBsBuilder.invoke("id", 3l);
sql = SqlHelper.getFormatMapperSql(tbBlobsMapper.getObject(), "updateByPrimaryKeyWithBLOBs", tbBlobsWithBLOBsBuilder.invoke("build")); sql = SqlHelper.getFormatMapperSql(tbBlobsMapper.getObject(), "updateByPrimaryKeyWithBLOBs", tbBlobsWithBLOBsBuilder.invoke("build"));
Assert.assertEquals(sql, "update tb_blobs set field1 = 'null', inc_f1 = inc_f1 + 100 , inc_f2 = 50, inc_f3 = null , field2 = 'null', field3 = 'blob' where id = 3"); Assert.assertEquals(sql, "update tb_blobs set field1 = 'null', inc_f1 = inc_f1 + 100 , inc_f2 = 50, inc_f3 = null , field2 = 'null', field3 = 'blob' where id = 3");
// 执行 // 执行
tbBlobsWithBLOBsBuilder.invoke("incF3", 10l); tbBlobsWithBLOBsBuilder.invoke("incF3", 10l);
// 测试自增字段没有配置自增参数 // 测试自增字段没有配置自增参数
sql = SqlHelper.getFormatMapperSql(tbBlobsMapper.getObject(), "updateByPrimaryKeyWithBLOBs", tbBlobsWithBLOBsBuilder.invoke("build")); sql = SqlHelper.getFormatMapperSql(tbBlobsMapper.getObject(), "updateByPrimaryKeyWithBLOBs", tbBlobsWithBLOBsBuilder.invoke("build"));
Assert.assertEquals(sql, "update tb_blobs set field1 = 'null', inc_f1 = inc_f1 + 100 , inc_f2 = 50, inc_f3 = 10 , field2 = 'null', field3 = 'blob' where id = 3"); Assert.assertEquals(sql, "update tb_blobs set field1 = 'null', inc_f1 = inc_f1 + 100 , inc_f2 = 50, inc_f3 = 10 , field2 = 'null', field3 = 'blob' where id = 3");
result = tbBlobsMapper.invoke("updateByPrimaryKeyWithBLOBs", tbBlobsWithBLOBsBuilder.invoke("build")); result = tbBlobsMapper.invoke("updateByPrimaryKeyWithBLOBs", tbBlobsWithBLOBsBuilder.invoke("build"));
Assert.assertEquals(result, 1); Assert.assertEquals(result, 1);
} catch (Exception e) {
e.printStackTrace();
Assert.assertTrue(false);
}
} }
}); });
} }
......
src/test/java/com/itfsw/mybatis/generator/plugins/LimitPluginTest.java
View file @ ba776224
...@@ -65,38 +65,33 @@ public class LimitPluginTest { ...@@ -65,38 +65,33 @@ public class LimitPluginTest {
MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/LimitPlugin/mybatis-generator.xml"); MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/LimitPlugin/mybatis-generator.xml");
tool.generate(new AbstractShellCallback() { tool.generate(new AbstractShellCallback() {
@Override @Override
public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) { public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) throws Exception {
try { // 1. 测试limit 方法
// 1. 测试limit 方法 ObjectUtil tbMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbMapper")));
ObjectUtil tbMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbMapper")));
ObjectUtil tbExample = new ObjectUtil(loader, packagz + ".TbExample"); ObjectUtil tbExample = new ObjectUtil(loader, packagz + ".TbExample");
tbExample.invoke("limit", 5); tbExample.invoke("limit", 5);
// 调用limit(5)方法 // 调用limit(5)方法
String sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectByExample", tbExample.getObject()); String sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectByExample", tbExample.getObject());
Assert.assertEquals(sql, "select id, field1 from tb limit 5"); Assert.assertEquals(sql, "select id, field1 from tb limit 5");
// 调用limit(1, 5)方法 // 调用limit(1, 5)方法
tbExample.invoke("limit", 1, 5); tbExample.invoke("limit", 1, 5);
sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectByExample", tbExample.getObject()); sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectByExample", tbExample.getObject());
Assert.assertEquals(sql, "select id, field1 from tb limit 1, 5"); Assert.assertEquals(sql, "select id, field1 from tb limit 1, 5");
// 执行一次看结果 // 执行一次看结果
List list = (List) tbMapper.invoke("selectByExample", tbExample.getObject()); List list = (List) tbMapper.invoke("selectByExample", tbExample.getObject());
Assert.assertEquals(list.size(), 5); Assert.assertEquals(list.size(), 5);
Assert.assertEquals(new ObjectUtil(list.get(0)).get("id"), 2l); Assert.assertEquals(new ObjectUtil(list.get(0)).get("id"), 2l);
// 2. 测试page 方法 // 2. 测试page 方法
tbExample.invoke("page", 2, 3); tbExample.invoke("page", 2, 3);
sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectByExample", tbExample.getObject()); sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectByExample", tbExample.getObject());
Assert.assertEquals(sql, "select id, field1 from tb limit 6, 3"); Assert.assertEquals(sql, "select id, field1 from tb limit 6, 3");
// 执行 // 执行
list = (List) tbMapper.invoke("selectByExample", tbExample.getObject()); list = (List) tbMapper.invoke("selectByExample", tbExample.getObject());
Assert.assertEquals(list.size(), 3); Assert.assertEquals(list.size(), 3);
Assert.assertEquals(new ObjectUtil(list.get(0)).get("id"), 7l); Assert.assertEquals(new ObjectUtil(list.get(0)).get("id"), 7l);
} catch (Exception e) {
e.printStackTrace();
Assert.assertTrue(false);
}
} }
}); });
} }
......
src/test/java/com/itfsw/mybatis/generator/plugins/LogicalDeletePluginTest.java
View file @ ba776224
...@@ -16,8 +16,8 @@ ...@@ -16,8 +16,8 @@
package com.itfsw.mybatis.generator.plugins; package com.itfsw.mybatis.generator.plugins;
import com.itfsw.mybatis.generator.plugins.tools.DBHelper; import com.itfsw.mybatis.generator.plugins.tools.*;
import com.itfsw.mybatis.generator.plugins.tools.MyBatisGeneratorTool; import org.apache.ibatis.session.SqlSession;
import org.junit.Assert; import org.junit.Assert;
import org.junit.BeforeClass; import org.junit.BeforeClass;
import org.junit.Test; import org.junit.Test;
...@@ -67,4 +67,25 @@ public class LogicalDeletePluginTest { ...@@ -67,4 +67,25 @@ public class LogicalDeletePluginTest {
Assert.assertEquals(tool.getWarnings().get(0), "itfsw(逻辑删除插件):tb没有找到您配置的逻辑删除值，请全局或者局部配置logicalDeleteValue和logicalUnDeleteValue值！"); Assert.assertEquals(tool.getWarnings().get(0), "itfsw(逻辑删除插件):tb没有找到您配置的逻辑删除值，请全局或者局部配置logicalDeleteValue和logicalUnDeleteValue值！");
} }
/**
* 测试 logicalDeleteByExample
*/
@Test
public void testLogicalDeleteByExample() throws IOException, XMLParserException, InvalidConfigurationException, InterruptedException, SQLException {
MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/LogicalDeletePlugin/mybatis-generator.xml");
tool.generate(new AbstractShellCallback() {
@Override
public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) throws Exception{
ObjectUtil tbMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbMapper")));
ObjectUtil tbExample = new ObjectUtil(loader, packagz + ".TbExample");
ObjectUtil criteria = new ObjectUtil(tbExample.invoke("createCriteria"));
criteria.invoke("andIdEqualTo", 1l);
String sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "logicalDeleteByExample", tbExample.getObject());
Assert.assertEquals(sql, "select field1 from tb WHERE ( id = '3' ) order by field2 asc limit 1");
}
});
}
} }
src/test/java/com/itfsw/mybatis/generator/plugins/SelectSelectivePluginTest.java
View file @ ba776224
...@@ -51,56 +51,51 @@ public class SelectSelectivePluginTest { ...@@ -51,56 +51,51 @@ public class SelectSelectivePluginTest {
MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/SelectSelectivePlugin/mybatis-generator.xml"); MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/SelectSelectivePlugin/mybatis-generator.xml");
tool.generate(new AbstractShellCallback() { tool.generate(new AbstractShellCallback() {
@Override @Override
public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) { public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) throws Exception {
try { // 1. 测试sql
// 1. 测试sql ObjectUtil tbMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbMapper")));
ObjectUtil tbMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbMapper")));
ObjectUtil tbExample = new ObjectUtil(loader, packagz + ".TbExample");
ObjectUtil tbExample = new ObjectUtil(loader, packagz + ".TbExample"); ObjectUtil criteria = new ObjectUtil(tbExample.invoke("createCriteria"));
ObjectUtil criteria = new ObjectUtil(tbExample.invoke("createCriteria")); criteria.invoke("andIdLessThan", 100l);
criteria.invoke("andIdLessThan", 100l); tbExample.set("orderByClause", "field2 asc");
tbExample.set("orderByClause", "field2 asc");
ObjectUtil columnField1 = new ObjectUtil(loader, packagz + ".Tb$Column#field1");
ObjectUtil columnField1 = new ObjectUtil(loader, packagz + ".Tb$Column#field1"); // java 动态参数不能有两个会冲突，最后一个封装成Array!!!必须使用反射创建指定类型数组，不然调用invoke对了可变参数会检查类型！
// java 动态参数不能有两个会冲突，最后一个封装成Array!!!必须使用反射创建指定类型数组，不然调用invoke对了可变参数会检查类型！ Object columns1 = Array.newInstance(columnField1.getCls(), 1);
Object columns1 = Array.newInstance(columnField1.getCls(), 1); Array.set(columns1, 0, columnField1.getObject());
Array.set(columns1, 0, columnField1.getObject());
String sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectByExampleSelective", tbExample.getObject(), columns1);
String sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectByExampleSelective", tbExample.getObject(), columns1); Assert.assertEquals(sql, "select field1 from tb WHERE ( id < '100' ) order by field2 asc");
Assert.assertEquals(sql, "select field1 from tb WHERE ( id < '100' ) order by field2 asc");
ObjectUtil columnField2 = new ObjectUtil(loader, packagz + ".Tb$Column#field2");
ObjectUtil columnField2 = new ObjectUtil(loader, packagz + ".Tb$Column#field2"); Object columns2 = Array.newInstance(columnField1.getCls(), 2);
Object columns2 = Array.newInstance(columnField1.getCls(), 2); Array.set(columns2, 0, columnField1.getObject());
Array.set(columns2, 0, columnField1.getObject()); Array.set(columns2, 1, columnField2.getObject());
Array.set(columns2, 1, columnField2.getObject());
sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectByExampleSelective", tbExample.getObject(), columns2);
sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectByExampleSelective", tbExample.getObject(), columns2); Assert.assertEquals(sql, "select field1 , field2 from tb WHERE ( id < '100' ) order by field2 asc");
Assert.assertEquals(sql, "select field1 , field2 from tb WHERE ( id < '100' ) order by field2 asc");
// 2. 执行sql
// 2. 执行sql List list = (List) tbMapper.invoke("selectByExampleSelective", tbExample.getObject(), columns1);
List list = (List) tbMapper.invoke("selectByExampleSelective", tbExample.getObject(), columns1); Assert.assertEquals(list.size(), 3);
Assert.assertEquals(list.size(), 3); int index = 0;
int index = 0; for (Object obj : list) {
for (Object obj : list) { ObjectUtil objectUtil = new ObjectUtil(obj);
ObjectUtil objectUtil = new ObjectUtil(obj); // 没有查询这两个字段
// 没有查询这两个字段 if (objectUtil.get("id") != null || objectUtil.get("field2") != null) {
if (objectUtil.get("id") != null || objectUtil.get("field2") != null) { Assert.assertTrue(false);
Assert.assertTrue(false);
}
if (index == 0) {
Assert.assertEquals(objectUtil.get("field1"), "fd1");
} else if (index == 1) {
Assert.assertNull(objectUtil.get("field1"));
} else {
Assert.assertEquals(objectUtil.get("field1"), "fd3");
}
index++;
} }
} catch (Exception e) { if (index == 0) {
e.printStackTrace(); Assert.assertEquals(objectUtil.get("field1"), "fd1");
Assert.assertTrue(false); } else if (index == 1) {
Assert.assertNull(objectUtil.get("field1"));
} else {
Assert.assertEquals(objectUtil.get("field1"), "fd3");
}
index++;
} }
} }
}); });
...@@ -115,43 +110,35 @@ public class SelectSelectivePluginTest { ...@@ -115,43 +110,35 @@ public class SelectSelectivePluginTest {
MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/SelectSelectivePlugin/mybatis-generator.xml"); MyBatisGeneratorTool tool = MyBatisGeneratorTool.create("scripts/SelectSelectivePlugin/mybatis-generator.xml");
tool.generate(new AbstractShellCallback() { tool.generate(new AbstractShellCallback() {
@Override @Override
public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) { public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) throws Exception {
try { // 1. 测试sql
// 1. 测试sql ObjectUtil tbMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbMapper")));
ObjectUtil tbMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbMapper")));
ObjectUtil columnField1 = new ObjectUtil(loader, packagz + ".TbKeys$Column#field1");
ObjectUtil columnField1 = new ObjectUtil(loader, packagz + ".TbKeys$Column#field1"); // java 动态参数不能有两个会冲突，最后一个封装成Array!!!必须使用反射创建指定类型数组，不然调用invoke对了可变参数会检查类型！
// java 动态参数不能有两个会冲突，最后一个封装成Array!!!必须使用反射创建指定类型数组，不然调用invoke对了可变参数会检查类型！ Object columns1 = Array.newInstance(columnField1.getCls(), 1);
Object columns1 = Array.newInstance(columnField1.getCls(), 1); Array.set(columns1, 0, columnField1.getObject());
Array.set(columns1, 0, columnField1.getObject());
String sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectByPrimaryKeySelective", 1, columns1);
String sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectByPrimaryKeySelective", 1, columns1); Assert.assertEquals(sql, "select field1 from tb where id = 1");
Assert.assertEquals(sql, "select field1 from tb where id = 1");
// 2. 测试xxxKey
// 2. 测试xxxKey ObjectUtil tbKeysMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbKeysMapper")));
ObjectUtil tbKeysMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbKeysMapper"))); ObjectUtil tbKeysKey = new ObjectUtil(loader, packagz + ".TbKeysKey");
ObjectUtil tbKeysKey = new ObjectUtil(loader, packagz + ".TbKeysKey"); tbKeysKey.set("key1", 1l);
tbKeysKey.set("key1", 1l); tbKeysKey.set("key2", "2");
tbKeysKey.set("key2", "2");
ObjectUtil columnField2 = new ObjectUtil(loader, packagz + ".TbKeys$Column#field2");
ObjectUtil columnField2 = new ObjectUtil(loader, packagz + ".TbKeys$Column#field2"); // java 动态参数不能有两个会冲突，最后一个封装成Array!!!必须使用反射创建指定类型数组，不然调用invoke对了可变参数会检查类型！
// java 动态参数不能有两个会冲突，最后一个封装成Array!!!必须使用反射创建指定类型数组，不然调用invoke对了可变参数会检查类型！ Object columns2 = Array.newInstance(columnField2.getCls(), 1);
Object columns2 = Array.newInstance(columnField2.getCls(), 1); Array.set(columns2, 0, columnField2.getObject());
Array.set(columns2, 0, columnField2.getObject());
sql = SqlHelper.getFormatMapperSql(tbKeysMapper.getObject(), "selectByPrimaryKeySelective", tbKeysKey.getObject(), columns2);
sql = SqlHelper.getFormatMapperSql(tbKeysMapper.getObject(), "selectByPrimaryKeySelective", tbKeysKey.getObject(), columns2); Assert.assertEquals(sql, "select field2 from tb_keys where key1 = 1 and key2 = '2'");
Assert.assertEquals(sql, "select field2 from tb_keys where key1 = 1 and key2 = '2'");
// 3. 执行sql
// 3. 执行sql Object tbKeys = tbKeysMapper.invoke("selectByPrimaryKeySelective", tbKeysKey.getObject(), columns1);
Object tbKeys = tbKeysMapper.invoke("selectByPrimaryKeySelective", tbKeysKey.getObject(), columns1); Assert.assertEquals(new ObjectUtil(tbKeys).get("field1"), "fd1");
Assert.assertEquals(new ObjectUtil(tbKeys).get("field1"), "fd1");
} catch (Exception e) {
e.printStackTrace();
Assert.assertTrue(false);
} finally {
sqlSession.close();
}
} }
}); });
} }
...@@ -176,34 +163,28 @@ public class SelectSelectivePluginTest { ...@@ -176,34 +163,28 @@ public class SelectSelectivePluginTest {
tool = MyBatisGeneratorTool.create("scripts/SelectSelectivePlugin/mybatis-generator-with-SelectOneByExamplePlugin.xml"); tool = MyBatisGeneratorTool.create("scripts/SelectSelectivePlugin/mybatis-generator-with-SelectOneByExamplePlugin.xml");
tool.generate(new AbstractShellCallback() { tool.generate(new AbstractShellCallback() {
@Override @Override
public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) { public void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) throws Exception {
try { // 1. 测试sql
// 1. 测试sql ObjectUtil tbMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbMapper")));
ObjectUtil tbMapper = new ObjectUtil(sqlSession.getMapper(loader.loadClass(packagz + ".TbMapper")));
ObjectUtil tbExample = new ObjectUtil(loader, packagz + ".TbExample");
ObjectUtil tbExample = new ObjectUtil(loader, packagz + ".TbExample"); ObjectUtil criteria = new ObjectUtil(tbExample.invoke("createCriteria"));
ObjectUtil criteria = new ObjectUtil(tbExample.invoke("createCriteria")); criteria.invoke("andIdEqualTo", 3l);
criteria.invoke("andIdEqualTo", 3l); tbExample.set("orderByClause", "field2 asc");
tbExample.set("orderByClause", "field2 asc");
ObjectUtil columnField1 = new ObjectUtil(loader, packagz + ".Tb$Column#field1");
ObjectUtil columnField1 = new ObjectUtil(loader, packagz + ".Tb$Column#field1"); // java 动态参数不能有两个会冲突，最后一个封装成Array!!!必须使用反射创建指定类型数组，不然调用invoke对了可变参数会检查类型！
// java 动态参数不能有两个会冲突，最后一个封装成Array!!!必须使用反射创建指定类型数组，不然调用invoke对了可变参数会检查类型！ Object columns1 = Array.newInstance(columnField1.getCls(), 1);
Object columns1 = Array.newInstance(columnField1.getCls(), 1); Array.set(columns1, 0, columnField1.getObject());
Array.set(columns1, 0, columnField1.getObject());
String sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectOneByExampleSelective", tbExample.getObject(), columns1);
String sql = SqlHelper.getFormatMapperSql(tbMapper.getObject(), "selectOneByExampleSelective", tbExample.getObject(), columns1); Assert.assertEquals(sql, "select field1 from tb WHERE ( id = '3' ) order by field2 asc limit 1");
Assert.assertEquals(sql, "select field1 from tb WHERE ( id = '3' ) order by field2 asc limit 1");
// 2. 执行sql
// 2. 执行sql Object result = tbMapper.invoke("selectOneByExampleSelective", tbExample.getObject(), columns1);
Object result = tbMapper.invoke("selectOneByExampleSelective", tbExample.getObject(), columns1); ObjectUtil tb = new ObjectUtil(result);
ObjectUtil tb = new ObjectUtil(result); Assert.assertEquals(tb.get("field1"), "fd3");
Assert.assertEquals(tb.get("field1"), "fd3"); Assert.assertNull(tb.get("field2"));
Assert.assertNull(tb.get("field2"));
} catch (Exception e) {
e.printStackTrace();
Assert.assertTrue(false);
}
} }
}); });
} }
......
src/test/java/com/itfsw/mybatis/generator/plugins/tools/AbstractShellCallback.java
View file @ ba776224
...@@ -71,8 +71,9 @@ public abstract class AbstractShellCallback implements ShellCallback { ...@@ -71,8 +71,9 @@ public abstract class AbstractShellCallback implements ShellCallback {
* @param sqlSession * @param sqlSession
* @param loader * @param loader
* @param packagz * @param packagz
* @throws Exception
*/ */
public abstract void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz); public abstract void reloadProject(SqlSession sqlSession, ClassLoader loader, String packagz) throws Exception;
/* (non-Javadoc) /* (non-Javadoc)
* @see org.mybatis.generator.api.ShellCallback#getDirectory(java.lang.String, java.lang.String) * @see org.mybatis.generator.api.ShellCallback#getDirectory(java.lang.String, java.lang.String)
......
src/test/resources/scripts/LogicalDeletePlugin/mybatis-generator.xml 0 → 100644
View file @ ba776224
<?xml version="1.0" encoding="UTF-8"?>
<!--
~ Copyright (c) 2017.
~
~ Licensed under the Apache License, Version 2.0 (the "License");
~ you may not use this file except in compliance with the License.
~ You may obtain a copy of the License at
~
~ http://www.apache.org/licenses/LICENSE-2.0
~
~ Unless required by applicable law or agreed to in writing, software
~ distributed under the License is distributed on an "AS IS" BASIS,
~ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
~ See the License for the specific language governing permissions and
~ limitations under the License.
-->
<!DOCTYPE generatorConfiguration
PUBLIC "-//mybatis.org//DTD MyBatis Generator Configuration 1.0//EN"
"http://mybatis.org/dtd/mybatis-generator-config_1_0.dtd">
<generatorConfiguration>
<properties resource="db.properties"/>
<!--导入属性配置 -->
<context id="default" targetRuntime="MyBatis3">
<!-- 插件 -->
<plugin type="com.itfsw.mybatis.generator.plugins.LogicalDeletePlugin">
<property name="logicalDeleteColumn" value="del_flag"/>
<property name="logicalDeleteValue" value="1"/>
<property name="logicalUnDeleteValue" value="0"/>
</plugin>
<!--jdbc的数据库连接 -->
<jdbcConnection driverClass="${driver}" connectionURL="${url}" userId="${username}" password="${password}" />
<!-- Model模型生成器,用来生成含有主键key的类，记录类 以及查询Example类
targetPackage 指定生成的model生成所在的包名
targetProject 指定在该项目下所在的路径 -->
<javaModelGenerator targetPackage="" targetProject="">
<!-- 是否对model添加 构造函数 -->
<property name="constructorBased" value="true"/>
<!-- 给Model添加一个父类 -->
<!--<property name="rootClass" value="com.itfsw.base"/>-->
</javaModelGenerator>
<!--Mapper映射文件生成所在的目录 为每一个数据库的表生成对应的SqlMap文件 -->
<sqlMapGenerator targetPackage="" targetProject="" />
<!-- 客户端代码，生成易于使用的针对Model对象和XML配置文件 的代码
type="ANNOTATEDMAPPER",生成Java Model 和基于注解的Mapper对象
type="MIXEDMAPPER",生成基于注解的Java Model 和相应的Mapper对象
type="XMLMAPPER",生成SQLMap XML文件和独立的Mapper接口 -->
<javaClientGenerator targetPackage="" targetProject="" type="XMLMAPPER"/>
<!-- +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 要自动生成的表 +++++++++++++++++++++++++++++++++++++++++++++++++++++++ -->
<table tableName="tb">
<generatedKey column="id" sqlStatement="MySql" identity="true"/>
</table>
</context>
</generatorConfiguration>
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment