feat: add shared runtime and codegen support

2026-09-05 02:33:58 +08:00
parent a6011cd01d
commit 8aa4c79604
8 changed files with 125 additions and 776 deletions
@@ -0,0 +1,80 @@
+#nullable enable
+
+using System;
+using System.Threading;
+using ShrinkSDK.Runtime;
+
+namespace ShrinkEventBus
+{
+    public static class ShrinkEventBusRuntime
+    {
+        private static IShrinkMainThreadDispatcher _mainThreadDispatcher = CreateDefaultDispatcher();
+
+        public static IShrinkMainThreadDispatcher MainThreadDispatcher =>
+            Volatile.Read(ref _mainThreadDispatcher);
+
+        public static void ConfigureMainThreadDispatcher(IShrinkMainThreadDispatcher dispatcher)
+        {
+            if (dispatcher == null)
+                throw new ArgumentNullException(nameof(dispatcher));
+            Volatile.Write(ref _mainThreadDispatcher, dispatcher);
+        }
+
+        private static IShrinkMainThreadDispatcher CreateDefaultDispatcher()
+        {
+#if UNITY_5_3_OR_NEWER
+            return new ShrinkUnityMainThreadDispatcher();
+#else
+            return new ShrinkSynchronizationContextDispatcher(
+                SynchronizationContext.Current,
+                Thread.CurrentThread.ManagedThreadId);
+#endif
+        }
+    }
+
+#if UNITY_5_3_OR_NEWER
+    internal sealed class ShrinkUnityMainThreadDispatcher : IShrinkMainThreadDispatcher
+    {
+        public bool IsMainThread => Cysharp.Threading.Tasks.PlayerLoopHelper.IsMainThread;
+
+        public bool TryPost(Action action)
+        {
+            if (action == null)
+                throw new ArgumentNullException(nameof(action));
+            Cysharp.Threading.Tasks.PlayerLoopHelper.AddContinuation(
+                Cysharp.Threading.Tasks.PlayerLoopTiming.Update,
+                action);
+            return true;
+        }
+    }
+#else
+    internal sealed class ShrinkSynchronizationContextDispatcher : IShrinkMainThreadDispatcher
+    {
+        private readonly SynchronizationContext? _context;
+        private readonly int _threadId;
+
+        public ShrinkSynchronizationContextDispatcher(SynchronizationContext? context, int threadId)
+        {
+            _context = context;
+            _threadId = threadId;
+        }
+
+        public bool IsMainThread => Thread.CurrentThread.ManagedThreadId == _threadId;
+
+        public bool TryPost(Action action)
+        {
+            if (action == null)
+                throw new ArgumentNullException(nameof(action));
+            if (IsMainThread)
+            {
+                action();
+                return true;
+            }
+            if (_context == null)
+                return false;
+            _context.Post(static state => ((Action)state!).Invoke(), action);
+            return true;
+        }
+    }
+#endif
+}