MapWrapper.java

  1. /*
  2.  *    Copyright 2009-2021 the original author or authors.
  3.  *
  4.  *    Licensed under the Apache License, Version 2.0 (the "License");
  5.  *    you may not use this file except in compliance with the License.
  6.  *    You may obtain a copy of the License at
  7.  *
  8.  *       http://www.apache.org/licenses/LICENSE-2.0
  9.  *
  10.  *    Unless required by applicable law or agreed to in writing, software
  11.  *    distributed under the License is distributed on an "AS IS" BASIS,
  12.  *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13.  *    See the License for the specific language governing permissions and
  14.  *    limitations under the License.
  15.  */
  16. package org.apache.ibatis.reflection.wrapper;

  17. import java.util.HashMap;
  18. import java.util.List;
  19. import java.util.Map;

  20. import org.apache.ibatis.reflection.MetaObject;
  21. import org.apache.ibatis.reflection.SystemMetaObject;
  22. import org.apache.ibatis.reflection.factory.ObjectFactory;
  23. import org.apache.ibatis.reflection.property.PropertyTokenizer;

  24. /**
  25.  * @author Clinton Begin
  26.  */
  27. public class MapWrapper extends BaseWrapper {

  28.   private final Map<String, Object> map;

  29.   public MapWrapper(MetaObject metaObject, Map<String, Object> map) {
  30.     super(metaObject);
  31.     this.map = map;
  32.   }

  33.   @Override
  34.   public Object get(PropertyTokenizer prop) {
  35.     if (prop.getIndex() != null) {
  36.       Object collection = resolveCollection(prop, map);
  37.       return getCollectionValue(prop, collection);
  38.     } else {
  39.       return map.get(prop.getName());
  40.     }
  41.   }

  42.   @Override
  43.   public void set(PropertyTokenizer prop, Object value) {
  44.     if (prop.getIndex() != null) {
  45.       Object collection = resolveCollection(prop, map);
  46.       setCollectionValue(prop, collection, value);
  47.     } else {
  48.       map.put(prop.getName(), value);
  49.     }
  50.   }

  51.   @Override
  52.   public String findProperty(String name, boolean useCamelCaseMapping) {
  53.     return name;
  54.   }

  55.   @Override
  56.   public String[] getGetterNames() {
  57.     return map.keySet().toArray(new String[0]);
  58.   }

  59.   @Override
  60.   public String[] getSetterNames() {
  61.     return map.keySet().toArray(new String[0]);
  62.   }

  63.   @Override
  64.   public Class<?> getSetterType(String name) {
  65.     PropertyTokenizer prop = new PropertyTokenizer(name);
  66.     if (prop.hasNext()) {
  67.       MetaObject metaValue = metaObject.metaObjectForProperty(prop.getIndexedName());
  68.       if (metaValue == SystemMetaObject.NULL_META_OBJECT) {
  69.         return Object.class;
  70.       } else {
  71.         return metaValue.getSetterType(prop.getChildren());
  72.       }
  73.     } else {
  74.       if (map.get(name) != null) {
  75.         return map.get(name).getClass();
  76.       } else {
  77.         return Object.class;
  78.       }
  79.     }
  80.   }

  81.   @Override
  82.   public Class<?> getGetterType(String name) {
  83.     PropertyTokenizer prop = new PropertyTokenizer(name);
  84.     if (prop.hasNext()) {
  85.       MetaObject metaValue = metaObject.metaObjectForProperty(prop.getIndexedName());
  86.       if (metaValue == SystemMetaObject.NULL_META_OBJECT) {
  87.         return Object.class;
  88.       } else {
  89.         return metaValue.getGetterType(prop.getChildren());
  90.       }
  91.     } else {
  92.       if (map.get(name) != null) {
  93.         return map.get(name).getClass();
  94.       } else {
  95.         return Object.class;
  96.       }
  97.     }
  98.   }

  99.   @Override
  100.   public boolean hasSetter(String name) {
  101.     return true;
  102.   }

  103.   @Override
  104.   public boolean hasGetter(String name) {
  105.     PropertyTokenizer prop = new PropertyTokenizer(name);
  106.     if (prop.hasNext()) {
  107.       if (map.containsKey(prop.getIndexedName())) {
  108.         MetaObject metaValue = metaObject.metaObjectForProperty(prop.getIndexedName());
  109.         if (metaValue == SystemMetaObject.NULL_META_OBJECT) {
  110.           return true;
  111.         } else {
  112.           return metaValue.hasGetter(prop.getChildren());
  113.         }
  114.       } else {
  115.         return false;
  116.       }
  117.     } else {
  118.       return map.containsKey(prop.getName());
  119.     }
  120.   }

  121.   @Override
  122.   public MetaObject instantiatePropertyValue(String name, PropertyTokenizer prop, ObjectFactory objectFactory) {
  123.     HashMap<String, Object> map = new HashMap<>();
  124.     set(prop, map);
  125.     return MetaObject.forObject(map, metaObject.getObjectFactory(), metaObject.getObjectWrapperFactory(), metaObject.getReflectorFactory());
  126.   }

  127.   @Override
  128.   public boolean isCollection() {
  129.     return false;
  130.   }

  131.   @Override
  132.   public void add(Object element) {
  133.     throw new UnsupportedOperationException();
  134.   }

  135.   @Override
  136.   public <E> void addAll(List<E> element) {
  137.     throw new UnsupportedOperationException();
  138.   }

  139. }