feat: add shared runtime and codegen support

2026-09-05 02:34:04 +08:00
parent 005ebf43cf
commit eb7fa55656
5 changed files with 26 additions and 182 deletions
@@ -1,13 +1,7 @@
 #nullable enable

-using System;
 using System.Collections.Generic;
-using System.IO;
-using System.Linq;
-using Mono.Cecil;
-using Mono.Cecil.Cil;
-using Mono.Cecil.Pdb;
-using ShrinkEventBus.CodeGen;
+using ShrinkShared.CodeGen;
 using Unity.CompilationPipeline.Common.Diagnostics;
 using Unity.CompilationPipeline.Common.ILPostProcessing;

@@ -15,170 +9,14 @@ namespace ShrinkNetwork.Integration.CodeGen
 {
    public sealed class NetworkEventILPostProcessor : ILPostProcessor
    {
-        private const string IntegrationAssembly = "ShrinkNetwork.Integration.EventBus";
-
        public override ILPostProcessor GetInstance() => this;

        public override bool WillProcess(ICompiledAssembly compiledAssembly) =>
-            compiledAssembly.References.Any(reference =>
-                string.Equals(Path.GetFileNameWithoutExtension(reference), IntegrationAssembly,
-                    StringComparison.Ordinal));
+            UnityShrinkCodeGenAdapter.ReferencesAny(compiledAssembly, "ShrinkNetwork.Integration.EventBus");

-        public override ILPostProcessResult Process(ICompiledAssembly compiledAssembly)
-        {
-            var diagnostics = new List<DiagnosticMessage>();
-            if (!WillProcess(compiledAssembly))
-                return new ILPostProcessResult(compiledAssembly.InMemoryAssembly, diagnostics);
-
-            var assembly = ReadAssembly(compiledAssembly);
-            try
-            {
-                InjectRegistrations(assembly.MainModule);
-            }
-            catch (Exception exception)
-            {
-                diagnostics.Add(new DiagnosticMessage
-                {
-                    DiagnosticType = DiagnosticType.Error,
-                    MessageData = $"[ShrinkNetwork.EventBus.CodeGen] {exception.Message}"
-                });
-            }
-            return WriteAssembly(assembly, diagnostics);
-        }
-
-        private static void InjectRegistrations(ModuleDefinition module)
-        {
-            var networkEventAttribute = ResolveType(module,
-                "ShrinkNetwork.Integration.ShrinkNetworkEventAttribute", IntegrationAssembly);
-            var messageAttribute = ResolveType(module,
-                "ShrinkNetwork.ShrinkNetworkMessageAttribute", "ShrinkNetwork.Runtime");
-            var registryType = ResolveType(module,
-                "ShrinkNetwork.Integration.ShrinkNetworkEventRegistry", IntegrationAssembly);
-            if (networkEventAttribute == null || messageAttribute == null || registryType == null)
-                return;
-
-            var registrations = GetAllTypes(module.Types)
-                .Where(type => !type.IsAbstract && HasAttribute(type, networkEventAttribute))
-                .Select(type => new
-                {
-                    Type = type,
-                    Message = type.CustomAttributes.FirstOrDefault(attribute =>
-                        attribute.AttributeType.FullName == messageAttribute.FullName)
-                })
-                .Where(item => item.Message != null)
-                .ToArray();
-            if (registrations.Length == 0)
-                return;
-
-            var registerDefinition = registryType.Resolve()?.Methods.Single(method =>
-                method.Name == "Register" && method.HasGenericParameters)
-                ?? throw new InvalidOperationException("ShrinkNetworkEventRegistry.Register<T> was not found.");
-            var registerOpen = module.ImportReference(registerDefinition);
-
-            var bootstrap = new TypeDefinition("ShrinkNetwork.Integration.Generated",
-                "ShrinkGeneratedNetworkEventBindings",
-                TypeAttributes.Abstract | TypeAttributes.Sealed | TypeAttributes.NotPublic,
-                module.TypeSystem.Object);
-            module.Types.Add(bootstrap);
-            var initialize = new MethodDefinition("Register",
-                MethodAttributes.Assembly | MethodAttributes.Static | MethodAttributes.HideBySig,
-                module.TypeSystem.Void);
-            bootstrap.Methods.Add(initialize);
-            var il = initialize.Body.GetILProcessor();
-
-            foreach (var item in registrations)
-            {
-                var attribute = item.Message!;
-                if (attribute.ConstructorArguments.Count == 0)
-                    throw new InvalidOperationException(
-                        $"[ShrinkNetworkMessage] on {item.Type.FullName} has no opcode.");
-
-                var opcode = Convert.ToInt32(attribute.ConstructorArguments[0].Value);
-                var route = attribute.ConstructorArguments.Count > 1
-                    ? attribute.ConstructorArguments[1].Value as string
-                    : null;
-                var registerClosed = new GenericInstanceMethod(registerOpen);
-                registerClosed.GenericArguments.Add(module.ImportReference(item.Type));
-                il.Emit(OpCodes.Ldc_I4, opcode);
-                if (route == null)
-                    il.Emit(OpCodes.Ldnull);
-                else
-                    il.Emit(OpCodes.Ldstr, route);
-                il.Emit(OpCodes.Call, registerClosed);
-            }
-            il.Emit(OpCodes.Ret);
-            InjectModuleInitializer(module, initialize);
-        }
-
-        private static void InjectModuleInitializer(ModuleDefinition module, MethodReference register)
-        {
-            var moduleType = module.Types.First(type => type.Name == "<Module>");
-            var initializer = moduleType.Methods.FirstOrDefault(method => method.Name == ".cctor");
-            if (initializer == null)
-            {
-                initializer = new MethodDefinition(".cctor",
-                    MethodAttributes.Private | MethodAttributes.Static | MethodAttributes.HideBySig |
-                    MethodAttributes.SpecialName | MethodAttributes.RTSpecialName,
-                    module.TypeSystem.Void);
-                initializer.Body.GetILProcessor().Emit(OpCodes.Ret);
-                moduleType.Methods.Add(initializer);
-            }
-            var il = initializer.Body.GetILProcessor();
-            il.InsertBefore(initializer.Body.Instructions[0], il.Create(OpCodes.Call, register));
-        }
-
-        private static bool HasAttribute(ICustomAttributeProvider provider, TypeReference attributeType) =>
-            provider.CustomAttributes.Any(attribute =>
-                attribute.AttributeType.FullName == attributeType.FullName);
-
-        private static IEnumerable<TypeDefinition> GetAllTypes(IEnumerable<TypeDefinition> roots)
-        {
-            foreach (var type in roots)
-            {
-                yield return type;
-                foreach (var nested in GetAllTypes(type.NestedTypes))
-                    yield return nested;
-            }
-        }
-
-        private static TypeReference? ResolveType(ModuleDefinition module, string fullName,
-            string assemblyName)
-        {
-            var reference = module.AssemblyReferences.FirstOrDefault(item => item.Name == assemblyName);
-            var definition = reference == null ? null : module.AssemblyResolver.Resolve(reference);
-            var type = definition?.MainModule.GetType(fullName);
-            return type == null ? null : module.ImportReference(type);
-        }
-
-        private static AssemblyDefinition ReadAssembly(ICompiledAssembly compiledAssembly)
-        {
-            var resolver = new PostProcessorAssemblyResolver(compiledAssembly);
-            var parameters = new ReaderParameters
-            {
-                SymbolStream = new MemoryStream(compiledAssembly.InMemoryAssembly.PdbData.ToArray()),
-                SymbolReaderProvider = new PdbReaderProvider(),
-                AssemblyResolver = resolver,
-                ReflectionImporterProvider = new PostProcessorReflectionImporterProvider(),
-                ReadingMode = ReadingMode.Immediate
-            };
-            var assembly = AssemblyDefinition.ReadAssembly(
-                new MemoryStream(compiledAssembly.InMemoryAssembly.PeData.ToArray()), parameters);
-            resolver.AddAssemblyDefinitionBeingOperatedOn(assembly);
-            return assembly;
-        }
-
-        private static ILPostProcessResult WriteAssembly(AssemblyDefinition assembly,
-            List<DiagnosticMessage> diagnostics)
-        {
-            var pe = new MemoryStream();
-            var pdb = new MemoryStream();
-            assembly.Write(pe, new WriterParameters
-            {
-                SymbolWriterProvider = new PdbWriterProvider(),
-                SymbolStream = pdb,
-                WriteSymbols = true
-            });
-            return new ILPostProcessResult(new InMemoryAssembly(pe.ToArray(), pdb.ToArray()), diagnostics);
-        }
+        public override ILPostProcessResult Process(ICompiledAssembly compiledAssembly) =>
+            WillProcess(compiledAssembly)
+                ? UnityShrinkCodeGenAdapter.Process(compiledAssembly, "ShrinkNetwork.EventBus.CodeGen")
+                : new ILPostProcessResult(compiledAssembly.InMemoryAssembly, new List<DiagnosticMessage>());
    }
 }
@@ -3,7 +3,8 @@
  "rootNamespace": "ShrinkNetwork.Integration.CodeGen",
  "references": [
    "ShrinkNetwork.Integration.EventBus",
-    "Unity.ShrinkEventBus.CodeGen"
+        "Unity.ShrinkEventBus.CodeGen",
+        "Unity.ShrinkShared.CodeGen"
  ],
  "includePlatforms": ["Editor"],
  "excludePlatforms": [],
@@ -5,6 +5,7 @@
    "ShrinkNetwork.Runtime",
    "ShrinkContext.Core.Runtime",
    "ShrinkEventBus.Runtime",
+    "ShrinkRuntime.Abstractions",
    "UniTask"
  ],
  "optionalUnityReferences": [],
@@ -5,7 +5,7 @@ using System.Linq;
 using System.Threading;
 using Cysharp.Threading.Tasks;
 using ShrinkEventBus;
-using UnityEngine;
+using ShrinkSDK.Runtime;

 namespace ShrinkNetwork.Integration
 {
@@ -24,7 +24,9 @@ namespace ShrinkNetwork.Integration
        private static Dictionary<Type, ShrinkNetworkEventRegistration> _networkEventTypes = new();
        private static bool _initialized;

-        [RuntimeInitializeOnLoadMethod(RuntimeInitializeLoadType.SubsystemRegistration)]
+#if UNITY_5_3_OR_NEWER
+        [UnityEngine.RuntimeInitializeOnLoadMethod(UnityEngine.RuntimeInitializeLoadType.SubsystemRegistration)]
+#endif
        private static void ResetStaticState()
        {
            EventBus.Posted -= HandleLocalEventPosted;
@@ -215,9 +217,8 @@ namespace ShrinkNetwork.Integration
                }
                catch (Exception ex)
                {
-                    Debug.LogException(ex);
-                    Debug.LogError(
-                        $"[ShrinkNetwork.Integration] 序列化网络事件失败: {eventArgs.GetType().FullName}");
+                    ShrinkRuntimeServices.Logger.Error(
+                        $"[ShrinkNetwork.Integration] 序列化网络事件失败: {eventArgs.GetType().FullName}", ex);
                    continue;
                }

@@ -236,9 +237,8 @@ namespace ShrinkNetwork.Integration
                    }
                    catch (Exception ex)
                    {
-                        Debug.LogException(ex);
-                        Debug.LogError(
-                            $"[ShrinkNetwork.Integration] 转发网络事件失败: {eventArgs.GetType().FullName}, Session={session.SessionId}");
+                        ShrinkRuntimeServices.Logger.Error(
+                            $"[ShrinkNetwork.Integration] 转发网络事件失败: {eventArgs.GetType().FullName}, Session={session.SessionId}", ex);
                    }
                }
            }
@@ -349,7 +349,7 @@ namespace ShrinkNetwork.Integration
            }
            catch (Exception ex)
            {
-                Debug.LogException(ex);
+                ShrinkRuntimeServices.Logger.Error("[ShrinkNetwork.Integration] 处理网络事件请求失败。", ex);
                return new ShrinkNetworkEventResultResponse
                {
                    ErrorCode = ShrinkRpcErrorCode.HandlerException,
@@ -481,6 +481,7 @@ namespace ShrinkNetwork.Integration
    /// handlers. Use ShrinkNetworkDispatchQueue when a hard per-frame budget is
    /// required instead of an unbounded PlayerLoop backlog.
    /// </summary>
+#if UNITY_5_3_OR_NEWER
    public sealed class ShrinkNetworkUnityMainThreadDispatchScheduler : IShrinkNetworkDispatchScheduler
    {
        public async UniTask<bool> ScheduleAsync(Func<UniTask> callback)
@@ -493,4 +494,5 @@ namespace ShrinkNetwork.Integration
            return true;
        }
    }
+#endif
 }
@@ -1,13 +1,15 @@
 {
  "name": "com.cneicy.shrink-network-integration-eventbus",
-  "version": "0.1.3",
+  "version": "0.2.0",
  "displayName": "ShrinkNetwork - EventBus Integration",
  "description": "ShrinkNetwork 与 ShrinkEventBus 2.0 的生成式桥接层，让 IShrinkEvent 网络事件可广播、裁决并在远端重新进入命名 Bus。",
  "unity": "2022.3",
  "dependencies": {
-    "com.cneicy.shrink-network": "0.2.1",
-    "com.cneicy.shrink-context-core": "0.1.0",
-    "com.cneicy.shrink-eventbus": "2.0.1"
+    "com.cneicy.shrink-network": "0.3.0",
+    "com.cneicy.shrink-context-core": "0.2.0",
+    "com.cneicy.shrink-eventbus": "2.1.0",
+    "com.cneicy.shrink-shared-codegen": "0.1.0",
+    "com.cneicy.shrink-runtime-abstractions": "0.1.0"
  },
  "keywords": [
    "network",