feat: add NativeAotCounter sample and simplify AOT registration API

Add AddRestateGenerated(JsonSerializerContext?) parameter to auto-configure
JsonSerde when a context is provided, reducing AOT setup from two calls to one.
Remove unused AddRestateAot(services, definitions, Type[]) overload.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: stevefan1999-personal

samples/NativeAotCounter/AppJsonContext.cs

@@ -0,0 +1,9 @@
+using System.Text.Json.Serialization;
+
+namespace NativeAotCounter;
+
+[JsonSerializable(typeof(int))]
+[JsonSourceGenerationOptions(
+    PropertyNamingPolicy = JsonKnownNamingPolicy.CamelCase,
+    DefaultIgnoreCondition = JsonIgnoreCondition.WhenWritingNull)]
+internal partial class AppJsonContext : JsonSerializerContext;

samples/NativeAotCounter/CounterObject.cs

@@ -0,0 +1,34 @@
+using Restate.Sdk;
+
+namespace NativeAotCounter;
+
+/// <summary>
+///     A Virtual Object that maintains a durable counter per key, compiled with NativeAOT.
+/// </summary>
+[VirtualObject]
+public sealed class CounterObject
+{
+    private static readonly StateKey<int> Count = new("count");
+
+    [Handler]
+    public async Task<int> Add(ObjectContext ctx, int delta)
+    {
+        var current = await ctx.Get(Count);
+        var next = current + delta;
+        ctx.Set(Count, next);
+        return next;
+    }
+
+    [Handler]
+    public Task Reset(ObjectContext ctx)
+    {
+        ctx.ClearAll();
+        return Task.CompletedTask;
+    }
+
+    [SharedHandler]
+    public async Task<int> Get(SharedObjectContext ctx)
+    {
+        return await ctx.Get(Count);
+    }
+}

samples/NativeAotCounter/NativeAotCounter.csproj

@@ -0,0 +1,16 @@
+<Project Sdk="Microsoft.NET.Sdk">
+
+    <PropertyGroup>
+        <OutputType>Exe</OutputType>
+        <IsPackable>false</IsPackable>
+        <PublishAot>true</PublishAot>
+    </PropertyGroup>
+
+    <ItemGroup>
+        <ProjectReference Include="../../src/Restate.Sdk/Restate.Sdk.csproj"/>
+        <ProjectReference Include="../../src/Restate.Sdk.Generators/Restate.Sdk.Generators.csproj"
+                          OutputItemType="Analyzer"
+                          ReferenceOutputAssembly="false"/>
+    </ItemGroup>
+
+</Project>

samples/NativeAotCounter/Program.cs

@@ -0,0 +1,11 @@
+using NativeAotCounter;
+using Restate.Sdk.Generated;
+using Restate.Sdk.Hosting;
+
+// NativeAOT-compatible Restate endpoint with a Virtual Object.
+// Publish with: dotnet publish -c Release
+await RestateHost
+    .CreateBuilder()
+    .WithPort(9086)
+    .BuildAot(services => services.AddRestateGenerated(AppJsonContext.Default))
+    .RunAsync();

samples/NativeAotGreeter/Program.cs

@@ -1,5 +1,4 @@
 using NativeAotGreeter;
-using Restate.Sdk;
 using Restate.Sdk.Generated;
 using Restate.Sdk.Hosting;
 
@@ -9,10 +8,5 @@
 await RestateHost
     .CreateBuilder()
     .WithPort(9085)
-    .BuildAot(services =>
-    {
-        // Wire the source-generated JSON context for AOT-safe serialization.
-        JsonSerde.Configure(AppJsonContext.Default.Options);
-        services.AddRestateGenerated();
-    })
+    .BuildAot(services => services.AddRestateGenerated(AppJsonContext.Default))
     .RunAsync();

src/Restate.Sdk.Generators/Emitters/RegistrationEmitter.cs

@@ -18,18 +18,24 @@ public static string Generate(ImmutableArray<ServiceInfo> services)
         sb.AppendLine();
         sb.AppendLine("/// <summary>");
         sb.AppendLine("///     Source-generated AOT-friendly service registration.");
-        sb.AppendLine("///     Eliminates reflection-based DI registration for NativeAOT compatibility.");
         sb.AppendLine("/// </summary>");
         sb.AppendLine("internal static class RestateRegistration");
         sb.AppendLine("{");
         sb.AppendLine("    /// <summary>");
         sb.AppendLine("    ///     Registers all discovered Restate services into the DI container without reflection.");
-        sb.AppendLine("    ///     Call this from <c>RestateHostBuilder.BuildAot()</c> or directly in your DI setup.");
         sb.AppendLine("    /// </summary>");
         sb.AppendLine("    public static global::Microsoft.Extensions.DependencyInjection.IServiceCollection AddRestateGenerated(");
-        sb.AppendLine("        this global::Microsoft.Extensions.DependencyInjection.IServiceCollection services)");
+        sb.AppendLine("        this global::Microsoft.Extensions.DependencyInjection.IServiceCollection services,");
+        sb.AppendLine("        global::System.Text.Json.Serialization.JsonSerializerContext? jsonContext = null)");
         sb.AppendLine("    {");
 
+        // Configure JsonSerde when a context is provided
+        sb.AppendLine("        if (jsonContext is not null)");
+        sb.AppendLine("        {");
+        sb.AppendLine("            global::Restate.Sdk.JsonSerde.Configure(jsonContext.Options);");
+        sb.AppendLine("        }");
+        sb.AppendLine();
+
         // Build the definitions array
         sb.AppendLine("        var definitions = new global::Restate.Sdk.Endpoint.ServiceDefinition[]");
         sb.AppendLine("        {");
@@ -46,7 +52,6 @@ public static string Generate(ImmutableArray<ServiceInfo> services)
         sb.AppendLine("        };");
         sb.AppendLine();
 
-        // Call the public AddRestateAot method (definitions only — no Type[] needed)
         sb.AppendLine("        global::Restate.Sdk.Hosting.RestateServiceCollectionExtensions.AddRestateAot(services, definitions);");
         sb.AppendLine();
 

src/Restate.Sdk/Hosting/RestateServiceCollectionExtensions.cs

@@ -63,33 +63,4 @@ ServiceDefinition[] definitions
 
         return services;
     }
-
-    /// <inheritdoc cref="AddRestateAot(IServiceCollection, ServiceDefinition[])" />
-    /// <param name="services">The service collection.</param>
-    /// <param name="definitions">Service definitions resolved from the source-generated registry.</param>
-    /// <param name="serviceTypes">The CLR types of the service implementations for reflection-based DI registration.</param>
-    [EditorBrowsable(EditorBrowsableState.Never)]
-    [UnconditionalSuppressMessage(
-        "AOT",
-        "IL2072",
-        Justification = "Service types are passed from source-generated code using typeof(T), which preserves the type metadata."
-    )]
-    [UnconditionalSuppressMessage(
-        "Trimming",
-        "IL2062",
-        Justification = "Service types are passed from source-generated code using typeof(T), which preserves the type metadata."
-    )]
-    public static IServiceCollection AddRestateAot(
-        this IServiceCollection services,
-        ServiceDefinition[] definitions,
-        Type[] serviceTypes
-    )
-    {
-        AddRestateAot(services, definitions);
-
-        foreach (var type in serviceTypes)
-            services.TryAddScoped(type);
-
-        return services;
-    }
 }