Fixes issue

apache · Sep 21, 2015 · 8cba153 · 8cba153
1 parent 8b792d4
commit 8cba153
Show file tree

Hide file tree

Showing 2 changed files with 144 additions and 115 deletions.
diff --git a/stack/core/src/main/java/org/apache/usergrid/persistence/cassandra/EntityManagerImpl.java b/stack/core/src/main/java/org/apache/usergrid/persistence/cassandra/EntityManagerImpl.java
@@ -1391,6 +1391,16 @@ public void updateProperties( UUID entityId, Map<String, Object> properties ) th
 
         batchUpdateProperties( m, entity, properties, timestampUuid );
 
+        //after updating properties, we want to ensure that re-add ourselves to the entity Id sets
+
+        final CollectionInfo collection = Schema.getDefaultSchema().getCollection( Application.ENTITY_TYPE,
+                defaultCollectionName( entity.getType() ) );
+
+        final EntityRef applicationRef = getApplicationRef();
+
+        getRelationManager( applicationRef ).batchAddEntityToCollection( m,  collection.getName(),  applicationRef, entity, timestampUuid);
+
+
         batchExecute( m, CassandraService.RETRY_COUNT );
     }