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

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐