public void Write(ArrayType arrayType, object value)
{
var collection = (IList)value;
writer.Write7BitEncodedInt(collection.Count);
for (var i = 0; i < collection.Count; i++)
{
Write(arrayType.UnderlyingType, collection[i]);
}
}
public void WriteValue(object data, ClickHouseType databaseType)
{
switch (databaseType.TypeCode)
{
case ClickHouseTypeCode.UInt8:
writer.Write(Convert.ToByte(data));
break;
case ClickHouseTypeCode.UInt16:
writer.Write(Convert.ToUInt16(data));
break;
case ClickHouseTypeCode.UInt32:
writer.Write(Convert.ToUInt32(data));
break;
case ClickHouseTypeCode.UInt64:
writer.Write(Convert.ToUInt64(data));
break;
case ClickHouseTypeCode.Int8:
writer.Write(Convert.ToSByte(data));
break;
case ClickHouseTypeCode.Int16:
writer.Write(Convert.ToInt16(data));
break;
case ClickHouseTypeCode.Int32:
writer.Write(Convert.ToInt32(data));
break;
case ClickHouseTypeCode.Int64:
writer.Write(Convert.ToInt64(data));
break;
case ClickHouseTypeCode.Float32:
writer.Write(Convert.ToSingle(data));
break;
case ClickHouseTypeCode.Float64:
writer.Write(Convert.ToDouble(data));
break;
case ClickHouseTypeCode.Decimal:
var dti = (DecimalType)databaseType;
var value = new BigInteger(MathUtils.ShiftDecimalPlaces(Convert.ToDecimal(data), dti.Scale));
var dbytes = new byte[dti.Size];
value.ToByteArray().CopyTo(dbytes, 0);
writer.Write(dbytes);
break;
case ClickHouseTypeCode.String:
writer.Write(Convert.ToString(data));
break;
case ClickHouseTypeCode.FixedString:
var @string = (string)data;
var stringInfo = (FixedStringType)databaseType;
var stringBytes = new byte[stringInfo.Length];
Encoding.UTF8.GetBytes(@string, 0, @string.Length, stringBytes, 0);
writer.Write(stringBytes);
break;
case ClickHouseTypeCode.Array:
var arrayTypeInfo = (ArrayType)databaseType;
var collection = (IList)data;
writer.Write7BitEncodedInt(collection.Count);
for (var i = 0; i < collection.Count; i++)
{
WriteValue(collection[i], arrayTypeInfo.UnderlyingType);
}
break;
case ClickHouseTypeCode.Nullable:
var nullableTypeInfo = (NullableType)databaseType;
if (data == null || data is DBNull)
{
writer.Write((byte)1);
}
else
{
writer.Write((byte)0);
WriteValue(data, nullableTypeInfo.UnderlyingType);
}
break;
case ClickHouseTypeCode.Tuple:
var tupleType = (TupleType)databaseType;
var tuple = (ITuple)data;
for (var i = 0; i < tuple.Length; i++)
{
WriteValue(tuple[i], tupleType.UnderlyingTypes[i]);
}
break;
case ClickHouseTypeCode.Nested:
var nestedType = (NestedType)databaseType;
var tuples = ((IEnumerable)data).Cast <ITuple>().ToList();
writer.Write7BitEncodedInt(tuples.Count);
foreach (var ntuple in tuples)
{
for (int i = 0; i < ntuple.Length; i++)
{
WriteValue(ntuple[i], nestedType.UnderlyingTypes[i]);
}
}
break;
case ClickHouseTypeCode.UUID:
var guid = ExtractGuid(data);
var bytes = guid.ToByteArray();
Array.Reverse(bytes, 8, 8);
writer.Write(bytes, 6, 2);
writer.Write(bytes, 4, 2);
writer.Write(bytes, 0, 4);
writer.Write(bytes, 8, 8);
break;
case ClickHouseTypeCode.IPv4:
var address4 = ExtractIPAddress(data);
if (address4.AddressFamily != System.Net.Sockets.AddressFamily.InterNetwork)
{
throw new ArgumentException($"Expected IPv4, got {address4.ToString()}");
}
var ipv4bytes = address4.GetAddressBytes();
Array.Reverse(ipv4bytes);
writer.Write(ipv4bytes, 0, ipv4bytes.Length);
break;
case ClickHouseTypeCode.IPv6:
var address6 = ExtractIPAddress(data);
if (address6.AddressFamily != System.Net.Sockets.AddressFamily.InterNetworkV6)
{
throw new ArgumentException($"Expected IPv4, got {address6.ToString()}");
}
var ipv6bytes = address6.GetAddressBytes();
writer.Write(ipv6bytes, 0, ipv6bytes.Length);
break;
case ClickHouseTypeCode.Date:
var days = (ushort)(((DateTime)data).Date - TypeConverter.DateTimeEpochStart).TotalDays;
writer.Write(days);
break;
case ClickHouseTypeCode.DateTime:
var dtType = (DateTimeType)databaseType;
var dto = dtType.ToDateTimeOffset((DateTime)data);
var seconds = (uint)(dto.UtcDateTime - TypeConverter.DateTimeEpochStart).TotalSeconds;
writer.Write(seconds);
break;
case ClickHouseTypeCode.DateTime64:
var dt64type = (DateTime64Type)databaseType;
var dto64 = dt64type.ToDateTimeOffset((DateTime)data);
var ticks = (dto64.UtcDateTime - TypeConverter.DateTimeEpochStart).Ticks;
// 7 is a 'magic constant' - Log10 of TimeSpan.TicksInSecond
writer.Write(MathUtils.ShiftDecimalPlaces(ticks, dt64type.Scale - 7));
break;
case ClickHouseTypeCode.Nothing:
break;
case ClickHouseTypeCode.Enum8:
var enum8TypeInfo = (EnumType)databaseType;
var enum8Index = data is string enum8Str ? (sbyte)enum8TypeInfo.Lookup(enum8Str) : Convert.ToSByte(data);
writer.Write(enum8Index);
break;
case ClickHouseTypeCode.Enum16:
var enum16TypeInfo = (EnumType)databaseType;
var enum16Index = data is string enum16Str ? (short)enum16TypeInfo.Lookup(enum16Str) : Convert.ToInt16(data);
writer.Write(enum16Index);
break;
case ClickHouseTypeCode.LowCardinality:
var lcCardinality = (LowCardinalityType)databaseType;
WriteValue(data, lcCardinality.UnderlyingType);
break;
default:
throw new NotImplementedException($"{databaseType.TypeCode} not supported yet");
}
}
