src/Npgsql/Internal/Converters/Temporal/DateConverters.cs - test-logs - dev-npgsql/npgsql

src/Npgsql/Internal/Converters/Temporal/DateConverters.cs
91 строка3 KB
Nino Floris
Remove PgBufferedConverter virtual dispatch hop (#6574)
15 май 2026, 18:24
15 май 2026, 18:2478d37a1
using System;
using Npgsql.Properties;

// ReSharper disable once CheckNamespace
namespace Npgsql.Internal.Converters;

sealed class DateOnlyDateConverter(bool dateTimeInfinityConversions) : PgBufferedConverter<DateOnly>
{
    static readonly DateOnly BaseValue = new(2000, 1, 1);

    public override bool CanConvert(DataFormat format, out BufferRequirements bufferRequirements)
    {
        bufferRequirements = BufferRequirements.CreateFixedSize(sizeof(int));
        return format is DataFormat.Binary;
    }

    public override DateOnly Read(PgReader reader)
        => reader.ReadInt32() switch
        {
            int.MaxValue => dateTimeInfinityConversions
                ? DateOnly.MaxValue
                : throw new InvalidCastException(NpgsqlStrings.CannotReadInfinityValue),
            int.MinValue => dateTimeInfinityConversions
                ? DateOnly.MinValue
                : throw new InvalidCastException(NpgsqlStrings.CannotReadInfinityValue),
            var value => BaseValue.AddDays(value)
        };

    public override void Write(PgWriter writer, DateOnly value)
    {
        if (dateTimeInfinityConversions)
        {
            if (value == DateOnly.MaxValue)
            {
                writer.WriteInt32(int.MaxValue);
                return;
            }

            if (value == DateOnly.MinValue)
            {
                writer.WriteInt32(int.MinValue);
                return;
            }
        }

        writer.WriteInt32(value.DayNumber - BaseValue.DayNumber);
    }
}

sealed class DateTimeDateConverter(bool dateTimeInfinityConversions) : PgBufferedConverter<DateTime>
{
    static readonly DateTime BaseValue = new(2000, 1, 1, 0, 0, 0);

    public override bool CanConvert(DataFormat format, out BufferRequirements bufferRequirements)
    {
        bufferRequirements = BufferRequirements.CreateFixedSize(sizeof(int));
        return format is DataFormat.Binary;
    }

    public override DateTime Read(PgReader reader)
        => reader.ReadInt32() switch
        {
            int.MaxValue => dateTimeInfinityConversions
                ? DateTime.MaxValue
                : throw new InvalidCastException(NpgsqlStrings.CannotReadInfinityValue),
            int.MinValue => dateTimeInfinityConversions
                ? DateTime.MinValue
                : throw new InvalidCastException(NpgsqlStrings.CannotReadInfinityValue),
            var value => BaseValue + TimeSpan.FromDays(value)
        };

    public override void Write(PgWriter writer, DateTime value)
    {
        if (dateTimeInfinityConversions)
        {
            if (value == DateTime.MaxValue)
            {
                writer.WriteInt32(int.MaxValue);
                return;
            }

            if (value == DateTime.MinValue)
            {
                writer.WriteInt32(int.MinValue);
                return;
            }
        }

        writer.WriteInt32((value.Date - BaseValue).Days);
    }
}