initial import

example/graceful/dns/rpc/gracefulrpc.go

@@ -0,0 +1,51 @@
+package main
+
+import (
+	"context"
+	"flag"
+	"fmt"
+	"os"
+	"time"
+
+	"zero/core/conf"
+	"zero/example/graceful/dns/rpc/graceful"
+	"zero/rpcx"
+
+	"google.golang.org/grpc"
+)
+
+var configFile = flag.String("f", "etc/config.json", "the config file")
+
+type GracefulServer struct{}
+
+func NewGracefulServer() *GracefulServer {
+	return &GracefulServer{}
+}
+
+func (gs *GracefulServer) Grace(ctx context.Context, req *graceful.Request) (*graceful.Response, error) {
+	fmt.Println("=>", req)
+
+	time.Sleep(time.Millisecond * 10)
+
+	hostname, err := os.Hostname()
+	if err != nil {
+		return nil, err
+	}
+
+	return &graceful.Response{
+		Host: hostname,
+	}, nil
+}
+
+func main() {
+	flag.Parse()
+
+	var c rpcx.RpcServerConf
+	conf.MustLoad(*configFile, &c)
+
+	server := rpcx.MustNewServer(c, func(grpcServer *grpc.Server) {
+		graceful.RegisterGraceServiceServer(grpcServer, NewGracefulServer())
+	})
+	defer server.Stop()
+	server.Start()
+}