Add grpc check example. (#306)

* Add grpc check example.

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>

* add missing package.

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
This commit is contained in:
Bo-Yi Wu
2017-11-14 00:18:48 -06:00
committed by GitHub
parent 25bfe420b0
commit c6e3b7417e
6 changed files with 80 additions and 34 deletions

View File

@@ -1,11 +1,12 @@
package main
import (
"context"
"log"
"time"
"github.com/appleboy/gorush/rpc/proto"
"github.com/appleboy/gorush/rpc"
"golang.org/x/net/context"
"google.golang.org/grpc"
)
@@ -21,10 +22,16 @@ func main() {
}
defer conn.Close()
c := proto.NewHealthClient(conn)
r, err := c.Check(context.Background(), &proto.HealthCheckRequest{})
if err != nil {
log.Fatalf("could not greet: %v", err)
client := rpc.NewGrpcHealthClient(conn)
for {
ok, err := client.Check(context.Background())
if !ok || err != nil {
log.Printf("can't connect grpc server: %v, code: %v\n", err, grpc.Code(err))
} else {
log.Println("connect the grpc server successfully")
}
<-time.After(time.Second)
}
log.Printf("Health: %d\n", r.GetStatus())
}