com.github.sadikovi.netflowlib.record.RecordMaterializer.java Source code

Introduction

Here is the source code for com.github.sadikovi.netflowlib.record.RecordMaterializer.java

Source

/*
 * Copyright 2016 sadikovi
 *
 * 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 com.github.sadikovi.netflowlib.record;

import io.netty.buffer.ByteBuf;

import com.github.sadikovi.netflowlib.predicate.Columns.Column;
import com.github.sadikovi.netflowlib.predicate.Inspectors.ValueInspector;
import com.github.sadikovi.netflowlib.predicate.Operators.FilterPredicate;

/**
 * [[RecordMaterializer]] interface provides all necessary methods to parse single record and return
 * either array of values that match list and order of pruned columns, or null, if record does not
 * pass predicate. Predicate check is optional and should depend on implementation.
 */
public abstract class RecordMaterializer {
    RecordMaterializer() {
    }

    public abstract Object[] processRecord(ByteBuf buffer);

    /** Read buffer bytes sequence for column offset */
    public Object readField(Column column, ByteBuf buffer) {
        Class<?> type = column.getColumnType();
        if (type.equals(Byte.class)) {
            return buffer.getByte(column.getColumnOffset());
        } else if (type.equals(Short.class)) {
            return buffer.getUnsignedByte(column.getColumnOffset());
        } else if (type.equals(Integer.class)) {
            return buffer.getUnsignedShort(column.getColumnOffset());
        } else if (type.equals(Long.class)) {
            return buffer.getUnsignedInt(column.getColumnOffset());
        } else {
            throw new UnsupportedOperationException("Unsupported read type " + type);
        }
    }

    public void updateValueInspector(Column column, ByteBuf buffer, ValueInspector vi) {
        Class<?> type = column.getColumnType();
        if (type.equals(Byte.class)) {
            vi.update(buffer.getByte(column.getColumnOffset()));
        } else if (type.equals(Short.class)) {
            vi.update(buffer.getUnsignedByte(column.getColumnOffset()));
        } else if (type.equals(Integer.class)) {
            vi.update(buffer.getUnsignedShort(column.getColumnOffset()));
        } else if (type.equals(Long.class)) {
            vi.update(buffer.getUnsignedInt(column.getColumnOffset()));
        } else {
            throw new UnsupportedOperationException("Unsupported read type " + type);
        }
    }
}