org.trustedanalytics.examples.hbase.services.ConversionsService.java Source code

Introduction

Here is the source code for org.trustedanalytics.examples.hbase.services.ConversionsService.java

Source

/**
 * Copyright (c) 2016 Intel Corporation
 *
 * 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.
 */
package org.trustedanalytics.examples.hbase.services;

import org.apache.hadoop.hbase.HColumnDescriptor;
import org.apache.hadoop.hbase.HTableDescriptor;
import org.apache.hadoop.hbase.client.Result;
import org.springframework.stereotype.Service;
import org.trustedanalytics.examples.hbase.model.ColumnFamilyValue;
import org.trustedanalytics.examples.hbase.model.ColumnValue;
import org.trustedanalytics.examples.hbase.model.RowValue;
import org.trustedanalytics.examples.hbase.model.TableDescription;

import java.util.Collection;
import java.util.List;
import java.util.NavigableMap;
import java.util.stream.Collectors;

@Service
public class ConversionsService {

    public TableDescription constructTableDescription(HTableDescriptor htd) {
        TableDescription result = null;

        if (htd != null) {
            String name = htd.getNameAsString();
            Collection<HColumnDescriptor> families = htd.getFamilies();

            List<String> familiesNames = families.stream().map(HColumnDescriptor::getNameAsString)
                    .collect(Collectors.toList());

            result = new TableDescription(name, familiesNames);
        }

        return result;
    }

    public RowValue constructRowValue(Result r) {
        RowValue result = null;

        if (r != null && !r.isEmpty()) {
            String rowKey = new String(r.getRow());

            // Map<family,Map<qualifier,value>>
            NavigableMap<byte[], NavigableMap<byte[], byte[]>> valueMap = r.getNoVersionMap();

            List<ColumnFamilyValue> families = valueMap.entrySet().stream()
                    .map(b -> new ColumnFamilyValue(new String(b.getKey()), constructColumnValues(b.getValue())))
                    .collect(Collectors.toList());

            result = new RowValue(rowKey, families);
        }

        return result;
    }

    private List<ColumnValue> constructColumnValues(NavigableMap<byte[], byte[]> map) {
        List<ColumnValue> result = null;
        result = map.entrySet().stream().map(i -> new ColumnValue(new String(i.getKey()), new String(i.getValue())))
                .collect(Collectors.toList());
        return result;
    }

}