[Merged by Bors] - hare/oracle: reset oracle when node enters synced state

CHANGELOG.md

@@ -23,6 +23,7 @@ The network will start using the new algorithm at layer 18_000 (2023-09-14 20:00
 * [#4923](https://github.com/spacemeshos/go-spacemesh/pull/4923) Faster ballot eligibility validation. Improves sync speed. 
 * [#4934](https://github.com/spacemeshos/go-spacemesh/pull/4934) Ensure state is synced before participating in tortoise consensus.
 * [#4939](https://github.com/spacemeshos/go-spacemesh/pull/4939) Make sure to fetch data from peers that are already connected.
+* [#4936](https://github.com/spacemeshos/go-spacemesh/pull/4936) Use correct hare active set after node was synced. Otherwise applied layer may lag slightly behind the rest.
 
 ## v1.1.2
 

hare/eligibility/oracle.go

@@ -60,14 +60,20 @@
 
 // Oracle is the hare eligibility oracle.
 type Oracle struct {
-	lock           sync.Mutex
+	mu           sync.Mutex
+	activesCache activeSetCache
+	fallback     map[types.EpochID][]types.ATXID
+	sync         system.SyncStateProvider
+	// NOTE(dshulyak) on switch from synced to not synced reset the cache
+	// to cope with https://github.com/spacemeshos/go-spacemesh/issues/4552
+	// until graded oracle is implemented
+	synced bool
+
 	beacons        system.BeaconGetter
 	cdb            *datastore.CachedDB
 	vrfSigner      *signing.VRFSigner
 	vrfVerifier    vrfVerifier
 	layersPerEpoch uint32
-	activesCache   activeSetCache
-	fallback       map[types.EpochID][]types.ATXID
 	cfg            config.Config
 	log.Log
 }
@@ -110,6 +116,27 @@
 	Layer  types.LayerID
 }
 
+func (o *Oracle) SetSync(sync system.SyncStateProvider) {
+	o.mu.Lock()
+	defer o.mu.Unlock()
+	o.sync = sync
+}
+
+func (o *Oracle) resetCacheOnSynced(ctx context.Context) {
+	if o.sync == nil {
+		return
+	}
+	synced := o.synced
+	o.synced = o.sync.IsSynced(ctx)
+	if !synced && o.synced {
+		ac, err := lru.New[types.EpochID, *cachedActiveSet](activesCacheSize)
+		if err != nil {
+			o.Log.With().Fatal("failed to create lru cache for active set", log.Err(err))
+		}
+		o.activesCache = ac
+	}
+}
+
 // buildVRFMessage builds the VRF message used as input for the BLS (msg=Beacon##Layer##Round).
 func (o *Oracle) buildVRFMessage(ctx context.Context, layer types.LayerID, round uint32) ([]byte, error) {
 	beacon, err := o.beacons.GetBeacon(layer.GetEpoch())
@@ -341,8 +368,9 @@
 		log.Stringer("target_epoch", targetEpoch),
 	)
 
-	o.lock.Lock()
-	defer o.lock.Unlock()
+	o.mu.Lock()
+	defer o.mu.Unlock()
+	o.resetCacheOnSynced(ctx)
 	if value, exists := o.activesCache.Get(targetEpoch); exists {
 		return value, nil
 	}
@@ -462,8 +490,8 @@
 		epoch,
 		log.Int("size", len(activeSet)),
 	)
-	o.lock.Lock()
-	defer o.lock.Unlock()
+	o.mu.Lock()
+	defer o.mu.Unlock()
 	if _, ok := o.fallback[epoch]; ok {
 		o.Log.With().Debug("fallback active set already exists", epoch)
 		return

hare/eligibility/oracle_test.go

@@ -869,6 +869,35 @@ func TestActiveSetDD(t *testing.T) {
 	}
 }
 
+func TestResetCache(t *testing.T) {
+	oracle := defaultOracle(t)
+	ctrl := gomock.NewController(t)
+
+	prev := oracle.activesCache
+	prev.Add(1, nil)
+
+	oracle.resetCacheOnSynced(context.Background())
+	require.Equal(t, prev, oracle.activesCache)
+
+	sync := mocks.NewMockSyncStateProvider(ctrl)
+	oracle.SetSync(sync)
+
+	sync.EXPECT().IsSynced(gomock.Any()).Return(false)
+	oracle.resetCacheOnSynced(context.Background())
+	require.Equal(t, prev, oracle.activesCache)
+
+	sync.EXPECT().IsSynced(gomock.Any()).Return(true)
+	oracle.resetCacheOnSynced(context.Background())
+	require.NotEqual(t, prev, oracle.activesCache)
+
+	prev = oracle.activesCache
+	prev.Add(1, nil)
+
+	sync.EXPECT().IsSynced(gomock.Any()).Return(true)
+	oracle.resetCacheOnSynced(context.Background())
+	require.Equal(t, prev, oracle.activesCache)
+}
+
 func FuzzVrfMessageConsistency(f *testing.F) {
 	tester.FuzzConsistency[VrfMessage](f)
 }

node/node.go

@@ -735,6 +735,7 @@ func (app *App) initServices(ctx context.Context) error {
 	)
 	// TODO(dshulyak) this needs to be improved, but dependency graph is a bit complicated
 	beaconProtocol.SetSyncState(newSyncer)
+	app.hOracle.SetSync(newSyncer)
 
 	hareOutputCh := make(chan hare.LayerOutput, app.Config.HARE.LimitConcurrent)
 	app.blockGen = blocks.NewGenerator(app.cachedDB, executor, msh, fetcherWrapped, app.certifier, patrol,