Add helper to create additional container indexes

Author: bbimber

discvrcore/src/org/labkey/discvrcore/DiscvrCoreController.java

@@ -302,4 +302,36 @@ public URLHelper getRedirectURL(Object o) throws Exception
             return DetailsURL.fromString("admin/manageFileRoot.view", getContainer()).getActionURL();
         }
     }
+
+    @UtilityAction(label = "Add Custom Core.Container Indexes", description = "Provides a mechanism to truncate the query and dataset audit tables for a container")
+    @RequiresPermission(AdminPermission.class)
+    public static class AddCustomIndexesAction extends ConfirmAction<Object>
+    {
+        @Override
+        public ModelAndView getConfirmView(Object o, BindException errors) throws Exception
+        {
+            setTitle("Add Custom Core.Container Indexes");
+
+            return HtmlView.of("This action will add custom indexes to core.contains. Only do this if you are absolutely certain about the consequences. Do you want to continue?");
+        }
+
+        @Override
+        public boolean handlePost(Object o, BindException errors) throws Exception
+        {
+            return DiscvrCoreManager.get().addCoreContainersIndexes();
+        }
+
+        @Override
+        public void validateCommand(Object o, Errors errors)
+        {
+
+        }
+
+        @NotNull
+        @Override
+        public URLHelper getSuccessURL(Object o)
+        {
+            return PageFlowUtil.urlProvider(PipelineUrls.class).urlBegin(getContainer());
+        }
+    }
 }

discvrcore/src/org/labkey/discvrcore/DiscvrCoreManager.java

@@ -16,8 +16,26 @@
 
 package org.labkey.discvrcore;
 
+import org.apache.commons.lang3.StringUtils;
+import org.apache.logging.log4j.Logger;
+import org.labkey.api.collections.CaseInsensitiveHashSet;
+import org.labkey.api.data.CoreSchema;
+import org.labkey.api.data.SQLFragment;
+import org.labkey.api.data.SqlExecutor;
+import org.labkey.api.data.TableInfo;
+import org.labkey.api.util.logging.LogHelper;
+
+import java.sql.DatabaseMetaData;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.util.Arrays;
+import java.util.List;
+import java.util.Set;
+
 public class DiscvrCoreManager
 {
+    private static final Logger _log = LogHelper.getLogger(DiscvrCoreManager.class, "Messages from DiscvrCoreManager");
+
     private static final DiscvrCoreManager _instance = new DiscvrCoreManager();
 
     private DiscvrCoreManager()
@@ -29,4 +47,59 @@ public static DiscvrCoreManager get()
     {
         return _instance;
     }
+
+    public boolean addCoreContainersIndexes()
+    {
+        try
+        {
+            TableInfo ti = CoreSchema.getInstance().getTableInfoContainers();
+            addCustomIndex(ti, Arrays.asList("EntityId", "RowId", "Type", "Parent"));
+            addCustomIndex(ti, Arrays.asList("Parent", "EntityId", "Type", "RowId"));
+
+            return true;
+        }
+        catch (Exception e)
+        {
+            _log.error("Unable to create container indexes", e);
+            return false;
+        }
+    }
+
+    private void addCustomIndex(TableInfo ti, List<String> columnNames) throws Exception
+    {
+        String idxName = getIndexName(ti.getName(), columnNames);
+        if (doesIndexExist(ti, idxName))
+        {
+            return;
+        }
+
+        createIndex(ti, idxName, columnNames);
+    }
+
+    private String getIndexName(String tableName, List<String> indexCols)
+    {
+        return "IDX_discvr_" + tableName + "_" + StringUtils.join(indexCols, "_");
+    }
+
+    private boolean doesIndexExist(TableInfo ti, String indexName) throws SQLException
+    {
+        Set<String> indexNames = new CaseInsensitiveHashSet();
+        DatabaseMetaData meta = ti.getSchema().getScope().getConnection().getMetaData();
+        try (ResultSet rs = meta.getIndexInfo(ti.getSchema().getScope().getDatabaseName(), ti.getSchema().getName(), ti.getName(), false, false))
+        {
+            while (rs.next())
+            {
+                indexNames.add(rs.getString("INDEX_NAME"));
+            }
+        }
+
+        return indexNames.contains(indexName);
+    }
+
+    private void createIndex(TableInfo realTable, String indexName, List<String> columns)
+    {
+        _log.info("Creating index on column(s): " + StringUtils.join(columns, ", ") + " for table: " + realTable.getName());
+        SQLFragment sql = new SQLFragment("CREATE NONCLUSTERED INDEX " + indexName + " ON " + realTable.getSelectName() + "(" + StringUtils.join(columns, ", ") + ")");
+        new SqlExecutor(realTable.getSchema()).execute(sql);
+    }
 }