ModelUtil.java

package io.lumify.core.util;

import com.altamiracorp.bigtable.model.ModelSession;
import io.lumify.core.model.artifactThumbnails.BigTableArtifactThumbnail;
import io.lumify.core.model.audit.Audit;
import io.lumify.core.user.User;

import java.util.Arrays;
import java.util.List;

public class ModelUtil {
    private static final LumifyLogger LOGGER = LumifyLoggerFactory.getLogger(ModelUtil.class);

    private static final List<String> tables = Arrays.asList(
            BigTableArtifactThumbnail.TABLE_NAME,
            Audit.TABLE_NAME,
            // TODO refactor to config file info. But since this is only for development this is low priority
            "lumify_securegraph_d",
            "lumify_securegraph_v",
            "lumify_securegraph_e",
            "lumify_securegraph_m",
            "lumify_userNotifications",
            "lumify_systemNotifications");

    public static void initializeTables(ModelSession modelSession, User user) {
        for (String table : tables) {
            modelSession.initializeTable(table, user.getModelUserContext());
        }
    }

    public static void deleteTables(ModelSession modelSession, User user) {
        LOGGER.debug("BEGIN deleting tables");
        for (String table : tables) {
            modelSession.deleteTable(table, user.getModelUserContext());
        }
        for (String table : modelSession.getTableList(user.getModelUserContext())) {
            if (table.startsWith("lumify_")) {
                modelSession.deleteTable(table, user.getModelUserContext());
            }
        }
        LOGGER.debug("END deleting tables");
    }
}