Get bridge data passed through

src/Temporalio/Api/Activity/V1/Message.cs

@@ -50,7 +50,6 @@ static MessageReflection() {
 
   }
   #region Messages
-  [global::System.Diagnostics.DebuggerDisplayAttribute("{ToString(),nq}")]
   public sealed partial class ActivityOptions : pb::IMessage<ActivityOptions>
   #if !GOOGLE_PROTOBUF_REFSTRUCT_COMPATIBILITY_MODE
       , pb::IBufferMessage
@@ -396,11 +395,7 @@ public void MergeFrom(pb::CodedInputStream input) {
     #else
       uint tag;
       while ((tag = input.ReadTag()) != 0) {
-      if ((tag & 7) == 4) {
-        // Abort on any end group tag.
-        return;
-      }
-      switch(tag) {
+        switch(tag) {
           default:
             _unknownFields = pb::UnknownFieldSet.MergeFieldFrom(_unknownFields, input);
             break;
@@ -457,11 +452,7 @@ public void MergeFrom(pb::CodedInputStream input) {
     void pb::IBufferMessage.InternalMergeFrom(ref pb::ParseContext input) {
       uint tag;
       while ((tag = input.ReadTag()) != 0) {
-      if ((tag & 7) == 4) {
-        // Abort on any end group tag.
-        return;
-      }
-      switch(tag) {
+        switch(tag) {
           default:
             _unknownFields = pb::UnknownFieldSet.MergeFieldFrom(_unknownFields, ref input);
             break;

src/Temporalio/Bridge/CustomMetricMeter.cs

@@ -7,11 +7,10 @@ namespace Temporalio.Bridge
     /// <summary>
     /// Core wrapper for a custom metric meter implementation.
     /// </summary>
-    internal class CustomMetricMeter
+    internal class CustomMetricMeter : NativeInvokeableClass<Interop.CustomMetricMeter>
     {
         private readonly Temporalio.Runtime.ICustomMetricMeter meter;
         private readonly Temporalio.Runtime.CustomMetricMeterOptions options;
-        private readonly List<GCHandle> handles = new();
 
         /// <summary>
         /// Initializes a new instance of the <see cref="CustomMetricMeter" /> class.
@@ -38,20 +37,9 @@ public unsafe CustomMetricMeter(
                 meter_free = FunctionPointer<Interop.CustomMetricMeterMeterFreeCallback>(Free),
             };
 
-            // Pin the metric meter pointer and set it as the first handle
-            var interopMeterHandle = GCHandle.Alloc(interopMeter, GCHandleType.Pinned);
-            handles.Insert(0, interopMeterHandle);
-            Ptr = (Interop.CustomMetricMeter*)interopMeterHandle.AddrOfPinnedObject();
-
-            // Add handle for ourself
-            handles.Add(GCHandle.Alloc(this));
+            PinCallbackHolder(interopMeter);
         }
 
-        /// <summary>
-        /// Gets the pointer to the native metric meter.
-        /// </summary>
-        internal unsafe Interop.CustomMetricMeter* Ptr { get; private init; }
-
         private static unsafe string? GetStringOrNull(Interop.ByteArrayRef bytes) =>
             (int)bytes.size == 0 ? null : GetString(bytes);
 
@@ -220,23 +208,5 @@ private unsafe void RecordMetricDuration(void* metric, ulong valueMs, void* attr
         }
 
         private unsafe void FreeAttributes(void* attributes) => GCHandle.FromIntPtr(new(attributes)).Free();
-
-        private unsafe void Free(Interop.CustomMetricMeter* meter)
-        {
-            // Free in order which frees function pointers first then meter handles
-            foreach (var handle in handles)
-            {
-                handle.Free();
-            }
-        }
-
-        // Similar to Scope.FunctionPointer
-        private IntPtr FunctionPointer<T>(T func)
-            where T : Delegate
-        {
-            var handle = GCHandle.Alloc(func);
-            handles.Add(handle);
-            return Marshal.GetFunctionPointerForDelegate(handle.Target!);
-        }
     }
-}
+}

src/Temporalio/Bridge/CustomSlotSupplier.cs

@@ -0,0 +1,52 @@
+using System.Threading.Tasks;
+
+namespace Temporalio.Bridge
+{
+    /// <summary>
+    /// Core wrapper for a user-defined custom slot supplier.
+    /// </summary>
+    internal class CustomSlotSupplier : NativeInvokeableClass<Interop.CustomSlotSupplierCallbacks>
+    {
+        private readonly Temporalio.Worker.Tuning.ICustomSlotSupplier userSupplier;
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="CustomSlotSupplier" /> class.
+        /// </summary>
+        /// <param name="userSupplier">User's slot supplier implementation'.</param>
+        internal unsafe CustomSlotSupplier(Temporalio.Worker.Tuning.ICustomSlotSupplier userSupplier)
+        {
+            this.userSupplier = userSupplier;
+
+            var interopCallbacks = new Interop.CustomSlotSupplierCallbacks
+            {
+                reserve = FunctionPointer<Interop.CustomReserveSlotCallback>(Reserve),
+                try_reserve = FunctionPointer<Interop.CustomTryReserveSlotCallback>(TryReserve),
+                mark_used = FunctionPointer<Interop.CustomMarkSlotUsedCallback>(MarkUsed),
+                release = FunctionPointer<Interop.CustomReleaseSlotCallback>(Release),
+            };
+
+            PinCallbackHolder(interopCallbacks);
+        }
+
+        private void Reserve(Interop.SlotReserveCtx ctx)
+        {
+            // TODO: Need to call callback with result that will put it in a channel to await in Rust
+            var reserveTask = Task.Run(() => userSupplier.ReserveSlotAsync(new(ctx)));
+        }
+
+        private void TryReserve(Interop.SlotReserveCtx ctx)
+        {
+            userSupplier.TryReserveSlot(new(ctx));
+        }
+
+        private void MarkUsed(Interop.SlotMarkUsedCtx ctx)
+        {
+            userSupplier.MarkSlotUsed(new(ctx));
+        }
+
+        private void Release(Interop.SlotReleaseCtx ctx)
+        {
+            userSupplier.ReleaseSlot(new(ctx));
+        }
+    }
+}