/
grpc.go
68 lines (62 loc) · 1.75 KB
/
grpc.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
package client
import (
"context"
"github.com/tuihub/librarian/internal/conf"
"github.com/tuihub/librarian/internal/lib/libapp"
mapper "github.com/tuihub/protos/pkg/librarian/mapper/v1"
miner "github.com/tuihub/protos/pkg/librarian/miner/v1"
searcher "github.com/tuihub/protos/pkg/librarian/searcher/v1"
"github.com/go-kratos/kratos/v2/middleware/recovery"
"github.com/go-kratos/kratos/v2/transport/grpc"
)
func NewMapperClient(c *conf.Consul) (mapper.LibrarianMapperServiceClient, error) {
r, err := libapp.NewDiscovery(c)
if err != nil {
return nil, err
}
conn, err := grpc.DialInsecure(
context.Background(),
grpc.WithEndpoint("discovery:///mapper"),
grpc.WithDiscovery(r),
grpc.WithNodeFilter(libapp.NewNodeFilter()),
grpc.WithMiddleware(
recovery.Recovery(),
),
)
cli := mapper.NewLibrarianMapperServiceClient(conn)
return cli, err
}
func NewSearcherClient(c *conf.Consul) (searcher.LibrarianSearcherServiceClient, error) {
r, err := libapp.NewDiscovery(c)
if err != nil {
return nil, err
}
conn, err := grpc.DialInsecure(
context.Background(),
grpc.WithEndpoint("discovery:///searcher"),
grpc.WithDiscovery(r),
grpc.WithNodeFilter(libapp.NewNodeFilter()),
grpc.WithMiddleware(
recovery.Recovery(),
),
)
cli := searcher.NewLibrarianSearcherServiceClient(conn)
return cli, err
}
func NewMinerClient(c *conf.Consul) (miner.LibrarianMinerServiceClient, error) {
r, err := libapp.NewDiscovery(c)
if err != nil {
return nil, err
}
conn, err := grpc.DialInsecure(
context.Background(),
grpc.WithEndpoint("discovery:///porter"),
grpc.WithDiscovery(r),
grpc.WithNodeFilter(libapp.NewNodeFilter()),
grpc.WithMiddleware(
recovery.Recovery(),
),
)
cli := miner.NewLibrarianMinerServiceClient(conn)
return cli, err
}