diff --git a/src/main/java/com/alibaba/druid/mapping/Entity.java b/src/main/java/com/alibaba/druid/mapping/Entity.java index 2274170204..6ba44d6fef 100644 --- a/src/main/java/com/alibaba/druid/mapping/Entity.java +++ b/src/main/java/com/alibaba/druid/mapping/Entity.java @@ -3,7 +3,7 @@ package com.alibaba.druid.mapping; import java.util.LinkedHashMap; import java.util.Map; -public class Entity { +public class Entity extends MappingObject { private String name; private String description; diff --git a/src/main/java/com/alibaba/druid/mapping/MappingObject.java b/src/main/java/com/alibaba/druid/mapping/MappingObject.java new file mode 100644 index 0000000000..479bb0c25f --- /dev/null +++ b/src/main/java/com/alibaba/druid/mapping/MappingObject.java @@ -0,0 +1,37 @@ +package com.alibaba.druid.mapping; + +import java.util.HashMap; +import java.util.Map; + + +public class MappingObject { + private Map attributes; + + public Map getAttributes() { + if (attributes == null) { + attributes = new HashMap(1); + } + + return attributes; + } + + public Object getAttribute(String name) { + if (attributes == null) { + return null; + } + + return attributes.get(name); + } + + public void putAttribute(String name, Object value) { + if (attributes == null) { + attributes = new HashMap(1); + } + + attributes.put(name, value); + } + + public Map getAttributesDirect() { + return attributes; + } +} diff --git a/src/main/java/com/alibaba/druid/mapping/Property.java b/src/main/java/com/alibaba/druid/mapping/Property.java index 7b52e10c78..88c08895d0 100644 --- a/src/main/java/com/alibaba/druid/mapping/Property.java +++ b/src/main/java/com/alibaba/druid/mapping/Property.java @@ -1,6 +1,6 @@ package com.alibaba.druid.mapping; -public class Property { +public class Property extends MappingObject { private String name; private String desciption; private String dbColumnName;