net.e6tech.elements.cassandra.driver.v4.ColumnMetadataV4 Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of elements-cassandra Show documentation
elements-cassandra
The newest version!
/*
 * Copyright 2015-2019 Futeh Kao
 *
 * 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 net.e6tech.elements.cassandra.driver.v4;

import net.e6tech.elements.cassandra.driver.datatype.DataType;
import net.e6tech.elements.cassandra.driver.datatype.ListType;
import net.e6tech.elements.cassandra.driver.datatype.MapType;
import net.e6tech.elements.cassandra.driver.datatype.SetType;
import net.e6tech.elements.cassandra.driver.metadata.AbstractColumnMetadata;
import net.e6tech.elements.cassandra.generator.Generator;
import net.e6tech.elements.cassandra.generator.TypeDescriptorImpl;

import java.math.BigDecimal;
import java.math.BigInteger;
import java.net.InetAddress;
import java.time.Instant;
import java.time.LocalDate;
import java.time.LocalTime;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.UUID;

import static com.datastax.oss.protocol.internal.ProtocolConstants.DataType.*;

public class ColumnMetadataV4 extends AbstractColumnMetadata {

    public ColumnMetadataV4(Generator generator, com.datastax.oss.driver.api.core.metadata.schema.ColumnMetadata metadata) {
        setName(metadata.getName().toString());
        com.datastax.oss.driver.api.core.type.DataType dataType = metadata.getType();
        DataType d = createDataType(generator, dataType);
        setDataType(d);
    }

    protected static DataType createDataType(Generator generator, com.datastax.oss.driver.api.core.type.DataType dataType) {
        TypeDescriptorImpl impl = new TypeDescriptorImpl();
        Class type;
        switch (dataType.getProtocolCode()) {
            case ASCII: type = String.class; break;
            case BIGINT: type = Long.class; break;
            case BLOB: type = byte[].class; break;
            case BOOLEAN: type = Boolean.class; break;
            case COUNTER: type = Long.class; break;
            case DECIMAL: type = BigDecimal.class; break;
            case DOUBLE: type = Double.class; break;
            case FLOAT: type = Float.class; break;
            case INT: type = Integer.class; break;
            case TIMESTAMP: type = Instant.class; break;
            case UUID: type = UUID.class; break;
            case VARCHAR: type = String.class; break;
            case VARINT: type = BigInteger.class; break;
            case TIMEUUID: type = UUID.class; impl.setTimeBased(true); break;
            case INET: type = InetAddress.class; break;
            case DATE: type = LocalDate.class; break;
            case TIME: type = LocalTime.class; break;
            case SMALLINT: type = Short.class; break;
            case TINYINT: type = Byte.class; break;
            case LIST: type = List.class; break;
            case MAP: type = Map.class; break;
            case SET: type = Set.class; break;
            default:
                type = null;
        }
        if (type == null)
            return null;

        DataType d = DataType.create(generator, type, impl);

        if (dataType instanceof com.datastax.oss.driver.api.core.type.ListType) {
            com.datastax.oss.driver.api.core.type.ListType l = (com.datastax.oss.driver.api.core.type.ListType) dataType;
            impl.setFrozen(l.isFrozen());
            ((ListType) d).setComponentType(createDataType(generator, l.getElementType()));
        } else if (dataType instanceof com.datastax.oss.driver.api.core.type.SetType) {
            com.datastax.oss.driver.api.core.type.SetType l = (com.datastax.oss.driver.api.core.type.SetType) dataType;
            impl.setFrozen(l.isFrozen());
            ((SetType) d).setComponentType(createDataType(generator, l.getElementType()));
        } else if (dataType instanceof com.datastax.oss.driver.api.core.type.MapType) {
            com.datastax.oss.driver.api.core.type.MapType l = (com.datastax.oss.driver.api.core.type.MapType) dataType;
            impl.setFrozen(l.isFrozen());
            ((MapType) d).setKeyType(createDataType(generator, l.getKeyType()));
            ((MapType) d).setValueType(createDataType(generator, l.getValueType()));
        }

        return d;
    }
}