ColumnType.cs

Go to the documentation of this file.

namespace kinetica.Records;

    public enum ColumnBaseType
    {
        Integer,
        Long,
        Float,
        Double,
        String,
        Bytes
    }

    public enum ColumnType
    {
        // Base types
        Integer,
        Long,
        Float,
        Double,
        String,
        Bytes,

        // Integer subtypes
        Boolean,
        Int8,
        Int16,

        // Long subtypes
        Timestamp,

        // String subtypes
        Date,
        DateTime,
        Time,
        Decimal,
        Ipv4,
        Uuid,
        Char1,
        Char2,
        Char4,
        Char8,
        Char16,
        Char32,
        Char64,
        Char128,
        Char256,

        // Bytes subtypes
        Wkt,

        // Special types
        Array,
        Vector,
        Json
    }

    public static class ColumnTypeExtensions
    {
        public static ColumnBaseType GetBaseType(this ColumnType columnType)
        {
            return columnType switch
            {
                ColumnType.Integer or ColumnType.Boolean or ColumnType.Int8 or ColumnType.Int16
                    => ColumnBaseType.Integer,

                ColumnType.Long or ColumnType.Timestamp
                    => ColumnBaseType.Long,

                ColumnType.Float
                    => ColumnBaseType.Float,

                ColumnType.Double
                    => ColumnBaseType.Double,

                ColumnType.String or ColumnType.Date or ColumnType.DateTime or ColumnType.Time or
                ColumnType.Decimal or ColumnType.Ipv4 or ColumnType.Uuid or
                ColumnType.Char1 or ColumnType.Char2 or ColumnType.Char4 or ColumnType.Char8 or
                ColumnType.Char16 or ColumnType.Char32 or ColumnType.Char64 or ColumnType.Char128 or
                ColumnType.Char256 or ColumnType.Array or ColumnType.Vector or ColumnType.Json
                    => ColumnBaseType.String,

                ColumnType.Bytes or ColumnType.Wkt
                    => ColumnBaseType.Bytes,

                _ => ColumnBaseType.String
            };
        }

        public static string GetAvroTypeName(this ColumnBaseType baseType)
        {
            return baseType switch
            {
                ColumnBaseType.Integer => "int",
                ColumnBaseType.Long => "long",
                ColumnBaseType.Float => "float",
                ColumnBaseType.Double => "double",
                ColumnBaseType.String => "string",
                ColumnBaseType.Bytes => "bytes",
                _ => "string"
            };
        }

        public static string GetAvroTypeName(this ColumnType columnType)
        {
            return columnType.GetBaseType().GetAvroTypeName();
        }

        public static ColumnType FromAvroAndProperties(string avroType, IEnumerable<string>? properties)
        {
            if (properties != null)
            {
                foreach (var prop in properties)
                {
                    var result = prop.ToLowerInvariant() switch
                    {
                        "boolean" => ColumnType.Boolean,
                        "int8" => ColumnType.Int8,
                        "int16" => ColumnType.Int16,
                        "timestamp" => ColumnType.Timestamp,
                        "date" => ColumnType.Date,
                        "datetime" => ColumnType.DateTime,
                        "time" => ColumnType.Time,
                        "decimal" => ColumnType.Decimal,
                        "ipv4" => ColumnType.Ipv4,
                        "uuid" => ColumnType.Uuid,
                        "char1" => ColumnType.Char1,
                        "char2" => ColumnType.Char2,
                        "char4" => ColumnType.Char4,
                        "char8" => ColumnType.Char8,
                        "char16" => ColumnType.Char16,
                        "char32" => ColumnType.Char32,
                        "char64" => ColumnType.Char64,
                        "char128" => ColumnType.Char128,
                        "char256" => ColumnType.Char256,
                        "wkt" => ColumnType.Wkt,
                        "array" => ColumnType.Array,
                        "vector" => ColumnType.Vector,
                        "json" => ColumnType.Json,
                        _ => (ColumnType?)null
                    };

                    if (result.HasValue)
                        return result.Value;
                }
            }

            // Fall back to base type
            return avroType.ToLowerInvariant() switch
            {
                "int" => ColumnType.Integer,
                "long" => ColumnType.Long,
                "float" => ColumnType.Float,
                "double" => ColumnType.Double,
                "string" => ColumnType.String,
                "bytes" => ColumnType.Bytes,
                _ => ColumnType.String
            };
        }

        public static string? GetPropertyName(this ColumnType columnType)
        {
            return columnType switch
            {
                ColumnType.Boolean => "boolean",
                ColumnType.Int8 => "int8",
                ColumnType.Int16 => "int16",
                ColumnType.Timestamp => "timestamp",
                ColumnType.Date => "date",
                ColumnType.DateTime => "datetime",
                ColumnType.Time => "time",
                ColumnType.Decimal => "decimal",
                ColumnType.Ipv4 => "ipv4",
                ColumnType.Uuid => "uuid",
                ColumnType.Char1 => "char1",
                ColumnType.Char2 => "char2",
                ColumnType.Char4 => "char4",
                ColumnType.Char8 => "char8",
                ColumnType.Char16 => "char16",
                ColumnType.Char32 => "char32",
                ColumnType.Char64 => "char64",
                ColumnType.Char128 => "char128",
                ColumnType.Char256 => "char256",
                ColumnType.Wkt => "wkt",
                ColumnType.Array => "array",
                ColumnType.Vector => "vector",
                ColumnType.Json => "json",
                _ => null
            };
        }
    }