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