From 91fc64d208659216267836d3ee3e0c28c9ab8374 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Florian=20Kr=C3=B6nert?= <Florian.Kroenert@gmx.de>
Date: Tue, 24 Oct 2017 23:35:02 +0200
Subject: [PATCH] Added version specific partials

---
 .../Xrm.Oss.UnitOfWork/UpdateContext.2016.cs  | 27 +++++++++++++++++++
 src/lib/Xrm.Oss.UnitOfWork/UpdateContext.cs   | 18 +++++++++++++
 2 files changed, 45 insertions(+)
 create mode 100644 src/lib/Xrm.Oss.UnitOfWork/UpdateContext.2016.cs

diff --git a/src/lib/Xrm.Oss.UnitOfWork/UpdateContext.2016.cs b/src/lib/Xrm.Oss.UnitOfWork/UpdateContext.2016.cs
new file mode 100644
index 0000000..346938f
--- /dev/null
+++ b/src/lib/Xrm.Oss.UnitOfWork/UpdateContext.2016.cs
@@ -0,0 +1,27 @@
+using System;
+using Microsoft.Xrm.Sdk;
+using Microsoft.Xrm.Sdk.Messages;
+
+namespace Xrm.Oss.UnitOfWork
+{
+    public partial class UpdateContext<T> : IDisposable where T : Entity
+    {
+        public bool AddToTransaction(ExecuteTransactionRequest transaction)
+        {
+            if (transaction.Requests == null)
+            {
+                transaction.Requests = new OrganizationRequestCollection();
+            }
+
+            var updateRequest = GetUpdateRequest();
+
+            if (updateRequest == null)
+            {
+                return false;
+            }
+
+            transaction.Requests.Add(updateRequest);
+            return true;
+        }
+    }
+}
diff --git a/src/lib/Xrm.Oss.UnitOfWork/UpdateContext.cs b/src/lib/Xrm.Oss.UnitOfWork/UpdateContext.cs
index 8e22236..4106084 100644
--- a/src/lib/Xrm.Oss.UnitOfWork/UpdateContext.cs
+++ b/src/lib/Xrm.Oss.UnitOfWork/UpdateContext.cs
@@ -213,6 +213,24 @@ public UpdateRequest GetUpdateRequest()
             return null;
         }
 
+        public bool AddToExecuteMultiple(ExecuteMultipleRequest executeMultiple)
+        {
+            if (executeMultiple.Requests == null)
+            {
+                executeMultiple.Requests = new OrganizationRequestCollection();
+            }
+
+            var updateRequest = GetUpdateRequest();
+
+            if (updateRequest == null)
+            {
+                return false;
+            }
+
+            executeMultiple.Requests.Add(updateRequest);
+            return true;
+        }
+
         public void Dispose()
         {
             Dispose(true);