浏览代码

添加 setTypeMapping

James 4 年之前
父节点
当前提交
19018a9c4a
共有 1 个文件被更改,包括 8 次插入0 次删除
  1. 8 0
      src/main/java/com/jfinal/plugin/activerecord/generator/TypeMapping.java

+ 8 - 0
src/main/java/com/jfinal/plugin/activerecord/generator/TypeMapping.java

@@ -87,6 +87,14 @@ public class TypeMapping {
 		put("java.time.LocalTime", "java.time.LocalTime");
 	}};
 	
+	public void setMapping(Class<?> from, Class<?> to) {
+		map.put(from.getName(), to.getName());
+	}
+	
+	public void setMapping(String from, String to) {
+		map.put(from, to);
+	}
+	
 	public String getType(String typeString) {
 		return map.get(typeString);
 	}