// 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.

#pragma once

#include <random>

#include <string>

namespace doris {

#ifndef ARRAY_SIZE

#define ARRAY_SIZE(a) (sizeof(a) / sizeof((a)[0]))

#endif

struct AuthInfo {

    std::string user;

    std::string passwd;

    std::string cluster;

    std::string user_ip;

    // -1 as unset

    int64_t auth_code = -1; // deprecated

    std::string token;

};

template <class T>

void set_request_auth(T* req, const AuthInfo& auth) {

    req->user = auth.user; // always set user, because it may be used by FE

    // auth code is deprecated and should be removed in 3.1

    if (auth.auth_code != -1) {

        // if auth_code is set, no need to set other info

        req->__set_auth_code(auth.auth_code);

        // user name and passwd is unused, but they are required field.

        // so they have to be set.

        req->passwd = "";

    } else if (auth.token != "") {

        req->__isset.token = true;

        req->token = auth.token;

    } else {

        req->passwd = auth.passwd;

        if (!auth.cluster.empty()) {

            req->__set_cluster(auth.cluster);

        }

        req->__set_user_ip(auth.user_ip);

    }

}

Unexecuted instantiation: _ZN5doris16set_request_authINS_20TLoadTxnBeginRequestEEEvPT_RKNS_8AuthInfoE

Unexecuted instantiation: _ZN5doris16set_request_authINS_18TLoadTxn2PCRequestEEEvPT_RKNS_8AuthInfoE

_ZN5doris16set_request_authINS_21TLoadTxnCommitRequestEEEvPT_RKNS_8AuthInfoE

Line

Count

Source

40

1

void set_request_auth(T* req, const AuthInfo& auth) {

41

1

    req->user = auth.user; // always set user, because it may be used by FE

42

    // auth code is deprecated and should be removed in 3.1

43

1

    if (auth.auth_code != -1) {

44

        // if auth_code is set, no need to set other info

45

1

        req->__set_auth_code(auth.auth_code);

46

        // user name and passwd is unused, but they are required field.

47

        // so they have to be set.

48

1

        req->passwd = "";

49

1

    } else if (auth.token != "") {

50

0

        req->__isset.token = true;

51

0

        req->token = auth.token;

52

0

    } else {

53

0

        req->passwd = auth.passwd;

54

0

        if (!auth.cluster.empty()) {

55

0

            req->__set_cluster(auth.cluster);

56

0

        }

57

0

        req->__set_user_ip(auth.user_ip);

58

0

    }

59

1

}

Unexecuted instantiation: _ZN5doris16set_request_authINS_23TLoadTxnRollbackRequestEEEvPT_RKNS_8AuthInfoE

Unexecuted instantiation: _ZN5doris16set_request_authINS_21TStreamLoadPutRequestEEEvPT_RKNS_8AuthInfoE

// This is the threshold used to periodically release the memory occupied by the expression.

// RELEASE_CONTEXT_COUNTER should be power of 2

// GCC will optimize the modulo operation to &(release_context_counter - 1)

// _conjunct_ctxs will free local alloc after this probe calculations

static constexpr int RELEASE_CONTEXT_COUNTER = 1 << 7;

static_assert((RELEASE_CONTEXT_COUNTER & (RELEASE_CONTEXT_COUNTER - 1)) == 0,

              "should be power of 2");

template <typename To, typename From>

To convert_to(From from) {

    union {

        From _from;

        To _to;

    };

    _from = from;

    return _to;

}

inline bool random_bool_slow(double probability_of_true = 0.5) {

    // Due to an unknown JNI bug, we cannot use thread_local variables here.

    static std::random_device seed;

    static std::mt19937 gen(seed());

    std::bernoulli_distribution d(probability_of_true);

    return d(gen);

}

} // namespace doris