From f93cab8e1cfba7ed6b519a2c4df0a60aa22492cb Mon Sep 17 00:00:00 2001
From: Canvas <canvasfull@gmail.com>
Date: Mon, 9 Oct 2023 17:22:18 +0800
Subject: [PATCH] optimize spine attach node logic.

---
 cocos/spine/attach-util.ts | 73 +++++++++++++++-----------------------
 cocos/spine/skeleton.ts    |  5 +--
 2 files changed, 31 insertions(+), 47 deletions(-)

diff --git a/cocos/spine/attach-util.ts b/cocos/spine/attach-util.ts
index f39449e258f..bbbb10b90d0 100644
--- a/cocos/spine/attach-util.ts
+++ b/cocos/spine/attach-util.ts
@@ -35,67 +35,50 @@ const tempMat4 = new Mat4();
  * @class sp.AttachUtil
  */
 export class AttachUtil {
-    protected _inited = false;
-    protected skeletonBones: spine.Bone[] | null = [];
+    protected isInitialized = false;
+    protected skeletonBones: spine.Bone[] | FrameBoneInfo[] | undefined;
+    protected socketNodes: Map<number, Node> | undefined;
+    private keysToDelete:number[] = [];
 
     constructor () {
-        this._inited = false;
+        this.isInitialized = false;
     }
 
     init (skeletonComp: Skeleton): void {
-        this._inited = true;
-        this.skeletonBones = skeletonComp._skeleton.bones;
+        this.isInitialized = false;
+        if (!skeletonComp || skeletonComp.socketNodes.size === 0) return;
+        const isCached = skeletonComp.isAnimationCached();
+        this.skeletonBones = isCached && skeletonComp._curFrame ? skeletonComp._curFrame.boneInfos : skeletonComp._skeleton.bones;
+        if (!this.skeletonBones || this.skeletonBones.length < 1) return;
+        this.socketNodes = skeletonComp.socketNodes;
+        if (this.socketNodes.size <= 0) return;
+        this.isInitialized = true;
+        this._syncAttachedNode();
     }
 
     reset (): void {
-        this._inited = false;
-        this.skeletonBones = null;
+        this.isInitialized = false;
+        this.skeletonBones = undefined;
+        this.socketNodes = undefined;
+        this.keysToDelete.length = 0;
     }
 
-    _syncAttachedNode (skeletonComp: Skeleton): void {
-        if (!this._inited || skeletonComp!.socketNodes.size === 0) {
-            return;
-        }
-        const isCached = skeletonComp!.isAnimationCached();
-        const boneInfos = isCached && skeletonComp!._curFrame ? skeletonComp!._curFrame.boneInfos : this.skeletonBones;
-        
-        if (!boneInfos || boneInfos.length < 1) {
-            return;
-        }
-
-        const sockets = skeletonComp!.sockets;
-        const socketNodes = skeletonComp!.socketNodes;
-        //const keysToDelete = [];
-
-        
-        for (let l = sockets.length - 1; l >= 0; l--) {
-            const sock = sockets[l];
-            const boneNode = sock.target;
-            if (!boneNode) continue;
-            const bone = boneInfos[sock.boneIndex];
-            if (bone) this.matrixHandle(boneNode, bone);
-        }
-
-        /*
-    
+    _syncAttachedNode (): void {
+        if(!this.isInitialized) return;
+        const socketNodes = this.socketNodes!;
         for (const [boneIdx, boneNode] of socketNodes) {
-            
-            /*
             if (!boneNode || !boneNode.isValid) {
-                keysToDelete.push(boneIdx);
+                this.keysToDelete.push(boneIdx);
                 continue;
             }
-            */
-    
-            //const bone = boneInfos[boneIdx];
-            //if (bone) this.matrixHandle(boneNode, bone);
-        //}
-        */
-        /*
-        for (const boneIdx of keysToDelete) {
+            const bone =  this.skeletonBones![boneIdx];
+            if (bone) this.matrixHandle(boneNode, bone);
+        }
+        if (this.keysToDelete.length <= 0) return;
+        for (const boneIdx of this.keysToDelete) {
             socketNodes.delete(boneIdx);
         }
-        */
+        this.keysToDelete.length = 0;
     }
 
     matrixHandle (node: Node, bone: any): void {
diff --git a/cocos/spine/skeleton.ts b/cocos/spine/skeleton.ts
index 26632761295..ec3fc83119b 100644
--- a/cocos/spine/skeleton.ts
+++ b/cocos/spine/skeleton.ts
@@ -559,7 +559,7 @@ export class Skeleton extends UIRenderer {
         }
         this._sockets = val;
         this._updateSocketBindings();
-        this.syncAttachedNode();
+        this.attachUtil.init(this); 
     }
 
     /**
@@ -1268,7 +1268,7 @@ export class Skeleton extends UIRenderer {
      */
     public syncAttachedNode (): void {
         // sync attached node matrix
-        this.attachUtil._syncAttachedNode(this);
+        this.attachUtil._syncAttachedNode();
     }
 
     /**
@@ -1295,6 +1295,7 @@ export class Skeleton extends UIRenderer {
             this._cacheMode = cacheMode;
             //this.setSkin(this.defaultSkin);
             this._instance.isCache = this.isAnimationCached();
+            this.attachUtil.init(this);
             this._updateSkeletonData();
             this.setSkin(this.defaultSkin);
             this._updateUseTint();