From efa1f8c1f034bdcaa532aec1d9bb265934490740 Mon Sep 17 00:00:00 2001
From: pandx <dxpan2002@163.com>
Date: Fri, 3 Jan 2025 10:16:38 +0800
Subject: [PATCH] refine remove logic.

---
 scheduler/schedule.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/scheduler/schedule.cpp b/scheduler/schedule.cpp
index 3312312..79c3bb9 100644
--- a/scheduler/schedule.cpp
+++ b/scheduler/schedule.cpp
@@ -670,11 +670,15 @@ void SensorScheduler::ClearScheduleCfg(int short_addr) {
         ShortAddrCfg::ClearCfg();
         UpdateCfg::ClearCfg();
         UpgradeCfg::ClearCfg();
+        wave_feature_set_inst::instance().RemoveAllFeatureCfg();
+        wave_feature_set_inst::instance().RemoveAllWaveCfg();
     } else {
         UpdateConfigResult(short_addr, 0);
         UpgradeResult(short_addr, kUpgradeSuccess);
         short_addr_map_.erase(short_addr);
         ShortAddrCfg::WriteCfg(short_addr_map_);
+        wave_feature_set_inst::instance().RemoveFeatureCfg(short_addr);
+        wave_feature_set_inst::instance().RemoveWaveCfg(short_addr);
     }
     
 }