Skip to content

Commit 402dc66

Browse files
authored
Merge pull request #1 from aktsk/feature/add-load_targets
Add `load_targets` to Dataset for customized loading
2 parents 56d3753 + 2764026 commit 402dc66

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

lib/simple_master/storage/dataset.rb

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ def run_after_load
3030
attr_reader :diff
3131
attr_accessor :tables
3232
attr_accessor :cache
33+
attr_accessor :load_targets
3334

3435
def initialize(table_class: DEFAULT_TABLE, loader: nil)
3536
@table_class = table_class
@@ -59,7 +60,7 @@ def reload
5960
def load
6061
memsize do
6162
cache.clear
62-
targets = SimpleMaster.targets
63+
targets = @load_targets || SimpleMaster.targets
6364

6465
tables = targets.map(&:base_class).uniq.map { table(_1) }
6566

0 commit comments

Comments
 (0)