src/Npgsql/Internal/Converters/Geometric/CubeConverter.cs - main - dev-npgsql/npgsql

src/Npgsql/Internal/Converters/Geometric/CubeConverter.cs
87 строк3 KB
Nino Floris
Introduce bind operation on PgConverter surface (#6566)
11 май 2026, 20:01
11 май 2026, 20:010f4cfd8
using System.Threading;
using System.Threading.Tasks;
using NpgsqlTypes;

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

sealed class CubeConverter : PgStreamingConverter<NpgsqlCube>
{
    const uint PointBit = 0x80000000;
    const int DimMask = 0x7fffffff;

    public override NpgsqlCube Read(PgReader reader)
        => Read(async: false, reader, CancellationToken.None).GetAwaiter().GetResult();

    public override ValueTask<NpgsqlCube> ReadAsync(PgReader reader, CancellationToken cancellationToken = default)
        => Read(async: true, reader, cancellationToken);

    async ValueTask<NpgsqlCube> Read(bool async, PgReader reader, CancellationToken cancellationToken)
    {
        if (reader.ShouldBuffer(sizeof(int)))
            await reader.Buffer(async, sizeof(int), cancellationToken).ConfigureAwait(false);

        var header = reader.ReadInt32();
        var dim = header & DimMask;
        var point = (header & PointBit) != 0;

        var lowerLeft = new double[dim];
        for (var i = 0; i < dim; i++)
        {
            if (reader.ShouldBuffer(sizeof(double)))
                await reader.Buffer(async, sizeof(double), cancellationToken).ConfigureAwait(false);
            lowerLeft[i] = reader.ReadDouble();
        }

        if (point)
            return new NpgsqlCube(lowerLeft);

        var upperRight = new double[dim];
        for (var i = 0; i < dim; i++)
        {
            if (reader.ShouldBuffer(sizeof(double)))
                await reader.Buffer(async, sizeof(double), cancellationToken).ConfigureAwait(false);
            upperRight[i] = reader.ReadDouble();
        }

        return new NpgsqlCube(lowerLeft, upperRight);
    }

    protected override Size BindValue(in BindContext context, NpgsqlCube value, ref object? writeState)
        => sizeof(int) + sizeof(double) * (value.IsPoint ? value.Dimensions : value.Dimensions * 2);

    public override void Write(PgWriter writer, NpgsqlCube value)
        => Write(async: false, writer, value, CancellationToken.None).GetAwaiter().GetResult();

    public override ValueTask WriteAsync(PgWriter writer, NpgsqlCube value, CancellationToken cancellationToken = default)
        => Write(async: true, writer, value, cancellationToken);

    async ValueTask Write(bool async, PgWriter writer, NpgsqlCube value, CancellationToken cancellationToken)
    {
        if (writer.ShouldFlush(sizeof(int)))
            await writer.Flush(async, cancellationToken).ConfigureAwait(false);

        var header = value.Dimensions;
        if (value.IsPoint)
            header |= 1 << 31;

        writer.WriteInt32(header);

        for (var i = 0; i < value.Dimensions; i++)
        {
            if (writer.ShouldFlush(sizeof(double)))
                await writer.Flush(async, cancellationToken).ConfigureAwait(false);
            writer.WriteDouble(value.LowerLeft[i]);
        }

        if (value.IsPoint)
            return;

        for (var i = 0; i < value.Dimensions; i++)
        {
            if (writer.ShouldFlush(sizeof(double)))
                await writer.Flush(async, cancellationToken).ConfigureAwait(false);
            writer.WriteDouble(value.UpperRight[i]);
        }
    }
}