Go-Micro rpc调用注册的服务

xiaohai 2021-05-09 22:31:20 2449人围观 标签: Go 
简介Go-Micro rpc调用注册的服务

1、目录结构

micro-home:项目目录
    service:服务目录
        model:proto生成的目录
        proto:proto定义目录
    main.go:入口
    make_proto.bat:生成proto

2、要实现调用rpc服务,就直接把服务端的proto文件拷贝的客户端

3、拷贝make_proto.bat到客户端

4、main.go内容

package main

import (
    "github.com/gin-gonic/gin"
    "github.com/micro/go-micro"
    "github.com/micro/go-micro/registry"
    "github.com/micro/go-micro/web"
    "github.com/micro/go-plugins/registry/consul"
    services "micro-home/service/model"
    "strconv"
)

func InitMiddleware(service services.StudentListService) gin.HandlerFunc {
    return func(context *gin.Context) {
        context.Keys = make(map[string]interface{})
        context.Keys["student_service"] = service
        context.Next()
    }
}

func main() {
    register := consul.NewRegistry(func(options *registry.Options) {
        options.Addrs = []string{
            "192.168.1.171:8500",
            "192.168.1.177:8500",
            "192.168.1.178:8500",
        }
    })
    //微服务需要注册
    homeService := micro.NewService(micro.Name("micro-home-client"), micro.Registry(register))
    //实例化服务
    studentListService := services.NewStudentListService("student-service", homeService.Client())

    routers := gin.Default()

    routers.Use(InitMiddleware(studentListService)) //使用中间件,将服务传递到context中

    routers.GET("/", func(context *gin.Context) {
        service := context.Keys["student_service"].(services.StudentListService)
        number := context.DefaultQuery("number", "0")
        num, err := strconv.Atoi(number)
        if err != nil {
            num = 1
        }
        if num <= 0 {
            num = 1
        }
        studentListResponse, err := service.GetStudentListService(context, &services.StudentListRequest{
            Num: int32(num),
        })
        if err != nil {
            context.JSON(200, gin.H{
                "status": "",
                "data":   err.Error(),
            })
        } else {
            context.JSON(200, gin.H{
                "status": studentListResponse.Status,
                "data":   studentListResponse.Data,
            })
        }
    })

    service := web.NewService(
        web.Name("home-service-api"),
        web.Address(":8081"),
        web.Handler(routers),
        web.Registry(register),
    )

    service.Run()
}