Exemplo n.º 1
0
func init() {
	dvid.SetLogMode(dvid.WarningMode)
}
Exemplo n.º 2
0
func main() {
	flag.BoolVar(showHelp, "h", false, "Show help message")
	flag.Usage = usage
	flag.Parse()

	if flag.NArg() == 0 {
		flag.Usage()
		os.Exit(0)
	}

	help := strings.ToLower(flag.Args()[0]) == "help"

	if help && flag.NArg() == 1 {
		*showHelp = true
	}
	if *showHelp || flag.NArg() == 0 {
		flag.Usage()
		os.Exit(0)
	}

	if help && flag.NArg() == 2 && strings.ToLower(flag.Args()[0]) == "server" {
		if err := DoCommand(dvid.Command([]string{"help"})); err != nil {
			fmt.Printf("Unable to get 'help' from DVID server at %q.\n%v\n", rpcAddress, err)
		}
		os.Exit(0)
	}

	if *runVerbose {
		dvid.SetLogMode(dvid.DebugMode)
	}
	if *readonly {
		server.SetReadOnly(true)
	}

	if *cpuprofile != "" {
		f, err := os.Create(*cpuprofile)
		if err != nil {
			log.Fatal(err)
		}
		pprof.StartCPUProfile(f)
		defer pprof.StopCPUProfile()
	}

	// Determine numer of logical CPUs on local machine and unless overridden, use
	// all of them.
	numCPU := runtime.NumCPU()
	if *useCPU != 0 {
		dvid.NumCPU = *useCPU
	} else if flag.NArg() >= 1 && flag.Args()[0] == "serve" {
		dvid.NumCPU = numCPU
	} else {
		dvid.NumCPU = 1
	}
	runtime.GOMAXPROCS(dvid.NumCPU)

	command := dvid.Command(flag.Args())
	if err := DoCommand(command); err != nil {
		fmt.Fprintln(os.Stderr, err.Error())
		os.Exit(1)
	}
}