Go Kratos V2 gRpc + ETCD 适配 V1 - 适合Kratos V1+ETCD 升级到 V2 的中间过程

Kratos V1 升级到 V2 过程中，如果服务数量较多，不适合同时升级到V2，那么可以考虑先将部分服务升级到V2同时也支持V1的调用，逐步实现所有服务升级到V2；下面是具体实现方式；

服务注册 go 1.17

​
require (
	github.com/go-kratos/kratos v1.0.1
	github.com/go-kratos/kratos/v2 v2.2.1
	google.golang.org/grpc v1.44.0
    go.etcd.io/etcd/client/v3 v3.5.0
)

​

1. provider/internal/server 添加 registrar.go 文件

package server

import (
	pb "provider/api/provider"

	"github.com/go-kratos/kratos/v2/log"
	clientv3 "go.etcd.io/etcd/client/v3"

	"github.com/go-kratos/kratos/pkg/naming/etcd"
	"github.com/go-kratos/kratos/pkg/conf/env"
	"github.com/go-kratos/kratos/pkg/naming"

	"google.golang.org/grpc"
	"provider/internal/conf"
	"time"
	"context"
)


type V1Registrar struct {}

func RegistrarServer(c *conf.Server, logger log.Logger) (*V1Registrar, func(),error) {
	// ETCD源地址
	endpoints := []string{"127.0.0.1:2379"}

	// ETCD配置信息
	etcdCfg := clientv3.Config{
		Endpoints: endpoints,
		DialTimeout: time.Second,
		DialOptions: []grpc.DialOption{grpc.WithBlock()},
	}

	builder, err := etcd.New(&etcdCfg)
	if err != nil {
		panic(err)
	}

	ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
	defer cancel()

	clean,err := builder.Register(ctx, &naming.Instance{
		Region:   env.Region,
		Zone:     env.Zone,
		Env:      env.DeployEnv,
		AppID:    pb.AppName,
		Hostname: env.Hostname,
		Addrs:    []string{"grpc://" + c.Grpc.Addr, "http://" + c.Http.Addr},
		Version:  "V1.0.0",
	})

	return new(V1Registrar),clean,err
}

2.provider/internal/server/server.go 文件修改

package server
 
import (
	"github.com/google/wire"
)
 
// ProviderSet is server providers.
var ProviderSet = wire.NewSet(NewHTTPServer, NewGRPCServer, RegistrarServer)

3.provider/cmd/provider/main.go 文件 newApp 方法修改

func newApp(logger log.Logger, hs *http.Server, gs *grpc.Server, registrar *server.V1Registrar) *kratos.App {
	return kratos.New(
		kratos.ID(id),
		kratos.Name(Name),
		kratos.Version(Version),
		kratos.Metadata(map[string]string{}),
		kratos.Logger(logger),
		kratos.Server(
			hs,
			gs,
		),

		// with registrar
		//kratos.Registrar(registrar),
	)
}

4.provider/cmd/provider 中执行 wire 命令，服务注册完成；

服务发现 go 1.17

require (
	github.com/go-kratos/kratos v1.0.1
	google.golang.org/grpc v1.44.0
    go.etcd.io/etcd/client/v3 v3.5.0
)

1.provider/api/provider 文件下添加 client.go 文件

package provider

import (
	"context"
	"fmt"
	"github.com/go-kratos/kratos/pkg/naming/etcd"
	"github.com/go-kratos/kratos/pkg/net/rpc/warden"
	"github.com/go-kratos/kratos/pkg/net/rpc/warden/resolver"
	xtime "github.com/go-kratos/kratos/pkg/time"
	clientv3 "go.etcd.io/etcd/client/v3"
	"os"
	"strings"
	"time"
)

// AppName
const AppName = "provider"


func Client() (ProviderServiceClient,error) {
	endpoints := []string{"127.0.0.1:2379"}
	resolver.Register(etcd.Builder(&clientv3.Config {
		Endpoints: endpoints,
	}))

	client := warden.NewClient(&warden.ClientConfig {
		Dial:    xtime.Duration(time.Second * 2),
		Timeout: xtime.Duration(time.Second * 30),
	})

	conn, err := client.Dial(context.Background(), fmt.Sprintf("etcd://default/%s", AppName))
	if err != nil {
		return nil, err
	}

	return NewProviderServiceClient(conn),err
}

2. 旧版本服务 import 此 api 包即可使用etcd完成服务发现；

    备注：旧版本服务需去掉 go.etcd.io/etcd v0.0.0-xxxxxx  引用并清理 go.sum 中 go.etcd.io/etcd 记录；替换 go.etcd.io/etcd/clientv3 为 clientv3 "go.etcd.io/etcd/client/v3"