fix: initial update of donut clusters

dhis2 · Jan 6, 2020 · cc9eb90 · cc9eb90
1 parent f87de9c
commit cc9eb90
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/layers/DonutCluster.js b/src/layers/DonutCluster.js
@@ -78,7 +78,7 @@ class DonutCluster extends Cluster {
 
         const mapgl = this.getMapGL()
 
-        mapgl.on('data', this.onData)
+        mapgl.on('sourcedata', this.onSourceData)
         mapgl.on('move', this.updateClusters)
         mapgl.on('moveend', this.updateClusters)
 
@@ -90,7 +90,7 @@ class DonutCluster extends Cluster {
 
         const mapgl = this.getMapGL()
 
-        mapgl.off('data', this.onData)
+        mapgl.off('sourcedata', this.onSourceData)
         mapgl.off('move', this.updateClusters)
         mapgl.off('moveend', this.updateClusters)
 
@@ -102,8 +102,9 @@ class DonutCluster extends Cluster {
         this.clusters = {}
     }
 
-    onData = evt => {
-        if (evt.sourceId === this.getId() && evt.isSourceLoaded) {
+    onSourceData = evt => {
+        if (evt.sourceId === this.getId() && this.getSourceFeatures().length) {
+            this.getMapGL().off('sourcedata', this.onSourceData)
             this.updateClusters()
         }
     }
@@ -154,6 +155,11 @@ class DonutCluster extends Cluster {
             }
         }
     }
+
+    // Returns source features
+    getSourceFeatures() {
+        return this.getMapGL().querySourceFeatures(this.getId())
+    }
 }
 
 export default DonutCluster