fix drain compatability (#6613) (#6716)

Author: vporyadke

ydb/core/mind/hive/tx__load_everything.cpp

@@ -318,6 +318,11 @@ class TTxLoadEverything : public TTransactionBase<THive> {
                 node.Statistics = nodeRowset.GetValueOrDefault<Schema::Node::Statistics>();
                 node.Name = nodeRowset.GetValueOrDefault<Schema::Node::Name>();
                 node.BecomeUpOnRestart = nodeRowset.GetValueOrDefault<Schema::Node::BecomeUpOnRestart>(false);
+                if (node.BecomeUpOnRestart) {
+                    // If a node must become up on restart, it must have been down
+                    // That was not persisted to avoid issues with downgrades
+                    node.Down = true;
+                }
                 if (nodeRowset.HaveValue<Schema::Node::Location>()) {
                     auto location = nodeRowset.GetValue<Schema::Node::Location>();
                     if (location.HasDataCenter()) {

ydb/core/mind/hive/tx__switch_drain.cpp

@@ -38,7 +38,10 @@ class TTxSwitchDrainOn : public TTransactionBase<THive> {
                     }
                     node->SetDown(true);
                     if (Settings.Persist) {
-                        db.Table<Schema::Node>().Key(NodeId).Update<Schema::Node::Down, Schema::Node::BecomeUpOnRestart>(true, node->BecomeUpOnRestart);
+                        db.Table<Schema::Node>().Key(NodeId).Update<Schema::Node::BecomeUpOnRestart>(node->BecomeUpOnRestart);
+                        if (Settings.DownPolicy == NKikimrHive::DRAIN_POLICY_KEEP_DOWN) {
+                            db.Table<Schema::Node>().Key(NodeId).Update<Schema::Node::Down>(true);
+                        }
                     }
                 }
                 Self->StartHiveDrain(NodeId, std::move(Settings));