risedev/task/
prometheus_service.rs

1// Copyright 2025 RisingWave Labs
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 std::env;
16use std::path::{Path, PathBuf};
17use std::process::Command;
18
19use anyhow::{Result, anyhow};
20
21use super::{ExecuteContext, Task};
22use crate::util::stylized_risedev_subcmd;
23use crate::{PrometheusConfig, PrometheusGen};
24
25pub struct PrometheusService {
26    config: PrometheusConfig,
27}
28
29impl PrometheusService {
30    pub fn new(config: PrometheusConfig) -> Result<Self> {
31        Ok(Self { config })
32    }
33
34    fn prometheus_path(&self) -> Result<PathBuf> {
35        let prefix_bin = env::var("PREFIX_BIN")?;
36        Ok(Path::new(&prefix_bin).join("prometheus").join("prometheus"))
37    }
38
39    fn prometheus(&self) -> Result<Command> {
40        Ok(Command::new(self.prometheus_path()?))
41    }
42
43    /// Apply command args according to config
44    pub fn apply_command_args(cmd: &mut Command, config: &PrometheusConfig) -> Result<()> {
45        cmd.arg(format!(
46            "--web.listen-address={}:{}",
47            config.listen_address, config.port
48        ));
49        cmd.arg("--storage.tsdb.retention.time=30d");
50        Ok(())
51    }
52}
53
54impl Task for PrometheusService {
55    fn execute(&mut self, ctx: &mut ExecuteContext<impl std::io::Write>) -> anyhow::Result<()> {
56        ctx.service(self);
57        ctx.pb.set_message("starting...");
58
59        let path = self.prometheus_path()?;
60        if !path.exists() {
61            return Err(anyhow!(
62                "prometheus binary not found in {:?}\nDid you enable monitoring feature in `{}`?",
63                path,
64                stylized_risedev_subcmd("configure")
65            ));
66        }
67
68        let prefix_config = env::var("PREFIX_CONFIG")?;
69        let prefix_data = env::var("PREFIX_DATA")?;
70
71        fs_err::write(
72            Path::new(&prefix_config).join("prometheus.yml"),
73            PrometheusGen.gen_prometheus_yml(&self.config),
74        )?;
75
76        let mut cmd = self.prometheus()?;
77
78        Self::apply_command_args(&mut cmd, &self.config)?;
79
80        cmd.arg(format!(
81            "--config.file={}",
82            Path::new(&prefix_config)
83                .join("prometheus.yml")
84                .to_string_lossy()
85        ))
86        .arg(format!(
87            "--storage.tsdb.path={}",
88            Path::new(&prefix_data).join("prometheus").to_string_lossy()
89        ));
90
91        ctx.run_command(ctx.tmux_run(cmd)?)?;
92
93        ctx.pb.set_message("started");
94
95        Ok(())
96    }
97
98    fn id(&self) -> String {
99        self.config.id.clone()
100    }
101}
risedev/task/prometheus_service.rs

risedev/task/
prometheus_service.rs
