custom hashcode and comparison for ErgoHeader

ergoplatform · Jun 5, 2024 · 0d6b2e0 · 0d6b2e0
1 parent 1092b08
commit 0d6b2e0
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/data/shared/src/main/scala/org/ergoplatform/ErgoHeader.scala b/data/shared/src/main/scala/org/ergoplatform/ErgoHeader.scala
@@ -104,9 +104,9 @@ case class ErgoHeader(override val version: ErgoHeader.Version,
                       override val unparsedBytes: Array[Byte],
                       _bytes: Array[Byte]) extends
   HeaderWithoutPow(version, parentId, ADProofsRoot, stateRoot, transactionsRoot, timestamp,
-      nBits, height, extensionRoot, votes, unparsedBytes) {
+    nBits, height, extensionRoot, votes, unparsedBytes) {
 
-    lazy val bytes = if (_bytes != null) {
+    lazy val bytes = if(_bytes != null) {
         _bytes
     } else {
         ErgoHeader.sigmaSerializer.toBytes(this)
@@ -116,6 +116,12 @@ case class ErgoHeader(override val version: ErgoHeader.Version,
 
     lazy val id = Colls.fromArray(serializedId)
 
+    override def hashCode(): Int = id.hashCode()
+
+    override def equals(other: Any): Boolean = other match {
+        case h: ErgoHeader => h.id == this.id
+        case _ => false
+    }
 }