2020-01-22 13:02:06 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
2020-11-27 13:00:45 +00:00
|
|
|
"errors"
|
2020-01-22 13:02:06 +00:00
|
|
|
"fmt"
|
2020-11-27 13:00:45 +00:00
|
|
|
"io"
|
|
|
|
"time"
|
2020-01-22 13:02:06 +00:00
|
|
|
|
2020-11-08 18:28:33 +00:00
|
|
|
"github.com/sirupsen/logrus"
|
|
|
|
|
2021-07-17 20:40:06 +00:00
|
|
|
"go.linka.cloud/grpc/client"
|
|
|
|
"go.linka.cloud/grpc/registry/mdns"
|
|
|
|
"go.linka.cloud/grpc/service"
|
2020-01-22 13:02:06 +00:00
|
|
|
)
|
|
|
|
|
2020-11-27 13:00:45 +00:00
|
|
|
type GreeterHandler struct{
|
|
|
|
UnimplementedGreeterServer
|
|
|
|
}
|
2020-01-22 13:02:06 +00:00
|
|
|
|
|
|
|
func hello(name string) string {
|
|
|
|
return fmt.Sprintf("Hello %s !", name)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (g *GreeterHandler) SayHello(ctx context.Context, req *HelloRequest) (*HelloReply, error) {
|
|
|
|
return &HelloReply{Message: hello(req.Name)}, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (g *GreeterHandler) SayHelloStream(req *HelloStreamRequest, s Greeter_SayHelloStreamServer) error {
|
|
|
|
for i := int64(0); i < req.Count; i++ {
|
2020-11-27 13:00:45 +00:00
|
|
|
if err := s.Send(&HelloReply{Message: fmt.Sprintf("Hello %s (%d)!", req.Name, i+1)}); err != nil {
|
2020-01-22 13:02:06 +00:00
|
|
|
return err
|
|
|
|
}
|
2020-11-27 13:00:45 +00:00
|
|
|
time.Sleep(time.Second)
|
2020-01-22 13:02:06 +00:00
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func main() {
|
2020-11-08 18:28:33 +00:00
|
|
|
name := "greeter"
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
|
|
done := make(chan struct{})
|
|
|
|
ready := make(chan struct{})
|
2020-01-22 13:02:06 +00:00
|
|
|
defer cancel()
|
2020-01-22 20:20:18 +00:00
|
|
|
var svc service.Service
|
2020-01-22 13:02:06 +00:00
|
|
|
var err error
|
2020-01-22 20:20:18 +00:00
|
|
|
svc, err = service.New(
|
|
|
|
service.WithContext(ctx),
|
2020-11-08 18:28:33 +00:00
|
|
|
service.WithName(name),
|
|
|
|
service.WithVersion("v0.0.1"),
|
|
|
|
service.WithAddress("0.0.0.0:9991"),
|
|
|
|
service.WithRegistry(mdns.NewRegistry()),
|
2020-08-07 08:59:27 +00:00
|
|
|
service.WithReflection(true),
|
|
|
|
service.WithSecure(true),
|
2020-01-22 20:20:18 +00:00
|
|
|
service.WithAfterStart(func() error {
|
2020-01-22 13:02:06 +00:00
|
|
|
fmt.Println("Server listening on", svc.Options().Address())
|
2020-11-08 18:28:33 +00:00
|
|
|
close(ready)
|
2020-01-22 13:02:06 +00:00
|
|
|
return nil
|
|
|
|
}),
|
2020-01-22 20:20:18 +00:00
|
|
|
service.WithAfterStop(func() error {
|
2020-01-22 13:02:06 +00:00
|
|
|
fmt.Println("Stopping server")
|
2020-11-08 18:28:33 +00:00
|
|
|
close(done)
|
2020-01-22 13:02:06 +00:00
|
|
|
return nil
|
|
|
|
}),
|
|
|
|
)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
2020-11-27 13:00:45 +00:00
|
|
|
RegisterGreeterServer(svc, &GreeterHandler{})
|
2020-11-08 18:28:33 +00:00
|
|
|
go func() {
|
|
|
|
if err := svc.Start(); err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
<-ready
|
|
|
|
s, err := client.New(
|
2020-11-27 13:00:45 +00:00
|
|
|
client.WithName("greeter"),
|
|
|
|
client.WithVersion("v0.0.1"),
|
2020-11-08 18:28:33 +00:00
|
|
|
client.WithRegistry(mdns.NewRegistry()),
|
|
|
|
client.WithSecure(true),
|
|
|
|
)
|
|
|
|
if err != nil {
|
|
|
|
logrus.Fatal(err)
|
|
|
|
}
|
2020-11-27 13:00:45 +00:00
|
|
|
g := NewGreeterClient(s)
|
2020-11-08 18:28:33 +00:00
|
|
|
defer cancel()
|
|
|
|
res, err := g.SayHello(context.Background(), &HelloRequest{Name: "test"})
|
|
|
|
if err != nil {
|
|
|
|
logrus.Fatal(err)
|
2020-01-22 13:02:06 +00:00
|
|
|
}
|
2020-11-08 18:28:33 +00:00
|
|
|
logrus.Infof("received message: %s", res.Message)
|
2020-11-27 13:00:45 +00:00
|
|
|
stream, err := g.SayHelloStream(context.Background(), &HelloStreamRequest{Name: "test", Count: 10})
|
|
|
|
if err != nil {
|
|
|
|
logrus.Fatal(err)
|
|
|
|
}
|
|
|
|
for {
|
|
|
|
m, err := stream.Recv()
|
|
|
|
if errors.Is(err, io.EOF) {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
if err != nil {
|
|
|
|
logrus.Fatal(err)
|
|
|
|
}
|
|
|
|
logrus.Infof("received stream message: %s", m.Message)
|
|
|
|
}
|
2020-11-08 18:28:33 +00:00
|
|
|
cancel()
|
|
|
|
<-done
|
2020-01-22 13:02:06 +00:00
|
|
|
}
|