Add plugin sample

samples/core/SqlServer/Plugin/AugmentationTranslator.cs

@@ -0,0 +1,24 @@
+using System;
+using System.Collections.Generic;
+using System.Linq.Expressions;
+using System.Reflection;
+using Microsoft.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore.Diagnostics;
+using Microsoft.EntityFrameworkCore.Query;
+using Microsoft.EntityFrameworkCore.Query.SqlExpressions;
+
+namespace SqlServer.Plugin;
+internal class AugmentationTranslator : IMethodCallTranslator
+{
+    public SqlExpression Translate(SqlExpression instance, MethodInfo method, IReadOnlyList<SqlExpression> arguments, IDiagnosticsLogger<DbLoggerCategory.Query> logger)
+    {
+        switch (method.Name)
+        {
+            case nameof(DbFunctionsExtensions.Augment):
+                var argument = arguments[1];
+                return new SqlBinaryExpression(ExpressionType.Add, argument, new SqlConstantExpression(Expression.Constant(1), null), argument.Type, null);
+            default:
+                return null;
+        }
+    }
+}

samples/core/SqlServer/Plugin/Blog.cs

@@ -0,0 +1,9 @@
+using System.ComponentModel.DataAnnotations.Schema;
+
+namespace SqlServer.Plugin;
+public class Blog
+{
+    [DatabaseGeneratedAttribute(DatabaseGeneratedOption.None)]
+    public int BlogId { get; set; }
+    public string Url { get; set; }
+}

samples/core/SqlServer/Plugin/BloggingContext.cs

@@ -0,0 +1,13 @@
+using Microsoft.EntityFrameworkCore;
+
+namespace SqlServer.Plugin;
+public class BloggingContext : DbContext
+{
+    public DbSet<Blog> Blogs { get; set; }
+
+    protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
+        => optionsBuilder.UseSqlServer(
+            @"Server=(localdb)\mssqllocaldb;Database=EFSaving.Basics;Trusted_Connection=True"
+            , o => o.UseAugmentation()
+            );
+}

samples/core/SqlServer/Plugin/CustomPlugin.cs

@@ -0,0 +1,26 @@
+using System.Linq;
+using Microsoft.EntityFrameworkCore;
+
+namespace SqlServer.Plugin;
+
+public class CustomPlugin
+{
+    public static void Run()
+    {
+        using (var context = new BloggingContext())
+        {
+            context.Database.EnsureDeleted();
+            context.Database.EnsureCreated();
+        }
+
+        using (var context = new BloggingContext())
+        {
+            context.Blogs.Add(new Blog { BlogId = 100, Url = "http://blog1.somesite.com" });
+
+            context.SaveChanges();
+
+            var simple = context.Blogs.Select(b => EF.Functions.Augment(b.BlogId)).First();
+            var withCast = context.Blogs.Select(b => EF.Functions.Augment((long)b.BlogId)).First();
+        }
+    }
+}

samples/core/SqlServer/Plugin/DbFunctionsExtensions.cs

@@ -0,0 +1,10 @@
+using System;
+using Microsoft.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore.Diagnostics;
+
+namespace SqlServer.Plugin;
+public static class DbFunctionsExtensions
+{
+    public static T Augment<T>(this DbFunctions _, T number)
+        => throw new InvalidOperationException(CoreStrings.FunctionOnClient(nameof(Augment)));
+}

samples/core/SqlServer/Plugin/PluginSetup.cs

@@ -0,0 +1,83 @@
+using System.Collections.Generic;
+using Microsoft.EntityFrameworkCore.Infrastructure;
+using Microsoft.EntityFrameworkCore.Query;
+using Microsoft.Extensions.DependencyInjection;
+
+namespace SqlServer.Plugin;
+public class AugmentationDbFunctionsExtensions : IDbContextOptionsExtension
+{
+    private ExtensionInfo _info;
+
+    public virtual DbContextOptionsExtensionInfo Info
+        => _info ??= new ExtensionInfo(this);
+
+    public void ApplyServices(IServiceCollection services) => services.AddAugmentationExtension();
+
+    public void Validate(IDbContextOptions options)
+    {
+    }
+
+    private sealed class ExtensionInfo : DbContextOptionsExtensionInfo
+    {
+        public ExtensionInfo(IDbContextOptionsExtension extension)
+            : base(extension)
+        {
+        }
+
+        private new AugmentationDbFunctionsExtensions Extension
+            => (AugmentationDbFunctionsExtensions)base.Extension;
+
+        public override bool IsDatabaseProvider
+            => false;
+
+        public override int GetServiceProviderHashCode()
+            => 0;
+
+        public override bool ShouldUseSameServiceProvider(DbContextOptionsExtensionInfo other)
+            => other is ExtensionInfo;
+
+        public override void PopulateDebugInfo(IDictionary<string, string> debugInfo)
+            => debugInfo["SqlServer:" + nameof(SqlServerExtensions.UseAugmentation)] = "1";
+
+        public override string LogFragment
+            => "using Augmentation";
+    }
+}
+
+public static class AugmentationServiceCollectionExtensions
+{
+    public static IServiceCollection AddAugmentationExtension(
+        this IServiceCollection serviceCollection)
+    {
+        new EntityFrameworkRelationalServicesBuilder(serviceCollection)
+            .TryAdd<IMethodCallTranslatorPlugin, AugmentationTranslatorPlugin>();
+
+        return serviceCollection;
+    }
+}
+
+public class AugmentationTranslatorPlugin : IMethodCallTranslatorPlugin
+{
+    public IEnumerable<IMethodCallTranslator> Translators { get; }
+
+    public AugmentationTranslatorPlugin()
+    {
+        var list = new List<IMethodCallTranslator>
+        {
+            new AugmentationTranslator()
+        };
+        Translators = list;
+    }
+}
+
+public static class SqlServerExtensions
+{
+    public static SqlServerDbContextOptionsBuilder UseAugmentation(this SqlServerDbContextOptionsBuilder optionsBuilder)
+    {
+        var coreOptionsBuilder = ((IRelationalDbContextOptionsBuilderInfrastructure)optionsBuilder).OptionsBuilder;
+        var extension = coreOptionsBuilder.Options.FindExtension<AugmentationDbFunctionsExtensions>() ?? new AugmentationDbFunctionsExtensions();
+        ((IDbContextOptionsBuilderInfrastructure)coreOptionsBuilder).AddOrUpdateExtension(extension);
+
+        return optionsBuilder;
+    }
+}

samples/core/SqlServer/Program.cs

@@ -4,6 +4,7 @@ public class Program
     {
         private static void Main()
         {
+            Plugin.CustomPlugin.Run();
         }
     }
 }