mirror of https://github.com/linka-cloud/grpc.git
57 lines
1.2 KiB
Go
57 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"gitlab.bertha.cloud/partitio/lab/grpc/service"
|
|
)
|
|
|
|
type GreeterHandler struct{}
|
|
|
|
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++ {
|
|
if err := s.Send(&HelloReply{Message: hello(req.Name)}); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func main() {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
var svc service.Service
|
|
var err error
|
|
svc, err = service.New(
|
|
service.WithContext(ctx),
|
|
service.WithName("Greeting"),
|
|
service.WithReflection(true),
|
|
service.WithSecure(true),
|
|
service.WithAfterStart(func() error {
|
|
fmt.Println("Server listening on", svc.Options().Address())
|
|
return nil
|
|
}),
|
|
service.WithAfterStop(func() error {
|
|
fmt.Println("Stopping server")
|
|
return nil
|
|
}),
|
|
)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
RegisterGreeterServer(svc.Server(), &GreeterHandler{})
|
|
if err := svc.Start(); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|