fqlite.base.SqliteInternalRow Maven / Gradle / Ivy

Go to download
package fqlite.base;

import java.nio.BufferUnderflowException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Set;

public class SqliteInternalRow {
    private List rowData = new ArrayList<>();
    private long offset = 0L;
    private String tableName = "";
    private String recordType = "";
    private String lineSuffix = "";
    private Map colIdx = null;

    void setColumnNamesMap(Map colIdx) {
        this.colIdx = colIdx;
    }

    public Set getColumnNames() {
        if (colIdx != null) {
            return colIdx.keySet();
        }
        return null;
    }

    public int getColumnIndex(String colName) {
        if (colIdx != null) {
            if (colIdx.containsKey(colName)) {
                return colIdx.get(colName);
            }
        }
        return -1;
    }

    public void append(SqliteElementData elementData) {
        rowData.add(elementData);
    }
    
    public List getRowData() {
        return rowData;
    }
    
    public long getOffset() {
        return offset;
    }
    
    public void setOffset(long offset) {
        this.offset = offset;
    }
    
    public String getTableName() {
        return tableName;
    }
    
    public void setTableName(String tableName) {
        this.tableName = tableName;
    }
    
    public String getRecordType() {
        return recordType;
    }
    
    public void setRecordType(String recordType) {
        this.recordType = recordType;
    }
    
    public String getLineSuffix() {
        return lineSuffix;
    }
    
    public void setLineSuffix(String lineSuffix) {
        this.lineSuffix = lineSuffix;
    }
    
    public String toString() {
        StringBuilder builder = new StringBuilder();
        
        builder.append(tableName).append(";");
        builder.append(recordType).append(";");
        builder.append(offset);
        for (SqliteElementData elementData : rowData) {
            builder.append(";").append(elementData.toString());
        }
        builder.append(lineSuffix);
        builder.append("\n");
        
        return builder.toString();
    }

    public long getIntValue(String col) {
        if (colIdx == null || !colIdx.containsKey(col)) {
            return Long.MIN_VALUE;
        }
        int idx = colIdx.get(col);
        if (idx >= rowData.size()) {
            return Long.MIN_VALUE;
        }
        try {
            return rowData.get(idx).getIntValue();
        } catch (BufferUnderflowException|ArrayIndexOutOfBoundsException e) {
        }
        return Long.MIN_VALUE;
    }

    public String getTextValue(String col) {
        if (colIdx == null || !colIdx.containsKey(col)) {
            return null;
        }
        int idx = colIdx.get(col);
        if (idx >= rowData.size()) {
            return null;
        }
        return rowData.get(idx).getTextValue();
    }

    public double getFloatValue(String col) {
        if (colIdx == null || !colIdx.containsKey(col)) {
            return Double.NaN;
        }
        int idx = colIdx.get(col);
        if (idx >= rowData.size()) {
            return Double.NaN;
        }
        return rowData.get(idx).getFloatValue();
    }

    public byte[] getBlobValue(String col) {
        if (colIdx == null || !colIdx.containsKey(col)) {
            return null;
        }
        int idx = colIdx.get(col);
        if (idx >= rowData.size()) {
            return null;
        }
        return rowData.get(idx).getBlobValue();
    }

    public boolean isDeletedRow() {
        if (recordType != null) {
            if (recordType.contains(Global.DELETED_RECORD_IN_PAGE) ||
                recordType.contains(Global.FREELIST_ENTRY) ||
                recordType.contains(Global.UNALLOCATED_SPACE)) {
                return true;
            }
        }
        return false;
    }

    public SqliteRow decodeRow() {
        SqliteRow result = new SqliteRow();
        result.setTableName(tableName);
        result.setDeletedRow(isDeletedRow());
        result.setColumnNamesMap(colIdx);
        Object [] data = new Object[rowData.size()];
        for (int i = 0; i < data.length; i++) {
            SqliteElementData d = rowData.get(i);
            if (d != null) {
                data[i] = d.getObject();
            }
        }
        result.setRowData(data);
        
        return result;
    }

}