mirror of
https://gitea.invidious.io/iv-org/shard-spectator.git
synced 2024-08-15 00:53:35 +00:00
Add lazy utility
This commit is contained in:
parent
fb0423ed02
commit
5cac4aa5a1
2 changed files with 25 additions and 10 deletions
|
@ -1,6 +1,6 @@
|
|||
require "./abstract_expression"
|
||||
require "./label"
|
||||
require "./wrapper"
|
||||
require "./lazy"
|
||||
|
||||
module Spectator
|
||||
# Represents a block from a test.
|
||||
|
@ -10,8 +10,7 @@ module Spectator
|
|||
# or nil if one isn't available.
|
||||
class Block(T) < AbstractExpression
|
||||
# Cached value returned from the block.
|
||||
# Is nil if the block hasn't been called.
|
||||
@wrapper : Wrapper(T)?
|
||||
@value = Lazy(T).new
|
||||
|
||||
# Creates the block expression from a proc.
|
||||
# The *proc* will be called to evaluate the value of the expression.
|
||||
|
@ -34,13 +33,7 @@ module Spectator
|
|||
# The block is lazily evaluated and the value retrieved only once.
|
||||
# Afterwards, the value is cached and returned by successive calls to this method.
|
||||
def value
|
||||
if (wrapper = @wrapper)
|
||||
wrapper.value
|
||||
else
|
||||
call.tap do |value|
|
||||
@wrapper = Wrapper.new(value)
|
||||
end
|
||||
end
|
||||
@value.get { call }
|
||||
end
|
||||
|
||||
# Evaluates the block and returns the value from it.
|
||||
|
|
22
src/spectator/lazy.cr
Normal file
22
src/spectator/lazy.cr
Normal file
|
@ -0,0 +1,22 @@
|
|||
require "./wrapper"
|
||||
|
||||
module Spectator
|
||||
# Lazily stores a value.
|
||||
struct Lazy(T)
|
||||
@wrapper : Wrapper(T)?
|
||||
|
||||
# Retrieves the value, if it was previously fetched.
|
||||
# On the first invocation of this method, it will yield.
|
||||
# The block should return the value to store.
|
||||
# Subsequent calls will return the same value and not yield.
|
||||
def get(&block : -> T)
|
||||
if (wrapper = @wrapper)
|
||||
wrapper.value
|
||||
else
|
||||
yield.tap do |value|
|
||||
@wrapper = Wrapper.new(value)
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
Loading…
Reference in a new issue