datahub/wherehows-api/app/utils/SampleData.java

/**
 * Copyright 2015 LinkedIn Corp. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 */
package utils;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.fasterxml.jackson.databind.node.ObjectNode;
import com.fasterxml.jackson.databind.node.TextNode;
import org.apache.commons.lang3.StringUtils;

import java.util.Arrays;
import java.util.Iterator;

public class SampleData
{
    private final static String MEMBER_ID = "memberId";
    private final static String TREE_ID = "treeId";
    private final static String TRACKING_ID = "trackingId";
    private final static String IP_AS_BYTES_1 = "ipAsBytes";
    private final static String IP_AS_BYTES = "ip_as_bytes";
    private final static String ATTACHMENTS = "attachments";
    private final static String PAYLOAD = "payload";
    private final static String MEDIA = "MEDIA";
    private final static String HEADER = "header";
    private final static String GUID = "guid";
    private final static String AUDITHEADER = "auditHeader";
    private final static String MESSAGE_ID = "messageId";
    private final static String REQUEST = "request";
    private final static String REQUEST_HEADER = "requestHeader";

    public static String convertToHexString(JsonNode node)
    {
        String result = null;
        if (node != null)
        {
            if (node.isArray())
            {
                Iterator<JsonNode> arrayIterator = node.elements();
                StringBuilder sb = new StringBuilder();
                while (arrayIterator.hasNext()) {
                    JsonNode elementNode = arrayIterator.next();
                    if (elementNode != null)
                    {
                        String text = elementNode.asText();
                        if (StringUtils.isNotBlank(text))
                        {
                            byte[] bytes = text.getBytes();
                            if (bytes != null && bytes.length > 0)
                            {
                                sb.append(String.format("%02x", text.getBytes()[0]));
                            }
                        }
                    }
                }
                result = sb.toString();
            }
            else
            {
                result = node.asText();
            }
        }

        return result;
    }

    public static JsonNode secureSampleData(JsonNode sample) {

        String[] nameArray = new String[]{"member_sk", "membersk", "member_id", "memberid", "mem_sk", "mem_id"};
        if (sample != null && sample.has("sample")) {
            JsonNode sampleNode = sample.get("sample");
            if (sampleNode != null && sampleNode.has("columnNames") && sampleNode.has("data")) {
                JsonNode namesNode = sampleNode.get("columnNames");
                JsonNode dataNode = sampleNode.get("data");
                if (namesNode != null && namesNode.isArray()) {
                    for (int i = 0; i < namesNode.size(); i++) {
                        if (Arrays.asList(nameArray).contains(namesNode.get(i).asText().toLowerCase())) {
                            if (dataNode != null && dataNode.isArray()) {
                                for (JsonNode node : dataNode) {
                                    JsonNode valueNode = node.get(i);
                                    ((ArrayNode) node).set(i, new TextNode("********"));
                                }
                            }
                        }
                    }
                }
            }
            else
            {
                int index = 0;
                Iterator<JsonNode> arrayIterator = sampleNode.elements();
                while (arrayIterator.hasNext()) {
                    JsonNode sampleRowNode = arrayIterator.next();
                    if (sampleRowNode != null)
                    {
                        if (sampleRowNode.has(MEMBER_ID))
                        {
                            ((ObjectNode) sampleRowNode).set(MEMBER_ID, new TextNode("********"));
                        }
                        if (sampleRowNode.has(TREE_ID))
                        {
                            JsonNode treeIdNode = sampleRowNode.get(TREE_ID);
                            String convertedValue = convertToHexString(treeIdNode);
                            ((ObjectNode) sampleRowNode).set(TREE_ID, new TextNode(convertedValue));
                        }
                        if (sampleRowNode.has(TRACKING_ID))
                        {
                            JsonNode trackingIdNode = sampleRowNode.get(TRACKING_ID);
                            String convertedValue = convertToHexString(trackingIdNode);
                            ((ObjectNode) sampleRowNode).set(TRACKING_ID, new TextNode(convertedValue));
                        }
                        if (sampleRowNode.has(IP_AS_BYTES))
                        {
                            JsonNode ipNode = sampleRowNode.get(IP_AS_BYTES);
                            String convertedValue = convertToHexString(ipNode);
                            ((ObjectNode) sampleRowNode).set(IP_AS_BYTES, new TextNode(convertedValue));
                        }
                        if (sampleRowNode.has(IP_AS_BYTES_1))
                        {
                            JsonNode ipNode = sampleRowNode.get(IP_AS_BYTES_1);
                            String convertedValue = convertToHexString(ipNode);
                            ((ObjectNode) sampleRowNode).set(IP_AS_BYTES_1, new TextNode(convertedValue));
                        }
                        if (sampleRowNode.has(ATTACHMENTS))
                        {
                            JsonNode attachmentNode = sampleRowNode.get(ATTACHMENTS);
                            if (attachmentNode.has(PAYLOAD))
                            {
                                JsonNode payloadNode = attachmentNode.get(PAYLOAD);
                                String value = "** " + Integer.toString(payloadNode.size()) + " bytes binary data **";
                                ((ObjectNode) attachmentNode).set(PAYLOAD, new TextNode(value));
                            }
                        }
                        if (sampleRowNode.has(MEDIA))
                        {
                            JsonNode mediaNode = sampleRowNode.get(MEDIA);
                            String convertedValue = convertToHexString(mediaNode);
                            ((ObjectNode) sampleRowNode).set(MEDIA, new TextNode(convertedValue));
                        }
                        if (sampleRowNode.has(HEADER))
                        {
                            JsonNode headerNode = sampleRowNode.get(HEADER);
                            if (headerNode != null)
                            {
                                if (headerNode.has(MEMBER_ID))
                                {
                                    ((ObjectNode) headerNode).set(MEMBER_ID, new TextNode("********"));
                                }
                                if (headerNode.has(GUID))
                                {
                                    JsonNode guidNode = headerNode.get(GUID);
                                    String convertedValue = convertToHexString(guidNode);
                                    ((ObjectNode) headerNode).set(GUID, new TextNode(convertedValue));
                                }
                                if (headerNode.has(TREE_ID))
                                {
                                    JsonNode headerTreeIdNode = headerNode.get(TREE_ID);
                                    String convertedValue = convertToHexString(headerTreeIdNode);
                                    ((ObjectNode) headerNode).set(TREE_ID, new TextNode(convertedValue));
                                }
                                if (headerNode.has(AUDITHEADER))
                                {
                                    JsonNode auditHeaderNode = headerNode.get(AUDITHEADER);
                                    if (auditHeaderNode != null && auditHeaderNode.has(MESSAGE_ID))
                                    {
                                        JsonNode messageIdNode = auditHeaderNode.get(MESSAGE_ID);
                                        String convertedValue = convertToHexString(messageIdNode);
                                        ((ObjectNode) auditHeaderNode).set(MESSAGE_ID, new TextNode(convertedValue));
                                    }
                                }
                            }
                        }
                        if (sampleRowNode.has(REQUEST))
                        {
                            JsonNode requestNode = sampleRowNode.get(REQUEST);
                            if (requestNode != null && requestNode.has(ATTACHMENTS))
                            {
                                JsonNode attachmentsNode = requestNode.get(ATTACHMENTS);
                                if (attachmentsNode != null && attachmentsNode.has(PAYLOAD))
                                {
                                    JsonNode payloadNode = attachmentsNode.get(PAYLOAD);
                                    String value = "** " +
                                        Integer.toString(payloadNode.size()) +
                                        " bytes binary data **";
                                    ((ObjectNode) attachmentsNode).set(PAYLOAD, new TextNode(value));
                                }
                            }
                        }
                        if (sampleRowNode.has(REQUEST_HEADER))
                        {
                            JsonNode requestHeaderNode = sampleRowNode.get(REQUEST_HEADER);
                            if (requestHeaderNode != null && requestHeaderNode.has(IP_AS_BYTES))
                            {
                                JsonNode ipNode = requestHeaderNode.get(IP_AS_BYTES);
                                String convertedValue = convertToHexString(ipNode);
                                ((ObjectNode) requestHeaderNode).set(IP_AS_BYTES, new TextNode(convertedValue));
                            }
                            if (requestHeaderNode != null && requestHeaderNode.has(IP_AS_BYTES_1))
                            {
                                JsonNode ipNode = requestHeaderNode.get(IP_AS_BYTES_1);
                                String convertedValue = convertToHexString(ipNode);
                                ((ObjectNode) requestHeaderNode).set(IP_AS_BYTES_1, new TextNode(convertedValue));
                            }
                        }
                    }
                }
            }
        }
        return sample;
    }
}