cm: database: persist disableRebalance flag for launcher instance info

Signed-off-by: Mykola Kobets <mykola_kobets@epam.com>
Reviewed-by: Mykhailo Lohvynenko <mykhailo_lohvynenko@epam.com>
Reviewed-by: Oleksandr Grytsov <oleksandr_grytsov@epam.com>
Reviewed-by: Mykola Solianko <mykola_solianko@epam.com>

Author: mykola-kobets-epam

src/cm/database/database.cpp

@@ -593,8 +593,8 @@ Error Database::AddInstance(const launcher::InstanceInfo& info)
         FromAos(info, row);
         *mSession << "INSERT INTO launcher_instances (itemID, subjectID, instance, type, preinstalled, manifestDigest, "
                      "nodeID, prevNodeID, runtimeID, uid, gid, timestamp, state, isUnitSubject, version, ownerID, "
-                     "subjectType, labels, priority) "
-                     "VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?);",
+                     "subjectType, labels, priority, disableRebalancing) "
+                     "VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?);",
             bind(row), now;
     } catch (const std::exception& e) {
         return AOS_ERROR_WRAP(common::utils::ToAosError(e));
@@ -613,16 +613,17 @@ Error Database::UpdateInstance(const launcher::InstanceInfo& info)
         statement
             << "UPDATE launcher_instances SET manifestDigest = ?, nodeID = ?, prevNodeID = ?, runtimeID = ?, uid = ?, "
                "gid = ?, timestamp = ?, state = ?, isUnitSubject = ?, ownerID = ?, subjectType = ?, labels = ?, "
-               "priority = ? "
+               "priority = ?, disableRebalancing = ? "
                "WHERE itemID = ? AND subjectID = ? AND instance = ? "
                "AND type = ? AND preinstalled = ? AND version = ?;",
             bind(info.mManifestDigest.CStr()), bind(info.mNodeID.CStr()), bind(info.mPrevNodeID.CStr()),
             bind(info.mRuntimeID.CStr()), bind(info.mUID), bind(info.mGID), bind(info.mTimestamp.UnixNano()),
             bind(info.mState.ToString().CStr()), bind(info.mIsUnitSubject), bind(info.mOwnerID.CStr()),
             bind(info.mSubjectType.ToString().CStr()), bind(SerializeLabels(info.mLabels)), bind(info.mPriority),
-            bind(info.mInstanceIdent.mItemID.CStr()), bind(info.mInstanceIdent.mSubjectID.CStr()),
-            bind(info.mInstanceIdent.mInstance), bind(info.mInstanceIdent.mType.ToString().CStr()),
-            bind(info.mInstanceIdent.mPreinstalled), bind(info.mVersion.CStr());
+            bind(info.mDisableRebalancing), bind(info.mInstanceIdent.mItemID.CStr()),
+            bind(info.mInstanceIdent.mSubjectID.CStr()), bind(info.mInstanceIdent.mInstance),
+            bind(info.mInstanceIdent.mType.ToString().CStr()), bind(info.mInstanceIdent.mPreinstalled),
+            bind(info.mVersion.CStr());
 
         if (statement.execute() != 1) {
             return ErrorEnum::eNotFound;
@@ -643,7 +644,7 @@ Error Database::LoadActiveInstances(Array<launcher::InstanceInfo>& instances) co
 
         *mSession << "SELECT itemID, subjectID, instance, type, preinstalled, manifestDigest, nodeID, prevNodeID, "
                      "runtimeID, uid, gid, timestamp, state, isUnitSubject, version, ownerID, subjectType, labels, "
-                     "priority "
+                     "priority, disableRebalancing "
                      "FROM launcher_instances;",
             into(rows), now;
 
@@ -1089,6 +1090,7 @@ void Database::CreateTables()
                  "subjectType TEXT,"
                  "labels TEXT,"
                  "priority INTEGER,"
+                 "disableRebalancing INTEGER,"
                  "PRIMARY KEY(itemID,subjectID,instance,type,preinstalled,version)"
                  ");",
         now;
@@ -1223,6 +1225,7 @@ void Database::FromAos(const launcher::InstanceInfo& src, LauncherInstanceInfoRo
     dst.set<ToInt(LauncherInstanceInfoColumns::eSubjectType)>(src.mSubjectType.ToString().CStr());
     dst.set<ToInt(LauncherInstanceInfoColumns::eLabels)>(SerializeLabels(src.mLabels));
     dst.set<ToInt(LauncherInstanceInfoColumns::ePriority)>(src.mPriority);
+    dst.set<ToInt(LauncherInstanceInfoColumns::eDisableRebalancing)>(src.mDisableRebalancing);
 }
 
 void Database::ToAos(const LauncherInstanceInfoRow& src, launcher::InstanceInfo& dst)
@@ -1260,7 +1263,8 @@ void Database::ToAos(const LauncherInstanceInfoRow& src, launcher::InstanceInfo&
     AOS_ERROR_CHECK_AND_THROW(AOS_ERROR_WRAP(err), "failed to parse subject type");
 
     DeserializeLabels(src.get<ToInt(LauncherInstanceInfoColumns::eLabels)>(), dst.mLabels);
-    dst.mPriority = src.get<ToInt(LauncherInstanceInfoColumns::ePriority)>();
+    dst.mPriority           = src.get<ToInt(LauncherInstanceInfoColumns::ePriority)>();
+    dst.mDisableRebalancing = src.get<ToInt(LauncherInstanceInfoColumns::eDisableRebalancing)>();
 }
 
 void Database::FromAos(const imagemanager::ItemInfo& src, ImageManagerItemInfoRow& dst)

src/cm/database/database.hpp

@@ -383,11 +383,12 @@ class Database : public DatabaseItf, public networkmanager::StorageItf {
         eOwnerID,
         eSubjectType,
         eLabels,
-        ePriority
+        ePriority,
+        eDisableRebalancing
     };
     using LauncherInstanceInfoRow = Poco::Tuple<std::string, std::string, uint64_t, std::string, bool, std::string,
         std::string, std::string, std::string, uint32_t, uint32_t, uint64_t, std::string, bool, std::string,
-        std::string, std::string, std::string, size_t>;
+        std::string, std::string, std::string, size_t, bool>;
 
     enum class ImageManagerItemInfoColumns : int {
         eItemID = 0,

src/cm/database/tests/database.cpp

@@ -109,7 +109,7 @@ launcher::InstanceInfo CreateLauncherInstanceInfo(const char* itemID, const char
     const char* manifestDigest, const char* nodeID, UpdateItemType itemType = UpdateItemTypeEnum::eService,
     launcher::InstanceStateEnum state = launcher::InstanceStateEnum::eCached, bool isUnitSubject = false,
     const char* version = "1.0.0", const char* ownerID = "owner1", SubjectTypeEnum subjectType = SubjectTypeEnum::eUser,
-    size_t priority = 0, std::vector<const char*> labels = {})
+    size_t priority = 0, std::vector<const char*> labels = {}, bool disableRebalancing = false)
 {
     launcher::InstanceInfo info;
 
@@ -130,7 +130,8 @@ launcher::InstanceInfo CreateLauncherInstanceInfo(const char* itemID, const char
     for (const char* label : labels) {
         AOS_ERROR_CHECK_AND_THROW(info.mLabels.EmplaceBack(label), "can't add label");
     }
-    info.mPriority = priority;
+    info.mPriority           = priority;
+    info.mDisableRebalancing = disableRebalancing;
 
     return info;
 }
@@ -661,12 +662,12 @@ TEST_F(CMDatabaseTest, LauncherGetActiveInstances)
     ASSERT_TRUE(mDB.LoadActiveInstances(*emptyInstances).IsNone());
     EXPECT_EQ(emptyInstances->Size(), 0);
 
-    auto instance1
-        = CreateLauncherInstanceInfo("service1", "subject1", 0, "image1", "node1", UpdateItemTypeEnum::eService,
-            launcher::InstanceStateEnum::eActive, false, "1.0.0", "owner1", SubjectTypeEnum::eUser, 80, {"label4"});
+    auto instance1 = CreateLauncherInstanceInfo("service1", "subject1", 0, "image1", "node1",
+        UpdateItemTypeEnum::eService, launcher::InstanceStateEnum::eActive, false, "1.0.0", "owner1",
+        SubjectTypeEnum::eUser, 80, {"label4"}, true);
     auto instance2 = CreateLauncherInstanceInfo("service2", "subject2", 0, "image2", "node2",
         UpdateItemTypeEnum::eService, launcher::InstanceStateEnum::eDisabled, true, "2.0.0", "owner2",
-        SubjectTypeEnum::eUser, 200, {"label5", "label6", "label7"});
+        SubjectTypeEnum::eUser, 200, {"label5", "label6", "label7"}, false);
 
     // Add instances
     ASSERT_TRUE(mDB.AddInstance(instance1).IsNone());