com.amalto.core.storage.record.ViewSearchResultsWriter.java Source code

Introduction

Here is the source code for com.amalto.core.storage.record.ViewSearchResultsWriter.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.storage.record;

import java.io.BufferedWriter;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.Writer;

import javax.xml.XMLConstants;

import com.amalto.core.storage.SecuredStorage;
import org.apache.commons.lang.StringEscapeUtils;
import org.apache.commons.lang.StringUtils;
import org.talend.mdm.commmon.metadata.*;

import com.amalto.core.query.user.DateConstant;
import com.amalto.core.query.user.DateTimeConstant;
import com.amalto.core.query.user.TimeConstant;
import com.amalto.core.storage.StorageMetadataUtils;

public class ViewSearchResultsWriter implements DataRecordWriter {

    private SecuredStorage.UserDelegator delegator = SecuredStorage.UNSECURED;

    @Override
    public void write(DataRecord record, OutputStream output) throws IOException {
        Writer out = new BufferedWriter(new OutputStreamWriter(output, "UTF-8")); //$NON-NLS-1$
        write(record, out);
    }

    @Override
    public void write(DataRecord record, Writer writer) throws IOException {
        writer.write("<result xmlns:metadata=\"" + DataRecordReader.METADATA_NAMESPACE + "\"" //$NON-NLS-1$ //$NON-NLS-2$
                + " xmlns:xsi=\"" + XMLConstants.W3C_XML_SCHEMA_INSTANCE_NS_URI + "\">\n"); //$NON-NLS-1$ //$NON-NLS-2$
        for (FieldMetadata fieldMetadata : record.getSetFields()) {
            Object value = record.get(fieldMetadata);
            String name = fieldMetadata.getName();
            if (value != null) {
                writer.append("\t<").append(name).append(">"); //$NON-NLS-1$ //$NON-NLS-2$
                processValue(writer, fieldMetadata, value);
                writer.append("</").append(name).append(">\n"); //$NON-NLS-1$ //$NON-NLS-2$
            } else {
                writer.append("\t<").append(name).append("/>\n"); //$NON-NLS-1$//$NON-NLS-2$
            }
        }
        writer.append("</result>"); //$NON-NLS-1$
        writer.flush();
    }

    @Override
    public void setSecurityDelegator(SecuredStorage.UserDelegator delegator) {
        if (delegator == null) {
            throw new IllegalArgumentException("Delegator cannot be null.");
        }
        this.delegator = delegator;
    }

    private void processValue(Writer out, FieldMetadata fieldMetadata, Object value) throws IOException {
        if (value == null) {
            throw new IllegalArgumentException("Not supposed to write null values to XML."); //$NON-NLS-1$
        }
        if (delegator.hide(fieldMetadata)) {
            out.append(StringUtils.EMPTY);
            return;
        }
        String stringValue;
        TypeMetadata type = fieldMetadata.getType();
        if (fieldMetadata instanceof SimpleTypeFieldMetadata) {
            type = MetadataUtils.getSuperConcreteType(type);
        }
        if (Types.DATE.equals(type.getName())) {
            synchronized (DateConstant.DATE_FORMAT) {
                stringValue = (DateConstant.DATE_FORMAT).format(value);
            }
        } else if (Types.DATETIME.equals(type.getName())) {
            synchronized (DateTimeConstant.DATE_FORMAT) {
                stringValue = (DateTimeConstant.DATE_FORMAT).format(value);
            }
        } else if (Types.TIME.equals(type.getName())) {
            synchronized (TimeConstant.TIME_FORMAT) {
                stringValue = (TimeConstant.TIME_FORMAT).format(value);
            }
        } else if (value instanceof Object[]) {
            StringBuilder valueAsString = new StringBuilder();
            CompoundFieldMetadata compoundFields = (CompoundFieldMetadata) ((ReferenceFieldMetadata) fieldMetadata)
                    .getReferencedField();
            int i = 0;
            for (Object current : ((Object[]) value)) {
                valueAsString.append('[')
                        .append(StorageMetadataUtils.toString(current, compoundFields.getFields()[i++]))
                        .append(']');
            }
            stringValue = valueAsString.toString();
        } else {
            stringValue = String.valueOf(value);
        }
        if (fieldMetadata instanceof ReferenceFieldMetadata) {
            if (value instanceof DataRecord) {
                // TODO Replace with MetadataUtils.toString()
                DataRecord referencedRecord = (DataRecord) value;
                StringBuilder fkValueAsString = new StringBuilder();
                for (FieldMetadata keyField : referencedRecord.getType().getKeyFields()) {
                    fkValueAsString.append('[').append(referencedRecord.get(keyField)).append(']');
                }
                stringValue = fkValueAsString.toString();
            } else {
                if (!stringValue.startsWith("[")) { //$NON-NLS-1$
                    stringValue = "[" + StorageMetadataUtils.toString(value, //$NON-NLS-1$
                            ((ReferenceFieldMetadata) fieldMetadata).getReferencedField()) + ']';
                }
            }
        }
        if (fieldMetadata instanceof AliasedFieldMetadata
                && ((AliasedFieldMetadata) fieldMetadata).getAliasedField() instanceof ReferenceFieldMetadata) {
            if (!stringValue.startsWith("[")) { //$NON-NLS-1$
                stringValue = "[" + StorageMetadataUtils.toString(value, //$NON-NLS-1$
                        ((ReferenceFieldMetadata) ((AliasedFieldMetadata) fieldMetadata).getAliasedField())
                                .getReferencedField())
                        + ']';
            }
        }
        out.append(StringEscapeUtils.escapeXml(stringValue));
    }

}