@Override
  @SuppressWarnings({"rawtypes", "unchecked"})
  public Family createMappedForm(PersistentEntity entity) {
    ClassPropertyFetcher cpf = ClassPropertyFetcher.forClass(entity.getJavaClass());
    final Closure value = cpf.getStaticPropertyValue(GormProperties.MAPPING, Closure.class);
    if (value == null) {
      return new Region();
    }

    final Region family = new Region();
    AttributesFactory factory =
        new AttributesFactory() {
          @SuppressWarnings("unused")
          public void setRegion(String name) {
            family.setRegion(name);
          }
        };
    factory.setDataPolicy(defaultDataPolicy);

    MappingConfigurationBuilder builder = new MappingConfigurationBuilder(factory, KeyValue.class);
    builder.evaluate(value);
    entityToPropertyMap.put(entity, builder.getProperties());
    final RegionAttributes regionAttributes = factory.create();
    family.setRegionAttributes(regionAttributes);
    family.setCacheListeners(regionAttributes.getCacheListeners());
    family.setDataPolicy(regionAttributes.getDataPolicy());
    family.setCacheLoader(regionAttributes.getCacheLoader());
    family.setCacheWriter(regionAttributes.getCacheWriter());

    builder = new MappingConfigurationBuilder(family, KeyValue.class);
    builder.evaluate(value);
    return family;
  }
Exemplo n.º 2
0
 private static String getFamily(
     PersistentEntity persistentEntity, SimpleDBDomainClassMappedForm mappedForm) {
   String table = null;
   if (mappedForm != null) {
     table = mappedForm.getFamily();
   }
   if (table == null) table = persistentEntity.getJavaClass().getSimpleName();
   return table;
 }
  @SuppressWarnings("unchecked")
  @Override
  public Table createMappedForm(PersistentEntity entity) {
    Table table = super.createMappedForm(entity);
    CassandraPersistentEntity cassandraPersistentEntity = (CassandraPersistentEntity) entity;
    // read tableOptions
    ClassPropertyFetcher cpf = ClassPropertyFetcher.forClass(entity.getJavaClass());
    final Closure value = cpf.getStaticPropertyValue(TABLE_PROPERTIES, Closure.class);
    if (value != null) {
      MapConfigurationBuilder builder = new MapConfigurationBuilder();
      try {
        builder.evaluate(value);
      } catch (Exception e) {
        throw new IllegalMappingException(
            String.format("Error reading %s : %s", TABLE_PROPERTIES, e.toString()));
      }
      table.setTableProperties(builder.getProperties());
    }

    if (table.getKeyspace() == null) {
      table.setKeyspace(keyspace);
    }

    // additional static mapping block handling
    Map<String, Column> properties = entityToPropertyMap.get(entity);
    Object version = properties.get(MappingConfigurationBuilder.VERSION_KEY);
    if (version instanceof Boolean) {
      cassandraPersistentEntity.setVersion((Boolean) version);
    }

    Column idProperty = properties.get(IDENTITY_PROPERTY);
    Iterator<Entry<String, Column>> propertyIterator = properties.entrySet().iterator();

    while (propertyIterator.hasNext()) {
      Entry<String, Column> entry = propertyIterator.next();
      if (entry.getValue() instanceof Column) {
        String name = entry.getKey();
        Column column = entry.getValue();
        if (idProperty != null
            && idProperty.getName() != null
            && idProperty.getName().equals(name)) {
          // remove extra column created if id property in constraints block, as it conflicts with
          // the column created in mapping block.
          // constraints will be handled elsewhere in GORM
          propertyIterator.remove();
          continue;
        }

        if (column.getName() == null) {
          column.setName(name);
        }
        table.addColumn(column);
      }
    }

    return table;
  }
  @Override
  protected List<Object> retrieveAllEntities(
      PersistentEntity persistentEntity, Iterable<Serializable> keys) {

    Query query = session.createQuery(persistentEntity.getJavaClass());

    PersistentProperty identity = persistentEntity.getIdentity();
    if (keys instanceof List) {
      List actualKeys = new ArrayList();
      Iterator iterator = keys.iterator();
      while (iterator.hasNext()) {
        Object key = iterator.next();
        Object id = getIdentifierForKey(key);
        actualKeys.add(id);
      }
      query.in(identity.getName(), actualKeys);
    } else {
      List<Serializable> keyList = new ArrayList<Serializable>();
      for (Serializable key : keys) {
        keyList.add(key);
      }
      query.in(identity.getName(), keyList);
    }

    List<Object> entityResults = new ArrayList<Object>();
    Iterator<Serializable> keyIterator = keys.iterator();
    Map<Serializable, Object> resultMap = new HashMap<Serializable, Object>();
    for (Object o : query.list()) {
      if (o instanceof DBObject) {
        DBObject dbo = (DBObject) o;
        o =
            createObjectFromNativeEntry(
                getPersistentEntity(), (Serializable) dbo.get(MONGO_ID_FIELD), dbo);
      }
      resultMap.put(getObjectIdentifier(o), o);
    }
    while (keyIterator.hasNext()) {
      Object key = getIdentifierForKey(keyIterator.next());
      ConversionService conversionService = getMappingContext().getConversionService();
      key = conversionService.convert(key, identity.getType());
      Object o = resultMap.get(key);
      entityResults.add(o); // may add null, so entityResults list size matches input list size.
    }

    return entityResults;
  }
  public static org.hibernate.criterion.DetachedCriteria getHibernateDetachedCriteria(
      AbstractHibernateQuery hibernateQuery, QueryableCriteria<?> queryableCriteria) {

    String alias = queryableCriteria.getAlias();
    PersistentEntity persistentEntity = queryableCriteria.getPersistentEntity();
    Class targetClass = persistentEntity.getJavaClass();
    org.hibernate.criterion.DetachedCriteria detachedCriteria;

    if (alias != null) {
      detachedCriteria = org.hibernate.criterion.DetachedCriteria.forClass(targetClass, alias);
    } else {
      detachedCriteria = org.hibernate.criterion.DetachedCriteria.forClass(targetClass);
    }
    populateHibernateDetachedCriteria(
        new HibernateQuery(detachedCriteria, persistentEntity),
        detachedCriteria,
        queryableCriteria);
    return detachedCriteria;
  }
 public boolean isOwningEntity(PersistentEntity owner) {
   return domainClass.isOwningClass(owner.getJavaClass());
 }
 /**
  * @see #getPersistentProperties(Class, org.grails.datastore.mapping.model.MappingContext,
  *     org.grails.datastore.mapping.model.ClassMapping)
  */
 public List<PersistentProperty> getPersistentProperties(
     PersistentEntity entity, MappingContext context, ClassMapping classMapping) {
   return getPersistentProperties(entity.getJavaClass(), context, classMapping);
 }