meta_client/client/
util.rs

1// Copyright 2023 Greptime Team
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use api::v1::meta::{ErrorCode, ResponseHeader};
16use tonic::{Code, Status};
17
18pub(crate) fn is_unreachable(status: &Status) -> bool {
19    status.code() == Code::Unavailable || status.code() == Code::DeadlineExceeded
20}
21
22pub(crate) fn is_not_leader(header: &Option<ResponseHeader>) -> bool {
23    let Some(header) = header else {
24        return false;
25    };
26
27    let Some(err) = header.error.as_ref() else {
28        return false;
29    };
30
31    err.code == ErrorCode::NotLeader as i32
32}