proto modify

cmd/push/main.go

@@ -2,30 +2,18 @@ package main
 
 import (
 	"OpenIM/internal/push"
+	"OpenIM/internal/startrpc"
 	"OpenIM/pkg/common/config"
 	"OpenIM/pkg/common/constant"
 	"OpenIM/pkg/common/log"
-	"flag"
-	"fmt"
-	"sync"
 )
 
 func main() {
 	if err := config.InitConfig(); err != nil {
 		panic(err.Error())
 	}
-	defaultPorts := config.Config.RpcPort.OpenImPushPort
-	rpcPort := flag.Int("port", defaultPorts[0], "rpc listening port")
-	prometheusPort := flag.Int("prometheus_port", config.Config.Prometheus.MessageTransferPrometheusPort[0], "PushrometheusPort default listen port")
-	flag.Parse()
-	var wg sync.WaitGroup
-	wg.Add(1)
 	log.NewPrivateLog(constant.LogFileName)
-	fmt.Println("start push rpc server, port: ", *rpcPort, ", OpenIM version: ", constant.CurrentVersion, "\n")
-	pusher := push.Push{}
-	if err := pusher.Init(*rpcPort); err != nil {
+	if err := startrpc.Start(config.Config.RpcPort.OpenImAuthPort[0], config.Config.RpcRegisterName.OpenImAuthName, config.Config.Prometheus.AuthPrometheusPort[0], push.Start); err != nil {
 		panic(err.Error())
 	}
-	pusher.Run(*prometheusPort)
-	wg.Wait()
 }