From a67aebf526fc9d952abc880143e8bed14029c7dc Mon Sep 17 00:00:00 2001
From: mcrakhman <rakhmanov.m@gmail.com>
Date: Mon, 5 Dec 2022 19:11:16 +0100
Subject: [PATCH] Remove duplicated code

---
 common/commonspace/synctree/synctree.go | 31 +++++++++++--------------
 1 file changed, 13 insertions(+), 18 deletions(-)

diff --git a/common/commonspace/synctree/synctree.go b/common/commonspace/synctree/synctree.go
index cd874462..967e3dc9 100644
--- a/common/commonspace/synctree/synctree.go
+++ b/common/commonspace/synctree/synctree.go
@@ -100,13 +100,7 @@ func DeriveSyncTree(ctx context.Context, deps CreateDeps) (t SyncTree, err error
 	t = syncTree
 	syncTree.Lock()
 	defer syncTree.Unlock()
-	if syncTree.listener != nil {
-		syncTree.listener.Rebuild(syncTree)
-	}
-	syncTree.treeUsage.Add(1)
-	if syncTree.notifiable != nil {
-		syncTree.notifiable.UpdateHeads(objTree.ID(), objTree.Heads())
-	}
+	syncTree.afterBuild()
 
 	headUpdate := syncClient.CreateHeadUpdate(t, nil)
 	err = syncClient.BroadcastAsync(headUpdate)
@@ -136,13 +130,7 @@ func CreateSyncTree(ctx context.Context, deps CreateDeps) (t SyncTree, err error
 	syncTree.Lock()
 	defer syncTree.Unlock()
 	// TODO: refactor here because the code is duplicated, when we create a tree we should only create a storage and then build a tree
-	if syncTree.listener != nil {
-		syncTree.listener.Rebuild(syncTree)
-	}
-	syncTree.treeUsage.Add(1)
-	if syncTree.notifiable != nil {
-		syncTree.notifiable.UpdateHeads(objTree.ID(), objTree.Heads())
-	}
+	syncTree.afterBuild()
 
 	headUpdate := syncClient.CreateHeadUpdate(t, nil)
 	err = syncClient.BroadcastAsync(headUpdate)
@@ -240,10 +228,7 @@ func buildSyncTree(ctx context.Context, isFirstBuild bool, deps BuildDeps) (t Sy
 	t = syncTree
 	syncTree.Lock()
 	defer syncTree.Unlock()
-	if syncTree.listener != nil {
-		syncTree.listener.Rebuild(syncTree)
-	}
-	syncTree.treeUsage.Add(1)
+	syncTree.afterBuild()
 
 	headUpdate := syncTree.syncClient.CreateHeadUpdate(t, nil)
 	// here we will have different behaviour based on who is sending this update
@@ -353,3 +338,13 @@ func (s *syncTree) Ping() (err error) {
 	headUpdate := s.syncClient.CreateHeadUpdate(s, nil)
 	return s.syncClient.BroadcastAsyncOrSendResponsible(headUpdate)
 }
+
+func (s *syncTree) afterBuild() {
+	if s.listener != nil {
+		s.listener.Rebuild(s)
+	}
+	s.treeUsage.Add(1)
+	if s.notifiable != nil {
+		s.notifiable.UpdateHeads(s.ID(), s.Heads())
+	}
+}