fix: self relationship #363

dineug · Oct 31, 2024 · 2064697 · 2064697
1 parent 5f4da4b
commit 2064697
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/packages/erd-editor/src/utils/draw-relationship/pathFinding.ts b/packages/erd-editor/src/utils/draw-relationship/pathFinding.ts
@@ -22,8 +22,8 @@ export function getRelationshipPath(
 }
 
 function getPath(
-  start: RelationshipPoint,
-  end: RelationshipPoint
+  start: Relationship['start'],
+  end: Relationship['end']
 ): RelationshipPath['path'] {
   const line: PathLine = {
     start: {
@@ -44,6 +44,15 @@ function getPath(
     L: { x: 0, y: 0 },
     Q: { x: 0, y: 0 },
     d() {
+      if (start.tableId === end.tableId) {
+        return [
+          [
+            { x: this.M.x, y: this.M.y },
+            { x: this.L.x, y: this.L.y },
+          ],
+        ];
+      }
+
       const distanceX = this.M.x - this.L.x;
       const distanceY = this.M.y - this.L.y;
       const distanceHalfX = distanceX / 2;