Fix missing disconnect file (#162)

Signed-off-by: Congqi Xia <[email protected]>
milvus-io · Jul 4, 2023 · 350cd9a · 350cd9a
1 parent 7fa12fe
commit 350cd9a
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/states/disconnected.go b/states/disconnected.go
@@ -0,0 +1,32 @@
+package states
+
+import (
+	"github.com/milvus-io/birdwatcher/configs"
+	"github.com/spf13/cobra"
+)
+
+type disconnectState struct {
+	cmdState
+	config *configs.Config
+}
+
+func (s *disconnectState) SetupCommands() {
+	cmd := &cobra.Command{}
+
+	s.mergeFunctionCommands(cmd, s)
+
+	s.rootCmd = cmd
+	s.setupFn = s.SetupCommands
+}
+
+func getDisconnectedState(config *configs.Config) State {
+	state := &disconnectState{
+		cmdState: cmdState{
+			label: "Offline",
+		},
+		config: config,
+	}
+
+	state.SetupCommands()
+	return state
+}