parent
5b6bf88a27
commit
156c97181e
@ -1,44 +0,0 @@ |
||||
package main |
||||
|
||||
import ( |
||||
"context" |
||||
"fmt" |
||||
"io" |
||||
"log" |
||||
|
||||
"github.com/go-kratos/kratos/v2/middleware/recovery" |
||||
transgrpc "github.com/go-kratos/kratos/v2/transport/grpc" |
||||
pb "google.golang.org/grpc/health/grpc_health_v1" |
||||
) |
||||
|
||||
func main() { |
||||
conn, err := transgrpc.DialInsecure( |
||||
context.Background(), |
||||
transgrpc.WithEndpoint("127.0.0.1:9010"), |
||||
transgrpc.WithMiddleware( |
||||
recovery.Recovery(), |
||||
), |
||||
) |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
defer conn.Close() |
||||
|
||||
client := pb.NewHealthClient(conn) |
||||
stream, err := client.Watch(context.Background(), &pb.HealthCheckRequest{Service: "health"}) |
||||
if err != nil { |
||||
log.Fatal(err) |
||||
} |
||||
for { |
||||
res, err := stream.Recv() |
||||
if err == io.EOF { |
||||
fmt.Println("EOF") |
||||
break |
||||
} |
||||
if err != nil { |
||||
log.Fatalf("ListStr get stream err: %v", err) |
||||
} |
||||
// 打印返回值
|
||||
log.Println(res.Status) |
||||
} |
||||
} |
@ -1,59 +0,0 @@ |
||||
package main |
||||
|
||||
import ( |
||||
"context" |
||||
"fmt" |
||||
"log" |
||||
|
||||
"github.com/go-kratos/kratos/examples/helloworld/helloworld" |
||||
"github.com/go-kratos/kratos/v2" |
||||
"github.com/go-kratos/kratos/v2/errors" |
||||
"github.com/go-kratos/kratos/v2/middleware/recovery" |
||||
"github.com/go-kratos/kratos/v2/transport/grpc" |
||||
) |
||||
|
||||
// go build -ldflags "-X main.Version=x.y.z"
|
||||
var ( |
||||
// Name is the name of the compiled software.
|
||||
Name = "health" |
||||
// Version is the version of the compiled software.
|
||||
// Version = "v1.0.0"
|
||||
) |
||||
|
||||
// server is used to implement helloworld.GreeterServer.
|
||||
type server struct { |
||||
helloworld.UnimplementedGreeterServer |
||||
} |
||||
|
||||
// SayHello implements helloworld.GreeterServer
|
||||
func (s *server) SayHello(ctx context.Context, in *helloworld.HelloRequest) (*helloworld.HelloReply, error) { |
||||
if in.Name == "error" { |
||||
return nil, errors.BadRequest("custom_error", fmt.Sprintf("invalid argument %s", in.Name)) |
||||
} |
||||
if in.Name == "panic" { |
||||
panic("server panic") |
||||
} |
||||
return &helloworld.HelloReply{Message: fmt.Sprintf("Hello %+v", in.Name)}, nil |
||||
} |
||||
|
||||
func main() { |
||||
s := &server{} |
||||
grpcSrv := grpc.NewServer( |
||||
grpc.Address(":9010"), |
||||
grpc.Middleware( |
||||
recovery.Recovery(), |
||||
), |
||||
) |
||||
helloworld.RegisterGreeterServer(grpcSrv, s) |
||||
|
||||
app := kratos.New( |
||||
kratos.Name(Name), |
||||
kratos.Server( |
||||
grpcSrv, |
||||
), |
||||
) |
||||
|
||||
if err := app.Run(); err != nil { |
||||
log.Fatal(err) |
||||
} |
||||
} |
Loading…
Reference in new issue