Add ViewManager.getAllExportedViews().

This method currently returns all registered cumulative views.  We could add a
way to specify which views should be exported later.  The API is currently
needed for the StackDriver exporter.

The NoopViewManager also returns all cumulative views.  However, the exporter
could choose to not export data when stats collection isn't enabled.

3 files changed, 25 insertions, 0 deletions

diff --git a/impl_core/src/main/java/io/opencensus/implcore/stats/MeasureToViewMap.java b/impl_core/src/main/java/io/opencensus/implcore/stats/MeasureToViewMap.java
index bf3c8b25..0ee8f3ac 100644
--- a/impl_core/src/main/java/io/opencensus/implcore/stats/MeasureToViewMap.java
+++ b/impl_core/src/main/java/io/opencensus/implcore/stats/MeasureToViewMap.java
@@ -19,6 +19,7 @@ package io.opencensus.implcore.stats;
 import com.google.common.collect.HashMultimap;
 import com.google.common.collect.Maps;
 import com.google.common.collect.Multimap;
+import com.google.common.collect.Sets;
 import io.opencensus.common.Clock;
 import io.opencensus.common.Function;
 import io.opencensus.common.Functions;
@@ -64,6 +65,10 @@ final class MeasureToViewMap {
     return view == null ? null : view.toViewData(clock.now(), state);
   }
 
+  synchronized Collection<View> getAllViews() {
+    return Sets.newHashSet(registeredViews.values());
+  }
+
   @Nullable
   private synchronized MutableViewData getMutableViewData(View.Name viewName) {
     View view = registeredViews.get(viewName);
diff --git a/impl_core/src/main/java/io/opencensus/implcore/stats/StatsManager.java b/impl_core/src/main/java/io/opencensus/implcore/stats/StatsManager.java
index 1a855710..044fb5a8 100644
--- a/impl_core/src/main/java/io/opencensus/implcore/stats/StatsManager.java
+++ b/impl_core/src/main/java/io/opencensus/implcore/stats/StatsManager.java
@@ -24,6 +24,7 @@ import io.opencensus.stats.StatsCollectionState;
 import io.opencensus.stats.View;
 import io.opencensus.stats.ViewData;
 import io.opencensus.tags.TagContext;
+import java.util.Collection;
 import javax.annotation.Nullable;
 
 /** Object that stores all views and stats. */
@@ -55,6 +56,10 @@ final class StatsManager {
     return measureToViewMap.getView(viewName, clock, state.getInternal());
   }
 
+  Collection<View> getAllViews() {
+    return measureToViewMap.getAllViews();
+  }
+
   void record(TagContext tags, MeasureMapInternal measurementValues) {
     // TODO(songya): consider exposing No-op MeasureMap and use it when stats state is DISABLED, so
     // that we don't need to create actual MeasureMapImpl.
diff --git a/impl_core/src/main/java/io/opencensus/implcore/stats/ViewManagerImpl.java b/impl_core/src/main/java/io/opencensus/implcore/stats/ViewManagerImpl.java
index abc4f921..895f1dea 100644
--- a/impl_core/src/main/java/io/opencensus/implcore/stats/ViewManagerImpl.java
+++ b/impl_core/src/main/java/io/opencensus/implcore/stats/ViewManagerImpl.java
@@ -16,9 +16,13 @@
 
 package io.opencensus.implcore.stats;
 
+import com.google.common.collect.Sets;
 import io.opencensus.stats.View;
+import io.opencensus.stats.View.AggregationWindow;
 import io.opencensus.stats.ViewData;
 import io.opencensus.stats.ViewManager;
+import java.util.Collections;
+import java.util.Set;
 import javax.annotation.Nullable;
 
 /** Implementation of {@link ViewManager}. */
@@ -40,6 +44,17 @@ public final class ViewManagerImpl extends ViewManager {
     return statsManager.getView(viewName);
   }
 
+  @Override
+  public Set<View> getAllExportedViews() {
+    Set<View> exportedViews = Sets.newHashSet();
+    for (View view : statsManager.getAllViews()) {
+      if (view.getWindow() instanceof AggregationWindow.Cumulative) {
+        exportedViews.add(view);
+      }
+    }
+    return Collections.unmodifiableSet(exportedViews);
+  }
+
   void clearStats() {
     statsManager.clearStats();
   }