wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-05 22:18:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Metrics] Add a flag to disable stdout exporter (#4634)

Browse Source
This commit is contained in:
Wang Qing
2019-04-20 10:06:30 +08:00
committed by Robert Nishihara
parent d5df91b031
commit d951eb740f
2 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ray/raylet/main.cc
+6 -3
View File
@@ -21,7 +21,7 @@ int main(int argc, char *argv[]) {
ray::RayLogLevel::INFO,
/*log_dir=*/"");
ray::RayLog::InstallFailureSignalHandler();
RAY_CHECK(argc >= 14 && argc <= 18);
RAY_CHECK(argc >= 14 && argc <= 19);
const std::string raylet_socket_name = std::string(argv[1]);
const std::string store_socket_name = std::string(argv[2]);
@@ -39,16 +39,19 @@ int main(int argc, char *argv[]) {
const std::string redis_password = (argc >= 15 ? std::string(argv[14]) : "");
const std::string temp_dir = (argc >= 16 ? std::string(argv[15]) : "/tmp/ray");
const std::string disable_stats_str(argc >= 17 ? std::string(argv[16]) : "false");
const bool disable_stats = ("true" == disable_stats_str);
const bool disable_stats("true" == disable_stats_str);
const std::string stat_address =
(argc >= 18 ? std::string(argv[17]) : "127.0.0.1:8888");
const std::string disable_stdout_exporter_str(argc >= 19 ? std::string(argv[18])
: "true");
const bool disable_stdout_exporter("true" == disable_stdout_exporter_str);
// Initialize stats.
const ray::stats::TagsType global_tags = {
{ray::stats::JobNameKey, "raylet"},
{ray::stats::VersionKey, "0.7.0"},
{ray::stats::NodeAddressKey, node_ip_address}};
ray::stats::Init(stat_address, global_tags, disable_stats);
ray::stats::Init(stat_address, global_tags, disable_stats, disable_stdout_exporter);
// Configuration for the node manager.
ray::raylet::NodeManagerConfig node_manager_config;
src/ray/stats/stats.h
+5 -3
View File
@@ -23,7 +23,7 @@ namespace stats {
/// Initialize stats.
static void Init(const std::string &address, const TagsType &global_tags,
bool disable_stats = false) {
bool disable_stats = false, bool disable_stdout_exporter = true) {
StatsConfig::instance().SetIsDisableStats(disable_stats);
if (disable_stats) {
RAY_LOG(INFO) << "Disabled stats.";
@@ -36,8 +36,10 @@ static void Init(const std::string &address, const TagsType &global_tags,
static auto exporter =
std::make_shared<opencensus::exporters::stats::PrometheusExporter>();
// Enable stdout exporter by default.
opencensus::exporters::stats::StdoutExporter::Register();
if (!disable_stdout_exporter) {
// Enable stdout exporter by default.
opencensus::exporters::stats::StdoutExporter::Register();
}
// Enable prometheus exporter.
try {