TypeBuilder.cs

Go to the documentation of this file.

using System;
using System.Collections.Generic;

namespace kinetica.Records;

    public sealed class TypeBuilder
    {
        private readonly string _label;
        private readonly List<Column> _columns;
        private (string Name, ColumnType Type, List<string> Properties)? _currentColumn;

        public TypeBuilder(string label)
        {
            _label = label ?? throw new ArgumentNullException(nameof(label));
            _columns = new List<Column>();
            _currentColumn = null;
        }

        private void FinishCurrentColumn()
        {
            if (_currentColumn.HasValue)
            {
                var (name, type, props) = _currentColumn.Value;
                _columns.Add(Column.WithProperties(name, type, props));
                _currentColumn = null;
            }
        }

        #region Add Column Methods

        public TypeBuilder AddColumn(Column column)
        {
            FinishCurrentColumn();
            _columns.Add(column);
            return this;
        }

        public TypeBuilder AddIntColumn(string name)
        {
            FinishCurrentColumn();
            _currentColumn = (name, ColumnType.Integer, new List<string>());
            return this;
        }

        public TypeBuilder AddLongColumn(string name)
        {
            FinishCurrentColumn();
            _currentColumn = (name, ColumnType.Long, new List<string>());
            return this;
        }

        public TypeBuilder AddFloatColumn(string name)
        {
            FinishCurrentColumn();
            _currentColumn = (name, ColumnType.Float, new List<string>());
            return this;
        }

        public TypeBuilder AddDoubleColumn(string name)
        {
            FinishCurrentColumn();
            _currentColumn = (name, ColumnType.Double, new List<string>());
            return this;
        }

        public TypeBuilder AddStringColumn(string name)
        {
            FinishCurrentColumn();
            _currentColumn = (name, ColumnType.String, new List<string>());
            return this;
        }

        public TypeBuilder AddBytesColumn(string name)
        {
            FinishCurrentColumn();
            _currentColumn = (name, ColumnType.Bytes, new List<string>());
            return this;
        }

        public TypeBuilder AddTimestampColumn(string name)
        {
            FinishCurrentColumn();
            _currentColumn = (name, ColumnType.Timestamp, new List<string> { "timestamp" });
            return this;
        }

        public TypeBuilder AddDateColumn(string name)
        {
            FinishCurrentColumn();
            _currentColumn = (name, ColumnType.Date, new List<string> { "date" });
            return this;
        }

        public TypeBuilder AddDateTimeColumn(string name)
        {
            FinishCurrentColumn();
            _currentColumn = (name, ColumnType.DateTime, new List<string> { "datetime" });
            return this;
        }

        public TypeBuilder AddTimeColumn(string name)
        {
            FinishCurrentColumn();
            _currentColumn = (name, ColumnType.Time, new List<string> { "time" });
            return this;
        }

        public TypeBuilder AddBooleanColumn(string name)
        {
            FinishCurrentColumn();
            _currentColumn = (name, ColumnType.Boolean, new List<string> { "boolean" });
            return this;
        }

        public TypeBuilder AddInt8Column(string name)
        {
            FinishCurrentColumn();
            _currentColumn = (name, ColumnType.Int8, new List<string> { "int8" });
            return this;
        }

        public TypeBuilder AddInt16Column(string name)
        {
            FinishCurrentColumn();
            _currentColumn = (name, ColumnType.Int16, new List<string> { "int16" });
            return this;
        }

        public TypeBuilder AddDecimalColumn(string name)
        {
            FinishCurrentColumn();
            _currentColumn = (name, ColumnType.Decimal, new List<string> { "decimal" });
            return this;
        }

        public TypeBuilder AddIpv4Column(string name)
        {
            FinishCurrentColumn();
            _currentColumn = (name, ColumnType.Ipv4, new List<string> { "ipv4" });
            return this;
        }

        public TypeBuilder AddUuidColumn(string name)
        {
            FinishCurrentColumn();
            _currentColumn = (name, ColumnType.Uuid, new List<string> { "uuid" });
            return this;
        }

        public TypeBuilder AddCharColumn(string name, int length)
        {
            FinishCurrentColumn();
            var (type, prop) = length switch
            {
                1 => (ColumnType.Char1, "char1"),
                2 => (ColumnType.Char2, "char2"),
                4 => (ColumnType.Char4, "char4"),
                8 => (ColumnType.Char8, "char8"),
                16 => (ColumnType.Char16, "char16"),
                32 => (ColumnType.Char32, "char32"),
                64 => (ColumnType.Char64, "char64"),
                128 => (ColumnType.Char128, "char128"),
                256 => (ColumnType.Char256, "char256"),
                _ => throw new ArgumentException($"Invalid char length: {length}. Must be 1, 2, 4, 8, 16, 32, 64, 128, or 256.", nameof(length))
            };
            _currentColumn = (name, type, new List<string> { prop });
            return this;
        }

        public TypeBuilder AddWktColumn(string name)
        {
            FinishCurrentColumn();
            _currentColumn = (name, ColumnType.Wkt, new List<string> { "wkt" });
            return this;
        }

        public TypeBuilder AddJsonColumn(string name)
        {
            FinishCurrentColumn();
            _currentColumn = (name, ColumnType.Json, new List<string> { "json" });
            return this;
        }

        #endregion

        #region Property Modifiers

        public TypeBuilder PrimaryKey()
        {
            if (_currentColumn.HasValue)
            {
                var (name, type, props) = _currentColumn.Value;
                if (!props.Contains("primary_key"))
                    props.Add("primary_key");
                _currentColumn = (name, type, props);
            }
            return this;
        }

        public TypeBuilder ShardKey()
        {
            if (_currentColumn.HasValue)
            {
                var (name, type, props) = _currentColumn.Value;
                if (!props.Contains("shard_key"))
                    props.Add("shard_key");
                _currentColumn = (name, type, props);
            }
            return this;
        }

        public TypeBuilder Nullable()
        {
            if (_currentColumn.HasValue)
            {
                var (name, type, props) = _currentColumn.Value;
                if (!props.Contains("nullable"))
                    props.Add("nullable");
                _currentColumn = (name, type, props);
            }
            return this;
        }

        public TypeBuilder Data()
        {
            if (_currentColumn.HasValue)
            {
                var (name, type, props) = _currentColumn.Value;
                if (!props.Contains("data"))
                    props.Add("data");
                _currentColumn = (name, type, props);
            }
            return this;
        }

        public TypeBuilder WithProperty(string property)
        {
            if (_currentColumn.HasValue && !string.IsNullOrEmpty(property))
            {
                var (name, type, props) = _currentColumn.Value;
                if (!props.Contains(property))
                    props.Add(property);
                _currentColumn = (name, type, props);
            }
            return this;
        }

        #endregion

        #region Build Methods

        public Type Build()
        {
            FinishCurrentColumn();
            return new Type(_label, _columns);
        }

        #endregion
    }