dev-npgsql/npgsql: Npgsql is the .NET data provider for PostgreSQL.

src/Npgsql.NodaTime/Internal/DateHandler.cs
91 строка4 KB
Shay Rojansky
Fix reading of NodaTime DateInterval with infinite upper bound (#4718)
14 окт 2022, 16:43
14 окт 2022, 16:43ea3932d
using System;
using NodaTime;
using Npgsql.BackendMessages;
using Npgsql.Internal;
using Npgsql.Internal.TypeHandling;
using Npgsql.NodaTime.Properties;
using Npgsql.PostgresTypes;
using static Npgsql.NodaTime.Internal.NodaTimeUtils;
using BclDateHandler = Npgsql.Internal.TypeHandlers.DateTimeHandlers.DateHandler;

namespace Npgsql.NodaTime.Internal;

sealed partial class DateHandler : NpgsqlSimpleTypeHandler<LocalDate>,
    INpgsqlSimpleTypeHandler<DateTime>, INpgsqlSimpleTypeHandler<int>
#if NET6_0_OR_GREATER
    , INpgsqlSimpleTypeHandler<DateOnly>
#endif
{
    readonly BclDateHandler _bclHandler;

    internal DateHandler(PostgresType postgresType)
        : base(postgresType)
        => _bclHandler = new BclDateHandler(postgresType);

    public override LocalDate Read(NpgsqlReadBuffer buf, int len, FieldDescription? fieldDescription = null)
        => buf.ReadInt32() switch
        {
            int.MaxValue => DisableDateTimeInfinityConversions
                ? throw new InvalidCastException(NpgsqlNodaTimeStrings.CannotReadInfinityValue)
                : LocalDate.MaxIsoValue,
            int.MinValue => DisableDateTimeInfinityConversions
                ? throw new InvalidCastException(NpgsqlNodaTimeStrings.CannotReadInfinityValue)
                : LocalDate.MinIsoValue,
            var value => new LocalDate().PlusDays(value + 730119)
        };

    public override int ValidateAndGetLength(LocalDate value, NpgsqlParameter? parameter)
        => 4;

    public override void Write(LocalDate value, NpgsqlWriteBuffer buf, NpgsqlParameter? parameter)
    {
        if (!DisableDateTimeInfinityConversions)
        {
            if (value == LocalDate.MaxIsoValue)
            {
                buf.WriteInt32(int.MaxValue);
                return;
            }
            if (value == LocalDate.MinIsoValue)
            {
                buf.WriteInt32(int.MinValue);
                return;
            }
        }

        var totalDaysSinceEra = Period.Between(default, value, PeriodUnits.Days).Days;
        buf.WriteInt32(totalDaysSinceEra - 730119);
    }

    DateTime INpgsqlSimpleTypeHandler<DateTime>.Read(NpgsqlReadBuffer buf, int len, FieldDescription? fieldDescription)
        => _bclHandler.Read<DateTime>(buf, len, fieldDescription);

    int INpgsqlSimpleTypeHandler<DateTime>.ValidateAndGetLength(DateTime value, NpgsqlParameter? parameter)
        => _bclHandler.ValidateAndGetLength(value, parameter);

    void INpgsqlSimpleTypeHandler<DateTime>.Write(DateTime value, NpgsqlWriteBuffer buf, NpgsqlParameter? parameter)
        => _bclHandler.Write(value, buf, parameter);

    int INpgsqlSimpleTypeHandler<int>.Read(NpgsqlReadBuffer buf, int len, FieldDescription? fieldDescription)
        => _bclHandler.Read<int>(buf, len, fieldDescription);

    int INpgsqlSimpleTypeHandler<int>.ValidateAndGetLength(int value, NpgsqlParameter? parameter)
        => _bclHandler.ValidateAndGetLength(value, parameter);

    void INpgsqlSimpleTypeHandler<int>.Write(int value, NpgsqlWriteBuffer buf, NpgsqlParameter? parameter)
        => _bclHandler.Write(value, buf, parameter);

#if NET6_0_OR_GREATER
    DateOnly INpgsqlSimpleTypeHandler<DateOnly>.Read(NpgsqlReadBuffer buf, int len, FieldDescription? fieldDescription)
        => _bclHandler.Read<DateOnly>(buf, len, fieldDescription);

    public int ValidateAndGetLength(DateOnly value, NpgsqlParameter? parameter)
        => _bclHandler.ValidateAndGetLength(value, parameter);

    public void Write(DateOnly value, NpgsqlWriteBuffer buf, NpgsqlParameter? parameter)
        => _bclHandler.Write(value, buf, parameter);
#endif

    public override NpgsqlTypeHandler CreateRangeHandler(PostgresType pgRangeType)
        => new DateRangeHandler(pgRangeType, this);
}