risedev/task/
minio_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::MinioConfig;
23use crate::util::stylized_risedev_subcmd;
24
25pub struct MinioService {
26    config: MinioConfig,
27}
28
29impl MinioService {
30    pub fn new(config: MinioConfig) -> Result<Self> {
31        Ok(Self { config })
32    }
33
34    fn minio_path(&self) -> Result<PathBuf> {
35        let prefix_bin = env::var("PREFIX_BIN")?;
36        Ok(Path::new(&prefix_bin).join("minio"))
37    }
38
39    fn minio(&self) -> Result<Command> {
40        Ok(Command::new(self.minio_path()?))
41    }
42
43    /// Apply command args according to config
44    pub fn apply_command_args(cmd: &mut Command, config: &MinioConfig) -> Result<()> {
45        cmd.arg("server")
46            .arg("--address")
47            .arg(format!("{}:{}", config.listen_address, config.port))
48            .arg("--console-address")
49            .arg(format!("{}:{}", config.listen_address, config.console_port))
50            .env("MINIO_ROOT_USER", &config.root_user)
51            .env("MINIO_ROOT_PASSWORD", &config.root_password)
52            .env("MINIO_PROMETHEUS_AUTH_TYPE", "public")
53            // Allow MinIO to be used on root disk, bypass restriction.
54            // https://github.com/risingwavelabs/risingwave/pull/3012
55            // https://docs.min.io/minio/baremetal/installation/deploy-minio-single-node-single-drive.html#id3
56            .env("MINIO_CI_CD", "1");
57        if config.api_requests_max > 0 {
58            // Rate limit minio
59            cmd.env(
60                "MINIO_API_REQUESTS_MAX",
61                config.api_requests_max.to_string(),
62            );
63        }
64        if !config.api_requests_deadline.is_empty() {
65            // Rate limit minio
66            cmd.env("MINIO_API_REQUESTS_DEADLINE", &config.api_requests_deadline);
67        }
68
69        let provide_prometheus = config.provide_prometheus.as_ref().unwrap();
70        match provide_prometheus.len() {
71            0 => {}
72            1 => {
73                let prometheus = &provide_prometheus[0];
74                cmd.env(
75                    "MINIO_PROMETHEUS_URL",
76                    format!("http://{}:{}", prometheus.address, prometheus.port),
77                );
78            }
79            other_length => {
80                return Err(anyhow!("expected 0 or 1 prometheus, get {}", other_length));
81            }
82        }
83
84        Ok(())
85    }
86}
87
88impl Task for MinioService {
89    fn execute(&mut self, ctx: &mut ExecuteContext<impl std::io::Write>) -> anyhow::Result<()> {
90        ctx.service(self);
91        ctx.pb.set_message("starting...");
92
93        let path = self.minio_path()?;
94        if !path.exists() {
95            return Err(anyhow!(
96                "minio binary not found in {:?}\nDid you enable minio feature in `{}`?",
97                path,
98                stylized_risedev_subcmd("configure")
99            ));
100        }
101
102        let mut cmd = self.minio()?;
103
104        Self::apply_command_args(&mut cmd, &self.config)?;
105
106        let prefix_config = env::var("PREFIX_CONFIG")?;
107
108        let data_path = Path::new(&env::var("PREFIX_DATA")?).join(self.id());
109        fs_err::create_dir_all(&data_path)?;
110
111        cmd.arg("--config-dir")
112            .arg(Path::new(&prefix_config).join("minio"))
113            .arg(&data_path);
114
115        ctx.run_command(ctx.tmux_run(cmd)?)?;
116
117        ctx.pb.set_message("started");
118
119        Ok(())
120    }
121
122    fn id(&self) -> String {
123        self.config.id.clone()
124    }
125}
risedev/task/minio_service.rs

risedev/task/
minio_service.rs
