handler: Update NetworkInfo cache on GetNetworkInfo request

Signed-off-by: Evgenii Baidakov <[email protected]>
nspcc-dev · Oct 23, 2024 · 309119b · 309119b
1 parent 262e100
commit 309119b
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/handlers/api.go b/handlers/api.go
@@ -49,6 +49,7 @@ type SessionToken struct {
 
 type networkInfoGetter interface {
 	NetworkInfo(ctx context.Context) (netmap.NetworkInfo, error)
+	StoreNetworkInfo(ni netmap.NetworkInfo)
 }
 
 const (

diff --git a/handlers/network.go b/handlers/network.go
@@ -17,6 +17,8 @@ func (a *RestAPI) GetNetworkInfo(ctx echo.Context) error {
 		return ctx.JSON(http.StatusBadRequest, resp)
 	}
 
+	a.networkInfoGetter.StoreNetworkInfo(networkInfo)
+
 	var resp apiserver.NetworkInfoOK
 	resp.AuditFee = networkInfo.AuditFee()
 	resp.StoragePrice = networkInfo.StoragePrice()

diff --git a/internal/cache/networkinfo.go b/internal/cache/networkinfo.go
@@ -43,9 +43,19 @@ func (n *NetworkInfo) NetworkInfo(ctx context.Context) (netmap.NetworkInfo, erro
 		return netmap.NetworkInfo{}, fmt.Errorf("get network info: %w", err)
 	}
 
+	n.update(ni)
+
+	return ni, nil
+}
+
+func (n *NetworkInfo) update(ni netmap.NetworkInfo) {
 	n.ttl = time.Duration(int64(ni.EpochDuration())/2*ni.MsPerBlock()) * time.Millisecond
 	n.validUntil = time.Now().Add(n.ttl)
 	n.ni = ni
+}
 
-	return ni, nil
+func (n *NetworkInfo) StoreNetworkInfo(ni netmap.NetworkInfo) {
+	n.mu.Lock()
+	n.update(ni)
+	n.mu.Unlock()
 }