refactor main

.gitignore

@@ -1,8 +1,6 @@
 conf.yml
 *.out
 *.log
-*/**/main
-main
 dist
 .idea/**
 .vscode

dtmsvr/entry/main.go

@@ -0,0 +1,60 @@
+package entry
+
+import (
+	"flag"
+	"fmt"
+	"os"
+	"path/filepath"
+
+	"github.com/dtm-labs/dtm/dtmcli/logger"
+	"github.com/dtm-labs/dtm/dtmsvr"
+	"github.com/dtm-labs/dtm/dtmsvr/config"
+	"github.com/dtm-labs/dtm/dtmsvr/storage/registry"
+	"go.uber.org/automaxprocs/maxprocs"
+)
+
+func ver(version *string) {
+	if *version == "" {
+		*version = "0.0.0-dev"
+	}
+	fmt.Printf("dtm version: %s\n", *version)
+}
+
+func usage() {
+	cmd := filepath.Base(os.Args[0])
+	s := "Usage: %s [options]\n\n"
+	fmt.Fprintf(os.Stderr, s, cmd)
+	flag.PrintDefaults()
+}
+
+var isVersion = flag.Bool("v", false, "Show the version of dtm.")
+var isDebug = flag.Bool("d", false, "Set log level to debug.")
+var isHelp = flag.Bool("h", false, "Show the help information about dtm.")
+var isReset = flag.Bool("r", false, "Reset dtm server data.")
+var confFile = flag.String("c", "", "Path to the server configuration file.")
+
+func Main(version *string) {
+	flag.Parse()
+	if flag.NArg() > 0 || *isHelp {
+		usage()
+		return
+	} else if *isVersion {
+		ver(version)
+		return
+	}
+	logger.Infof("dtm version is: %s", *version)
+	config.MustLoadConfig(*confFile)
+	conf := &config.Config
+	if *isDebug {
+		conf.LogLevel = "debug"
+	}
+	logger.InitLog2(conf.LogLevel, conf.Log.Outputs, conf.Log.RotationEnable, conf.Log.RotationConfigJSON)
+	if *isReset {
+		dtmsvr.PopulateDB(false)
+	}
+	_, _ = maxprocs.Set(maxprocs.Logger(logger.Infof))
+	registry.WaitStoreUp()
+	dtmsvr.StartSvr()              // start dtmsvr api
+	go dtmsvr.CronExpiredTrans(-1) // start dtmsvr cron job
+	select {}
+}

dtmsvr/microservices/drivers.go

@@ -0,0 +1,9 @@
+package microservices
+
+// load the microserver driver
+import (
+	_ "github.com/dtm-labs/dtmdriver-gozero"
+	_ "github.com/dtm-labs/dtmdriver-kratos"
+	_ "github.com/dtm-labs/dtmdriver-polaris"
+	_ "github.com/dtm-labs/dtmdriver-protocol1"
+)

main.go

@@ -7,70 +7,13 @@
 package main
 
 import (
-	"flag"
+	"github.com/dtm-labs/dtm/dtmsvr/entry"
-	"fmt"
+	_ "github.com/dtm-labs/dtm/dtmsvr/microservices"
-	"os"
-	"path/filepath"
-
-	"go.uber.org/automaxprocs/maxprocs"
-
-	"github.com/dtm-labs/dtm/dtmcli/logger"
-	"github.com/dtm-labs/dtm/dtmsvr"
-	"github.com/dtm-labs/dtm/dtmsvr/config"
-	"github.com/dtm-labs/dtm/dtmsvr/storage/registry"
-
-	// load the microserver driver
-	_ "github.com/dtm-labs/dtmdriver-gozero"
-	_ "github.com/dtm-labs/dtmdriver-kratos"
-	_ "github.com/dtm-labs/dtmdriver-polaris"
-	_ "github.com/dtm-labs/dtmdriver-protocol1"
 )
 
-// Version declares version info
+// Version defines version info. It is set by -ldflags.
 var Version string
 
-func version() {
-	if Version == "" {
-		Version = "0.0.0-dev"
-	}
-	fmt.Printf("dtm version: %s\n", Version)
-}
-
-func usage() {
-	cmd := filepath.Base(os.Args[0])
-	s := "Usage: %s [options]\n\n"
-	fmt.Fprintf(os.Stderr, s, cmd)
-	flag.PrintDefaults()
-}
-
-var isVersion = flag.Bool("v", false, "Show the version of dtm.")
-var isDebug = flag.Bool("d", false, "Set log level to debug.")
-var isHelp = flag.Bool("h", false, "Show the help information about dtm.")
-var isReset = flag.Bool("r", false, "Reset dtm server data.")
-var confFile = flag.String("c", "", "Path to the server configuration file.")
-
 func main() {
-	flag.Parse()
+	entry.Main(&Version)
-	if flag.NArg() > 0 || *isHelp {
-		usage()
-		return
-	} else if *isVersion {
-		version()
-		return
-	}
-	logger.Infof("dtm version is: %s", Version)
-	config.MustLoadConfig(*confFile)
-	conf := &config.Config
-	if *isDebug {
-		conf.LogLevel = "debug"
-	}
-	logger.InitLog2(conf.LogLevel, conf.Log.Outputs, conf.Log.RotationEnable, conf.Log.RotationConfigJSON)
-	if *isReset {
-		dtmsvr.PopulateDB(false)
-	}
-	_, _ = maxprocs.Set(maxprocs.Logger(logger.Infof))
-	registry.WaitStoreUp()
-	dtmsvr.StartSvr()              // start dtmsvr api
-	go dtmsvr.CronExpiredTrans(-1) // start dtmsvr cron job
-	select {}
 }