Changeset 1134 for trunk/sources/HeuristicLab.Hive.Server.ADODataAccess/CachedDataAdapter.cs

Timestamp:

01/15/09 15:04:50 (16 years ago)

Author:

svonolfe

Message:

Further improved locking mechanism (avoid race conditions, performance) (#372)

File:

• trunk/sources/HeuristicLab.Hive.Server.ADODataAccess/CachedDataAdapter.cs (modified) (16 diffs)

trunk/sources/HeuristicLab.Hive.Server.ADODataAccess/CachedDataAdapter.cs

@@ -25 +25 @@
 using System.Text;
 using HeuristicLab.Hive.Contracts.BusinessObjects;
-using System.Runtime.CompilerServices;
 using System.Data;
+using System.Threading;
 
 namespace HeuristicLab.Hive.Server.ADODataAccess {
@@ -41 +41 @@
     private static bool cacheFilled = false;
 
+    private static ReaderWriterLock cacheLock =
+      new ReaderWriterLock();
+
     protected DataTable dataTable =
       new DataTable();
@@ -47 +50 @@
       new List<ICachedDataAdapter>();
 
-    [MethodImpl(MethodImplOptions.Synchronized)]
     protected CachedDataAdapter() {
+      cacheLock.AcquireWriterLock(Timeout.Infinite);
+
       if (!cacheFilled) {
         FillCache();
@@ -54 +58 @@
       }
 
+      cacheLock.ReleaseWriterLock();
+
       ServiceLocator.GetTransactionManager().OnUpdate +=
         new EventHandler(CachedDataAdapter_OnUpdate);
@@ -60 +66 @@
     protected virtual RowT FindSingleRow(Selector dbSelector,
       Selector cacheSelector) {
+      cacheLock.AcquireReaderLock(Timeout.Infinite);
+      
       RowT row =
          FindSingleRow(cacheSelector);
@@ -69 +77 @@
       }
 
+      cacheLock.ReleaseReaderLock();
+
       return row;
     }
@@ -74 +84 @@
     protected virtual IEnumerable<RowT> FindMultipleRows(Selector dbSelector,
         Selector cacheSelector) {
+      cacheLock.AcquireReaderLock(Timeout.Infinite);
+
       IList<RowT> result =
          new List<RowT>(cacheSelector());
@@ -85 +97 @@
         }
       }
+
+      cacheLock.ReleaseReaderLock();
 
       return result;
@@ -105 +119 @@
     protected virtual ICollection<ObjT> FindMultiple(Selector dbSelector,
       Selector cacheSelector) {
+      cacheLock.AcquireReaderLock(Timeout.Infinite);
+
       ICollection<ObjT> result =
         FindMultiple(cacheSelector);
@@ -110 +126 @@
       ICollection<ObjT> resultDb =
         FindMultiple(dbSelector);
+
+      cacheLock.ReleaseReaderLock();
 
       foreach (ObjT obj in resultDb) {
@@ -119 +137 @@
     }
 
-    [MethodImpl(MethodImplOptions.Synchronized)]
     protected abstract RowT InsertNewRowInCache(ObjT obj);
 
-    [MethodImpl(MethodImplOptions.Synchronized)]
     protected abstract void FillCache();
 
-    [MethodImpl(MethodImplOptions.Synchronized)]
     public abstract void SyncWithDb();
 
-    [MethodImpl(MethodImplOptions.Synchronized)]
     protected abstract bool PutInCache(ObjT obj);
 
     protected abstract RowT FindCachedById(long id);
 
-    [MethodImpl(MethodImplOptions.Synchronized)]
     void CachedDataAdapter_OnUpdate(object sender, EventArgs e) {
       foreach (ICachedDataAdapter parent in this.parentAdapters) {
@@ -140 +153 @@
 
       this.SyncWithDb();
-    }
-
-    [MethodImpl(MethodImplOptions.Synchronized)]
-    protected virtual void RemoveRowFromCache(RowT row) {
-      cache.Rows.Remove(row);
     }
 
@@ -167 +175 @@
     }
 
-    [MethodImpl(MethodImplOptions.Synchronized)]
+    private void AddToCache(RowT row) {
+      cacheLock.AcquireWriterLock(Timeout.Infinite);
+
+      cache.ImportRow(row);
+      row.Table.Rows.Remove(row);
+
+      cacheLock.ReleaseWriterLock();
+    }
+
+    private RowT AddToCache(ObjT obj) {
+      cacheLock.AcquireWriterLock(Timeout.Infinite);
+
+      RowT row =  InsertNewRowInCache(obj);
+
+      cacheLock.ReleaseWriterLock();
+
+      return row;
+    }
+
+    private void RemoveRowFromCache(RowT row) {
+      cacheLock.AcquireWriterLock(Timeout.Infinite);
+
+      dataTable.ImportRow(row);
+      cache.Rows.Remove(row);
+
+      cacheLock.ReleaseWriterLock();
+
+      UpdateRow(row);
+    }
+
     public override void Update(ObjT obj) {
       if (obj != null) {
@@ -175 +212 @@
         if (row == null) {
           if (PutInCache(obj)) {
-            row = InsertNewRowInCache(obj);
+            row = AddToCache(obj);
           } else {
             row = InsertNewRow(obj);
@@ -184 +221 @@
 
         obj.Id = (long)row[row.Table.PrimaryKey[0]];
+        LockRow(obj.Id);
 
         ConvertObj(obj, row);
@@ -191 +229 @@
 
         if (IsCached(row) &&
-            !PutInCache(obj)) {
-          //remove from cache
-          dataTable.ImportRow(row);
+            !PutInCache(obj)) {          
           RemoveRowFromCache(row);
-
-          UpdateRow(row);
         } else if (!IsCached(row) &&
           PutInCache(obj)) {
-          //add to cache
-          cache.ImportRow(row);
-          row.Table.Rows.Remove(row);
-        } 
+          AddToCache(row);
+        }
+
+        UnlockRow(obj.Id);
       }
     }