controller: DCS task: Node & DB

Add database support for the DCS task (set/get id). Add Node entry for the DCS task ID. Signed-off-by: itay elenzweig <[email protected]>
prplfoundation · Jan 20, 2020 · 5cc493c · 5cc493c
1 parent 734a2c5
commit 5cc493c
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 0 deletions.
diff --git a/controller/src/beerocks/master/db/db.cpp b/controller/src/beerocks/master/db/db.cpp
@@ -3320,6 +3320,29 @@ bool db::assign_rdkb_wlan_task_id(int new_task_id)
 
 int db::get_rdkb_wlan_task_id() { return rdkb_wlan_task_id; }
 
+bool db::assign_dynamic_channel_selection_task_id(const sMacAddr &mac, int new_task_id)
+{
+    auto n = get_node(mac);
+    if (!n) {
+        LOG(WARNING) << __FUNCTION__ << " - node " << network_utils::mac_to_string(mac)
+                     << " does not exist!";
+        return false;
+    }
+    n->dynamic_channel_selection_task_id = new_task_id;
+    return true;
+}
+
+int db::get_dynamic_channel_selection_task_id(const sMacAddr &mac)
+{
+    auto n = get_node(mac);
+    if (!n) {
+        LOG(WARNING) << __FUNCTION__ << " - node " << network_utils::mac_to_string(mac)
+                     << " does not exist!";
+        return -1;
+    }
+    return n->dynamic_channel_selection_task_id;
+}
+
 void db::lock() { db_mutex.lock(); }
 
 void db::unlock() { db_mutex.unlock(); }

diff --git a/controller/src/beerocks/master/db/db.h b/controller/src/beerocks/master/db/db.h
@@ -714,6 +714,9 @@ class db {
     bool assign_channel_selection_task_id(int new_task_id);
     int get_channel_selection_task_id();
 
+    bool assign_dynamic_channel_selection_task_id(const sMacAddr &mac, int new_task_id);
+    int get_dynamic_channel_selection_task_id(const sMacAddr &mac);
+
     void lock();
     void unlock();
 

diff --git a/controller/src/beerocks/master/db/node.h b/controller/src/beerocks/master/db/node.h
@@ -101,6 +101,7 @@ class node {
     int load_balancer_task_id                    = -1;
     int client_locating_task_id_new_connection   = -1;
     int client_locating_task_id_exist_connection = -1;
+    int dynamic_channel_selection_task_id        = -1;
 
     std::chrono::steady_clock::time_point measurement_sent_timestamp;
     int measurement_recv_delta  = 0;