github.com/pingcap/tiflow@v0.0.0-20240520035814-5bf52d54e205/pkg/cmd/cmd.go (about)

     1  // Copyright 2021 PingCAP, Inc.
     2  //
     3  // Licensed under the Apache License, Version 2.0 (the "License");
     4  // you may not use this file except in compliance with the License.
     5  // You may obtain a copy of the License at
     6  //
     7  //     http://www.apache.org/licenses/LICENSE-2.0
     8  //
     9  // Unless required by applicable law or agreed to in writing, software
    10  // distributed under the License is distributed on an "AS IS" BASIS,
    11  // See the License for the specific language governing permissions and
    12  // limitations under the License.
    13  
    14  package cmd
    15  
    16  import (
    17  	"os"
    18  
    19  	"github.com/pingcap/tiflow/pkg/cmd/cli"
    20  	"github.com/pingcap/tiflow/pkg/cmd/redo"
    21  	"github.com/pingcap/tiflow/pkg/cmd/server"
    22  	"github.com/pingcap/tiflow/pkg/cmd/version"
    23  	"github.com/spf13/cobra"
    24  )
    25  
    26  // NewCmd creates the root command.
    27  func NewCmd() *cobra.Command {
    28  	return &cobra.Command{
    29  		Use:   "cdc",
    30  		Short: "CDC",
    31  		Long:  `Change Data Capture`,
    32  		CompletionOptions: cobra.CompletionOptions{
    33  			DisableDefaultCmd: true,
    34  		},
    35  	}
    36  }
    37  
    38  // Run runs the root command.
    39  func Run() {
    40  	cmd := NewCmd()
    41  
    42  	cmd.SetOut(os.Stdout)
    43  	cmd.SetErr(os.Stderr)
    44  
    45  	cmd.AddCommand(server.NewCmdServer())
    46  	cmd.AddCommand(cli.NewCmdCli())
    47  	cmd.AddCommand(version.NewCmdVersion())
    48  	cmd.AddCommand(redo.NewCmdRedo())
    49  
    50  	if err := cmd.Execute(); err != nil {
    51  		cmd.PrintErrln(err)
    52  		os.Exit(1)
    53  	}
    54  }