Returns the client specified by the protocol in the given vector.
(self, vector, fetch_size=1024)
| 91 | assert num_batches_count == num_batches |
| 92 | |
| 93 | def __get_shell_client(self, vector, fetch_size=1024): |
| 94 | """Returns the client specified by the protocol in the given vector.""" |
| 95 | impalad = get_impalad_host_port(vector).split(":") |
| 96 | protocol = vector.get_value("protocol") |
| 97 | if protocol == 'hs2': |
| 98 | return ImpalaHS2Client(impalad, fetch_size, None) |
| 99 | elif protocol == 'hs2-http': |
| 100 | return ImpalaHS2Client(impalad, fetch_size, None, |
| 101 | use_http_base_transport=True, http_path='cliservice') |
| 102 | elif protocol == 'beeswax': |
| 103 | return ImpalaBeeswaxClient(impalad, fetch_size, None) |
no test coverage detected