feat(louvain): support returning the same cluster id

packages/graph/src/louvain.ts

@@ -1,6 +1,5 @@
 import getAdjMatrix from './adjacent-matrix'
-import { uniqueId } from './util';
-import { NodeConfig, ClusterData, GraphData } from './types';
+import { NodeConfig, ClusterData, GraphData, ClusterMap } from './types';
 
 const getModularity = (
   nodes: NodeConfig[],
@@ -41,12 +40,13 @@ const louvain = (
 ): ClusterData => {
   // the origin data
   const { nodes = [], edges = [] } = graphData;
+  let uniqueId = 1;
 
-  const clusters = {};
+  const clusters: ClusterMap = {};
   const nodeMap = {};
   // init the clusters and nodeMap
   nodes.forEach((node, i) => {
-    const cid: string = uniqueId();
+    const cid: string = String(uniqueId++);
     node.clusterId = cid;
     clusters[cid] = {
       id: cid,
@@ -210,6 +210,18 @@ const louvain = (
     }
   });
 
+  Object.keys(clusters).forEach((clusterId, index) => {
+    const cluster = clusters[clusterId];
+    const newId = String(index + 1);
+    if (newId === clusterId) {
+      return;
+    }
+    cluster.id = newId;
+    cluster.nodes = cluster.nodes.map(item => ({ id: item.id, clusterId: newId }));
+    clusters[newId] = cluster;
+    delete clusters[clusterId];
+  });
+
   // get the cluster edges
   const clusterEdges = [];
   const clusterEdgeMap = {};

packages/graph/src/types.ts

@@ -30,6 +30,10 @@ export interface ClusterData {
   clusterEdges: EdgeConfig[];
 }
 
+export interface ClusterMap {
+  [key: string]: Cluster
+}
+
 // 图算法回调方法接口定义
 export interface IAlgorithmCallbacks {
   enter?: (param: { current: string; previous: string }) => void;