Write publish message store (#380)

* made possible updating message before storing to db

made updating message Content possible on WritePublishMessageStoreBefore event

* added TracingHeaders to BrokerStoreEventData model

src/DotNetCore.CAP/Abstractions/CapPublisherBase.cs

@@ -68,7 +68,12 @@ namespace DotNetCore.CAP.Abstractions
 
             try
             {
-                operationId = s_diagnosticListener.WritePublishMessageStoreBefore(message);
+                var tracingResult = TracingBefore(message.Name, message.Content);
+                operationId = tracingResult.Item1;
+                
+                message.Content = tracingResult.Item2 != null
+                    ? Helper.AddTracingHeaderProperty(message.Content, tracingResult.Item2)
+                    : message.Content;
 
                 if (Transaction.Value?.DbTransaction == null)
                 {
@@ -101,6 +106,21 @@ namespace DotNetCore.CAP.Abstractions
             }
         }
         
+        private (Guid, TracingHeaders) TracingBefore(string topic, string values)
+        {
+            Guid operationId = Guid.NewGuid();
+            
+            var eventData = new BrokerStoreEventData(
+                operationId, 
+                "",
+                topic, 
+                values);
+
+            s_diagnosticListener.WritePublishMessageStoreBefore(eventData);
+
+            return (operationId, eventData.Headers);
+        }
+
         protected abstract Task ExecuteAsync(CapPublishedMessage message,
             ICapTransaction transaction = null,
             CancellationToken cancel = default(CancellationToken));

src/DotNetCore.CAP/Diagnostics/DiagnosticListenerExtensions.cs

@@ -38,26 +38,13 @@ namespace DotNetCore.CAP.Diagnostics
         //============================================================================
         //====================  Before publish store message      ====================
         //============================================================================
-        public static Guid WritePublishMessageStoreBefore(this DiagnosticListener @this,
-            CapPublishedMessage message,
-            [CallerMemberName] string operation = "")
+        public static void WritePublishMessageStoreBefore(this DiagnosticListener @this, BrokerStoreEventData eventData)
         {
             if (@this.IsEnabled(CapBeforePublishMessageStore))
             {
-                var operationId = Guid.NewGuid();
-
-                @this.Write(CapBeforePublishMessageStore, new
-                {
-                    OperationId = operationId,
-                    Operation = operation,
-                    MessageName = message.Name,
-                    MessageContent = message.Content
-                });
-
-                return operationId;
+                eventData.Headers = new TracingHeaders();
+                @this.Write(CapBeforePublishMessageStore, eventData);
             }
-
-            return Guid.Empty;
         }
 
         public static void WritePublishMessageStoreAfter(this DiagnosticListener @this,

src/DotNetCore.CAP/Diagnostics/EventData.Broker.Store.cs

+using System;
+
+namespace DotNetCore.CAP.Diagnostics
+{
+    public class BrokerStoreEventData : EventData
+    {
+        public BrokerStoreEventData(
+            Guid operationId, 
+            string operation,
+            string messageName, 
+            string messageContent) : base(operationId, operation)
+        {
+            MessageName = messageName;
+            MessageContent = messageContent;
+        }
+        
+        public string MessageName { get; set; }
+        public string MessageContent { get; set; }
+        
+        public TracingHeaders Headers { get; set; } 
+    }
+}
\ No newline at end of file