MappedDocumentMaterializer: allow performing final modifications to resource

Chris Santero · Chris Santero · commit 151c82c9fac7 · 2015-10-16T14:59:32.000-04:00
diff --git a/JSONAPI/Http/MappedDocumentMaterializer.cs b/JSONAPI/Http/MappedDocumentMaterializer.cs
@@ -1,9 +1,7 @@
 &#65279;using System;
-using System.Collections.Generic;
 using System.Linq;
 using System.Linq.Expressions;
 using System.Net.Http;
-using System.Reflection;
 using System.Threading;
 using System.Threading.Tasks;
 using JSONAPI.Core;
@@ -83,6 +81,8 @@ public virtual async Task<ISingleResourceDocument> GetRecordById(string id, Http
                 throw JsonApiException.CreateForNotFound(
                     string.Format("No record exists with type `{0}` and ID `{1}`.", ResourceTypeName, id));
 
+            await OnResourceFetched(primaryResource);
+
             var baseUrl = _baseUrlService.GetBaseUrl(request);
             return _singleResourceDocumentBuilder.BuildDocument(primaryResource, baseUrl, jsonApiPaths, null);
         }
@@ -112,7 +112,17 @@ protected virtual Expression<Func<TDto, object>>[] GetIncludePathsForSingleResou
         {
             return null;
         }
-
+        
+        /// <summary>
+        /// Hook for performing any final modifications to the resource before serialization
+        /// </summary>
+        /// <param name="resource"></param>
+        /// <returns></returns>
+        protected virtual Task OnResourceFetched(TDto resource)
+        {
+            return Task.FromResult(0);
+        }
+        
         private string ConvertToJsonKeyPath(Expression<Func<TDto, object>> expression)
         {
             var visitor = new PathVisitor(_resourceTypeRegistry);
