SchemaMetadataUpdater.cs

using System;
using NHibernate.Cfg;
using NHibernate.Engine;
using NHibernate.Mapping;
using System.Collections.Generic;
using System.Linq;

namespace NHibernate.Tool.hbm2ddl
{
	// Candidate to be exstensions of ISessionFactory and Configuration
	public static partial class SchemaMetadataUpdater
	{
		public static void Update(ISessionFactoryImplementor sessionFactory)
		{
			UpdateDialectKeywords(
				sessionFactory.Dialect,
				new SuppliedConnectionProviderConnectionHelper(sessionFactory.ConnectionProvider));
		}

		public static void Update(Configuration configuration, Dialect.Dialect dialect)
		{
			UpdateDialectKeywords(
				dialect,
				new ManagedProviderConnectionHelper(configuration.GetDerivedProperties()));
		}

		static void UpdateDialectKeywords(Dialect.Dialect dialect, IConnectionHelper connectionHelper)
		{
			dialect.RegisterKeywords(GetReservedWords(dialect, connectionHelper));
		}

		static IEnumerable<string> GetReservedWords(Dialect.Dialect dialect, IConnectionHelper connectionHelper)
		{
			connectionHelper.Prepare();
			try
			{
				var metaData = dialect.GetDataBaseSchema(connectionHelper.Connection);
				return metaData.GetReservedWords();
			}
			finally
			{
				connectionHelper.Release();
			}
		}

		// Since v5
		[Obsolete("Use the overload that passes dialect so keywords will be updated and persisted before auto-quoting")]
		public static void QuoteTableAndColumns(Configuration configuration)
		{
			// Instantiates a new instance of the dialect so doesn't benefit from the Update call.
			var dialect = Dialect.Dialect.GetDialect(configuration.GetDerivedProperties());
			Update(configuration, dialect);
			QuoteTableAndColumns(configuration, dialect);
		}

		public static void QuoteTableAndColumns(Configuration configuration, Dialect.Dialect dialect)
		{
			// We have to build the mappings in order to quote collection mappings that use a second pass command to be fully initialized
			configuration.BuildMappings();
			foreach (var cm in configuration.ClassMappings)
			{
				QuoteTable(cm.Table, dialect);
			}
			foreach (var cm in configuration.CollectionMappings)
			{
				QuoteTable(cm.Table, dialect);
				QuoteColumns(cm.Key, dialect);
				QuoteColumns(cm.Element, dialect);
			}
		}

		private static void QuoteTable(Table table, Dialect.Dialect dialect)
		{
			if (!table.IsQuoted && dialect.IsKeyword(table.Name))
			{
				table.IsQuoted = true;
			}

			QuoteColumns(table.ColumnIterator, dialect);
		}

		private static void QuoteColumns(IValue value, Dialect.Dialect dialect)
		{
			if (value == null)
				return;
			QuoteColumns(value.ColumnIterator.OfType<Column>(), dialect);
		}

		private static void QuoteColumns(IEnumerable<Column> columns, Dialect.Dialect dialect)
		{
			foreach (var column in columns)
			{
				if (!column.IsQuoted && dialect.IsKeyword(column.Name))
				{
					column.IsQuoted = true;
				}
			}
		}
	}
}