提交 e20c1973 编写于 作者: zlt2000's avatar zlt2000

修复代码生成器可能出现类名错误的问题

上级 b484f603
......@@ -172,7 +172,7 @@ public class GenUtils {
*/
public static String tableToJava(String tableName, String tablePrefix) {
if (StringUtils.isNotBlank(tablePrefix)) {
tableName = tableName.replace(tablePrefix, "");
tableName = tableName.substring(tablePrefix.length());
}
return columnToJava(tableName);
}
......
package com.central.generator.utils;
import org.assertj.core.api.Assertions;
import org.junit.Test;
import org.springframework.boot.test.context.SpringBootTest;
/**
* GenUtils测试用例
*
* @author zlt
* @date 2019/5/10
*/
@SpringBootTest
public class GenUtilsTest {
@Test
public void testTableToJava() {
String javaName = GenUtils.tableToJava("t_event_message", "t_");
Assertions.assertThat(javaName).isEqualTo("EventMessage");
}
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册