com.gemstone.gemfire.management.internal.cli.result.TabularResultData Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of gemfire-core Show documentation
SnappyData store based off Pivotal GemFireXD
The newest version!
/*
 * Copyright (c) 2010-2015 Pivotal Software, Inc. 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. See the License for the specific language governing
 * permissions and limitations under the License. See accompanying
 * LICENSE file.
 */
package com.gemstone.gemfire.management.internal.cli.result;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import com.gemstone.gemfire.management.internal.cli.json.GfJsonArray;
import com.gemstone.gemfire.management.internal.cli.json.GfJsonException;
import com.gemstone.gemfire.management.internal.cli.json.GfJsonObject;

/**
 * 
 * @author Abhishek Chaudhari
 * 
 * @since 7.0
 */
public class TabularResultData extends AbstractResultData {
  /*package*/TabularResultData() {
    super();
  }
  
  /*package*/TabularResultData(GfJsonObject gfJsonObject) {
    super(gfJsonObject);
  }

  public TabularResultData accumulate(String accumulateFor, Object value) {
    try {
      contentObject.accumulate(accumulateFor, value);
    } catch (GfJsonException e) {
      throw new ResultDataException(e.getMessage());
    }
    return this;
  }
  
  public GfJsonArray getHeaders() {
    try {
      return this.contentObject.names();
    } catch (GfJsonException e) {
      e.printStackTrace();
    }
    return null;
  }

  /**
   * @return the gfJsonObject
   */
  public GfJsonObject getGfJsonObject() {
    return gfJsonObject;
  }

  @Override
  public String getType() {
    return TYPE_TABULAR;
  }

  /**
   * 
   * @param headerText
   * @return this TabularResultData
   * @throws ResultDataException
   *           If the value is non-finite number or if the key is null.
   */
  public TabularResultData setHeader(String headerText) {
    return (TabularResultData) super.setHeader(headerText);
  }
  
  /**
   * 
   * @param footerText
   * @return this TabularResultData
   * @throws ResultDataException
   *           If the value is non-finite number or if the key is null.
   */
  public TabularResultData setFooter(String footerText) {
    return (TabularResultData) super.setFooter(footerText);
  }

  @Override
  public String getHeader() {
    return gfJsonObject.getString(RESULT_HEADER);
  }

  @Override
  public String getFooter() {
    return gfJsonObject.getString(RESULT_FOOTER);
  }
  
  public Map retrieveDataByValueInColumn(String columnName, String valueToSearch) {
    Map foundValues = Collections.emptyMap();
    try {
      GfJsonArray jsonArray = contentObject.getJSONArray(columnName);
      int size = jsonArray.size();
      int foundIndex = -1;
      for (int i = 0; i < size; i++) {
        Object object = jsonArray.get(i);
        if (object != null && object.equals(valueToSearch)) {
          foundIndex = i;
          break;
        }
      }
      
      if (foundIndex != -1) {
        foundValues = new LinkedHashMap();
        for (Iterator iterator = contentObject.keys(); iterator.hasNext();) {
          String storedColumnNames = (String) iterator.next();
          GfJsonArray storedColumnValues = contentObject.getJSONArray(storedColumnNames);
          foundValues.put(storedColumnNames, String.valueOf(storedColumnValues.get(foundIndex)));
        }
      }
    } catch (GfJsonException e) {
      throw new ResultDataException(e.getMessage());
    }
    return foundValues;
  }
  
  public List> retrieveAllDataByValueInColumn(String columnName, String valueToSearch) {
    List> foundValuesList = new ArrayList>();
    try {
      GfJsonArray jsonArray = contentObject.getJSONArray(columnName);
      int size = jsonArray.size();
      for (int i = 0; i < size; i++) {
        Object object = jsonArray.get(i);
        if (object != null && object.equals(valueToSearch)) {
          Map foundValues = new LinkedHashMap();

          for (Iterator iterator = contentObject.keys(); iterator.hasNext();) {
            String storedColumnNames = (String) iterator.next();
            GfJsonArray storedColumnValues = contentObject.getJSONArray(storedColumnNames);
            foundValues.put(storedColumnNames, String.valueOf(storedColumnValues.get(i)));
          }
          
          foundValuesList.add(foundValues);
        }
      }
    } catch (GfJsonException e) {
      throw new ResultDataException(e.getMessage());
    }
    return foundValuesList;
  }
  
  public List retrieveAllValues(String columnName) {
    List values = new ArrayList();

    try {
      GfJsonArray jsonArray = contentObject.getJSONArray(columnName);
      int size = jsonArray.size();
      for (int i = 0; i < size; i++) {
        values.add(String.valueOf(jsonArray.get(i)));
      }
    } catch (GfJsonException e) {
      throw new ResultDataException(e.getMessage());
    }
    return values;
  }
}