lib/file_scanner/loader.rb in file_scanner-2.1.1 vs lib/file_scanner/loader.rb in file_scanner-2.2.1

@@ -1,13 +1,16 @@
 module FileScanner
   class Loader
-    def initialize(path:, extensions: [])
+    def initialize(path:, extensions: [], limit: -1)
       @path = File.expand_path(path)
       @extensions = extensions
+      @limit = limit.to_i
     end
 
     def call
-      Dir.glob(files_path)
+      paths = Dir.glob(files_path)
+      return paths if @limit <= 0
+      paths.sample(@limit)
     end
 
     private def files_path
       File.join(@path, "**", extensions_path)
     end