14.Go-micro的裝飾器wrapper的初步使用(中間件)

使用go-micro官方提供的wrapper來對路由進行裝飾

package main

import (
    "context"
    "fmt"
    "github.com/micro/go-micro"
    "github.com/micro/go-micro/client"
    "github.com/micro/go-micro/registry"
    "github.com/micro/go-micro/web"
    "github.com/micro/go-plugins/registry/consul"
    "go-micro/Services"
    "go-micro/Weblib"
)

type logWrapper struct { //官方提供的例子，創建自己的struct，嵌套go-micro的client
    client.Client
}

//重寫Call方法
func (this *logWrapper) Call(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error {
    fmt.Println("調用接口") //這樣每一次調用調用接口時都會
    return this.Client.Call(ctx, req, rsp)
}

func NewLogWrapper(c client.Client) client.Client {
    return &logWrapper{c}
}

func main() {
    //下面兩局代碼是注冊rpcserver調用客戶端
    myService := micro.NewService(
        micro.Name("prodservice.client"),
        micro.WrapClient(NewLogWrapper), //在注冊時只需要傳入方法名即可，底層會自動給這個方法傳入client
    )
    prodService := Services.NewProdService("prodservice", myService.Client())//生成的這個客戶端綁定consul中存儲的prodservice服務，只要調用了prodservice接口就會調用我們上面注冊的中間件

    consulReg := consul.NewRegistry( //新建一個consul注冊的地址，也就是我們consul服務啟動的機器ip+端口
        registry.Addrs("localhost:8500"),
    )

    //其實下面這段代碼的作用就是啟動webserver的同事的時候把服務注冊進去
    httpserver := web.NewService( //go-micro很靈性的實現了注冊和反注冊，我們啟動后直接ctrl+c退出這個server，它會自動幫我們實現反注冊
        web.Name("httpprodservice"),                   //注冊進consul服務中的service名字
        web.Address(":8001"),                          //注冊進consul服務中的端口,也是這里我們gin的server地址
        web.Handler(Weblib.NewGinRouter(prodService)), //web.Handler()返回一個Option，我們直接把ginRouter穿進去，就可以和gin完美的結合
        web.Registry(consulReg),                       //注冊到哪個服務器上的consul中
    )
    httpserver.Init() //加了這句就可以使用命令行的形式去設置我們一些啟動的配置
    httpserver.Run()
}

來自為知筆記(Wiz)