Add Rollback to ISession and clear tracked documents.

Author: Credit Cards For People Under

Lucene.Net.Linq.Tests/Integration/SessionTests.cs

@@ -34,5 +34,23 @@ public void Query()
  var result = (from d in session.Query() where d.Name == "a" select d).Single();
  Assert.That(result.Scalar, Is.EqualTo(4));
  }
+ [Test]
+ public void RollbackDiscardsTrackedDocumentModifications()
+ {
+ var session = provider.OpenSession<SampleDocument>();
+ int originalScalar;
+
+ using (session)
+ {
+ var item = (from d in session.Query() where d.Name == "a" select d).Single();
+
+ originalScalar = item.Scalar;
+ item.Scalar = originalScalar + 1;
+
+ session.Rollback();
+ }
+
+ Assert.That(provider.AsQueryable<SampleDocument>().Single(doc => doc.Name == "a").Scalar, Is.EqualTo(originalScalar));
+ }
  }
 }

Lucene.Net.Linq/ISession.cs

@@ -12,5 +12,6 @@ public interface ISession<T> : IDisposable
  void Delete(params Query[] items);
  void DeleteAll();
  void Commit();
+ void Rollback();
  }
 }

Lucene.Net.Linq/LuceneSession.cs

@@ -88,6 +88,14 @@ public void Dispose()
  Commit();
  }
 
+ public void Rollback()
+ {
+ lock (sessionLock)
+ {
+ ClearPendingChanges();
+ }
+ }
+
  public void Commit()
  {
  lock (sessionLock)
@@ -190,6 +198,7 @@ private void ClearPendingChanges()
  deleteKeys.Clear();
  deleteQueries.Clear();
  additions.Clear();
+ documentTracker.Clear();
  }
 
  internal bool PendingChanges
@@ -252,6 +261,11 @@ public IEnumerable<T> FindModifiedDocuments()
  {
  return items.Where(t => !mapper.Equals(t.Item1, t.Item2)).Select(t => t.Item1);
  }
+
+ public void Clear()
+ {
+ items.Clear();
+ }
  }
  }
 }