Entity Framework: initial setup

Author: messieurMe

src/Ef.Ydb/src/Diagnostics/Internal/YdbCommandInterceptor.cs

@@ -0,0 +1,17 @@
+using System.Data.Common;
+using Microsoft.EntityFrameworkCore.Diagnostics;
+
+namespace Ef.Ydb.Diagnostics.Internal;
+
+// TODO: Temporary for debugging
+public class YdbCommandInterceptor : DbCommandInterceptor
+{
+    public override InterceptionResult<DbDataReader> ReaderExecuting(
+        DbCommand command,
+        CommandEventData eventData,
+        InterceptionResult<DbDataReader> result
+    )
+    {
+        return result;
+    }
+}

src/Ef.Ydb/src/Diagnostics/Internal/YdbLoggingDefinitions.cs

@@ -0,0 +1,7 @@
+using Microsoft.EntityFrameworkCore.Diagnostics;
+
+namespace Ef.Ydb.Diagnostics.Internal;
+
+public class YdbLoggingDefinitions : RelationalLoggingDefinitions
+{
+}

src/Ef.Ydb/src/Ef.Ydb.csproj

@@ -0,0 +1,24 @@
+<Project Sdk="Microsoft.NET.Sdk">
+
+    <PropertyGroup>
+        <TargetFramework>net8.0</TargetFramework>
+        <RootNamespace>Ef.Ydb</RootNamespace>
+        <Nullable>enable</Nullable>
+        <PackageId>Ef.Ydb</PackageId>
+    </PropertyGroup>
+
+    <ItemGroup>
+      <PackageReference Include="Microsoft.EntityFrameworkCore" Version="9.0.0" />
+      <PackageReference Include="Microsoft.EntityFrameworkCore.Relational" Version="9.0.0" />
+    </ItemGroup>
+
+    <ItemGroup>
+      <ProjectReference Include="..\..\Ydb.Sdk\src\Ydb.Sdk.csproj" />
+    </ItemGroup>
+
+    <ItemGroup>
+      <Reference Include="Microsoft.EntityFrameworkCore.Relational">
+        <HintPath>..\..\..\..\..\.nuget\packages\microsoft.entityframeworkcore.relational\9.0.0\lib\net8.0\Microsoft.EntityFrameworkCore.Relational.dll</HintPath>
+      </Reference>
+    </ItemGroup>
+</Project>

src/Ef.Ydb/src/Extensions/EfYdbContextOptionsBuilderExtensions.cs

@@ -0,0 +1,39 @@
+using System;
+using Ef.Ydb.Infrastructure;
+using Ef.Ydb.Infrastructure.Internal;
+using Microsoft.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore.Infrastructure;
+
+namespace Ef.Ydb.Extensions;
+
+public static class YdbContextOptionsBuilderExtensions
+{
+    public static DbContextOptionsBuilder UseEfYdb(
+        this DbContextOptionsBuilder optionsBuilder,
+        string? connectionString,
+        Action<YdbDbContextOptionsBuilder>? efYdbOptionsAction = null
+    )
+    {
+        var extension = GetOrCreateExtension(optionsBuilder).WithConnectionString(connectionString);
+        ((IDbContextOptionsBuilderInfrastructure)optionsBuilder).AddOrUpdateExtension(extension);
+
+        ConfigureWarnings(optionsBuilder);
+
+        efYdbOptionsAction?.Invoke(new YdbDbContextOptionsBuilder(optionsBuilder));
+        return optionsBuilder;
+    }
+
+    // TODO: Right now there are no arguments for constructor, so probably it's ok
+    private static YdbOptionsExtension GetOrCreateExtension(DbContextOptionsBuilder options)
+        => options.Options.FindExtension<YdbOptionsExtension>()
+           ?? new YdbOptionsExtension();
+
+    private static void ConfigureWarnings(DbContextOptionsBuilder optionsBuilder)
+    {
+        var coreOptionsExtension = optionsBuilder.Options.FindExtension<CoreOptionsExtension>()
+                                   ?? new CoreOptionsExtension();
+
+        coreOptionsExtension = RelationalOptionsExtension.WithDefaultWarningConfiguration(coreOptionsExtension);
+        ((IDbContextOptionsBuilderInfrastructure)optionsBuilder).AddOrUpdateExtension(coreOptionsExtension);
+    }
+}

src/Ef.Ydb/src/Extensions/YdbServiceCollectionExtensions.cs

@@ -0,0 +1,61 @@
+using Ef.Ydb.Diagnostics.Internal;
+using Ef.Ydb.Infrastructure;
+using Ef.Ydb.Infrastructure.Internal;
+using Ef.Ydb.Metadata.Conventions;
+using Ef.Ydb.Metadata.Internal;
+using Ef.Ydb.Migrations;
+using Ef.Ydb.Migrations.Internal;
+using Ef.Ydb.Query.Internal;
+using Ef.Ydb.Storage.Internal;
+using Ef.Ydb.Update.Internal;
+using Microsoft.EntityFrameworkCore.Diagnostics;
+using Microsoft.EntityFrameworkCore.Infrastructure;
+using Microsoft.EntityFrameworkCore.Metadata;
+using Microsoft.EntityFrameworkCore.Metadata.Conventions.Infrastructure;
+using Microsoft.EntityFrameworkCore.Migrations;
+using Microsoft.EntityFrameworkCore.Query;
+using Microsoft.EntityFrameworkCore.Storage;
+using Microsoft.EntityFrameworkCore.Update;
+using Microsoft.Extensions.DependencyInjection;
+
+namespace Ef.Ydb.Extensions;
+
+public static class YdbServiceCollectionExtensions
+{
+    public static IServiceCollection AddEntityFrameworkYdb(this IServiceCollection serviceCollection)
+    {
+        new EntityFrameworkYdbServicesBuilder(serviceCollection)
+            .TryAdd<LoggingDefinitions, YdbLoggingDefinitions>()
+            .TryAdd<IDatabaseProvider, DatabaseProvider<YdbOptionsExtension>>()
+            .TryAdd<IRelationalTypeMappingSource, YdbTypeMappingSource>()
+            .TryAdd<ISqlGenerationHelper, YdbSqlGenerationHelper>()
+            .TryAdd<IRelationalAnnotationProvider, YdbAnnotationProvider>()
+            .TryAdd<IModelValidator, YdbModelValidator>()
+            .TryAdd<IProviderConventionSetBuilder, YdbConventionSetBuilder>()
+            .TryAdd<IUpdateSqlGenerator, YdbUpdateSqlGenerator>()
+            .TryAdd<IModificationCommandFactory, YdbModificationCommandFactory>()
+            .TryAdd<IModificationCommandBatchFactory, YdbModificationCommandBatchFactory>()
+            .TryAdd<IRelationalConnection>(p => p.GetRequiredService<IYdbRelationalConnection>())
+            .TryAdd<IMigrationsSqlGenerator, YdbMigrationsSqlGenerator>()
+            .TryAdd<IRelationalDatabaseCreator, YdbDatabaseCreator>()
+            .TryAdd<IHistoryRepository, YdbHistoryRepository>()
+            .TryAdd<IQueryableMethodTranslatingExpressionVisitorFactory,
+                YdbQueryableMethodTranslatingExpressionVisitorFactory>()
+            .TryAdd<IMethodCallTranslatorProvider, YdbMethodCallTranslatorProvider>()
+            .TryAdd<IAggregateMethodCallTranslatorProvider, YdbAggregateMethodCallTranslatorProvider>()
+            .TryAdd<IMemberTranslatorProvider, YdbMemberTranslatorProvider>()
+            .TryAdd<IQuerySqlGeneratorFactory, YdbQuerySqlGeneratorFactory>()
+            .TryAdd<IRelationalSqlTranslatingExpressionVisitorFactory, YdbSqlTranslatingExpressionVisitorFactory>()
+            .TryAdd<IQueryTranslationPostprocessorFactory, YdbQueryTranslationPostprocessorFactory>()
+            .TryAdd<IRelationalParameterBasedSqlProcessorFactory, YdbParameterBasedSqlProcessorFactory>()
+            .TryAdd<ISqlExpressionFactory, YdbSqlExpressionFactory>()
+            .TryAdd<IQueryCompilationContextFactory, YdbQueryCompilationContextFactory>()
+            .TryAddProviderSpecificServices(
+                b => b
+                    .TryAddScoped<IYdbRelationalConnection, YdbRelationalConnection>()
+                    .TryAddScoped<IDbCommandInterceptor, YdbCommandInterceptor>())
+            .TryAddCoreServices();
+
+        return serviceCollection;
+    }
+}

src/Ef.Ydb/src/Infrastructure/EntityFrameworkYdbServicesBuilder.cs

@@ -0,0 +1,27 @@
+using System;
+using System.Collections.Generic;
+using Microsoft.EntityFrameworkCore.Infrastructure;
+using Microsoft.Extensions.DependencyInjection;
+
+namespace Ef.Ydb.Infrastructure;
+
+public class EntityFrameworkYdbServicesBuilder : EntityFrameworkRelationalServicesBuilder
+{
+    private static readonly IDictionary<Type, ServiceCharacteristics> YdbServices
+        = new Dictionary<Type, ServiceCharacteristics>
+        {
+            // TODO: Add items if required
+        };
+
+    public EntityFrameworkYdbServicesBuilder(IServiceCollection serviceCollection) : base(serviceCollection)
+    {
+    }
+
+    protected override ServiceCharacteristics GetServiceCharacteristics(Type serviceType)
+    {
+        var contains = YdbServices.TryGetValue(serviceType, out var characteristics);
+        return contains
+            ? characteristics
+            : base.GetServiceCharacteristics(serviceType);
+    }
+}

src/Ef.Ydb/src/Infrastructure/Internal/YdbModelValidator.cs

@@ -0,0 +1,14 @@
+using Microsoft.EntityFrameworkCore.Infrastructure;
+
+namespace Ef.Ydb.Infrastructure.Internal;
+
+// TODO: Not required for mvp
+public class YdbModelValidator : RelationalModelValidator
+{
+    public YdbModelValidator(
+        ModelValidatorDependencies dependencies,
+        RelationalModelValidatorDependencies relationalDependencies
+    ) : base(dependencies, relationalDependencies)
+    {
+    }
+}

src/Ef.Ydb/src/Infrastructure/Internal/YdbOptionsExtension.cs

@@ -0,0 +1,42 @@
+using System.Collections.Generic;
+using Ef.Ydb.Extensions;
+using Microsoft.EntityFrameworkCore.Infrastructure;
+using Microsoft.Extensions.DependencyInjection;
+
+namespace Ef.Ydb.Infrastructure.Internal;
+
+public class YdbOptionsExtension : RelationalOptionsExtension
+{
+    private DbContextOptionsExtensionInfo? _info;
+
+    public YdbOptionsExtension()
+    {
+    }
+
+    protected YdbOptionsExtension(YdbOptionsExtension copyFrom) : base(copyFrom)
+    {
+    }
+
+    protected override RelationalOptionsExtension Clone()
+        => new YdbOptionsExtension(this);
+
+    public override void ApplyServices(IServiceCollection services)
+        => services.AddEntityFrameworkYdb();
+
+    public override DbContextOptionsExtensionInfo Info
+        => _info ??= new ExtensionInfo(this);
+
+    private sealed class ExtensionInfo(IDbContextOptionsExtension extension) : RelationalExtensionInfo(extension)
+    {
+        private new YdbOptionsExtension Extension
+            => (YdbOptionsExtension)base.Extension;
+
+        public override bool IsDatabaseProvider => true;
+
+        // TODO: Right now it's stub
+        public override void PopulateDebugInfo(IDictionary<string, string> debugInfo)
+        {
+            debugInfo["Hello"] = "World!";
+        }
+    }
+}

src/Ef.Ydb/src/Infrastructure/YdbDbContextOptionsBuilder.cs

@@ -0,0 +1,13 @@
+using Ef.Ydb.Infrastructure.Internal;
+using Microsoft.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore.Infrastructure;
+
+namespace Ef.Ydb.Infrastructure;
+
+public class YdbDbContextOptionsBuilder
+    : RelationalDbContextOptionsBuilder<YdbDbContextOptionsBuilder, YdbOptionsExtension>
+{
+    public YdbDbContextOptionsBuilder(DbContextOptionsBuilder optionsBuilder) : base(optionsBuilder)
+    {
+    }
+}

src/Ef.Ydb/src/Metadata/Conventions/YdbConventionSetBuilder.cs

@@ -0,0 +1,14 @@
+using Microsoft.EntityFrameworkCore.Metadata.Conventions.Infrastructure;
+
+namespace Ef.Ydb.Metadata.Conventions;
+
+public class YdbConventionSetBuilder
+    : RelationalConventionSetBuilder
+{
+    public YdbConventionSetBuilder(
+        ProviderConventionSetBuilderDependencies dependencies,
+        RelationalConventionSetBuilderDependencies relationalDependencies
+    ) : base(dependencies, relationalDependencies)
+    {
+    }
+}