【feature】mapextend 根据 sourceId 判断 overlaylayer 优化

xiongjiaojiao · xiongjiaojiao · commit 9a408d5b5aae · 2024-07-24T18:48:41.000+08:00
diff --git a/src/mapboxgl/core/MapExtend.js b/src/mapboxgl/core/MapExtend.js
@@ -4,6 +4,7 @@
 import mapboxgl from 'mapbox-gl';
 import { decryptSources } from './decryptSource';
 import { getServiceKey } from '@supermapgis/iclient-common/util/EncryptRequest';
+import { CustomOverlayLayer } from '../overlay/Base';
 
 /**
  * @function MapExtend
@@ -38,34 +39,25 @@ export var MapExtend = (function () {
       }
   
       getSource(sourceId) {
-        const customOverlayerLayer = Object.values(this.overlayLayersManager).find(item => item.sourceId === sourceId);
+        const customOverlayerLayer = Object.values(this.overlayLayersManager).find(item => item.sourceId === sourceId && item instanceof CustomOverlayLayer);
         if (customOverlayerLayer) {
-          if (customOverlayerLayer.getSource) {
-            return customOverlayerLayer.getSource();
-          }
-          return;
+          return customOverlayerLayer.getSource();
         }
         return originMapProto.getSource.call(this, sourceId);
       }
   
       removeSource(sourceId) {
-        const customOverlayerLayer = Object.values(this.overlayLayersManager).find(item => item.sourceId === sourceId);
+        const customOverlayerLayer = Object.values(this.overlayLayersManager).find(item => item.sourceId === sourceId && item instanceof CustomOverlayLayer);
         if (customOverlayerLayer) {
-          if (customOverlayerLayer.removeSource) {
-            return customOverlayerLayer.removeSource();
-          }
-          return;
+          return customOverlayerLayer.removeSource();
         }
         return originMapProto.removeSource.call(this, sourceId);
       }
   
       isSourceLoaded(sourceId) {
-        const customOverlayerLayer = Object.values(this.overlayLayersManager).find(item => item.sourceId === sourceId);
+        const customOverlayerLayer = Object.values(this.overlayLayersManager).find(item => item.sourceId === sourceId && item instanceof CustomOverlayLayer);
         if (customOverlayerLayer) {
-          if (customOverlayerLayer.isSourceLoaded) {
-            return customOverlayerLayer.isSourceLoaded();
-          }
-          return;
+          return customOverlayerLayer.isSourceLoaded();
         }
         return originMapProto.isSourceLoaded.call(this, sourceId);
       }
@@ -284,9 +276,9 @@ export var MapExtend = (function () {
       }
   
       querySourceFeatures(sourceId, paramters) {
-        const customOverlayerLayer = Object.values(this.overlayLayersManager).find(item => item.sourceId === sourceId);
+        const customOverlayerLayer = Object.values(this.overlayLayersManager).find(item => item.sourceId === sourceId && item instanceof CustomOverlayLayer);
         if (customOverlayerLayer) {
-          if (customOverlayerLayer.query && customOverlayerLayer.querySourceFeatures) {
+          if (customOverlayerLayer.query) {
             return customOverlayerLayer.querySourceFeatures(paramters);
           }
           return;
diff --git a/test/mapboxgl/core/MapExtendSpec.js b/test/mapboxgl/core/MapExtendSpec.js
@@ -5,6 +5,7 @@ import cipher from 'node-forge/lib/cipher';
 import { MapExtend } from '../../../src/mapboxgl/core/MapExtend';
 import { decryptSources } from '../../../src/mapboxgl/core/decryptSource';
 import { EncryptRequest } from '../../../src/common/util/EncryptRequest';
+import { CustomOverlayLayer } from '../../../src/mapboxgl/overlay/Base';
 
 describe('MapExtend mapboxgl', () => {
   let originalTimeout, testDiv;
@@ -371,15 +372,24 @@ describe('MapExtend mapboxgl', () => {
       for (const key in options) {
         spyOn(options, key).and.callThrough();
       }
+      
+      class L7LayerTest extends CustomOverlayLayer {
+        constructor() {
+          const optionsData = {
+            id: 'l7_layer_1',
+            sourceId: 'l7_layer_1',
+            events: ['click'],
+            query: true,
+            interaction: true,
+            ...options
+          };
+          for (const key in optionsData) {
+            this[key] = optionsData;
+          }
+        }
+      }
       map.overlayLayersManager = {
-        l7_layer_1: {
-          id: 'l7_layer_1',
-          sourceId: 'l7_layer_1',
-          events: ['click'],
-          query: true,
-          interaction: true,
-          ...options
-        },
+        l7_layer_1: new L7LayerTest(),
         heatmap_1: { id: 'heatmap_1' }
       };
       expect(map.getSource('l7_layer_1')).not.toBeUndefined();
