Merge pull request #22 from balenamiaa/master

setindex for ref

Author: jw3126

Project.toml

@@ -1,7 +1,7 @@
 name = "Accessors"
 uuid = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697"
 authors = ["Takafumi Arakaki <[email protected]>", "Jan Weidner <[email protected]> and contributors"]
-version = "0.1.1"
+version = "0.1.2"
 
 [deps]
 Compat = "34da2185-b29b-5c13-b0c7-acf172513d20"

src/optics.jl

@@ -160,7 +160,7 @@ function _modify(f, obj, optic::ComposedOptic, ::ModifyBased)
     end
 end
 
-function _modify(f, obj, optic, ::SetBased)
+@inline function _modify(f, obj, optic, ::SetBased)
     set(obj, optic, f(optic(obj)))
 end
 

src/setindex.jl

@@ -2,6 +2,8 @@ Base.@propagate_inbounds function setindex(args...)
     Base.setindex(args...)
 end
 
+@inline setindex(::Base.RefValue, val) = Ref(val)
+
 Base.@propagate_inbounds function setindex(xs::AbstractArray, v, I...)
     T = promote_type(eltype(xs), typeof(v))
     ys = similar(xs, T)
@@ -20,3 +22,4 @@ Base.@propagate_inbounds function setindex(d0::AbstractDict, v, k)
     d[k] = v
     return d
 end
+

test/test_setindex.jl

@@ -15,6 +15,13 @@ Check that _type_ and value of `x` and `y` are equal.
     @test !(1.0 ==ₜ 1)
 end
 
+function ref_alloc_test()
+    ref = Ref((UInt(10), 100))
+    ref2 = @set ref[][1] *= -1
+
+    ref2[]
+end
+
 @testset "setindex" begin
     arr = [1,2,3]
     @test_throws MethodError Base.setindex(arr, 10, 1)
@@ -32,6 +39,21 @@ end
     @test d == Dict(:a => 1, :b => 2)
     @test Accessors.setindex(d, 30, "c") ==ₜ Dict(:a=>1, :b=>2, "c"=>30)
     @test Accessors.setindex(d, 10.0, :a) ==ₜ Dict(:a=>10.0, :b=>2.0)
+    
+    ref = Ref((; a = 1, b = 2, c = (; aa = 3)))
+    @test @set(ref[].a = 90)[] == (; a = 90, b = 2, c = (; aa = 3))
+    @test @set(ref[].b = "2")[] ==ₜ (; a = 1, b = "2", c = (; aa = 3))
+    @test @set(ref[].c.aa += 2)[] == (; a = 1, b = 2, c = (; aa = 5))
+
+    ref = Ref(1::Int)
+    @set ref[] = "no mutation"
+    @test ref[] === 1
+    @test typeof(ref) == Base.RefValue{Int}
+
+    if VERSION >= v"1.5.0"
+        _ = ref_alloc_test()
+        @test @allocated(ref_alloc_test()) == 0
+    end
 end
 
 end