Added db4o.

Added guards against most problematic areas.
2025-09-17 17:14:18 +02:00 · 2013-02-02 11:39:27 -08:00
parent b6a4e6c32c
commit 5dbaaee005
11 changed files with 396 additions and 6 deletions
--- a/NzbDrone.Core/Datastore/NoCahceRefrenceSystem.cs
+++ b/NzbDrone.Core/Datastore/NoCahceRefrenceSystem.cs
@@ -0,0 +1,116 @@
+using System.Linq;
+using Db4objects.Db4o;
+using Db4objects.Db4o.Foundation;
+using Db4objects.Db4o.Internal;
+using Db4objects.Db4o.Internal.References;
+
+namespace NzbDrone.Core.Datastore
+{
+    public class NoCahceRefrenceSystem : IReferenceSystem
+    {
+        private ObjectReference _hashCodeTree;
+        private ObjectReference _idTree;
+
+        internal NoCahceRefrenceSystem()
+        {
+
+        }
+
+        public virtual void AddNewReference(ObjectReference @ref)
+        {
+            AddReference(@ref);
+        }
+
+        public virtual void AddExistingReference(ObjectReference @ref)
+        {
+            AddReference(@ref);
+        }
+
+        public virtual void Commit()
+        {
+            Reset();
+        }
+
+
+
+        public virtual ObjectReference ReferenceForId(int id)
+        {
+            if (DTrace.enabled)
+                DTrace.GetYapobject.Log(id);
+            if (_idTree == null)
+                return null;
+            if (!ObjectReference.IsValidId(id))
+                return null;
+            else
+                return _idTree.Id_find(id);
+        }
+
+        public virtual ObjectReference ReferenceForObject(object obj)
+        {
+            if (_hashCodeTree == null)
+                return null;
+            else
+                return _hashCodeTree.Hc_find(obj);
+        }
+
+        public virtual void RemoveReference(ObjectReference @ref)
+        {
+            if (DTrace.enabled)
+                DTrace.ReferenceRemoved.Log(@ref.GetID());
+            if (_hashCodeTree != null)
+                _hashCodeTree = _hashCodeTree.Hc_remove(@ref);
+            if (_idTree == null)
+                return;
+            _idTree = _idTree.Id_remove(@ref);
+        }
+
+        public virtual void Rollback()
+        {
+            Reset();
+        }
+
+        public virtual void TraverseReferences(IVisitor4 visitor)
+        {
+            if (_hashCodeTree == null)
+                return;
+            _hashCodeTree.Hc_traverse(visitor);
+        }
+
+        public virtual void Discarded()
+        {
+        }
+
+
+        public void Reset()
+        {
+            _hashCodeTree = null;
+            _idTree = null;
+        }
+
+        private void AddReference(ObjectReference @ref)
+        {
+            @ref.Ref_init();
+            IdAdd(@ref);
+            HashCodeAdd(@ref);
+        }
+
+        private void HashCodeAdd(ObjectReference @ref)
+        {
+            if (_hashCodeTree == null)
+                _hashCodeTree = @ref;
+            else
+                _hashCodeTree = _hashCodeTree.Hc_add(@ref);
+        }
+
+        private void IdAdd(ObjectReference @ref)
+        {
+            if (DTrace.enabled)
+                DTrace.IdTreeAdd.Log(@ref.GetID());
+            if (_idTree == null)
+                _idTree = @ref;
+            else
+                _idTree = _idTree.Id_add(@ref);
+        }
+
+    }
+}