You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@usergrid.apache.org by mr...@apache.org on 2016/09/02 02:19:19 UTC

usergrid git commit: Fix UniqueValueManager tool to work with datastax driver in the unique value serialization strategy.

Repository: usergrid
Updated Branches:
  refs/heads/master aa0353d96 -> 1a080a2d7


Fix UniqueValueManager tool to work with datastax driver in the unique value serialization strategy.


Project: http://git-wip-us.apache.org/repos/asf/usergrid/repo
Commit: http://git-wip-us.apache.org/repos/asf/usergrid/commit/1a080a2d
Tree: http://git-wip-us.apache.org/repos/asf/usergrid/tree/1a080a2d
Diff: http://git-wip-us.apache.org/repos/asf/usergrid/diff/1a080a2d

Branch: refs/heads/master
Commit: 1a080a2d76457c957cf192e81890e2195b554022
Parents: aa0353d
Author: Michael Russo <mr...@apigee.com>
Authored: Thu Sep 1 19:18:46 2016 -0700
Committer: Michael Russo <mr...@apigee.com>
Committed: Thu Sep 1 19:18:46 2016 -0700

----------------------------------------------------------------------
 .../usergrid/tools/UniqueValueManager.java       | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/usergrid/blob/1a080a2d/stack/tools/src/main/java/org/apache/usergrid/tools/UniqueValueManager.java
----------------------------------------------------------------------
diff --git a/stack/tools/src/main/java/org/apache/usergrid/tools/UniqueValueManager.java b/stack/tools/src/main/java/org/apache/usergrid/tools/UniqueValueManager.java
index 6131866..6209784 100644
--- a/stack/tools/src/main/java/org/apache/usergrid/tools/UniqueValueManager.java
+++ b/stack/tools/src/main/java/org/apache/usergrid/tools/UniqueValueManager.java
@@ -23,11 +23,9 @@ import java.io.FileReader;
 import java.util.*;
 import java.util.concurrent.atomic.AtomicInteger;
 
-import com.netflix.astyanax.connectionpool.exceptions.ConnectionException;
-import com.netflix.astyanax.model.Column;
-import com.netflix.astyanax.model.ConsistencyLevel;
-import com.netflix.astyanax.util.RangeBuilder;
-import org.apache.usergrid.persistence.Entity;
+import com.datastax.driver.core.BatchStatement;
+import com.datastax.driver.core.ConsistencyLevel;
+import com.datastax.driver.core.Session;
 import org.apache.usergrid.persistence.EntityManager;
 import org.apache.usergrid.persistence.collection.serialization.MvccEntitySerializationStrategy;
 import org.apache.usergrid.persistence.collection.serialization.UniqueValueSerializationStrategy;
@@ -73,7 +71,7 @@ public class UniqueValueManager extends ToolBase {
     private final MultiTenantColumnFamily<ScopedRowKey<TypeField>, EntityVersion> CF_UNIQUE_VALUES =
         new MultiTenantColumnFamily<>( "Unique_Values_V2", ROW_KEY_SER, ENTITY_VERSION_SER );
 
-    private com.netflix.astyanax.Keyspace keyspace;
+    private Session session;
 
     private MvccEntitySerializationStrategy mvccEntitySerializationStrategy;
 
@@ -136,7 +134,7 @@ public class UniqueValueManager extends ToolBase {
                     "Example: 'b9398e88-ef7f-11e5-9e41-0a2cb9e6caa9|user|email|baasadmins@apigee.com'");
         }
 
-        keyspace = injector.getInstance(com.netflix.astyanax.Keyspace.class);
+        session = injector.getInstance(Session.class);
         mvccEntitySerializationStrategy = injector.getInstance(MvccEntitySerializationStrategy.class);
         uniqueValueSerializationStrategy = injector.getInstance(UniqueValueSerializationStrategy.class);
 
@@ -161,7 +159,7 @@ public class UniqueValueManager extends ToolBase {
 
                 UniqueValueSet uniqueValueSet = uniqueValueSerializationStrategy.load(
                         new ApplicationScopeImpl(new SimpleId(appUuid, "application")),
-                        ConsistencyLevel.valueOf(System.getProperty("usergrid.read.cl", "CL_LOCAL_QUORUM")), entityType,
+                        ConsistencyLevel.valueOf(System.getProperty("usergrid.read.cl", "LOCAL_QUORUM")), entityType,
                         Collections.singletonList(new StringField(fieldType, fieldValue)), false);
 
                 StringBuilder stringBuilder = new StringBuilder();
@@ -188,7 +186,10 @@ public class UniqueValueManager extends ToolBase {
                     uniqueValueSet.forEach(uniqueValue -> {
                         logger.info("DELETING UNIQUE VALUE");
                         try {
-                            uniqueValueSerializationStrategy.delete(new ApplicationScopeImpl(new SimpleId(appUuid, "application")), uniqueValue).execute();
+                            BatchStatement batchStatement = uniqueValueSerializationStrategy.
+                                deleteCQL(new ApplicationScopeImpl(new SimpleId(appUuid, "application")), uniqueValue);
+
+                            session.execute(batchStatement);
                         }
                         catch (Exception e) {
                             logger.error("Exception thrown for UV delete: " + e.getMessage());