mirror of
https://github.com/neondatabase/neon.git
synced 2026-05-25 00:50:36 +00:00
f64074c609
Currently it's included with minimal changes and lives aside of the main workspace. Later we may re-use and combine common parts with zenith control_plane. This change is mostly needed to unify cloud deployment pipeline: 1.1. build compute-tools image 1.2. build compute-node image based on the freshly built compute-tools 2. build zenith image So we can roll new compute image and new storage required by it to operate properly. Also it becomes easier to test console against some specific version of compute-node/-tools.
52 lines
1.4 KiB
Rust
52 lines
1.4 KiB
Rust
use std::fs::{File, OpenOptions};
|
|
use std::io;
|
|
use std::io::prelude::*;
|
|
use std::path::Path;
|
|
|
|
use anyhow::Result;
|
|
|
|
use crate::pg_helpers::PgOptionsSerialize;
|
|
use crate::zenith::ClusterSpec;
|
|
|
|
/// Check that `line` is inside a text file and put it there if it is not.
|
|
/// Create file if it doesn't exist.
|
|
pub fn line_in_file(path: &Path, line: &str) -> Result<bool> {
|
|
let mut file = OpenOptions::new()
|
|
.read(true)
|
|
.write(true)
|
|
.create(true)
|
|
.append(false)
|
|
.open(path)?;
|
|
let buf = io::BufReader::new(&file);
|
|
let mut count: usize = 0;
|
|
|
|
for l in buf.lines() {
|
|
if l? == line {
|
|
return Ok(false);
|
|
}
|
|
count = 1;
|
|
}
|
|
|
|
write!(file, "{}{}", "\n".repeat(count), line)?;
|
|
Ok(true)
|
|
}
|
|
|
|
/// Create or completely rewrite configuration file specified by `path`
|
|
pub fn write_postgres_conf(path: &Path, spec: &ClusterSpec) -> Result<()> {
|
|
// File::create() destroys the file content if it exists.
|
|
let mut postgres_conf = File::create(path)?;
|
|
|
|
write_zenith_managed_block(&mut postgres_conf, &spec.cluster.settings.as_pg_settings())?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
// Write Postgres config block wrapped with generated comment section
|
|
fn write_zenith_managed_block(file: &mut File, buf: &str) -> Result<()> {
|
|
writeln!(file, "# Managed by Zenith: begin")?;
|
|
writeln!(file, "{}", buf)?;
|
|
writeln!(file, "# Managed by Zenith: end")?;
|
|
|
|
Ok(())
|
|
}
|