Verify rwswitch status before update cfg

douban · Feb 2, 2024 · e3b905f · e3b905f
1 parent c101cf9
commit e3b905f
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/cassandra/prefix_switch.go b/cassandra/prefix_switch.go
@@ -32,6 +32,16 @@ const (
 	statusCr string = "br0w0cr1w0"
 )
 
+var (
+	allowRWStatus = map[string]bool{
+		statusBrw: true,
+		statusBrwCw: true,
+		statusBwCrw: true,
+		statusCrw: true,
+		statusCr: true,
+	}
+)
+
 type PrefixSwitcher struct {
 	trie *trie.Tree[rune, PrefixSwitchStatus]
 	defaultT PrefixSwitchStatus
@@ -309,6 +319,11 @@ func (s *PrefixSwitcher) LoadCfg(cfg *config.CassandraStoreCfg, cqlStore *Cassan
 }
 
 func (s *PrefixSwitcher) Upsert(cfg *config.CassandraStoreCfg, data map[string][]string, cqlStore *CassandraStore) error {
+	for rwStatus := range data {
+		if _, ok := allowRWStatus[rwStatus]; !ok {
+			return fmt.Errorf("%s is not a validate rwstatus", rwStatus)
+		}
+	}
 	dispatcherCfg := DisPatcherCfg(cfg.PrefixRWDispatcherCfg)
 	return dispatcherCfg.SaveToDB(data, cqlStore)
 }