metricsd_main.cc revision 9d3a4aeae2bd59ebe72fca44c4fa508c1e9f1333 
1/*
2 * Copyright (C) 2015 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#include <base/at_exit.h>
18#include <base/command_line.h>
19#include <base/files/file_path.h>
20#include <base/logging.h>
21#include <base/strings/string_util.h>
22#include <base/time/time.h>
23#include <brillo/flag_helper.h>
24#include <brillo/syslog_logging.h>
25
26#include "constants.h"
27#include "uploader/upload_service.h"
28
29
30int main(int argc, char** argv) {
31  DEFINE_bool(foreground, false, "Don't daemonize");
32
33  // Upload the metrics once and exit. (used for testing)
34  DEFINE_bool(uploader_test,
35              false,
36              "run the uploader once and exit");
37
38  // Upload Service flags.
39  DEFINE_int32(upload_interval_secs,
40               1800,
41               "Interval at which metrics_daemon sends the metrics. (needs "
42               "-uploader)");
43  DEFINE_string(server,
44                metrics::kMetricsServer,
45                "Server to upload the metrics to. (needs -uploader)");
46  DEFINE_string(private_directory, metrics::kMetricsdDirectory,
47                "Path to the private directory used by metricsd "
48                "(testing only)");
49  DEFINE_string(shared_directory, metrics::kSharedMetricsDirectory,
50                "Path to the shared metrics directory, used by "
51                "metrics_collector, metricsd and all metrics clients "
52                "(testing only)");
53
54  DEFINE_bool(logtostderr, false, "Log to standard error");
55  DEFINE_bool(logtosyslog, false, "Log to syslog");
56
57  brillo::FlagHelper::Init(argc, argv, "Brillo metrics daemon.");
58
59  int logging_location = (FLAGS_foreground ? brillo::kLogToStderr
60                          : brillo::kLogToSyslog);
61  if (FLAGS_logtosyslog)
62    logging_location = brillo::kLogToSyslog;
63
64  if (FLAGS_logtostderr)
65    logging_location = brillo::kLogToStderr;
66
67  // Also log to stderr when not running as daemon.
68  brillo::InitLog(logging_location | brillo::kLogHeader);
69
70  if (FLAGS_logtostderr && FLAGS_logtosyslog) {
71    LOG(ERROR) << "only one of --logtosyslog and --logtostderr can be set";
72    return 1;
73  }
74
75  if (!FLAGS_foreground && daemon(0, 0) != 0) {
76    return errno;
77  }
78
79  UploadService service(
80      FLAGS_server, base::TimeDelta::FromSeconds(FLAGS_upload_interval_secs),
81      base::FilePath(FLAGS_private_directory),
82      base::FilePath(FLAGS_shared_directory));
83
84  service.Run();
85}
86