Spring整合mongodb动态实现

来源:互联网 发布:快雪时晴帖 知乎 编辑:程序博客网 时间:2024/05/16 13:56

Spring整合mongodb的配置文件与上一篇文章没有区别,直接看接口BaseMongoDAO:

public interface BaseMongoDAO<T> {     /**      * 通过条件查询实体(集合)      *       * @param query      */      public List<T> find(Map<String, Object> gtMap,            Map<String, Object> ltMap, Map<String, Object> eqMap,            Map<String, Object> gteMap, Map<String, Object> lteMap,            Map<String, String> regexMap, Map<String, Object> inMap,            Map<String, Object> neMap) ;      /**      * 通过一定的条件查询一个实体      *       * @param query      * @return      */      public T findOne(Map<String, Object> gtMap,            Map<String, Object> ltMap, Map<String, Object> eqMap,            Map<String, Object> gteMap, Map<String, Object> lteMap,            Map<String, String> regexMap, Map<String, Object> inMap,            Map<String, Object> neMap) ;      /**      * 通过条件查询更新数据      *       * @param query      * @param update      * @return      */      public void update(Map<String, Object> gtMap,            Map<String, Object> ltMap, Map<String, Object> eqMap,            Map<String, Object> gteMap, Map<String, Object> lteMap,            Map<String, String> regexMap, Map<String, Object> inMap,            Map<String, Object> neMap, Map<String, Object> params) ;      /**      * 保存一个对象到mongodb      *       * @param entity      * @return      */      public T save(T entity) ;      /**      * 通过ID获取记录      *       * @param id      * @return      */      public T findById(String id) ;      /**      * 通过ID获取记录,并且指定了集合名(表的意思)      *       * @param id      * @param collectionName      *            集合名      * @return      */      public T findById(String id, String collectionName) ;      /**      * 分页查询      * @param page      * @param query      * @return      */      public List<T> findList(int skip, int limit);    /**      * 求数据总和      * @param query      * @return      */      public long count(Map<String, Object> gtMap,            Map<String, Object> ltMap, Map<String, Object> eqMap,            Map<String, Object> gteMap, Map<String, Object> lteMap,            Map<String, String> regexMap, Map<String, Object> inMap,            Map<String, Object> neMap);}

然后是接口的实现类BaseMongoDAOImpl:

public class BaseMongoDAOImpl<T> implements BaseMongoDAO<T> {    @Resource      private MongoTemplate mongoTemplate;    @Override    public List<T> find(Map<String, Object> gtMap,            Map<String, Object> ltMap, Map<String, Object> eqMap,            Map<String, Object> gteMap, Map<String, Object> lteMap,            Map<String, String> regexMap, Map<String, Object> inMap,            Map<String, Object> neMap) {        Criteria c = this.createCriteria(gtMap, ltMap, eqMap, gteMap, lteMap, regexMap, inMap, neMap);        Query q = this.createQuery(c);        return mongoTemplate.find(q, this.getEntityClass());     }    @Override    public T findOne(Map<String, Object> gtMap,            Map<String, Object> ltMap, Map<String, Object> eqMap,            Map<String, Object> gteMap, Map<String, Object> lteMap,            Map<String, String> regexMap, Map<String, Object> inMap,            Map<String, Object> neMap) {        Criteria c = this.createCriteria(gtMap, ltMap, eqMap, gteMap, lteMap, regexMap, inMap, neMap);        Query q = this.createQuery(c);        return mongoTemplate.findOne(q, this.getEntityClass());     }    @Override    public void update(Map<String, Object> gtMap,            Map<String, Object> ltMap, Map<String, Object> eqMap,            Map<String, Object> gteMap, Map<String, Object> lteMap,            Map<String, String> regexMap, Map<String, Object> inMap,            Map<String, Object> neMap, Map<String, Object> params) {        Criteria c = this.createCriteria(gtMap, ltMap, eqMap, gteMap, lteMap, regexMap, inMap, neMap);        Query q = this.createQuery(c);        Set<String> _set = null;        if (params != null && params.size() > 0) {            _set = params.keySet();        }        Update update = new Update().set(_set.toString(), params.get(_set));        mongoTemplate.findAndModify(q, update, this.getEntityClass());     }    @Override    public T save(T entity) {         mongoTemplate.insert(entity);           return entity;     }    @Override    public T findById(String id) {        return mongoTemplate.findById(id, this.getEntityClass());    }    @Override    public T findById(String id, String collectionName) {        return mongoTemplate.findById(id, this.getEntityClass(), collectionName);    }    @Override    public long count(Map<String, Object> gtMap,            Map<String, Object> ltMap, Map<String, Object> eqMap,            Map<String, Object> gteMap, Map<String, Object> lteMap,            Map<String, String> regexMap, Map<String, Object> inMap,            Map<String, Object> neMap) {        Criteria c = this.createCriteria(gtMap, ltMap, eqMap, gteMap, lteMap, regexMap, inMap, neMap);        Query q = this.createQuery(c);        return mongoTemplate.count(q, this.getEntityClass());    }    @Override    public List<T> findList(int skip, int limit) {        Query query = new Query();          query.with(new Sort(new Order(Direction.ASC, "_id")));          query.skip(skip).limit(limit);          return this.mongoTemplate.find(query, this.getEntityClass());    }    private Class<T> getEntityClass(){          return ReflectionUtils.getSuperClassGenricType(getClass());      }    //生成查询条件    public Criteria createCriteria(Map<String, Object> gtMap,            Map<String, Object> ltMap, Map<String, Object> eqMap,            Map<String, Object> gteMap, Map<String, Object> lteMap,            Map<String, String> regexMap, Map<String, Object> inMap,            Map<String, Object> neMap) {        Criteria c = new Criteria();        List<Criteria> listC= new ArrayList<Criteria>();        Set<String> _set = null;        if (gtMap != null && gtMap.size() > 0) {            _set = gtMap.keySet();            for (String _s : _set) {                listC.add( Criteria.where(_s).gt(gtMap.get(_s)));            }        }        if (ltMap != null && ltMap.size() > 0) {            _set = ltMap.keySet();            for (String _s : _set) {                listC.add(  Criteria.where(_s).lt(ltMap.get(_s)));            }        }        if (eqMap != null && eqMap.size() > 0) {            _set = eqMap.keySet();            for (String _s : _set) {                listC.add(  Criteria.where(_s).is(eqMap.get(_s)));            }        }        if (gteMap != null && gteMap.size() > 0) {            _set = gteMap.keySet();            for (String _s : _set) {                listC.add( Criteria.where(_s).gte(gteMap.get(_s)));            }        }        if (lteMap != null && lteMap.size() > 0) {            _set = lteMap.keySet();            for (String _s : _set) {                listC.add(  Criteria.where(_s).lte(lteMap.get(_s)));            }        }        if (regexMap != null && regexMap.size() > 0) {            _set = regexMap.keySet();            for (String _s : _set) {                listC.add(  Criteria.where(_s).regex(regexMap.get(_s)));            }        }        if (inMap != null && inMap.size() > 0) {            _set = inMap.keySet();            for (String _s : _set) {                listC.add(  Criteria.where(_s).in(inMap.get(_s)));            }        }        if (neMap != null && neMap.size() > 0) {            _set = neMap.keySet();            for (String _s : _set) {                listC.add( Criteria.where(_s).ne(neMap.get(_s)));            }        }        if(listC.size() > 0){            Criteria [] cs = new Criteria[listC.size()];            c.andOperator(listC.toArray(cs));        }        return c;    }    public Query createQuery(Criteria c) {        Query query = new Query(c);        return query;    }}

其中主要的区别是利用反射获取泛型的class类型方法ReflectionUtils:

private static Logger logger = LoggerFactory.getLogger(ReflectionUtils.class);      /**      * 调用Getter方法.      */      public static Object invokeGetterMethod(Object obj, String propertyName) {          String getterMethodName = "get" + StringUtils.capitalize(propertyName);          return invokeMethod(obj, getterMethodName, new Class[] {}, new Object[] {});      }      /**      * 调用Setter方法.使用value的Class来查找Setter方法.      */      public static void invokeSetterMethod(Object obj, String propertyName, Object value) {          invokeSetterMethod(obj, propertyName, value, null);      }      /**      * 调用Setter方法.      *       * @param propertyType 用于查找Setter方法,为空时使用value的Class替代.      */      public static void invokeSetterMethod(Object obj, String propertyName, Object value, Class<?> propertyType) {          Class<?> type = propertyType != null ? propertyType : value.getClass();          String setterMethodName = "set" + StringUtils.capitalize(propertyName);          invokeMethod(obj, setterMethodName, new Class[] { type }, new Object[] { value });      }      /**      * 直接读取对象属性值, 无视private/protected修饰符, 不经过getter函数.      */      public static Object getFieldValue(final Object obj, final String fieldName) {          Field field = getAccessibleField(obj, fieldName);          if (field == null) {              throw new IllegalArgumentException("Could not find field [" + fieldName + "] on target [" + obj + "]");          }          Object result = null;          try {              result = field.get(obj);          } catch (IllegalAccessException e) {              logger.error("不可能抛出的异常{}", e.getMessage());          }          return result;      }      /**      * 直接设置对象属性值, 无视private/protected修饰符, 不经过setter函数.      */      public static void setFieldValue(final Object obj, final String fieldName, final Object value) {          Field field = getAccessibleField(obj, fieldName);          if (field == null) {              throw new IllegalArgumentException("Could not find field [" + fieldName + "] on target [" + obj + "]");          }          try {              field.set(obj, value);          } catch (IllegalAccessException e) {              logger.error("不可能抛出的异常:{}", e.getMessage());          }      }      /**      * 循环向上转型, 获取对象的DeclaredField,   并强制设置为可访问.      *       * 如向上转型到Object仍无法找到, 返回null.      */      public static Field getAccessibleField(final Object obj, final String fieldName) {          Assert.notNull(obj, "object不能为空");          Assert.hasText(fieldName, "fieldName");          for (Class<?> superClass = obj.getClass(); superClass != Object.class; superClass = superClass.getSuperclass()) {              try {                  Field field = superClass.getDeclaredField(fieldName);                  field.setAccessible(true);                  return field;              } catch (NoSuchFieldException e) {//NOSONAR                  // Field不在当前类定义,继续向上转型              }          }          return null;      }      /**      * 直接调用对象方法, 无视private/protected修饰符.      * 用于一次性调用的情况.      */      public static Object invokeMethod(final Object obj, final String methodName, final Class<?>[] parameterTypes,              final Object[] args) {          Method method = getAccessibleMethod(obj, methodName, parameterTypes);          if (method == null) {              throw new IllegalArgumentException("Could not find method [" + methodName + "] on target [" + obj + "]");          }          try {              return method.invoke(obj, args);          } catch (Exception e) {              throw convertReflectionExceptionToUnchecked(e);          }      }      /**      * 循环向上转型, 获取对象的DeclaredMethod,并强制设置为可访问.      * 如向上转型到Object仍无法找到, 返回null.      *       * 用于方法需要被多次调用的情况. 先使用本函数先取得Method,然后调用Method.invoke(Object obj, Object... args)      */      public static Method getAccessibleMethod(final Object obj, final String methodName,              final Class<?>... parameterTypes) {          Assert.notNull(obj, "object不能为空");          for (Class<?> superClass = obj.getClass(); superClass != Object.class; superClass = superClass.getSuperclass()) {              try {                  Method method = superClass.getDeclaredMethod(methodName, parameterTypes);                  method.setAccessible(true);                  return method;              } catch (NoSuchMethodException e) {//NOSONAR                  // Method不在当前类定义,继续向上转型              }          }          return null;      }      /**      * 通过反射, 获得Class定义中声明的父类的泛型参数的类型.      * 如无法找到, 返回Object.class.      * eg.      * public UserDao extends HibernateDao<User>      *      * @param clazz The class to introspect      * @return the first generic declaration, or Object.class if cannot be determined      */      @SuppressWarnings({ "unchecked", "rawtypes" })      public static <T> Class<T> getSuperClassGenricType(final Class clazz) {          return getSuperClassGenricType(clazz, 0);      }      /**      * 通过反射, 获得Class定义中声明的父类的泛型参数的类型.      * 如无法找到, 返回Object.class.      *       * 如public UserDao extends HibernateDao<User,Long>      *      * @param clazz clazz The class to introspect      * @param index the Index of the generic ddeclaration,start from 0.      * @return the index generic declaration, or Object.class if cannot be determined      */      @SuppressWarnings("rawtypes")      public static Class getSuperClassGenricType(final Class clazz, final int index) {          Type genType = clazz.getGenericSuperclass();          if (!(genType instanceof ParameterizedType)) {              logger.warn(clazz.getSimpleName() + "'s superclass not ParameterizedType");              return Object.class;          }          Type[] params = ((ParameterizedType) genType).getActualTypeArguments();          if (index >= params.length || index < 0) {              logger.warn("Index: " + index + ", Size of " + clazz.getSimpleName() + "'s Parameterized Type: "                      + params.length);              return Object.class;          }          if (!(params[index] instanceof Class)) {              logger.warn(clazz.getSimpleName() + " not set the actual class on superclass generic parameter");              return Object.class;          }          return (Class) params[index];      }      /**      * 将反射时的checked exception转换为unchecked exception.      */      public static RuntimeException convertReflectionExceptionToUnchecked(Exception e) {          if (e instanceof IllegalAccessException || e instanceof IllegalArgumentException                  || e instanceof NoSuchMethodException) {              return new IllegalArgumentException("Reflection Exception.", e);          } else if (e instanceof InvocationTargetException) {              return new RuntimeException("Reflection Exception.", ((InvocationTargetException) e).getTargetException());          } else if (e instanceof RuntimeException) {              return (RuntimeException) e;          }          return new RuntimeException("Unexpected Checked Exception.", e);      }}

这样就可以动态的调用方法做CRUD操作,但是方法中传的参数Map中的键值需对应实体类中的属性名称,每次CRUD操作都需要用到反射获取泛型的class类型,对性能有影响。

0 0
原创粉丝点击