add node roles (#9413)

Author: adameat

ydb/core/base/board_replica.cpp

@@ -1,4 +1,5 @@
 #include "statestorage_impl.h"
+#include <ydb/core/node_whiteboard/node_whiteboard.h>
 #include <ydb/library/services/services.pb.h>
 #include <ydb/library/actors/core/interconnect.h>
 
@@ -17,7 +18,7 @@
 
 namespace NKikimr {
 
-class TBoardReplicaActor : public TActor<TBoardReplicaActor> {
+class TBoardReplicaActor : public TActorBootstrapped<TBoardReplicaActor> {
 
     using TOwnerIndex = TMap<TActorId, ui32, TActorId::TOrderedCmp>;
     using TPathIndex = TMap<TString, TSet<ui32>>;
@@ -419,9 +420,15 @@ class TBoardReplicaActor : public TActor<TBoardReplicaActor> {
     }
 
     TBoardReplicaActor()
-        : TActor(&TThis::StateWork)
     {}
 
+    void Bootstrap() {
+        auto localNodeId = SelfId().NodeId();
+        auto whiteboardId = NNodeWhiteboard::MakeNodeWhiteboardServiceId(localNodeId);
+        Send(whiteboardId, new NNodeWhiteboard::TEvWhiteboard::TEvSystemStateAddRole("StateStorageBoard"));
+        Become(&TThis::StateWork);
+    }
+
     STATEFN(StateWork) {
         switch (ev->GetTypeRewrite()) {
             hFunc(TEvStateStorage::TEvReplicaBoardPublish, Handle);

ydb/core/base/statestorage_replica.cpp

@@ -7,6 +7,7 @@
 #include <ydb/library/actors/core/interconnect.h>
 #include <ydb/library/actors/core/hfunc.h>
 #include <ydb/library/actors/core/log.h>
+#include <ydb/core/node_whiteboard/node_whiteboard.h>
 
 #include <util/generic/map.h>
 #include <util/generic/hash_set.h>
@@ -22,7 +23,7 @@
 
 namespace NKikimr {
 
-class TStateStorageReplica : public TActor<TStateStorageReplica> {
+class TStateStorageReplica : public TActorBootstrapped<TStateStorageReplica> {
     TIntrusivePtr<TStateStorageInfo> Info;
     const ui32 ReplicaIndex;
 
@@ -416,13 +417,19 @@ class TStateStorageReplica : public TActor<TStateStorageReplica> {
     }
 
     TStateStorageReplica(const TIntrusivePtr<TStateStorageInfo> &info, ui32 replicaIndex)
-        : TActor(&TThis::StateInit)
-        , Info(info)
+        : Info(info)
         , ReplicaIndex(replicaIndex)
     {
         Y_UNUSED(ReplicaIndex);
     }
 
+    void Bootstrap() {
+        auto localNodeId = SelfId().NodeId();
+        auto whiteboardId = NNodeWhiteboard::MakeNodeWhiteboardServiceId(localNodeId);
+        Send(whiteboardId, new NNodeWhiteboard::TEvWhiteboard::TEvSystemStateAddRole("StateStorage"));
+        Become(&TThis::StateInit);
+    }
+
     STATEFN(StateInit) {
         switch (ev->GetTypeRewrite()) {
             hFunc(TEvStateStorage::TEvReplicaLookup, Handle);

ydb/core/tablet/bootstrapper.cpp

@@ -3,6 +3,7 @@
 #include <ydb/core/base/tablet.h>
 #include <ydb/core/base/statestorage.h>
 #include <ydb/core/base/appdata.h>
+#include <ydb/core/node_whiteboard/node_whiteboard.h>
 
 #include <ydb/library/actors/core/interconnect.h>
 #include <ydb/library/actors/core/hfunc.h>
@@ -190,6 +191,10 @@ class TBootstrapper : public TActor<TBootstrapper> {
     }
 
     void BeginNewRound(const TActorContext &ctx) {
+        auto localNodeId = SelfId().NodeId();
+        auto whiteboardId = NNodeWhiteboard::MakeNodeWhiteboardServiceId(localNodeId);
+        Send(whiteboardId, new NNodeWhiteboard::TEvWhiteboard::TEvSystemStateAddRole("Bootstrapper"));
+
         if (BootstrapperInfo->OtherNodes.empty())
             return Boot(ctx);
 

ydb/core/tx/scheme_board/replica.cpp

@@ -8,6 +8,7 @@
 #include "replica.h"
 
 #include <ydb/core/scheme/scheme_pathid.h>
+#include <ydb/core/node_whiteboard/node_whiteboard.h>
 
 #include <ydb/library/services/services.pb.h>
 #include <ydb/library/yverify_stream/yverify_stream.h>
@@ -1269,6 +1270,9 @@ class TReplica: public TMonitorableActor<TReplica> {
 
     void Bootstrap() {
         TMonitorableActor::Bootstrap();
+        auto localNodeId = SelfId().NodeId();
+        auto whiteboardId = NNodeWhiteboard::MakeNodeWhiteboardServiceId(localNodeId);
+        Send(whiteboardId, new NNodeWhiteboard::TEvWhiteboard::TEvSystemStateAddRole("SchemeBoard"));
         Become(&TThis::StateWork);
     }