add signature to bundle events

NethermindEth · Sep 23, 2021 · 9f88015 · 9f88015
1 parent c02d691
commit 9f88015
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 3 deletions.
diff --git a/src/Nethermind/Nethermind.Mev/Data/BundleEventArgs.cs b/src/Nethermind/Nethermind.Mev/Data/BundleEventArgs.cs
@@ -16,16 +16,19 @@
 // 
 
 using System;
+using Nethermind.Core.Crypto;
 
 namespace Nethermind.Mev.Data
 {
     public class BundleEventArgs : EventArgs
     {
         public MevBundle MevBundle { get; }
-
-        public BundleEventArgs(MevBundle mevBundle)
+        public Signature? RelaySignature { get;  }
+
+        public BundleEventArgs(MevBundle mevBundle, Signature? relaySignature = null)
         {
             MevBundle = mevBundle;
+            RelaySignature = relaySignature;
         }
     }
 }
diff --git a/src/Nethermind/Nethermind.Mev/Data/MevMegabundle.cs b/src/Nethermind/Nethermind.Mev/Data/MevMegabundle.cs
@@ -39,7 +39,17 @@ public bool Equals(MevMegabundle? other)
             return Equals(Hash, other.Hash)
                 && Equals(RelaySignature, other.RelaySignature);
         }
+
+        public override bool Equals(object? obj)
+        {
+            if (ReferenceEquals(null, obj)) return false;
+            if (ReferenceEquals(this, obj)) return true;
+            if (obj.GetType() != GetType()) return false;
+            return Equals((MevMegabundle) obj);
+        }
 
+        public override int GetHashCode() => Hash.GetHashCode();
+
         public override string ToString() => $"Hash:{Hash}; Block:{BlockNumber}; Min:{MinTimestamp}; Max:{MaxTimestamp}; TxCount:{Transactions.Count}; RelaySignature:{RelaySignature};";
     }
 }
diff --git a/src/Nethermind/Nethermind.Mev/Source/BundlePool.cs b/src/Nethermind/Nethermind.Mev/Source/BundlePool.cs
@@ -326,7 +326,7 @@ private void OnNewBlock(object? sender, BlockEventArgs e)
                 UInt256 timestamp = _timestamper.UnixTime.Seconds;
                 IEnumerable<MevBundle> bundles = GetBundles(e.Block.Number + 1, UInt256.MaxValue, timestamp);
                 IEnumerable<MevBundle> megabundles = GetMegabundles(e.Block.Number + 1, UInt256.MaxValue, timestamp);
-                IEnumerable<MevBundle> allBundles = bundles.Concat(megabundles)
+                IEnumerable<MevBundle> allBundles = bundles.Concat(megabundles);
                 foreach (MevBundle bundle in allBundles)
                 {
                     SimulateBundle(bundle, e.Block.Header);