go_dreamfactory/stress/tests/grpc/main.go
2022-11-24 21:30:50 +08:00

48 lines
878 B
Go

// Package main grpc server
package main
import (
"context"
"fmt"
"log"
"net"
"google.golang.org/grpc"
pb "go_dreamfactory/stress/proto"
)
const (
// port 监听端口
port = ":8099"
)
// server is used to implement helloWorld.GreeterServer.
type server struct {
pb.UnimplementedApiServerServer
}
// HelloWorld hello world 接口
func (s *server) HelloWorld(_ context.Context, req *pb.Request) (rsp *pb.Response, err error) {
rsp = &pb.Response{
Code: 200,
Msg: "success",
Data: fmt.Sprintf("hello %s !", req.UserName),
}
return
}
// main 主函数
func main() {
fmt.Println("trpc server 启动中...")
lis, err := net.Listen("tcp", port)
if err != nil {
log.Fatalf("failed to listen: %v", err)
}
s := grpc.NewServer()
pb.RegisterApiServerServer(s, &server{})
if err := s.Serve(lis); err != nil {
log.Fatalf("failed to serve: %v", err)
}
}