com.tesora.dve.db.NativeResultHandler.java Source code

Introduction

Here is the source code for com.tesora.dve.db.NativeResultHandler.java

Source

package com.tesora.dve.db;

/*
 * #%L
 * Tesora Inc.
 * Database Virtualization Engine
 * %%
 * Copyright (C) 2011 - 2014 Tesora Inc.
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License, version 3,
 * as published by the Free Software Foundation.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 * 
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 * #L%
 */

import io.netty.util.CharsetUtil;

import java.io.Serializable;
import java.math.BigDecimal;
import java.math.BigInteger;
import java.sql.Time;
import java.sql.Timestamp;
import java.sql.Types;
import java.util.Date;

import com.tesora.dve.exceptions.PEException;
import com.tesora.dve.resultset.ColumnMetadata;

public abstract class NativeResultHandler implements Serializable {

    private static final long serialVersionUID = 1L;

    public String getObjectAsString(ColumnMetadata uc, Object obj) throws PEException {
        String colStr;
        byte[] b = getObjectAsBytes(uc, obj);
        //      try {
        if (uc.getDataType() == Types.BLOB || uc.getDataType() == Types.LONGVARBINARY) {
            colStr = new String(b, CharsetUtil.ISO_8859_1);
        } else {
            colStr = new String(b, CharsetUtil.UTF_8);
        }
        //      } catch (UnsupportedEncodingException e) {
        //         // Just return the default encoding
        //         colStr = new String(b);
        //      }
        return colStr;
    }

    public byte[] getObjectAsBytes(ColumnMetadata uc, Object obj) {
        byte[] ret = null;
        if (obj == null)
            return null;

        if (obj instanceof String) {
            ret = getStringAsBytes(uc, obj);
        } else if (obj instanceof byte[]) {
            ret = getBytesAsBytes(uc, obj);
        } else if (obj instanceof Byte) {
            ret = getByteAsBytes(uc, obj);
        } else if (obj instanceof Boolean) {
            ret = getBooleanAsBytes(uc, obj);
        } else if (obj instanceof Short) {
            ret = getShortAsBytes(uc, obj);
        } else if (obj instanceof Integer) {
            ret = getIntegerAsBytes(uc, obj);
        } else if (obj instanceof Long) {
            ret = getLongAsBytes(uc, obj);
        } else if (obj instanceof BigInteger) {
            ret = getBigIntegerAsBytes(uc, obj);
        } else if (obj instanceof BigDecimal) {
            ret = getBigDecimalAsBytes(uc, obj);
        } else if (obj instanceof Float) {
            ret = getFloatAsBytes(uc, obj);
        } else if (obj instanceof Double) {
            ret = getDoubleAsBytes(uc, obj);
        } else if (obj instanceof Date) {
            ret = getDateAsBytes(uc, obj);
        } else if (obj instanceof Time) {
            ret = getTimeAsBytes(uc, obj);
        } else if (obj instanceof Timestamp) {
            ret = getTimestampAsBytes(uc, obj);
        } else {
            throw new IllegalArgumentException("Unhandled type " + obj.getClass().getName());
        }

        return ret;
    }

    protected byte[] getBytesAsBytes(ColumnMetadata uc, Object obj) {
        return (byte[]) obj;
    }

    protected byte[] getByteAsBytes(ColumnMetadata uc, Object obj) {
        return obj.toString().getBytes();
    }

    protected byte[] getBooleanAsBytes(ColumnMetadata uc, Object obj) {
        return obj.toString().getBytes();
    }

    protected byte[] getShortAsBytes(ColumnMetadata uc, Object obj) {
        return obj.toString().getBytes();
    }

    protected byte[] getIntegerAsBytes(ColumnMetadata uc, Object obj) {
        return obj.toString().getBytes();
    }

    protected byte[] getLongAsBytes(ColumnMetadata uc, Object obj) {
        return obj.toString().getBytes();
    }

    protected byte[] getBigIntegerAsBytes(ColumnMetadata uc, Object obj) {
        return obj.toString().getBytes();
    }

    protected byte[] getBigDecimalAsBytes(ColumnMetadata uc, Object obj) {
        return obj.toString().getBytes();
    }

    protected byte[] getStringAsBytes(ColumnMetadata uc, Object obj) {
        return ((String) obj).getBytes();
    }

    protected byte[] getFloatAsBytes(ColumnMetadata uc, Object obj) {
        return obj.toString().getBytes();
    }

    protected byte[] getDoubleAsBytes(ColumnMetadata uc, Object obj) {
        return obj.toString().getBytes();
    }

    protected byte[] getDateAsBytes(ColumnMetadata uc, Object obj) {
        return obj.toString().getBytes();
    }

    protected byte[] getTimeAsBytes(ColumnMetadata uc, Object obj) {
        return obj.toString().getBytes();
    }

    protected byte[] getTimestampAsBytes(ColumnMetadata uc, Object obj) {
        return obj.toString().getBytes();
    }

}