a05dbd3f81
Supplement the documentation of be-clion-dev, avoid the problem of undefined DORIS_JAVA_HOME and inability to find jni.h when using clion development without directly compiling through build.sh Complete the classification of header files in pch.h and introduce some header files that are not frequently modified in doris. Separate the declaration and definition in common/config.h. If you need to modify the default configuration now, please modify it in common/config.cpp. gen_cpp/version.h is regenerated every time it is recompiled, which may cause PCH to fail, so now you need to get the version information indirectly rather than directly.
134 lines
3.8 KiB
C++
134 lines
3.8 KiB
C++
// 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.
|
|
|
|
#include "testutil/test_util.h"
|
|
|
|
#ifndef __APPLE__
|
|
#else
|
|
#include <mach-o/dyld.h>
|
|
#endif
|
|
|
|
#include <common/config.h>
|
|
#include <ctype.h>
|
|
#include <glog/logging.h>
|
|
#include <libgen.h>
|
|
#include <limits.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <strings.h>
|
|
#include <unistd.h>
|
|
|
|
#include <algorithm>
|
|
// IWYU pragma: no_include <bits/chrono.h>
|
|
#include <chrono> // IWYU pragma: keep
|
|
#include <random>
|
|
|
|
#include "gutil/strings/numbers.h"
|
|
#include "gutil/strings/substitute.h"
|
|
#include "olap/olap_common.h"
|
|
|
|
using strings::Substitute;
|
|
|
|
namespace doris {
|
|
|
|
static const char* const kSlowTestsEnvVar = "DORIS_ALLOW_SLOW_TESTS";
|
|
|
|
bool AllowSlowTests() {
|
|
return GetBooleanEnvironmentVariable(kSlowTestsEnvVar);
|
|
}
|
|
|
|
bool GetBooleanEnvironmentVariable(const char* env_var_name) {
|
|
const char* const e = getenv(env_var_name);
|
|
if ((e == nullptr) || (strlen(e) == 0) || (strcasecmp(e, "false") == 0) ||
|
|
(strcasecmp(e, "0") == 0) || (strcasecmp(e, "no") == 0)) {
|
|
return false;
|
|
}
|
|
if ((strcasecmp(e, "true") == 0) || (strcasecmp(e, "1") == 0) || (strcasecmp(e, "yes") == 0)) {
|
|
return true;
|
|
}
|
|
LOG(FATAL) << Substitute("$0: invalid value for environment variable $0", e, env_var_name);
|
|
return false; // unreachable
|
|
}
|
|
|
|
std::string GetCurrentRunningDir() {
|
|
char exe[PATH_MAX];
|
|
ssize_t r;
|
|
|
|
#ifdef __APPLE__
|
|
uint32_t size = PATH_MAX;
|
|
if (_NSGetExecutablePath(exe, &size) < 0) {
|
|
return std::string();
|
|
}
|
|
r = strlen(exe) + 1;
|
|
#else
|
|
if ((r = readlink("/proc/self/exe", exe, PATH_MAX)) < 0) {
|
|
return std::string();
|
|
}
|
|
#endif
|
|
|
|
if (r == PATH_MAX) {
|
|
r -= 1;
|
|
}
|
|
exe[r] = 0;
|
|
char* dir = dirname(exe);
|
|
return std::string(dir);
|
|
}
|
|
|
|
void InitConfig() {
|
|
std::string conf_file = std::string(getenv("DORIS_HOME")) + "/conf/be.conf";
|
|
if (!config::init(conf_file.c_str(), false)) {
|
|
fprintf(stderr, "Init config file failed, path %s\n", conf_file.c_str());
|
|
exit(-1);
|
|
}
|
|
}
|
|
|
|
bool equal_ignore_case(std::string lhs, std::string rhs) {
|
|
std::transform(lhs.begin(), lhs.end(), lhs.begin(), ::tolower);
|
|
std::transform(rhs.begin(), rhs.end(), rhs.begin(), ::tolower);
|
|
return lhs == rhs;
|
|
}
|
|
|
|
std::mt19937_64 rng_64(std::chrono::steady_clock::now().time_since_epoch().count());
|
|
|
|
int rand_rng_int(int l, int r) {
|
|
std::uniform_int_distribution<int> u(l, r);
|
|
return u(rng_64);
|
|
}
|
|
char rand_rng_char() {
|
|
return (rand_rng_int(0, 1) ? 'a' : 'A') + rand_rng_int(0, 25);
|
|
}
|
|
std::string rand_rng_string(size_t length) {
|
|
string s;
|
|
while (length--) {
|
|
s += rand_rng_char();
|
|
}
|
|
return s;
|
|
}
|
|
std::string rand_rng_by_type(FieldType fieldType) {
|
|
if (fieldType == FieldType::OLAP_FIELD_TYPE_CHAR) {
|
|
return rand_rng_string(rand_rng_int(1, 8));
|
|
} else if (fieldType == FieldType::OLAP_FIELD_TYPE_VARCHAR) {
|
|
return rand_rng_string(rand_rng_int(1, 128));
|
|
} else if (fieldType == FieldType::OLAP_FIELD_TYPE_STRING) {
|
|
return rand_rng_string(rand_rng_int(1, 100000));
|
|
} else {
|
|
return std::to_string(rand_rng_int(1, 1000000));
|
|
}
|
|
}
|
|
|
|
} // namespace doris
|