Fixing tests and code

Author: Denis Rosa

src/main/java/org/couchbase/quickstart/runners/DBSetupRunner.java

@@ -0,0 +1,61 @@
+package org.couchbase.quickstart.runners;
+
+import com.couchbase.client.core.error.*;
+import com.couchbase.client.java.Bucket;
+import com.couchbase.client.java.Cluster;
+import com.couchbase.client.java.json.JsonObject;
+import com.couchbase.client.java.manager.collection.CollectionManager;
+import com.couchbase.client.java.manager.collection.CollectionSpec;
+import com.couchbase.client.java.query.QueryResult;
+import org.couchbase.quickstart.configs.CollectionNames;
+import org.couchbase.quickstart.configs.DBProperties;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.CommandLineRunner;
+import org.springframework.stereotype.Component;
+
+
+/**
+ * This class run after the application startup. It automatically setup all indexes needed
+ */
+@Component
+public class DBSetupRunner implements CommandLineRunner {
+
+    @Autowired
+    private Bucket bucket;
+    @Autowired
+    private Cluster cluster;
+    @Autowired
+    private DBProperties props;
+
+    @Override
+    public void run(String... args) {
+
+        try {
+            cluster.queryIndexes().createPrimaryIndex(props.getBucketName());
+        } catch (Exception e) {
+            System.out.println("Primary index already exists on bucket "+props.getBucketName());
+        }
+
+        CollectionManager collectionManager = bucket.collections();
+        try {
+            CollectionSpec spec = CollectionSpec.create(CollectionNames.PROFILE, bucket.defaultScope().name());
+            collectionManager.createCollection(spec);
+        } catch (CollectionExistsException e){
+            System.out.println(String.format("Collection <%s> already exists", CollectionNames.PROFILE));
+        } catch (Exception e) {
+            System.out.println(String.format("Generic error <%s>",e.getMessage()));
+        }
+
+        try {
+            final QueryResult result = cluster.query("CREATE PRIMARY INDEX default_profile_index ON "+props.getBucketName()+"._default."+ CollectionNames.PROFILE);
+            for (JsonObject row : result.rowsAsObject()){
+                System.out.println(String.format("Index Creation Status %s",row.getObject("meta").getString("status")));
+            }
+        } catch (IndexExistsException e){
+            System.out.println(String.format("Collection's primary index already exists"));
+        } catch (Exception e){
+            System.out.println(String.format("General error <%s> when trying to create index ",e.getMessage()));
+        }
+
+    }
+}