Merge pull request #50 from JuliaData/jps/cmp-dref

Add == and hash for DRef

Project.toml

@@ -2,7 +2,7 @@ name = "MemPool"
 uuid = "f9f48841-c794-520a-933b-121f7ba6ed94"
 license = "MIT"
 desc = "a simple distributed data store"
-version = "0.3.3"
+version = "0.3.4"
 
 [deps]
 DataStructures = "864edb3b-99cc-5e75-8d2d-829cb0a9cfe8"

src/datastore.jl

@@ -12,6 +12,8 @@ mutable struct DRef
         d
     end
 end
+Base.:(==)(d1::DRef, d2::DRef) = (d1.owner == d2.owner) && (d1.id == d2.id)
+Base.hash(d::DRef, h::UInt) = hash(d.id, hash(d.owner, h))
 
 function Serialization.serialize(io::AbstractSerializer, d::DRef)
     Serialization.serialize_cycle_header(io, d)

test/runtests.jl

@@ -122,6 +122,24 @@ end
 end
 =#
 
+@testset "DRef equality" begin
+    d1 = poolset(1)
+    iob = IOBuffer()
+    serialize(iob, d1)
+    seek(iob, 0)
+    d2 = deserialize(iob)
+    @test d1 == d2
+    @test hash(d1) == hash(d2)
+
+    d = Dict{DRef, Int}()
+    d[d1] = 1
+    @test haskey(d, d1)
+    @test haskey(d, d2)
+    @test d[d2] == 1
+
+    pooldelete(d1)
+end
+
 @testset "set-get-delete" begin
     r1 = poolset([1,2])
     r2 = poolset(["abc","def"], 2)