risedev/task/
postgres_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 crate::PostgresConfig;
16use crate::task::docker_service::{DockerService, DockerServiceConfig};
17
18impl DockerServiceConfig for PostgresConfig {
19    fn id(&self) -> String {
20        self.id.clone()
21    }
22
23    fn is_user_managed(&self) -> bool {
24        self.user_managed
25    }
26
27    fn image(&self) -> String {
28        self.image.clone()
29    }
30
31    fn args(&self) -> Vec<String> {
32        // Enable CDC.
33        [
34            "-c",
35            "wal_level=logical",
36            "-c",
37            "max_replication_slots=30",
38            "-c",
39            "log_statement=all",
40        ]
41        .map(String::from)
42        .to_vec()
43    }
44
45    fn envs(&self) -> Vec<(String, String)> {
46        vec![
47            ("POSTGRES_HOST_AUTH_METHOD".to_owned(), "trust".to_owned()),
48            ("POSTGRES_USER".to_owned(), self.user.clone()),
49            ("POSTGRES_PASSWORD".to_owned(), self.password.clone()),
50            ("POSTGRES_DB".to_owned(), self.database.clone()),
51            (
52                "POSTGRES_INITDB_ARGS".to_owned(),
53                "--encoding=UTF-8 --lc-collate=C --lc-ctype=C".to_owned(),
54            ),
55        ]
56    }
57
58    fn ports(&self) -> Vec<(String, String)> {
59        vec![(self.port.to_string(), "5432".to_owned())]
60    }
61
62    fn data_path(&self) -> Option<String> {
63        self.persist_data
64            .then(|| "/var/lib/postgresql/data".to_owned())
65    }
66
67    fn latency_ms(&self) -> u32 {
68        self.latency_ms
69    }
70}
71
72/// Docker-backed PostgreSQL service.
73pub type PostgresService = DockerService<PostgresConfig>;
risedev/task/postgres_service.rs

risedev/task/
postgres_service.rs
