| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package common |
| |
| import ( |
| "context" |
| ) |
| |
| import ( |
| "github.com/dubbogo/gost/log/logger" |
| ) |
| |
| import ( |
| greet "dubbo.apache.org/dubbo-go/v3/protocol/triple/internal/proto" |
| "dubbo.apache.org/dubbo-go/v3/protocol/triple/internal/proto/triple_gen/greettriple" |
| ) |
| |
| func TestClient(cli greettriple.GreetService) { |
| if err := testUnary(cli); err != nil { |
| logger.Error(err) |
| } |
| |
| if err := testBidiStream(cli); err != nil { |
| logger.Error(err) |
| } |
| |
| if err := testClientStream(cli); err != nil { |
| logger.Error(err) |
| } |
| |
| if err := testServerStream(cli); err != nil { |
| logger.Error(err) |
| } |
| } |
| |
| func testUnary(cli greettriple.GreetService) error { |
| logger.Info("start to test TRIPLE unary call") |
| resp, err := cli.Greet(context.Background(), &greet.GreetRequest{Name: "triple"}) |
| if err != nil { |
| return err |
| } |
| logger.Infof("TRIPLE unary call resp: %s", resp.Greeting) |
| return nil |
| } |
| |
| func testBidiStream(cli greettriple.GreetService) error { |
| logger.Info("start to test TRIPLE bidi stream") |
| stream, err := cli.GreetStream(context.Background()) |
| if err != nil { |
| return err |
| } |
| if err := stream.Send(&greet.GreetStreamRequest{Name: "triple"}); err != nil { |
| return err |
| } |
| resp, err := stream.Recv() |
| if err != nil { |
| return err |
| } |
| logger.Infof("TRIPLE bidi stream resp: %s", resp.Greeting) |
| if err := stream.CloseRequest(); err != nil { |
| return err |
| } |
| if err := stream.CloseResponse(); err != nil { |
| return err |
| } |
| return nil |
| } |
| |
| func testClientStream(cli greettriple.GreetService) error { |
| logger.Info("start to test TRIPLE client stream") |
| stream, err := cli.GreetClientStream(context.Background()) |
| if err != nil { |
| return err |
| } |
| for i := 0; i < 5; i++ { |
| if err := stream.Send(&greet.GreetClientStreamRequest{Name: "triple"}); err != nil { |
| return err |
| } |
| } |
| resp, err := stream.CloseAndRecv() |
| if err != nil { |
| return err |
| } |
| logger.Infof("TRIPLE client stream resp: %s", resp.Greeting) |
| return nil |
| } |
| |
| func testServerStream(cli greettriple.GreetService) error { |
| logger.Info("start to test TRIPLE server stream") |
| stream, err := cli.GreetServerStream(context.Background(), &greet.GreetServerStreamRequest{Name: "triple"}) |
| if err != nil { |
| return err |
| } |
| for stream.Recv() { |
| logger.Infof("TRIPLE server stream resp: %s", stream.Msg().Greeting) |
| } |
| if stream.Err() != nil { |
| return err |
| } |
| if err := stream.Close(); err != nil { |
| return err |
| } |
| return nil |
| } |