go+gin+grpc整合demo
pb/hello.proto:
syntax = "proto3";
option go_package = "./;pb";
package pb;
service Greeter{
rpc SayHello (HelloRequest) returns (HelloReply) {}
}
// 请求数据格式
message HelloRequest{
string name = 1;
}
// 返回数据格式
message HelloReply{
string name = 1;
}
grpc/main.go:
package main
import (
"context"
"fmt"
pb "gin_gprc/pb"
"log"
"net"
"google.golang.org/grpc"
"google.golang.org/grpc/reflection"
)
const (
port = ":50051"
)
// 服务对象
type server struct {
pb.UnimplementedGreeterServer
}
// sayhello实现服务的接口 在proto中定义的所有服务都是接口
func (s *server) SayHello(ctx context.Context, in *pb.HelloRequest) (*pb.HelloReply, error) {
fmt.Println("请求数据........")
return &pb.HelloReply{Name: "Hello " + in.Name}, nil
}
func main() {
lis, err := net.Listen("tcp", port)
if err != nil {
log.Fatalf("failed to listen: %v", err)
}
// 起一个服务
s := grpc.NewServer()
pb.RegisterGreeterServer(s, &server{})
// 注册反射服务 这个服务是CLI使用的 跟服务本身没有关系
fmt.Printf("\"开始提供服务\": %v\n", "开始提供服务")
reflection.Register(s)
if err := s.Serve(lis); err != nil {
log.Fatalf("failed to serve %v", err)
}
}
gin/main.go:
package main
import (
"fmt"
pb "gin_gprc/pb"
"log"
"net/http"
"github.com/gin-gonic/gin"
"google.golang.org/grpc"
)
func main() {
// 建立grpc连接充当客户端
conn, err := grpc.Dial("localhost:50051", grpc.WithInsecure())
if err != nil {
log.Fatalf("did not connect :%v",err)
}
defer conn.Close()
client := pb.NewGreeterClient(conn)
// http服务端
r := gin.Default()
r.GET("/rest/n/:name", func(c *gin.Context){
name := c.Param("name")
// 接收到参数后向grpc服务端请求数据
req := &pb.HelloRequest{Name: name}
res, err := client.SayHello(c, req)
if err != nil{
c.JSON(http.StatusInternalServerError,gin.H{
"error":err.Error(),
})
return
}
c.JSON(http.StatusOK, gin.H{
"result":fmt.Sprint(res.Name),
})
})
if err := r.Run(":8052"); err != nil {
log.Fatalf("could not run sercer :%v", err)
}
}
然后启动grpc服务端,再启动http服务端,在浏览器请求:http://127.0.0.1:8052/rest/n/xiaobai