com.jci.item.repo.ItemRepoImpl.java Source code

Introduction

Here is the source code for com.jci.item.repo.ItemRepoImpl.java

Source

/**
 * (C) Copyright 2016 Johnson Controls, Inc
 * Use or Copying of all or any part of this program, except as
 * permitted by License Agreement, is prohibited.
 */
package com.jci.item.repo;

import java.io.IOException;
import java.net.URISyntaxException;
import java.security.InvalidKeyException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;

import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Repository;
//import org.springframework.transaction.annotation.Transactional;

import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.jci.item.azure.AzureStorage;
import com.jci.item.azure.data.DataHelper;
import com.jci.item.azure.data.DataUtil;
import com.jci.item.azure.data.ResultSet;
import com.jci.item.azure.query.PaginationParam;
import com.jci.item.azure.query.ScrollingParam;
import com.jci.item.utils.QueryBuilder;
import com.microsoft.azure.storage.ResultContinuation;
import com.microsoft.azure.storage.ResultSegment;
import com.microsoft.azure.storage.StorageException;
import com.microsoft.azure.storage.table.CloudTable;
import com.microsoft.azure.storage.table.DynamicTableEntity;
import com.microsoft.azure.storage.table.EntityProperty;
import com.microsoft.azure.storage.table.TableQuery;

/**
 * <p>
 * <strong> The  Item Repository Implementation Class.</strong>
 * <p>
 *
 * @author csonisk
 */
@Repository
public class ItemRepoImpl implements ItemRepo {
    /** The Constant LOG. */
    // NO_UCD (unused code)
    private static final Logger LOG = LoggerFactory.getLogger(ItemRepoImpl.class);

    /** The counter. */
    static int counter = 0;

    /** The batch size. */
    final int batchSize = 15;

    /** The azure storage. */
    @Autowired
    private AzureStorage azureStorage;

    /* (non-Javadoc)
     * @see com.jci.item.repo.ItemRepo#getSegmentedResultSet(com.jci.item.azure.query.ScrollingParam, com.jci.item.azure.data.DataHelper)
     */
    @Override
    public ResultSet getSegmentedResultSet(ScrollingParam param, DataHelper request)
            throws InvalidKeyException, URISyntaxException, StorageException {
        ResultContinuation continuationToken = DataUtil.getContinuationToken(param);
        PaginationParam pagination = new PaginationParam();
        if (continuationToken != null) {
            pagination.setLastPartition(param.getPartition());
            pagination.setLastRow(param.getRow());
        }

        // Create the query
        String whereCondition = QueryBuilder.partitionWhereCondition(request.getPartitionValue());

        if (StringUtils.isBlank(whereCondition)) {
            return null;
        }

        TableQuery<DynamicTableEntity> query = TableQuery.from(DynamicTableEntity.class).where(whereCondition)
                .take(param.getSize());
        CloudTable table = azureStorage.getTable(request.getTableName());

        // segmented query
        ResultSegment<DynamicTableEntity> response = table.executeSegmented(query, continuationToken);

        // next continuation token
        continuationToken = response.getContinuationToken();
        if (continuationToken != null) {
            pagination.setNextPartition(continuationToken.getNextPartitionKey());
            pagination.setNextRow(continuationToken.getNextRowKey());
        }

        HashMap<String, Object> hashmap;

        List<HashMap<String, Object>> series = new ArrayList<>();
        DynamicTableEntity row;
        EntityProperty ep;
        ObjectMapper mapper = new ObjectMapper();
        TypeReference<HashMap<String, Object>> typeRef = new TypeReference<HashMap<String, Object>>() {
        };
        Iterator<DynamicTableEntity> rows = response.getResults().iterator();

        while (rows.hasNext()) {
            row = rows.next();
            HashMap<String, EntityProperty> map = row.getProperties();
            hashmap = new HashMap<>();
            for (String key : map.keySet()) {
                ep = map.get(key);
                if ("ItemJsonString".equalsIgnoreCase(key)) {
                    try {
                        hashmap = mapper.readValue(ep.getValueAsString(), typeRef);
                        hashmap.put("id", row.getRowKey());
                        series.add(hashmap);
                    } catch (IOException e) {
                        LOG.error("### Exception in   ####", e);
                    }
                }
            }
        }
        return new ResultSet(series, pagination);
    }
}