From f453e48707134eabebc9ae60d1bda13167ad1ad5 Mon Sep 17 00:00:00 2001 From: Jacob Beck Date: Tue, 10 Sep 2019 14:00:59 -0600 Subject: [PATCH] populate the cache on compile, too --- core/dbt/task/compile.py | 4 ++++ core/dbt/task/run.py | 3 --- core/dbt/task/runnable.py | 3 +++ 3 files changed, 7 insertions(+), 3 deletions(-) diff --git a/core/dbt/task/compile.py b/core/dbt/task/compile.py index b25015dfe86..1f29ef6b48f 100644 --- a/core/dbt/task/compile.py +++ b/core/dbt/task/compile.py @@ -9,6 +9,10 @@ class CompileTask(GraphRunnableTask): def raise_on_first_error(self): return True + def before_run(self, adapter, selected_uids): + with adapter.connection_named('master'): + self.populate_adapter_cache(adapter) + def build_query(self): return { "include": self.args.models, diff --git a/core/dbt/task/run.py b/core/dbt/task/run.py index 5ba75f4d778..b4f053ac897 100644 --- a/core/dbt/task/run.py +++ b/core/dbt/task/run.py @@ -56,9 +56,6 @@ def __init__(self, args, config): def raise_on_first_error(self): return False - def populate_adapter_cache(self, adapter): - adapter.set_relations_cache(self.manifest) - def get_hook_sql(self, adapter, hook, idx, num_hooks, extra_context): compiled = compile_node(adapter, self.config, hook, self.manifest, extra_context) diff --git a/core/dbt/task/runnable.py b/core/dbt/task/runnable.py index 60b95c49ad9..501f724187d 100644 --- a/core/dbt/task/runnable.py +++ b/core/dbt/task/runnable.py @@ -52,6 +52,9 @@ def _runtime_initialize(self): self.load_manifest() self.compile_manifest() + def populate_adapter_cache(self, adapter): + adapter.set_relations_cache(self.manifest) + class GraphRunnableTask(ManifestTask): def __init__(self, args, config):