X Tutup
#region License // The PostgreSQL License // // Copyright (C) 2018 The Npgsql Development Team // // Permission to use, copy, modify, and distribute this software and its // documentation for any purpose, without fee, and without a written // agreement is hereby granted, provided that the above copyright notice // and this paragraph and the following two paragraphs appear in all copies. // // IN NO EVENT SHALL THE NPGSQL DEVELOPMENT TEAM BE LIABLE TO ANY PARTY // FOR DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES, // INCLUDING LOST PROFITS, ARISING OUT OF THE USE OF THIS SOFTWARE AND ITS // DOCUMENTATION, EVEN IF THE NPGSQL DEVELOPMENT TEAM HAS BEEN ADVISED OF // THE POSSIBILITY OF SUCH DAMAGE. // // THE NPGSQL DEVELOPMENT TEAM SPECIFICALLY DISCLAIMS ANY WARRANTIES, // INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY // AND FITNESS FOR A PARTICULAR PURPOSE. THE SOFTWARE PROVIDED HEREUNDER IS // ON AN "AS IS" BASIS, AND THE NPGSQL DEVELOPMENT TEAM HAS NO OBLIGATIONS // TO PROVIDE MAINTENANCE, SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS. #endregion using System; using System.Collections.Concurrent; using System.Collections.Generic; using System.Data.Common; using System.Runtime.InteropServices.ComTypes; using System.Threading.Tasks; using Npgsql.PostgresTypes; namespace Npgsql { /// /// Base class for implementations which provide information about PostgreSQL and PostgreSQL-like databases /// (e.g. type definitions, capabilities...). /// public abstract class NpgsqlDatabaseInfo { #region Fields internal static ConcurrentDictionary Cache = new ConcurrentDictionary(); static readonly List Factories = new List { new PostgresMinimalDatabaseInfoFactory(), new PostgresDatabaseInfoFactory() }; #endregion Fields #region General database info /// /// The hostname of IP address of the database. /// public string Host { get; protected set; } /// /// The TCP port of the database. /// public int Port { get; protected set; } /// /// The database name. /// public string Name { get; protected set; } /// /// The version of the PostgreSQL database we're connected to, as reported in the "server_version" parameter. /// Exposed via . /// public Version Version { get; protected set; } #endregion General database info #region Supported capabilities and features /// /// Whether the backend supports range types. /// public virtual bool SupportsRangeTypes => Version >= new Version(9, 2, 0); /// /// Whether the backend supports enum types. /// public virtual bool SupportsEnumTypes => Version >= new Version(8, 3, 0); /// /// Whether the backend supports the CLOSE ALL statement. /// public virtual bool SupportsCloseAll => Version >= new Version(8, 3, 0); /// /// Whether the backend supports advisory locks. /// public virtual bool SupportsAdvisoryLocks => Version >= new Version(8, 2, 0); /// /// Whether the backend supports the DISCARD SEQUENCES statement. /// public virtual bool SupportsDiscardSequences => Version >= new Version(9, 4, 0); /// /// Whether the backend supports the UNLISTEN statement. /// public virtual bool SupportsUnlisten => Version >= new Version(6, 4, 0); // overridden by PostgresDatabase /// /// Whether the backend supports the DISCARD TEMP statement. /// public virtual bool SupportsDiscardTemp => Version >= new Version(8, 3, 0); /// /// Whether the backend supports the DISCARD statement. /// public virtual bool SupportsDiscard => Version >= new Version(8, 3, 0); /// /// Reports whether the backend uses the newer integer timestamp representation. /// public virtual bool HasIntegerDateTimes { get; protected set; } = true; /// /// Whether the database supports transactions. /// public virtual bool SupportsTransactions { get; protected set; } = true; #endregion Supported capabilities and features #region Types internal IReadOnlyList BaseTypes { get; private set; } internal IReadOnlyList ArrayTypes { get; private set; } internal IReadOnlyList RangeTypes { get; private set; } internal IReadOnlyList EnumTypes { get; private set; } internal IReadOnlyList CompositeTypes { get; private set; } internal IReadOnlyList DomainTypes { get; private set; } /// /// Indexes backend types by their type OID. /// internal Dictionary ByOID { get; } = new Dictionary(); /// /// Indexes backend types by their PostgreSQL name, including namespace (e.g. pg_catalog.int4). /// Only used for enums and composites. /// internal Dictionary ByFullName { get; } = new Dictionary(); /// /// Indexes backend types by their PostgreSQL name, not including namespace. /// If more than one type exists with the same name (i.e. in different namespaces) this /// table will contain an entry with a null value. /// Only used for enums and composites. /// internal Dictionary ByName { get; } = new Dictionary(); internal void ProcessTypes() { var baseTypes = new List(); var arrayTypes = new List(); var rangeTypes = new List(); var enumTypes = new List(); var compositeTypes = new List(); var domainTypes = new List(); foreach (var type in GetTypes()) { ByOID[type.OID] = type; ByFullName[type.FullName] = type; // If more than one type exists with the same partial name, we place a null value. // This allows us to detect this case later and force the user to use full names only. ByName[type.Name] = ByName.ContainsKey(type.Name) ? null : type; switch (type) { case PostgresBaseType baseType: baseTypes.Add(baseType); continue; case PostgresArrayType arrayType: arrayTypes.Add(arrayType); continue; case PostgresRangeType rangeType: rangeTypes.Add(rangeType); continue; case PostgresEnumType enumType: enumTypes.Add(enumType); continue; case PostgresCompositeType compositeType: compositeTypes.Add(compositeType); continue; case PostgresDomainType domainType: domainTypes.Add(domainType); continue; default: throw new ArgumentOutOfRangeException(); } } BaseTypes = baseTypes; ArrayTypes = arrayTypes; RangeTypes = rangeTypes; EnumTypes = enumTypes; CompositeTypes = compositeTypes; DomainTypes = domainTypes; } /// /// Provides all PostgreSQL types detected in this database. /// /// protected abstract IEnumerable GetTypes(); #endregion Types #region Misc /// /// Parses a PostgreSQL server version (e.g. 10.1, 9.6.3) and returns a CLR Version. /// protected static Version ParseServerVersion(string value) { var versionString = value.Trim(); for (var idx = 0; idx != versionString.Length; ++idx) { var c = value[idx]; if (!char.IsDigit(c) && c != '.') { versionString = versionString.Substring(0, idx); break; } } if (!versionString.Contains(".")) versionString += ".0"; return new Version(versionString); } #endregion Misc #region Factory management /// /// Registers a new database info factory, which is used to load information about databases. /// public static void RegisterFactory(INpgsqlDatabaseInfoFactory factory) { if (factory == null) throw new ArgumentNullException(nameof(factory)); Factories.Insert(0, factory); } internal static async Task Load(NpgsqlConnection conn, NpgsqlTimeout timeout, bool async) { foreach (var factory in Factories) { var dbInfo = await factory.Load(conn, timeout, async); if (dbInfo != null) { dbInfo.ProcessTypes(); return dbInfo; } } // Should never be here throw new NpgsqlException("No DatabaseInfoFactory could be found for this connection"); } // For tests internal static void ResetFactories() { Factories.Clear(); Factories.Add(new PostgresMinimalDatabaseInfoFactory()); Factories.Add(new PostgresDatabaseInfoFactory()); } #endregion Factory management } }
X Tutup