66 lines
1.5 KiB
Go
66 lines
1.5 KiB
Go
package main
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"git.gqnotes.com/guoqiang/grpcexercises/consuldemo/pb"
|
||
"git.gqnotes.com/guoqiang/grpcexercises/consuldemo/pkg/consul"
|
||
"google.golang.org/grpc"
|
||
"google.golang.org/grpc/credentials/insecure"
|
||
"log"
|
||
"math/rand"
|
||
)
|
||
|
||
func main() {
|
||
// 通过consul服务发现，调用grpc接口
|
||
consulAddress := "localhost:8500"
|
||
serviceName := "service-grpcdemo"
|
||
|
||
// 初始化consul客户端
|
||
if err := consul.InitConsulClient(consulAddress); err != nil {
|
||
log.Fatalf("init consul client failed:%+v", err)
|
||
}
|
||
|
||
// 获取consul客户端
|
||
client := consul.GetConsulClient()
|
||
|
||
// 获取服务
|
||
services, _, err := client.Health().Service(serviceName, "", true, nil)
|
||
|
||
if err != nil {
|
||
log.Fatalf("get service failed:%+v", err)
|
||
}
|
||
|
||
// 随机获取一个服务
|
||
service := services[rand.Intn(len(services))]
|
||
|
||
// 拼接服务地址
|
||
address := fmt.Sprintf("%s:%d", service.Service.Address, service.Service.Port)
|
||
|
||
log.Printf("grpc client call address:%s", address)
|
||
|
||
// 连接grpc服务
|
||
conn, err := grpc.Dial(address, grpc.WithTransportCredentials(insecure.NewCredentials()))
|
||
|
||
if err != nil {
|
||
log.Fatalf("grpc client dial failed:%+v", err)
|
||
}
|
||
|
||
defer conn.Close()
|
||
|
||
// 创建grpc客户端
|
||
grpcClient := pb.NewGreetServiceClient(conn)
|
||
|
||
// 调用grpc接口
|
||
resp, err := grpcClient.Greet(context.Background(), &pb.GreetRequest{
|
||
Greeting: "zhangsan",
|
||
})
|
||
|
||
if err != nil {
|
||
log.Fatalf("grpc client call greet failed:%+v", err)
|
||
return
|
||
}
|
||
|
||
log.Printf("grpc client call greet success:%+v", resp)
|
||
}
|