blog_20171201_use_glog_with_cobra2.go

package main

import (
	goflag "flag"

	"github.com/golang/glog"
	"github.com/spf13/cobra"
	flag "github.com/spf13/pflag"
)

var (
	rootCmd = &cobra.Command{
		Long: "Use glog with cobra.",
		PersistentPreRun: func(cmd *cobra.Command, args []string) {
			// For cobra + glog flags. Available to all subcommands.
			goflag.Parse()
		},
	}

	s1 string
	s2 string
)

func init() {
	rootCmd.AddCommand(
		subCmd1(),
		subCmd2(),
	)

	flag.CommandLine.AddGoFlagSet(goflag.CommandLine)
}

func subCmd1() *cobra.Command {
	c := &cobra.Command{
		Use:   "sub1",
		Short: "test subcmd1",
		Long:  "Test sub command 1.",
		Run: func(cmd *cobra.Command, args []string) {
			glog.Infof("hello %v from subCmd1", s1)
		},
	}

	c.Flags().SortFlags = false
	c.Flags().StringVar(&s1, "target", "world", "say hello to")
	return c
}

func subCmd2() *cobra.Command {
	c := &cobra.Command{
		Use:   "sub2",
		Short: "test subcmd2",
		Long:  "Test sub command 2.",
		Run: func(cmd *cobra.Command, args []string) {
			glog.Infof("hello %v from subCmd2", s2)
		},
	}

	c.Flags().SortFlags = false
	c.Flags().StringVar(&s2, "target", "parallel world", "say hello to")
	return c
}

func main() {
	err := rootCmd.Execute()
	if err != nil {
		glog.Error(err)
	}
}