risedev/task/
task_redis_ready_check.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::io::Write;
16
17use anyhow::Result;
18
19use crate::{ExecuteContext, RedisConfig, Task};
20
21pub struct RedisReadyCheckTask {
22    config: RedisConfig,
23}
24
25impl RedisReadyCheckTask {
26    pub fn new(config: RedisConfig) -> Result<Self> {
27        Ok(Self { config })
28    }
29}
30
31impl Task for RedisReadyCheckTask {
32    fn execute(&mut self, ctx: &mut ExecuteContext<impl Write>) -> Result<()> {
33        ctx.pb.set_message("waiting for online...");
34        let client = redis::Client::open(
35            "redis://".to_owned() + &format!("{}:{}", self.config.address, self.config.port),
36        )?;
37
38        ctx.wait(|| {
39            let _ = client.get_connection()?;
40            Ok(())
41        })?;
42
43        ctx.complete_spin();
44        Ok(())
45    }
46}
risedev/task/task_redis_ready_check.rs

risedev/task/
task_redis_ready_check.rs
