imposm3/cmd/imposm/main.go

package main

import (
	"fmt"
	golog "log"
	"os"
	"runtime"
	"strings"

	"github.com/omniscale/imposm3"
	"github.com/omniscale/imposm3/cache/query"
	"github.com/omniscale/imposm3/config"
	"github.com/omniscale/imposm3/import_"
	"github.com/omniscale/imposm3/logging"
	"github.com/omniscale/imposm3/stats"
	"github.com/omniscale/imposm3/update"
)

var log = logging.NewLogger("")

func PrintCmds() {
	fmt.Fprintf(os.Stderr, "Usage: %s COMMAND [args]\n\n", os.Args[0])
	fmt.Println("Available commands:")
	fmt.Println("\timport")
	fmt.Println("\tdiff")
	fmt.Println("\trun")
	fmt.Println("\tquery-cache")
	fmt.Println("\tversion")
}

func Main(usage func()) {
	golog.SetFlags(golog.LstdFlags | golog.Lshortfile)
	if os.Getenv("GOMAXPROCS") == "" {
		runtime.GOMAXPROCS(runtime.NumCPU())
	}

	if len(os.Args) <= 1 {
		usage()
		logging.Shutdown()
		os.Exit(1)
	}

	if strings.HasSuffix(os.Args[0], "imposm3") {
		fmt.Println("WARNING: Use imposm binary instead of imposm3!")
	}

	switch os.Args[1] {
	case "import":
		opts := config.ParseImport(os.Args[2:])
		if opts.Base.Httpprofile != "" {
			stats.StartHttpPProf(opts.Base.Httpprofile)
		}
		import_.Import(opts)
	case "diff":
		opts, files := config.ParseDiffImport(os.Args[2:])

		if opts.Httpprofile != "" {
			stats.StartHttpPProf(opts.Httpprofile)
		}
		update.Diff(opts, files)
	case "run":
		opts := config.ParseRunImport(os.Args[2:])

		if opts.Httpprofile != "" {
			stats.StartHttpPProf(opts.Httpprofile)
		}
		update.Run(opts)
	case "query-cache":
		query.Query(os.Args[2:])
	case "version":
		fmt.Println(imposm3.Version)
		os.Exit(0)
	default:
		usage()
		log.Fatalf("invalid command: '%s'", os.Args[1])
	}
	logging.Shutdown()
	os.Exit(0)

}

func main() {
	Main(PrintCmds)
}