60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
package rpc
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/appleboy/gorush/core"
|
|
"github.com/appleboy/gorush/rpc/proto"
|
|
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
// generate protobuffs
|
|
// protoc --go_out=plugins=grpc,import_path=proto:. *.proto
|
|
|
|
type healthClient struct {
|
|
client proto.HealthClient
|
|
conn *grpc.ClientConn
|
|
}
|
|
|
|
// NewGrpcHealthClient returns a new grpc Client.
|
|
func NewGrpcHealthClient(conn *grpc.ClientConn) core.Health {
|
|
client := new(healthClient)
|
|
client.client = proto.NewHealthClient(conn)
|
|
client.conn = conn
|
|
return client
|
|
}
|
|
|
|
func (c *healthClient) Close() error {
|
|
return c.conn.Close()
|
|
}
|
|
|
|
func (c *healthClient) Check(ctx context.Context) (bool, error) {
|
|
var res *proto.HealthCheckResponse
|
|
var err error
|
|
req := new(proto.HealthCheckRequest)
|
|
|
|
res, err = c.client.Check(ctx, req)
|
|
if err == nil {
|
|
if res.GetStatus() == proto.HealthCheckResponse_SERVING {
|
|
return true, nil
|
|
}
|
|
return false, nil
|
|
}
|
|
switch status.Code(err) {
|
|
case
|
|
codes.Aborted,
|
|
codes.DataLoss,
|
|
codes.DeadlineExceeded,
|
|
codes.Internal,
|
|
codes.Unavailable:
|
|
// non-fatal errors
|
|
default:
|
|
return false, err
|
|
}
|
|
|
|
return false, err
|
|
}
|