Как то работает, но плохо...

2025-03-11 19:42:10 +03:00
parent 76f395dce5
commit 4033ae53eb
16 changed files with 148 additions and 102 deletions
@@ -70,7 +70,7 @@ namespace mROA.Implementation.Backend

                var context = command.ObjectId.ContextId != -1
                    ? contextRepository.GetObject<object>(command.ObjectId)
-                    : contextRepository.GetSingleObject(invoker.SuitableType);
+                    : contextRepository.GetSingleObject(invoker.SuitableType, command.ObjectId.OwnerId);

                if (context == null)
                    throw new NullReferenceException("Instance can't be null");
@@ -23,26 +23,20 @@ namespace mROA.Implementation.Backend

        // [CanBeNull]
        private Dictionary<int, object?> _singletons;
-        private object?[] _storage;
+        private IStorage<object> _storage;


        public ContextRepository()
        {
-            _storage = new object[StartupSize];
+            _storage = new ExtensibleStorage<object>();
        }

        public int HostId { get; set; }

        public int ResisterObject<T>(object o, IEndPointContext context)
        {
-            if (!_lastIndexFinder.IsCompleted)
-                _lastIndexFinder.Wait();
+            var last = _storage.Place(o);

-            _storage[_lastIndexFinder.Result] = o;
-
-
-            var last = _lastIndexFinder.Result;
-            _lastIndexFinder = Task.Run(FindLastIndex);
            EventBinders.OfType<IEventBinder<T>>().FirstOrDefault()
                ?.BindEvents((T)o, context, _representationModuleProducer!, last);

@@ -51,18 +45,22 @@ namespace mROA.Implementation.Backend

        public void ClearObject(ComplexObjectIdentifier id)
        {
-            _storage[id.ContextId] = null;
-            _lastIndexFinder = Task.FromResult(id.ContextId);
+            _storage.Free(id.ContextId);
        }

        public T GetObject<T>(ComplexObjectIdentifier id)
        {
-            return id.ContextId == -1 || _storage.Length <= id.ContextId
-                ? throw new NullReferenceException("Cannot find that object. It is null")
-                : (T)_storage[id.ContextId]!;
+            var value = _storage.GetValue(id.ContextId);
+            
+            if (value == null)
+            {
+                throw new NullReferenceException("Cannot find that object. It is null");
+            }
+
+            return (T)value;
        }

-        public object GetSingleObject(Type type)
+        public object GetSingleObject(Type type, int ownerId)
        {
            return _singletons.GetValueOrDefault(type.GetHashCode()) ??
                   throw new ArgumentException("Unregistered singleton type");
@@ -70,7 +68,7 @@ namespace mROA.Implementation.Backend

        public int GetObjectIndex<T>(object o, IEndPointContext context)
        {
-            var index = Array.IndexOf(_storage, o);
+            var index = _storage.GetIndex(o);
            return index == -1 ? ResisterObject<T>(o, context) : index;
        }

@@ -93,19 +91,5 @@ namespace mROA.Implementation.Backend
                        i.GetCustomAttributes(typeof(SharedObjectInterfaceAttribute), true).Length > 0)!.GetHashCode(),
                    Activator.CreateInstance);
        }
-
-        private int FindLastIndex()
-        {
-            for (var i = 0; i < _storage.Length; i++)
-            {
-                if (_storage[i] is null)
-                    return i;
-            }
-
-            var nextStorage = new object[_storage.Length + GrowSize];
-            Array.Copy(_storage, nextStorage, _storage.Length);
-            _storage = nextStorage;
-            return _storage.Length;
-        }
    }
 }
@@ -8,6 +8,7 @@ namespace mROA.Implementation.Backend
        private IConnectionHub? _hub;

        private IContextRepository? _contextRepository;
+        private IContextRepository? _remoteContextRepository;
        private IMethodRepository? _methodRepository;
        private ISerializationToolkit? _serializationToolkit;
        private IExecuteModule? _executeModule;
@@ -26,8 +27,12 @@ namespace mROA.Implementation.Backend
                    _hub = connectionHub;
                    _hub.OnConnected += HubOnOnConnected;
                    break;
-                case IContextRepository contextRepository:
-                    _contextRepository = contextRepository;
+                case MultiClientContextRepository:
+                case ContextRepository:
+                    _contextRepository = dependency as IContextRepository;
+                    break;
+                case RemoteContextRepository remoteContextRepository:
+                    _remoteContextRepository = remoteContextRepository;
                    break;
                case IMethodRepository methodRepository:
                    _methodRepository = methodRepository;
@@ -52,6 +57,7 @@ namespace mROA.Implementation.Backend
            extractor.Inject(_methodRepository);
            extractor.Inject(_serializationToolkit);
            extractor.Inject(_executeModule);
+            extractor.Inject(_remoteContextRepository);
            _ = extractor.StartExtraction();
        }
    }
@@ -38,10 +38,10 @@ namespace mROA.Implementation.Backend
            return repository.GetObject<T>(id);
        }

-        public object GetSingleObject(Type type)
+        public object GetSingleObject(Type type, int ownerId)
        {
            var repository = GetRepositoryByClientId(TransmissionConfig.OwnershipRepository.GetOwnershipId());
-            return repository.GetSingleObject(type);
+            return repository.GetSingleObject(type, ownerId);
        }

        public int GetObjectIndex<T>(object o, IEndPointContext context)
@@ -0,0 +1,74 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using mROA.Abstract;
+
+namespace mROA.Implementation
+{
+    public class ComplexContextRepository : IContextRepository
+    {
+        private List<KeyValuePair<int, ExtensibleStorage<object>>> _storages = new();
+        public static object[] EventBinders = { };
+
+        private IRemoteObjectFactory? _remoteObjectFactory;
+        private IRepresentationModuleProducer? _representationModuleProducer;
+
+        public void Inject<T>(T dependency)
+        {
+            if (dependency is IRemoteObjectFactory remoteObjectFactory)
+            {
+                _remoteObjectFactory = remoteObjectFactory;
+            }
+
+            if (dependency is IRepresentationModuleProducer moduleProducer)
+            {
+                _representationModuleProducer = moduleProducer;
+            }
+        }
+
+        public int HostId { get; set; }
+
+        public int ResisterObject<T>(object o, IEndPointContext context)
+        {
+            var storageIndex = _storages.FindIndex(i => i.Key == context.OwnerId);
+            if (storageIndex == -1)
+            {
+                _storages.Add(
+                    new KeyValuePair<int, ExtensibleStorage<object>>(context.OwnerId, new ExtensibleStorage<object>()));
+                storageIndex = _storages.Count - 1;
+            }
+
+            var storage = _storages[storageIndex].Value;
+
+            var placedIndex = storage.Place(o);
+            EventBinders.OfType<IEventBinder<T>>().FirstOrDefault()
+                ?.BindEvents((T)o, context, _representationModuleProducer!, placedIndex);
+
+            return placedIndex;
+        }
+
+        public void ClearObject(ComplexObjectIdentifier id)
+        {
+            _storages.Find(i => i.Key == id.OwnerId).Value.Free(id.ContextId);
+        }
+
+        public T GetObject<T>(ComplexObjectIdentifier id)
+        {
+            throw new NotImplementedException();
+        }
+
+        public object GetSingleObject(Type type, int ownerId)
+        {
+            throw new NotImplementedException();
+        }
+
+        // public object GetSingleObject(Type type, int ownerId)
+        // {
+        // }
+
+        public int GetObjectIndex<T>(object o, IEndPointContext context)
+        {
+            throw new NotImplementedException();
+        }
+    }
+}
@@ -1,39 +0,0 @@
-using System;
-using mROA.Abstract;
-
-namespace mROA.Implementation
-{
-    public class ComplexRepository : IContextRepository
-    {
-        public void Inject<T>(T dependency)
-        {
-            throw new NotImplementedException();
-        }
-
-        public int HostId { get; set; }
-        public int ResisterObject<T>(object o, IEndPointContext context)
-        {
-            throw new NotImplementedException();
-        }
-
-        public void ClearObject(ComplexObjectIdentifier id)
-        {
-            throw new NotImplementedException();
-        }
-
-        public T GetObject<T>(ComplexObjectIdentifier id)
-        {
-            throw new NotImplementedException();
-        }
-
-        public object GetSingleObject(Type type)
-        {
-            throw new NotImplementedException();
-        }
-
-        public int GetObjectIndex<T>(object o, IEndPointContext context)
-        {
-            throw new NotImplementedException();
-        }
-    }
-}
@@ -5,16 +5,20 @@ using mROA.Abstract;

 namespace mROA.Implementation
 {
-    public class ExtensibleStorage<T> : IStorage<T>
+    public class ExtensibleStorage<T> : IStorage<T> where T : class
    {
        private const int StartupSize = 1024;
        private const int GrowSize = 128;
        private T?[] _array = new T?[StartupSize];
        private readonly LinkedList<int> _freePlaces = new(Enumerable.Range(0, StartupSize));

-        public T GetValue(int index)
+        public T? GetValue(int index)
        {
-            return _array[index]!;
+            if (index < 0 || index >= _array.Length)
+            {
+                return null;
+            }
+            return _array[index];
        }

        public int GetIndex(T value)
@@ -31,6 +35,7 @@ namespace mROA.Implementation

            var index = _freePlaces.First.Value;

+            _freePlaces.RemoveFirst();
            _array[index] = value;

            return index;
@@ -11,7 +11,8 @@ namespace mROA.Implementation.Frontend
 {
    public class RequestExtractor : IRequestExtractor
    {
-        private IContextRepository? _contextRepository;
+        private IContextRepository? _realContextRepository;
+        private IContextRepository? _remoteContextRepository;
        private IExecuteModule? _executeModule;
        private IMethodRepository? _methodRepository;
        private IRepresentationModule? _representationModule;
@@ -24,8 +25,12 @@ namespace mROA.Implementation.Frontend
                case IExecuteModule executeModule:
                    _executeModule = executeModule;
                    break;
-                case IContextRepository contextRepository:
-                    _contextRepository = contextRepository;
+                case MultiClientContextRepository :
+                case ContextRepository:
+                    _realContextRepository = dependency as IContextRepository;
+                    break;
+                case RemoteContextRepository remoteContextRepository:
+                    _remoteContextRepository = remoteContextRepository;
                    break;
                case IMethodRepository methodRepository:
                    _methodRepository = methodRepository;
@@ -47,7 +52,7 @@ namespace mROA.Implementation.Frontend
                    throw new NullReferenceException("Serializing toolkit is null.");
                if (_executeModule == null)
                    throw new NullReferenceException("Execute module is null.");
-                if (_contextRepository == null)
+                if (_realContextRepository == null)
                    throw new NullReferenceException("Context repository is null.");
                if (_representationModule == null)
                    throw new NullReferenceException("Representation module is null.");
@@ -89,14 +94,14 @@ namespace mROA.Implementation.Frontend
 #endif
                            var req = cancelRequest.Result;
                            tokenSource.Cancel();
-                            _executeModule.Execute(req, _contextRepository, _representationModule);
+                            _executeModule.Execute(req, _realContextRepository, _representationModule);
                        }
                        else if (defaultRequest.IsCompleted)
                        {
                            tokenSource.Cancel();
                            var request = defaultRequest.Result;

-                            var result = _executeModule.Execute(request, _contextRepository, _representationModule);
+                            var result = _executeModule.Execute(request, _realContextRepository, _representationModule);

                            var resultType = result switch
                            {
@@ -112,7 +117,7 @@ namespace mROA.Implementation.Frontend
                        {
                            tokenSource.Cancel();
                            var request = eventRequest.Result;
-                            _executeModule.Execute(request, _contextRepository, _representationModule);
+                            _executeModule.Execute(request, _remoteContextRepository!, _representationModule);
                        }
                    }
                }
@@ -1,11 +1,13 @@
 using System;
 using System.Collections.Generic;
+using System.Linq;
 using mROA.Abstract;

 namespace mROA.Implementation
 {
    public class RemoteContextRepository : IContextRepository
    {
+        private List<RemoteObjectBase> _producedRemoteEndpoints = new();
        public static Dictionary<Type, Type> RemoteTypes = new();
        private IRepresentationModuleProducer? _representationProducer;

@@ -23,6 +25,9 @@ namespace mROA.Implementation

        public T GetObject<T>(ComplexObjectIdentifier id)
        {
+            var index = _producedRemoteEndpoints.Find(i => i.Identifier.Equals(id));
+            if (index is not null)
+                return (T)(index as object);
            if (_representationProducer == null)
                throw new NullReferenceException("representation producer is not initialized");

@@ -31,18 +36,24 @@ namespace mROA.Implementation
                _representationProducer.Produce(TransmissionConfig.OwnershipRepository.GetOwnershipId());
            var remote = (T)Activator.CreateInstance(remoteType, id.ContextId,
                representationModule)!;
+
+            _producedRemoteEndpoints.Add((remote as RemoteObjectBase)!);
+
            return remote;
        }

-        public object GetSingleObject(Type type)
+        public object GetSingleObject(Type type, int ownerId)
        {
            if (_representationProducer == null)
                throw new NullReferenceException("representation producer is not initialized");
-
+            
            var representationModule =
                _representationProducer.Produce(TransmissionConfig.OwnershipRepository.GetOwnershipId());
-            return Activator.CreateInstance(RemoteTypes[type], -1,
-                representationModule)!;
+
+            _producedRemoteEndpoints.Add((Activator.CreateInstance(RemoteTypes[type], -1,
+                representationModule) as RemoteObjectBase)!);
+
+            return _producedRemoteEndpoints.Last();
        }

        public int GetObjectIndex<T>(object o, IEndPointContext context)