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

src/Npgsql/PostgresMinimalDatabaseInfo.cs

38 строк1 KB

Shay Rojansky

Apply file-scoped namespaces for 6.0 (#4298)

28 янв 2022, 15:13

28 янв 2022, 15:132e0ad2d

using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using System.Threading.Tasks;
using Npgsql.Internal;
using Npgsql.PostgresTypes;
using Npgsql.Util;
using NpgsqlTypes;

namespace Npgsql;

class PostgresMinimalDatabaseInfoFactory : INpgsqlDatabaseInfoFactory
{
    public Task<NpgsqlDatabaseInfo?> Load(NpgsqlConnector conn, NpgsqlTimeout timeout, bool async)
        => Task.FromResult(
            conn.Settings.ServerCompatibilityMode == ServerCompatibilityMode.NoTypeLoading
                ? (NpgsqlDatabaseInfo)new PostgresMinimalDatabaseInfo(conn)
                : null
        );
}

class PostgresMinimalDatabaseInfo : PostgresDatabaseInfo
{
    static readonly PostgresBaseType[] Types = typeof(NpgsqlDbType).GetFields()
        .Select(f => f.GetCustomAttribute<BuiltInPostgresType>())
        .OfType<BuiltInPostgresType>()
        .Select(a => new PostgresBaseType("pg_catalog", a.Name, a.OID))
        .ToArray();

    protected override IEnumerable<PostgresType> GetTypes() => Types;

    internal PostgresMinimalDatabaseInfo(NpgsqlConnector conn)
        : base(conn)
    {
        HasIntegerDateTimes = !conn.PostgresParameters.TryGetValue("integer_datetimes", out var intDateTimes) ||
                              intDateTimes == "on";
    }
}