Доступ к нескольким службам gRPC по одному соединению - PullRequest
Новая
       34

Доступ к нескольким службам gRPC по одному соединению

0 голосов
/ 03 октября 2018

Я использую сервер grpc, прослушивающий localhost:6000, предоставляющий 2 grpc службы: RegisterSubscriberServiceServer и RegisterDropperServiceServer.Поскольку обе эти службы доступны с номера localhost:6000, я хотел бы набрать этот адрес только из заглушки.

Сервер выглядит так: 

func main() {
    grpcServer := grpc.NewServer()
    pb.RegisterSubscriberServiceServer(grpcServer, &subscriberServer{})
    pb.RegisterDropperServiceServer(grpcServer, &dropperServer{})

    l, err := net.Listen("tcp", ":6000")
    if err != nil {
        log.Fatalf("failed to listen: %v", err)
    }

    log.Println("Listening on tcp://localhost:6000")
    grpcServer.Serve(l)
}

Зачем мне нужнонабирать разные сокеты для каждой grpc службы? 

type DropperRPC struct {
    conn      *grpc.ClientConn
    client pb.DropperServiceClient
    chunkSize int
}

type SubscriberRPC struct {
    conn      *grpc.ClientConn
    client pb.SubscriberServiceClient
    chunkSize int
}

func NewSubscriber() (c SubscriberRPC, err error) {
    c.conn, err = grpc.Dial("localhost:6000", grpc.WithInsecure())
    if err != nil {
        log.Fatal(err)
    }
    c.client = pb.NewSubscriberServiceClient(c.conn)
    return
}

func NewDropper() (c DropperRPC, err error) {
    c.conn, err = grpc.Dial("localhost:6000", grpc.WithInsecure())
    if err != nil {
        log.Fatal(err)
    }
    c.client = pb.NewDropperServiceClient(c.conn)
    return
}

И поскольку код в основном дублируется для размещения каждой службы, я не могу просто использовать Interface для уменьшения кода? 

func main() {
    c1, err := NewSubscriber()
    if err != nil {
        log.Fatal(err)
    }

    c2, err := NewDropper()
    if err != nil {
        log.Fatal(err)
    }

    cc1 := &c1
    cc2 := &c2
}

А затем используйте этот Interface для реализации клиентских функций grpc для каждой службы вместо создания struct для каждой службы.Я нашел cmux , но должен быть способ сделать это без использования внешних библиотек.

1 Ответ

0 голосов
/ 04 октября 2018

Почему мне нужно набирать разные сокеты для каждой службы grpc?

Вы не делаете.Вы можете создать одну grpc.ClientConn и передать ее нескольким pb.New*Client() функциям, и они будут использовать одни и те же соединения. 

func main() {
    cc, err := grpc.Dial("localhost:6000", grpc.WithInsecure())
    if err != nil {
        log.Fatal(err)
    }
    c1 := pb.NewSubscriberServiceClient(cc)
    c2 := pb.NewDropperServiceClient(cc)
}

И затем использовать этот интерфейс для реализации на стороне клиента.Функции grpc для каждой службы вместо создания новой структуры для каждой службы

Сгенерированный код в файле pb.go делает все необходимое для выполнения RPC.Вам не нужно реализовывать что-либо на стороне клиента, если у вас нет особой логики, которую вы хотите, чтобы происходило автоматически при каждом вызове.

Если у двух сервисов есть уникальные имена методов, вы можете поместить их в один и тот жеструктура, так что вам не нужно использовать их отдельно, для незначительного удобства: 

type SubscriberDropper struct {
    pb.SubscriberServiceClient
    pb.DropperServiceClient
}

func main() {
    // ... as above ...
    sd := &SubscriberDropper{c1, c2}
}
...