Method TestBackupRequestSelective

test/brpc_channel_unittest.cpp:1442–1489

    }

    void TestBackupRequestSelective(
        bool single_server, bool async, bool short_connection) {
        std::cout << " *** single=" << single_server
                  << " async=" << async
                  << " short=" << short_connection << std::endl;
        ASSERT_EQ(0, StartAccept(_ep));

        const size_t NCHANS = 8;
        brpc::SelectiveChannel channel;
        ASSERT_EQ(0, channel.Init("rr", NULL));
        for (size_t i = 0; i < NCHANS; ++i) {
            brpc::Channel* subchan = new brpc::Channel;
            SetUpChannel(subchan, single_server, short_connection);
            ASSERT_EQ(0, channel.AddChannel(subchan, NULL)) << "i=" << i;
        }

        brpc::Controller cntl;
        test::EchoRequest req;
        test::EchoResponse res;
        req.set_message(__FUNCTION__);
        cntl.set_backup_request_ms(20);
        cntl.set_timeout_ms(100);
        std::atomic<int> call_cnt(0);
        _svc.SetMockFunc([&call_cnt](google::protobuf::RpcController* cntl_base,
                                     const ::test::EchoRequest*,
                                     ::test::EchoResponse*,
                                     google::protobuf::Closure*) {
            brpc::Controller* cntl = static_cast<brpc::Controller*>(cntl_base);
            int see_cnt = call_cnt.fetch_add(1, std::memory_order_relaxed);
            if (see_cnt == 0) {
                LOG(INFO) << "slow node";
                bthread_usleep(30 * 1000);
            } else {
                LOG(INFO) << "normal node ";
                butil::IOBuf iobuf;
                iobuf.append("123");
                cntl->response_attachment().swap(iobuf);
            }
        });
        butil::Timer tm;
        tm.start();
        CallMethod(&channel, &cntl, &req, &res, async);
        tm.stop();
        EXPECT_FALSE(cntl.Failed());
        EXPECT_EQ(call_cnt.load(std::memory_order_relaxed), 2);
        EXPECT_EQ(cntl.response_attachment().to_string(), "123");
        StopAndJoin();
    }

    void TestBackupRequestSelectiveResponseRace() {
        ASSERT_EQ(0, StartAccept(_ep));