iv-org-mirror/shard-spectator
1
0
Fork 0
mirror of https://gitea.invidious.io/iv-org/shard-spectator.git synced 2024-08-15 00:53:35 +00:00
Code Issues Projects Releases Packages Wiki Activity

Add optional count to sample group

Browse source
This commit is contained in:
Michael Miller 2019-09-26 15:41:58 -06:00
parent 311b4b0f2f
commit 3d37a94aaf
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/spectator/dsl/groups.cr
View file

@ -26,7 +26,7 @@ module Spectator
context({{what}}) {{block}} context({{what}}) {{block}}
end end
macro sample(collection, &block) macro sample(collection, count = nil, &block)
{% name = block.args.empty? ? :value.id : block.args.first.id %} {% name = block.args.empty? ? :value.id : block.args.first.id %}
def %collection def %collection
@ -34,7 +34,11 @@ module Spectator
end end
def %to_a def %to_a
%collection.to_a {% if count %}
%collection.first({{count}})
{% else %}
%collection.to_a
{% end %}
end end
class Context%sample < {{@type.id}} class Context%sample < {{@type.id}}