com.amalto.core.integrity.DefaultCheckDataSource.java Source code

Introduction

Here is the source code for com.amalto.core.integrity.DefaultCheckDataSource.java

Source

/*
 * Copyright (C) 2006-2016 Talend Inc. - www.talend.com
 * 
 * This source code is available under agreement available at
 * %InstallDIR%\features\org.talend.rcp.branding.%PRODUCTNAME%\%PRODUCTNAME%license.txt
 * 
 * You should have received a copy of the agreement along with this program; if not, write to Talend SA 9 rue Pages
 * 92150 Suresnes, France
 */

package com.amalto.core.integrity;

import com.amalto.core.objects.ItemPOJO;
import com.amalto.core.objects.ItemPOJOPK;
import com.amalto.core.storage.StorageMetadataUtils;
import com.amalto.core.objects.datacluster.DataClusterPOJOPK;
import com.amalto.core.server.ServerContext;
import com.amalto.core.server.StorageAdmin;
import com.amalto.core.storage.Storage;
import com.amalto.core.util.Util;
import com.amalto.core.util.XtentisException;
import com.amalto.xmlserver.interfaces.IWhereItem;
import com.amalto.xmlserver.interfaces.WhereCondition;
import org.apache.commons.lang.StringUtils;
import org.apache.log4j.Logger;
import org.talend.mdm.commmon.metadata.*;

import java.util.*;

import static com.amalto.core.integrity.FKIntegrityCheckResult.*;

class DefaultCheckDataSource implements FKIntegrityCheckDataSource {

    private final static Logger logger = Logger.getLogger(DefaultCheckDataSource.class);

    public String getDataModel(String clusterName, String concept, String[] ids) throws XtentisException {
        String dataModel;
        try {
            ItemPOJOPK pk = new ItemPOJOPK(new DataClusterPOJOPK(clusterName), concept, ids);
            ItemPOJO item = Util.getItemCtrl2Local().getItem(pk);
            if (item == null) {
                String id = StringUtils.EMPTY;
                for (String currentIdValue : ids) {
                    id += "[" + currentIdValue + "]"; //$NON-NLS-1$ //$NON-NLS-2$
                }

                throw new RuntimeException("Document with id '" //$NON-NLS-1$
                        + id + "' (concept name: '" //$NON-NLS-1$
                        + concept + "') has already been deleted."); //$NON-NLS-1$
            } else {

                dataModel = item.getDataModelName();
            }
        } catch (Exception e) {
            throw new XtentisException(e);
        }
        return dataModel;
    }

    public long countInboundReferences(String clusterName, String[] ids, String fromTypeName,
            ReferenceFieldMetadata fromReference) throws XtentisException {
        // For the anonymous type and leave the type name empty
        if (fromTypeName == null || fromTypeName.trim().equals("")) { //$NON-NLS-1$
            return 0;
        }
        // Transform ids into the string format expected in base
        StringBuilder referencedId = new StringBuilder(); //$NON-NLS-1$
        for (String id : ids) {
            referencedId.append('[').append(id).append(']');
        }
        StorageAdmin storageAdmin = ServerContext.INSTANCE.get().getStorageAdmin();
        Storage storage = storageAdmin.get(clusterName, storageAdmin.getType(clusterName));
        MetadataRepository repository = storage.getMetadataRepository();
        ComplexTypeMetadata complexType = repository.getComplexType(fromTypeName);
        Set<List<FieldMetadata>> paths = StorageMetadataUtils.paths(complexType, fromReference);
        long inboundReferenceCount = 0;
        for (List<FieldMetadata> path : paths) {
            StringBuilder builder = new StringBuilder();
            builder.append(complexType.getName()).append('/');
            for (FieldMetadata fieldMetadata : path) {
                builder.append(fieldMetadata.getName()).append('/');
            }
            String leftPath = builder.toString();
            IWhereItem whereItem = new WhereCondition(leftPath, WhereCondition.EQUALS, referencedId.toString(),
                    WhereCondition.NO_OPERATOR);
            inboundReferenceCount += Util.getXmlServerCtrlLocal().countItems(clusterName, fromTypeName, whereItem);
        }
        return inboundReferenceCount;
    }

    public Set<ReferenceFieldMetadata> getForeignKeyList(String concept, String dataModel) throws XtentisException {
        // Get FK(s) to check
        MetadataRepository mr = ServerContext.INSTANCE.get().getMetadataRepositoryAdmin().get(dataModel);
        TypeMetadata type = mr.getType(concept);
        if (type != null) {
            return mr.accept(new InboundReferences(type));
        } else {
            logger.warn("Type '" + concept + "' does not exist anymore in data model '" + dataModel //$NON-NLS-1$//$NON-NLS-2$
                    + "'. No integrity check will be performed."); //$NON-NLS-1$
            return Collections.emptySet();
        }
    }

    public void resolvedConflict(Map<FKIntegrityCheckResult, Set<FieldMetadata>> checkResultToFields,
            FKIntegrityCheckResult conflictResolution) {
        if (logger.isInfoEnabled()) {
            logger.info("Found conflicts in data model relative to FK integrity checks"); //$NON-NLS-1$
            logger.info("= Forbidden deletes ="); //$NON-NLS-1$
            dumpFields(FORBIDDEN, checkResultToFields);
            logger.info("= Forbidden deletes (override allowed) ="); //$NON-NLS-1$
            dumpFields(FORBIDDEN_OVERRIDE_ALLOWED, checkResultToFields);
            logger.info("= Allowed deletes ="); //$NON-NLS-1$
            dumpFields(ALLOWED, checkResultToFields);
            logger.info("Conflict resolution: " + conflictResolution); //$NON-NLS-1$
        }
    }

    private static void dumpFields(FKIntegrityCheckResult checkResult,
            Map<FKIntegrityCheckResult, Set<FieldMetadata>> checkResultToFields) {
        Set<FieldMetadata> fields = checkResultToFields.get(checkResult);
        if (fields != null) {
            for (FieldMetadata fieldMetadata : fields) {
                logger.info(fieldMetadata.toString());
            }
        }
    }
}