From 425479172ca1e1824bbfea2a0ceae8b716f0110c Mon Sep 17 00:00:00 2001
From: Mikhail Rakhmanov <rakhmanov.m@gmail.com>
Date: Tue, 18 Mar 2025 13:51:06 +0100
Subject: [PATCH] Add build time parameter

---
 commonspace/object/tree/synctree/synctree.go  | 3 +++
 commonspace/object/tree/synctree/treestats.go | 8 +++++---
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/commonspace/object/tree/synctree/synctree.go b/commonspace/object/tree/synctree/synctree.go
index c76d6114..360f78d0 100644
--- a/commonspace/object/tree/synctree/synctree.go
+++ b/commonspace/object/tree/synctree/synctree.go
@@ -60,6 +60,7 @@ type syncTree struct {
 	onClose        func(id string)
 	isClosed       bool
 	isDeleted      bool
+	buildTime      time.Duration
 }
 
 var log = logger.NewNamed("common.commonspace.synctree")
@@ -114,6 +115,7 @@ func PutSyncTree(ctx context.Context, payload treestorage.TreeStorageCreatePaylo
 }
 
 func buildSyncTree(ctx context.Context, peerId string, deps BuildDeps) (t SyncTree, err error) {
+	buildStart := time.Now()
 	objTree, err := deps.BuildObjectTree(deps.TreeStorage, deps.AclList)
 	if err != nil {
 		return
@@ -126,6 +128,7 @@ func buildSyncTree(ctx context.Context, peerId string, deps BuildDeps) (t SyncTr
 		listener:       deps.Listener,
 		syncStatus:     deps.SyncStatus,
 		statsCollector: deps.StatsCollector,
+		buildTime:      time.Since(buildStart),
 	}
 	syncHandler := NewSyncHandler(syncTree, syncClient, deps.SpaceId)
 	syncTree.ObjectSyncHandler = syncHandler
diff --git a/commonspace/object/tree/synctree/treestats.go b/commonspace/object/tree/synctree/treestats.go
index 4c9cf4b1..5ee0f7da 100644
--- a/commonspace/object/tree/synctree/treestats.go
+++ b/commonspace/object/tree/synctree/treestats.go
@@ -5,19 +5,19 @@ import (
 )
 
 type TreeStatsCollector struct {
-	trees   map[string]SyncTree
+	trees   map[string]*syncTree
 	mutex   sync.Mutex
 	spaceId string
 }
 
 func NewTreeStatsCollector(spaceId string) *TreeStatsCollector {
 	return &TreeStatsCollector{
-		trees:   make(map[string]SyncTree),
+		trees:   make(map[string]*syncTree),
 		spaceId: spaceId,
 	}
 }
 
-func (t *TreeStatsCollector) Register(tree SyncTree) {
+func (t *TreeStatsCollector) Register(tree *syncTree) {
 	t.mutex.Lock()
 	defer t.mutex.Unlock()
 	t.trees[tree.Id()] = tree
@@ -35,6 +35,7 @@ func (t *TreeStatsCollector) Collect() []TreeStats {
 			Heads:           tree.Heads(),
 			ObjectId:        tree.Id(),
 			SpaceId:         t.spaceId,
+			BuildTimeSecs:   int(tree.buildTime.Seconds()),
 		})
 		tree.Unlock()
 	}
@@ -53,4 +54,5 @@ type TreeStats struct {
 	Heads           []string `json:"heads"`
 	ObjectId        string   `json:"object_id"`
 	SpaceId         string   `json:"space_id"`
+	BuildTimeSecs   int      `json:"build_time"`
 }