rename rpcx to zrpc

zrpc/internal/rpcpubserver.go

@@ -0,0 +1,29 @@
+package internal
+
+import "github.com/tal-tech/go-zero/core/discov"
+
+func NewRpcPubServer(etcdEndpoints []string, etcdKey, listenOn string, opts ...ServerOption) (Server, error) {
+	registerEtcd := func() error {
+		pubClient := discov.NewPublisher(etcdEndpoints, etcdKey, listenOn)
+		return pubClient.KeepAlive()
+	}
+	server := keepAliveServer{
+		registerEtcd: registerEtcd,
+		Server:       NewRpcServer(listenOn, opts...),
+	}
+
+	return server, nil
+}
+
+type keepAliveServer struct {
+	registerEtcd func() error
+	Server
+}
+
+func (ags keepAliveServer) Start(fn RegisterFn) error {
+	if err := ags.registerEtcd(); err != nil {
+		return err
+	}
+
+	return ags.Server.Start(fn)
+}