Nullability annotation for storage

src/EFCore.Abstractions/CommentAttribute.cs

@@ -8,15 +8,15 @@
 namespace Microsoft.EntityFrameworkCore
 {
     /// <summary>
-    ///     Marks a class, property or field with a comment which will be included in the SQL sent to the database .
+    ///     Marks a class, property or field with a comment to be set on the corresponding database table or column.
     /// </summary>
     [AttributeUsage(AttributeTargets.Class | AttributeTargets.Property | AttributeTargets.Field)]
     public sealed class CommentAttribute : Attribute
     {
         /// <summary>
         ///     Initializes a new instance of the <see cref="CommentAttribute" /> class.
         /// </summary>
-        /// <param name="comment">The comment.</param>
+        /// <param name="comment"> The comment. </param>
         public CommentAttribute([NotNull] string comment)
         {
             Check.NotEmpty(comment, nameof(comment));
@@ -25,7 +25,7 @@ public CommentAttribute([NotNull] string comment)
         }
 
         /// <summary>
-        ///     The Comment
+        ///     The comment to be configured.
         /// </summary>
         public string Comment { get; }
     }

src/EFCore.Cosmos/Storage/Internal/CosmosExecutionStrategy.cs

@@ -9,6 +9,8 @@
 using Microsoft.Azure.Cosmos;
 using Microsoft.EntityFrameworkCore.Storage;
 
+#nullable enable
+
 namespace Microsoft.EntityFrameworkCore.Cosmos.Storage.Internal
 {
     /// <summary>
@@ -97,24 +99,15 @@ public CosmosExecutionStrategy([NotNull] ExecutionStrategyDependencies dependenc
         ///     any release. You should only use it directly in your code with extreme caution and knowing that
         ///     doing so can result in application failures when updating to a new Entity Framework Core release.
         /// </summary>
-        protected override bool ShouldRetryOn(Exception exception)
+        protected override bool ShouldRetryOn(Exception? exception)
         {
-            if (exception is CosmosException cosmosException)
+            return exception switch
             {
-                return IsTransient(cosmosException.StatusCode);
-            }
-
-            if (exception is HttpException httpException)
-            {
-                return IsTransient(httpException.Response.StatusCode);
-            }
-
-            if (exception is WebException webException)
-            {
-                return IsTransient(((HttpWebResponse)webException.Response).StatusCode);
-            }
-
-            return false;
+                CosmosException cosmosException => IsTransient(cosmosException.StatusCode),
+                HttpException httpException => IsTransient(httpException.Response.StatusCode),
+                WebException webException => IsTransient(((HttpWebResponse)webException.Response!).StatusCode),
+                _ => false
+            };
 
             static bool IsTransient(HttpStatusCode statusCode)
                 => statusCode == HttpStatusCode.ServiceUnavailable
@@ -136,47 +129,54 @@ static bool IsTransient(HttpStatusCode statusCode)
                     ?? baseDelay;
         }
 
-        private static TimeSpan? GetDelayFromException(Exception exception)
+        private static TimeSpan? GetDelayFromException(Exception? exception)
         {
-            if (exception is CosmosException cosmosException)
+            switch (exception)
             {
-                return cosmosException.RetryAfter;
-            }
+                case CosmosException cosmosException:
+                    return cosmosException.RetryAfter;
 
-            if (exception is HttpException httpException)
-            {
-                if (httpException.Response.Headers.TryGetValues("x-ms-retry-after-ms", out var values)
-                    && TryParseMsRetryAfter(values.FirstOrDefault(), out var delay))
+                case HttpException httpException:
                 {
-                    return delay;
+                    if (httpException.Response.Headers.TryGetValues("x-ms-retry-after-ms", out var values)
+                        && TryParseMsRetryAfter(values.FirstOrDefault(), out var delay))
+                    {
+                        return delay;
+                    }
+
+                    if (httpException.Response.Headers.TryGetValues("Retry-After", out values)
+                        && TryParseRetryAfter(values.FirstOrDefault(), out delay))
+                    {
+                        return delay;
+                    }
+
+                    return null;
                 }
 
-                if (httpException.Response.Headers.TryGetValues("Retry-After", out values)
-                    && TryParseRetryAfter(values.FirstOrDefault(), out delay))
+                case WebException webException:
                 {
-                    return delay;
-                }
-            }
+                    var response = (HttpWebResponse)webException.Response!;
 
-            if (exception is WebException webException)
-            {
-                var response = (HttpWebResponse)webException.Response;
+                    var delayString = response.Headers.GetValues("x-ms-retry-after-ms")?.FirstOrDefault();
+                    if (TryParseMsRetryAfter(delayString, out var delay))
+                    {
+                        return delay;
+                    }
 
-                var delayString = response.Headers.GetValues("x-ms-retry-after-ms")?.FirstOrDefault();
-                if (TryParseMsRetryAfter(delayString, out var delay))
-                {
-                    return delay;
-                }
+                    delayString = response.Headers.GetValues("Retry-After")?.FirstOrDefault();
+                    if (TryParseRetryAfter(delayString, out delay))
+                    {
+                        return delay;
+                    }
 
-                delayString = response.Headers.GetValues("Retry-After")?.FirstOrDefault();
-                if (TryParseRetryAfter(delayString, out delay))
-                {
-                    return delay;
+                    return null;
                 }
+
+                default:
+                    return null;
             }
 
-            return null;
-            static bool TryParseMsRetryAfter(string delayString, out TimeSpan delay)
+            static bool TryParseMsRetryAfter(string? delayString, out TimeSpan delay)
             {
                 delay = default;
                 if (delayString == null)
@@ -193,7 +193,7 @@ static bool TryParseMsRetryAfter(string delayString, out TimeSpan delay)
                 return false;
             }
 
-            static bool TryParseRetryAfter(string delayString, out TimeSpan delay)
+            static bool TryParseRetryAfter(string? delayString, out TimeSpan delay)
             {
                 delay = default;
                 if (delayString == null)

src/EFCore.Cosmos/Storage/Internal/CosmosExecutionStrategyFactory.cs

@@ -40,7 +40,7 @@ public CosmosExecutionStrategyFactory([NotNull] ExecutionStrategyDependencies de
 
             Dependencies = dependencies;
 
-            _createExecutionStrategy = dependencies.Options?.FindExtension<CosmosOptionsExtension>()?.ExecutionStrategyFactory
+            _createExecutionStrategy = dependencies.Options.FindExtension<CosmosOptionsExtension>()?.ExecutionStrategyFactory
                 ?? CreateDefaultStrategy;
         }
 

src/EFCore.Design/Design/Internal/DesignTimeConnectionStringResolver.cs

@@ -6,6 +6,8 @@
 using Microsoft.EntityFrameworkCore.Storage.Internal;
 using Microsoft.Extensions.DependencyInjection;
 
+#nullable enable
+
 namespace Microsoft.EntityFrameworkCore.Design.Internal
 {
     /// <summary>
@@ -24,15 +26,15 @@ namespace Microsoft.EntityFrameworkCore.Design.Internal
     /// </summary>
     public class DesignTimeConnectionStringResolver : NamedConnectionStringResolverBase
     {
-        private readonly Func<IServiceProvider> _applicationServiceProviderAccessor;
+        private readonly Func<IServiceProvider>? _applicationServiceProviderAccessor;
 
         /// <summary>
         ///     This is an internal API that supports the Entity Framework Core infrastructure and not subject to
         ///     the same compatibility standards as public APIs. It may be changed or removed without notice in
         ///     any release. You should only use it directly in your code with extreme caution and knowing that
         ///     doing so can result in application failures when updating to a new Entity Framework Core release.
         /// </summary>
-        public DesignTimeConnectionStringResolver([CanBeNull] Func<IServiceProvider> applicationServiceProviderAccessor)
+        public DesignTimeConnectionStringResolver([CanBeNull] Func<IServiceProvider>? applicationServiceProviderAccessor)
         {
             _applicationServiceProviderAccessor = applicationServiceProviderAccessor;
         }
@@ -43,7 +45,7 @@ public DesignTimeConnectionStringResolver([CanBeNull] Func<IServiceProvider> app
         ///     any release. You should only use it directly in your code with extreme caution and knowing that
         ///     doing so can result in application failures when updating to a new Entity Framework Core release.
         /// </summary>
-        protected override IServiceProvider ApplicationServiceProvider
+        protected override IServiceProvider? ApplicationServiceProvider
             => _applicationServiceProviderAccessor?.Invoke();
     }
 }

src/EFCore.Relational/Extensions/RelationalDatabaseFacadeExtensions.cs

@@ -16,6 +16,8 @@
 using Microsoft.EntityFrameworkCore.Utilities;
 using Microsoft.Extensions.DependencyInjection;
 
+#nullable enable
+
 // ReSharper disable once CheckNamespace
 namespace Microsoft.EntityFrameworkCore
 {
@@ -177,7 +179,7 @@ public static int ExecuteSqlRaw(
         public static int ExecuteSqlInterpolated(
             [NotNull] this DatabaseFacade databaseFacade,
             [NotNull] FormattableString sql)
-            => ExecuteSqlRaw(databaseFacade, sql.Format, sql.GetArguments());
+            => ExecuteSqlRaw(databaseFacade, sql.Format, sql.GetArguments()!);
 
         /// <summary>
         ///     <para>
@@ -274,7 +276,7 @@ public static Task<int> ExecuteSqlInterpolatedAsync(
             [NotNull] this DatabaseFacade databaseFacade,
             [NotNull] FormattableString sql,
             CancellationToken cancellationToken = default)
-            => ExecuteSqlRawAsync(databaseFacade, sql.Format, sql.GetArguments(), cancellationToken);
+            => ExecuteSqlRawAsync(databaseFacade, sql.Format, sql.GetArguments()!, cancellationToken);
 
         /// <summary>
         ///     <para>
@@ -442,15 +444,15 @@ public static DbConnection GetDbConnection([NotNull] this DatabaseFacade databas
         /// </summary>
         /// <param name="databaseFacade"> The <see cref="DatabaseFacade" /> for the context. </param>
         /// <param name="connection"> The connection. </param>
-        public static void SetDbConnection([NotNull] this DatabaseFacade databaseFacade, [CanBeNull] DbConnection connection)
+        public static void SetDbConnection([NotNull] this DatabaseFacade databaseFacade, [CanBeNull] DbConnection? connection)
             => GetFacadeDependencies(databaseFacade).RelationalConnection.DbConnection = connection;
 
         /// <summary>
         ///     Gets the underlying connection string configured for this <see cref="DbContext" />.
         /// </summary>
         /// <param name="databaseFacade"> The <see cref="DatabaseFacade" /> for the context. </param>
         /// <returns> The connection string. </returns>
-        public static string GetConnectionString([NotNull] this DatabaseFacade databaseFacade)
+        public static string? GetConnectionString([NotNull] this DatabaseFacade databaseFacade)
             => GetFacadeDependencies(databaseFacade).RelationalConnection.ConnectionString;
 
         /// <summary>
@@ -463,7 +465,7 @@ public static string GetConnectionString([NotNull] this DatabaseFacade databaseF
         /// </summary>
         /// <param name="databaseFacade"> The <see cref="DatabaseFacade" /> for the context. </param>
         /// <param name="connectionString"> The connection string. </param>
-        public static void SetConnectionString([NotNull] this DatabaseFacade databaseFacade, [CanBeNull] string connectionString)
+        public static void SetConnectionString([NotNull] this DatabaseFacade databaseFacade, [CanBeNull] string? connectionString)
             => GetFacadeDependencies(databaseFacade).RelationalConnection.ConnectionString = connectionString;
 
         /// <summary>
@@ -552,9 +554,9 @@ public static Task<IDbContextTransaction> BeginTransactionAsync(
         /// <param name="databaseFacade"> The <see cref="DatabaseFacade" /> for the context. </param>
         /// <param name="transaction"> The <see cref="DbTransaction" /> to use. </param>
         /// <returns> A <see cref="IDbContextTransaction" /> that encapsulates the given transaction. </returns>
-        public static IDbContextTransaction UseTransaction(
+        public static IDbContextTransaction? UseTransaction(
             [NotNull] this DatabaseFacade databaseFacade,
-            [CanBeNull] DbTransaction transaction)
+            [CanBeNull] DbTransaction? transaction)
             => databaseFacade.UseTransaction(transaction, Guid.NewGuid());
 
         /// <summary>
@@ -564,9 +566,9 @@ public static IDbContextTransaction UseTransaction(
         /// <param name="transaction"> The <see cref="DbTransaction" /> to use. </param>
         /// <param name="transactionId"> The unique identifier for the transaction. </param>
         /// <returns> A <see cref="IDbContextTransaction" /> that encapsulates the given transaction. </returns>
-        public static IDbContextTransaction UseTransaction(
+        public static IDbContextTransaction? UseTransaction(
             [NotNull] this DatabaseFacade databaseFacade,
-            [CanBeNull] DbTransaction transaction,
+            [CanBeNull] DbTransaction? transaction,
             Guid transactionId)
         {
             var transactionManager = GetTransactionManager(databaseFacade);
@@ -587,9 +589,9 @@ public static IDbContextTransaction UseTransaction(
         /// <param name="cancellationToken"> A <see cref="CancellationToken" /> to observe while waiting for the task to complete. </param>
         /// <returns> A <see cref="Task" /> containing the <see cref="IDbContextTransaction" /> for the given transaction. </returns>
         /// <exception cref="OperationCanceledException"> If the <see cref="CancellationToken"/> is canceled. </exception>
-        public static Task<IDbContextTransaction> UseTransactionAsync(
+        public static Task<IDbContextTransaction?> UseTransactionAsync(
             [NotNull] this DatabaseFacade databaseFacade,
-            [CanBeNull] DbTransaction transaction,
+            [CanBeNull] DbTransaction? transaction,
             CancellationToken cancellationToken = default)
             => databaseFacade.UseTransactionAsync(transaction, Guid.NewGuid(), cancellationToken);
 
@@ -602,9 +604,9 @@ public static Task<IDbContextTransaction> UseTransactionAsync(
         /// <param name="cancellationToken"> A <see cref="CancellationToken" /> to observe while waiting for the task to complete. </param>
         /// <returns> A <see cref="Task" /> containing the <see cref="IDbContextTransaction" /> for the given transaction. </returns>
         /// <exception cref="OperationCanceledException"> If the <see cref="CancellationToken"/> is canceled. </exception>
-        public static Task<IDbContextTransaction> UseTransactionAsync(
+        public static Task<IDbContextTransaction?> UseTransactionAsync(
             [NotNull] this DatabaseFacade databaseFacade,
-            [CanBeNull] DbTransaction transaction,
+            [CanBeNull] DbTransaction? transaction,
             Guid transactionId,
             CancellationToken cancellationToken = default)
         {

src/EFCore.Relational/Query/ISqlExpressionFactory.cs

@@ -41,7 +41,7 @@ public interface ISqlExpressionFactory
         /// <param name="type"> The CLR type. </param>
         /// <returns> The type mapping, or <see langword="null" /> if none was found. </returns>
         [Obsolete("Use IRelationalTypeMappingSource directly.")]
-        RelationalTypeMapping FindMapping([NotNull] Type type);
+        RelationalTypeMapping? FindMapping([NotNull] Type type);
 
         /// <summary>
         ///     Applies type mapping to the given <see cref="SqlExpression" />.

src/EFCore.Relational/Query/Internal/BufferedDataReader.cs

@@ -1270,7 +1270,7 @@ private void InitializeFields()
                     for (var i = 0; i < _columns.Count; i++)
                     {
                         var column = _columns[i];
-                        if (!readerColumns.TryGetValue(column.Name, out var ordinal))
+                        if (!readerColumns.TryGetValue(column.Name!, out var ordinal))
                         {
                             throw new InvalidOperationException(RelationalStrings.FromSqlMissingColumn(column.Name));
                         }

src/EFCore.Relational/Query/Internal/FromSqlParameterExpandingExpressionVisitor.cs

@@ -129,7 +129,7 @@ public virtual SelectExpression Expand(
                             }
 
                             updatedFromSql = fromSql.Update(
-                                Expression.Constant(new CompositeRelationalParameter(parameterExpression.Name, subParameters)));
+                                Expression.Constant(new CompositeRelationalParameter(parameterExpression.Name!, subParameters)));
 
                             _visitedFromSqlExpressions[fromSql] = updatedFromSql;
                             break;

src/EFCore.Relational/Query/QuerySqlGenerator.cs

@@ -331,7 +331,7 @@ protected override Expression VisitColumn(ColumnExpression columnExpression)
             Check.NotNull(columnExpression, nameof(columnExpression));
 
             _relationalCommandBuilder
-                .Append(_sqlGenerationHelper.DelimitIdentifier(columnExpression.Table.Alias))
+                .Append(_sqlGenerationHelper.DelimitIdentifier(columnExpression.Table.Alias!))
                 .Append(".")
                 .Append(_sqlGenerationHelper.DelimitIdentifier(columnExpression.Name));
 
@@ -572,7 +572,7 @@ protected override Expression VisitSqlParameter(SqlParameterExpression sqlParame
                 _relationalCommandBuilder.AddParameter(
                     sqlParameterExpression.Name,
                     parameterNameInCommand,
-                    sqlParameterExpression.TypeMapping,
+                    sqlParameterExpression.TypeMapping!,
                     sqlParameterExpression.IsNullable);
             }
 

src/EFCore.Relational/Query/SqlExpressionFactory.cs

@@ -34,7 +34,7 @@ public SqlExpressionFactory([NotNull] SqlExpressionFactoryDependencies dependenc
             Check.NotNull(dependencies, nameof(dependencies));
 
             _typeMappingSource = dependencies.TypeMappingSource;
-            _boolTypeMapping = _typeMappingSource.FindMapping(typeof(bool));
+            _boolTypeMapping = _typeMappingSource.FindMapping(typeof(bool))!;
         }
 
         /// <inheritdoc />
@@ -1017,7 +1017,7 @@ public virtual RelationalTypeMapping GetTypeMappingForValue(object? value)
 
         /// <inheritdoc />
         [Obsolete("Use IRelationalTypeMappingSource directly.")]
-        public virtual RelationalTypeMapping FindMapping(Type type)
+        public virtual RelationalTypeMapping? FindMapping(Type type)
             => _typeMappingSource.FindMapping(Check.NotNull(type, nameof(type)));
     }
 }

src/EFCore.Relational/Query/SqlExpressions/ExceptExpression.cs

@@ -61,7 +61,7 @@ public virtual ExceptExpression Update([NotNull] SelectExpression source1, [NotN
             Check.NotNull(source2, nameof(source2));
 
             return source1 != Source1 || source2 != Source2
-                ? new ExceptExpression(Alias!, source1, source2, IsDistinct)
+                ? new ExceptExpression(Alias, source1, source2, IsDistinct)
                 : this;
         }