org.redkalex.source.pgsql.PgColumnDesc Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of redkale-plugins Show documentation
Redkale-Plugins -- java framework
There is a newer version: 2.7.7
/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package org.redkalex.source.pgsql;

import java.io.Serializable;
import java.math.BigDecimal;
import static java.nio.charset.StandardCharsets.UTF_8;
import java.sql.*;
import java.time.*;
import static java.time.format.DateTimeFormatter.*;
import java.time.format.DateTimeParseException;
import java.util.UUID;
import org.redkale.convert.json.JsonConvert;
import org.redkale.util.Utility;
import static org.redkalex.source.pgsql.PgsqlLDataSource.*;

/**
 *
 * @author zhangjx
 */
public class PgColumnDesc {

    final String name;

    final PgOid type;

    public PgColumnDesc(String name, PgOid type) {
        this.name = name;
        this.type = type;
    }

    public String getName() {
        return name;
    }

    public PgOid getType() {
        return type;
    }

    public Serializable getObject(byte[] value) {
        if (value == null) return null;
        switch (type) {
            case TEXT: // fallthrough
            case CHAR: // fallthrough
            case BPCHAR: // fallthrough
            case VARCHAR: return new String(value, UTF_8);
            case INT2: return Short.valueOf(new String(value, UTF_8));
            case INT4: return Integer.valueOf(new String(value, UTF_8));
            case INT8: return Long.valueOf(new String(value, UTF_8));
            case FLOAT4: return Float.valueOf(new String(value, UTF_8));
            case NUMERIC: // fallthrough
            case FLOAT8: return new BigDecimal(new String(value, UTF_8));
            case BIT: return Utility.hexToBin(new String(value, 2, value.length - 2, UTF_8))[0];
            case BIT_ARRAY:// fallthrough
            case BYTEA: return Utility.hexToBin(new String(value, 2, value.length - 2, UTF_8));
            case DATE: String date = new String(value, UTF_8);
                try {
                    return Date.valueOf(LocalDate.parse(date, ISO_LOCAL_DATE));
                } catch (DateTimeParseException e) {
                    throw new RuntimeException("Invalid date: " + date);
                }
            case TIMETZ: return Time.valueOf(OffsetTime.parse(new String(value, UTF_8), TIMEZ_FORMAT).toLocalTime());
            case TIME: return Time.valueOf(LocalTime.parse(new String(value, UTF_8), ISO_LOCAL_TIME));
            case TIMESTAMP: return Timestamp.valueOf(LocalDateTime.parse(new String(value, UTF_8), TIMESTAMP_FORMAT));
            case TIMESTAMPTZ: return Timestamp.valueOf(OffsetDateTime.parse(new String(value, UTF_8), TIMESTAMPZ_FORMAT).toLocalDateTime());
            case UUID: return UUID.fromString(new String(value, UTF_8));
            case BOOL: return 't' == value[0];

            case INT2_ARRAY:
            case INT4_ARRAY:
            case INT8_ARRAY:
            case FLOAT4_ARRAY:
            case FLOAT8_ARRAY:
            case TEXT_ARRAY:
            case CHAR_ARRAY:
            case BPCHAR_ARRAY:
            case VARCHAR_ARRAY:
            case TIMESTAMP_ARRAY:
            case TIMESTAMPTZ_ARRAY:
            case TIMETZ_ARRAY:
            case TIME_ARRAY:
            case BOOL_ARRAY:
                throw new IllegalStateException("Unsupported array type: " + type);
            default:
                throw new IllegalStateException("Unsupported array type: " + type);
        }
    }

    @Override
    public String toString() {
        return JsonConvert.root().convertTo(this);
    }
}