Drop/Create EHR Index Page

EHR_App/test/src/org/labkey/test/tests/EHR_AppTest.java

@@ -130,6 +130,7 @@ protected List<String> skipLinksForValidation()
         links.add("ehr-colonyOverview.view");
         links.add("ehr-updateTable.view");
         links.add("ehr-populateLookupData.view");
+        links.add("ehr-postgresMigration.view");
         return links;
     }
 

ehr/resources/views/ehrAdmin.html

@@ -11,6 +11,7 @@ <h4><b>Admin:</b></h4>
         <div><a class="labkey-text-link" href="<%=contextPath%><%=containerPath%>/ehr-cacheLivingAnimals.view">Cache Demographics On All Living Animals</a></div>
         <div><a class="labkey-text-link" href="<%=contextPath%><%=containerPath%>/ehr-cacheLivingAnimals.view?includeAll=true">Cache Demographics On All Animals</a></div>
         <div><a class="labkey-text-link" href="<%=contextPath%><%=containerPath%>/ehr-geneticCalculationSettings.view">Genetics Calculations</a></div>
+        <div><a class="labkey-text-link" href="<%=contextPath%><%=containerPath%>/ehr-postgresMigration.view">Postgres Migration</a></div>
     </div>
 
     <div class="lead">

ehr/resources/views/postgresMigration.html

@@ -0,0 +1,47 @@
+<div class="col-lg-3">
+    <div class="lead">
+        <div><a id="dropIndicesLink" class="labkey-text-link" href="#">Drop Indices</a></div>
+        <div><a id="addIndicesLink" class="labkey-text-link" href="#">Add Indices</a></div>
+    </div>
+</div>
+<script type="text/javascript" nonce="<%=scriptNonce%>">
+    function indexAction(operation) {
+        LABKEY.Ajax.request({
+            url: LABKEY.ActionURL.buildURL('ehr', 'updateEHRIndices'),
+            method: 'POST',
+            jsonData: {
+                operation: operation
+            },
+            success: function(response) {
+                var json = JSON.parse(response.responseText);
+                if (json.success) {
+                    alert('Success!');
+                } else {
+                    alert('Failure: ' + (json.message || 'Unknown error'));
+                }
+            },
+            failure: function(response) {
+                var message = 'Request failed';
+                try {
+                    var json = JSON.parse(response.responseText);
+                    if (json.exception) {
+                        message = json.exception;
+                    }
+                } catch (e) {
+                    // ignore parse error
+                }
+                alert('Failure: ' + message);
+            }
+        });
+    }
+
+    LABKEY.Utils.attachEventHandler('dropIndicesLink', 'click', function(e) {
+        e.preventDefault();
+        indexAction('drop');
+    }, 1);
+
+    LABKEY.Utils.attachEventHandler('addIndicesLink', 'click', function(e) {
+        e.preventDefault();
+        indexAction('add');
+    }, 1);
+</script>

ehr/src/org/labkey/ehr/EHRController.java

@@ -684,6 +684,70 @@ public ApiResponse execute(RecordDeleteForm form, BindException errors)
         }
     }
 
+    public static class UpdateEHRIndicesForm
+    {
+        private String _operation;
+
+        public String getOperation()
+        {
+            return _operation;
+        }
+
+        public void setOperation(String operation)
+        {
+            _operation = operation;
+        }
+    }
+
+    @RequiresPermission(AdminPermission.class)
+    public static class UpdateEHRIndicesAction extends MutatingApiAction<UpdateEHRIndicesForm>
+    {
+        @Override
+        public void validateForm(UpdateEHRIndicesForm form, Errors errors)
+        {
+            super.validateForm(form, errors);
+
+            if (form.getOperation() == null)
+            {
+                errors.reject(ERROR_MSG, "Operation parameter is required.");
+            }
+            else if (!"add".equalsIgnoreCase(form.getOperation()) && !"drop".equalsIgnoreCase(form.getOperation()))
+            {
+                errors.reject(ERROR_MSG, "Invalid operation. Must be 'add' or 'drop'.");
+            }
+        }
+
+        @Override
+        public ApiResponse execute(UpdateEHRIndicesForm form, BindException errors)
+        {
+            try
+            {
+                List<String> messages;
+
+                if ("drop".equalsIgnoreCase(form.getOperation()))
+                {
+                    messages = EHRManager.get().dropEHRIndices(getContainer(), getUser());
+                }
+                else
+                {
+                    messages = EHRManager.get().addEHRIndices(getContainer(), getUser());
+                }
+
+                Map<String, Object> response = new HashMap<>();
+                response.put("success", true);
+                response.put("messages", messages);
+                return new ApiSimpleResponse(response);
+            }
+            catch (Exception e)
+            {
+                Map<String, Object> response = new HashMap<>();
+                response.put("success", false);
+                response.put("message", e.getMessage());
+                return new ApiSimpleResponse(response);
+            }
+        }
+    }
+
     public static class AnimalDetailsForm
     {
         private String[] _animalIds;