Use exposure: node selection

dbt-labs · Sep 22, 2020 · 9bb68eb · 9bb68eb
1 parent 8bdbf1a
commit 9bb68eb
Show file tree

Hide file tree

Showing 5 changed files with 51 additions and 2 deletions.
diff --git a/src/app/components/graph/graph-launcher.js b/src/app/components/graph/graph-launcher.js
@@ -138,6 +138,8 @@ angular
 
                 if (node && node.resource_type == 'source') {
                     var nodes = graph.showFullGraph('source:' + node.source_name + "." + node.name);
+                } else if (node && node.resource_type == 'exposure') {
+                    var nodes = graph.showFullGraph('exposure:' + node.name);
                 } else {
                     var nodes = graph.showFullGraph(node_name);
                 }
@@ -149,6 +151,8 @@ angular
                 var node = selectorService.getViewNode();
                 if (node && node.resource_type == 'source') {
                     var nodes = graph.showVerticalGraph('source:' + node.source_name + "." + node.name, true);
+                } else if (node && node.resource_type == 'exposure') {
+                    var nodes = graph.showVerticalGraph('exposure:' + node.name, true);
                 } else {
                     var nodes = graph.showVerticalGraph(node.name, true);
                 }

diff --git a/src/app/graph/index.js b/src/app/graph/index.js
@@ -19,6 +19,8 @@ angular
         function getNodeSelector(node) {
             if (node && node.resource_type == 'source') {
                 return 'source:' + node.source_name + "." + node.name;
+            } else if (node && node.resource_type == 'exposure') {
+                return 'exposure:' + node.name;
             } else if (node.name)  {
                 return node.name;
             } else {

diff --git a/src/app/services/node_selection_service.js b/src/app/services/node_selection_service.js
@@ -51,10 +51,12 @@ angular
 
     service.resetSelection = function(node) {
         var include_selection;
-        if (node && _.includes(['model', 'seed', 'snapshot', 'exposure'], node.resource_type)) {
+        if (node && _.includes(['model', 'seed', 'snapshot'], node.resource_type)) {
             include_selection = '+' + node.name + '+';
         } else if (node && node.resource_type == 'source') {
             include_selection = '+source:' + node.source_name + "." + node.name + '+';
+        } else if (node && node.resource_type == 'exposure') {
+            include_selection = '+exposure:' + node.name;
         } else if (node && _.includes(['analysis', 'test'], node.resource_type)) {
             include_selection = '+' + node.name;
         } else {
@@ -87,6 +89,9 @@ angular
         if (node.resource_type == 'source') {
             pre += "source:"
             node_name = node.source_name + "." + node.name;
+        } else if (node.resource_type == 'exposure') {
+            pre += "exposure:"
+            node_name = node.name;
         } else {
             node_name = node.name;
         }

diff --git a/src/app/services/selector_matcher.js b/src/app/services/selector_matcher.js
@@ -8,6 +8,7 @@ var SELECTOR_TYPE = {
     FQN: 'fqn',
     TAG: 'tag',
     SOURCE: 'source',
+    EXPOSURE: 'exposure',
     PATH: 'path',
     PACKAGE: 'package',
     CONFIG: 'config',
@@ -20,6 +21,7 @@ NODE_MATCHERS[SELECTOR_TYPE.IMPLICIT] = getNodesByImplicitSelection;
 NODE_MATCHERS[SELECTOR_TYPE.FQN] = getNodesByFQN;
 NODE_MATCHERS[SELECTOR_TYPE.TAG] = getNodesByTag;
 NODE_MATCHERS[SELECTOR_TYPE.SOURCE] = getNodesBySource;
+NODE_MATCHERS[SELECTOR_TYPE.EXPOSURE] = getNodesByExposure;
 NODE_MATCHERS[SELECTOR_TYPE.PATH] = getNodesByPath;
 NODE_MATCHERS[SELECTOR_TYPE.PACKAGE] = getNodesByPackage;
 NODE_MATCHERS[SELECTOR_TYPE.CONFIG] = getNodesByConfig;
@@ -57,7 +59,7 @@ function getNodesByFQN(elements, qualified_name) {
         var node = el.data;
         var fqn = node.fqn;
 
-        if (!fqn || node.resource_type == 'source') {
+        if (!fqn || node.resource_type == 'source' || node.resource_type == 'exposure') {
             return;
         }
 
@@ -212,6 +214,27 @@ function getNodesBySource(elements, source) {
     return nodes;
 }
 
+function getNodesByExposure(elements, exposure) {
+    var nodes = [];
+    _.each(elements, function(node_obj) {
+        var node = node_obj.data;
+
+        if (node.resource_type != 'exposure') {
+            return;
+        }
+
+        var exposure_name = node.name;
+        var selected_exposure_name = exposure;
+
+        if (selected_exposure_name == '*') {
+            nodes.push(node_obj.data);
+        } else if (selected_exposure_name == exposure_name) {
+            nodes.push(node_obj.data);
+        }
+    })
+    return nodes;
+}
+
 function getNodesFromSpec(dag, pristine_nodes, maxHops, selector) {
     const matcher = NODE_MATCHERS[selector.selector_type];
     if (!matcher) {

diff --git a/src/app/services/selector_methods.test.js b/src/app/services/selector_methods.test.js
@@ -146,6 +146,21 @@ test("Test parsing specs (explicit source.table)", () => {
     });
 })
 
+test("Test parsing specs (explicit exposure)", () => {
+    expect(
+        selectors.parseSpec('+exposure:a')
+    ).toStrictEqual({
+        select_at: false,
+        select_children: false,
+        select_parents: true,
+        parents_depth: null,
+        children_depth: null,
+        selector_type: 'exposure',
+        selector_value: 'a',
+        raw: '+exposure:a',
+    });
+})
+
 test("Test parsing specs (scoped fqn)", () => {
     expect(
         selectors.parseSpec('a.b.c+')