cobra

cmd/msgtransfer/main.go

@@ -1,24 +1,16 @@
 package main
 
 import (
-	"OpenIM/internal/msgtransfer"
-	"OpenIM/pkg/common/config"
-	"OpenIM/pkg/common/constant"
-	"OpenIM/pkg/common/log"
-	"flag"
-	"sync"
+	"OpenIM/pkg/common/cmd"
+	"fmt"
+	"os"
 )
 
 func main() {
-	if err := config.InitConfig(""); err != nil {
-		panic(err.Error())
+	msgTransferCmd := cmd.NewMsgTransferCmd()
+	msgTransferCmd.AddPrometheusPortFlag()
+	if err := msgTransferCmd.Exec(); err != nil {
+		fmt.Println(err.Error())
+		os.Exit(1)
 	}
-	var wg sync.WaitGroup
-	wg.Add(1)
-	prometheusPort := flag.Int("prometheus_port", config.Config.Prometheus.MessageTransferPrometheusPort[0], "MessageTransferPrometheusPort default listen port")
-	log.NewPrivateLog(constant.LogFileName)
-	if err := msgtransfer.StartTransfer(*prometheusPort); err != nil {
-		panic(err)
-	}
-	wg.Wait()
 }