Move synapse_auto_compressor

This commit is contained in:
Erik Johnston
2021-09-28 16:05:38 +01:00
parent 70b73489c0
commit 08d312b66d
9 changed files with 46 additions and 45 deletions

42
Cargo.lock generated
View File

@@ -54,26 +54,6 @@ dependencies = [
"winapi", "winapi",
] ]
[[package]]
name = "auto_compressor"
version = "0.1.0"
dependencies = [
"anyhow",
"clap",
"env_logger",
"jemallocator",
"log",
"log-panics",
"openssl",
"postgres",
"postgres-openssl",
"pyo3",
"pyo3-log",
"rand",
"serial_test",
"synapse_compress_state",
]
[[package]] [[package]]
name = "autocfg" name = "autocfg"
version = "1.0.1" version = "1.0.1"
@@ -144,7 +124,6 @@ dependencies = [
name = "compressor_integration_tests" name = "compressor_integration_tests"
version = "0.1.0" version = "0.1.0"
dependencies = [ dependencies = [
"auto_compressor",
"env_logger", "env_logger",
"log", "log",
"openssl", "openssl",
@@ -154,6 +133,7 @@ dependencies = [
"serial_test", "serial_test",
"state-map", "state-map",
"string_cache", "string_cache",
"synapse_auto_compressor",
"synapse_compress_state", "synapse_compress_state",
] ]
@@ -1122,6 +1102,26 @@ dependencies = [
"unicode-xid", "unicode-xid",
] ]
[[package]]
name = "synapse_auto_compressor"
version = "0.1.0"
dependencies = [
"anyhow",
"clap",
"env_logger",
"jemallocator",
"log",
"log-panics",
"openssl",
"postgres",
"postgres-openssl",
"pyo3",
"pyo3-log",
"rand",
"serial_test",
"synapse_compress_state",
]
[[package]] [[package]]
name = "synapse_compress_state" name = "synapse_compress_state"
version = "0.1.0" version = "0.1.0"

View File

@@ -1,5 +1,5 @@
[workspace] [workspace]
members = ["auto_compressor", "compressor_integration_tests"] members = ["synapse_auto_compressor", "compressor_integration_tests"]
[package] [package]
authors = ["Erik Johnston"] authors = ["Erik Johnston"]

View File

@@ -3,7 +3,7 @@
This workspace contains experimental tools that attempt to reduce the number of This workspace contains experimental tools that attempt to reduce the number of
rows in the `state_groups_state` table inside of a Synapse Postgresql database. rows in the `state_groups_state` table inside of a Synapse Postgresql database.
# Automated tool: auto_compressor # Automated tool: synapse_auto_compressor
## Introduction: ## Introduction:
@@ -11,7 +11,7 @@ This tool is significantly more simple to use than the manual tool (described be
It scans through all of the rows in the `state_groups` database table from the start. When It scans through all of the rows in the `state_groups` database table from the start. When
it finds a group that hasn't been compressed, it runs the compressor for a while on that it finds a group that hasn't been compressed, it runs the compressor for a while on that
group's room, saving where it got up to. After compressing a number of these chunks it stops, group's room, saving where it got up to. After compressing a number of these chunks it stops,
saving where it got up to for the next run of the `auto_compressor`. saving where it got up to for the next run of the `synapse_auto_compressor`.
It creates three extra tables in the database: `state_compressor_state` which stores the It creates three extra tables in the database: `state_compressor_state` which stores the
information needed to stop and start the compressor for each room, `state_compressor_progress` information needed to stop and start the compressor for each room, `state_compressor_progress`
@@ -26,14 +26,15 @@ periodically.
This tool requires `cargo` to be installed. See https://www.rust-lang.org/tools/install This tool requires `cargo` to be installed. See https://www.rust-lang.org/tools/install
for instructions on how to do this. for instructions on how to do this.
To build `auto_compressor`, clone this repository and navigate to the `autocompressor/` To build `synapse_auto_compressor`, clone this repository and navigate to the
subdirectory. Then execute `cargo build`. `synapse_auto_compressor/` subdirectory. Then execute `cargo build`.
This will create an executable and store it in `auto_compressor/target/debug/auto_compressor`. This will create an executable and store it in
`synapse_auto_compressor/target/debug/auto_compressor`.
## Example usage ## Example usage
``` ```
$ auto_compressor -p postgresql://user:pass@localhost/synapse -c 500 -n 100 $ synapse_auto_compressor -p postgresql://user:pass@localhost/synapse -c 500 -n 100
``` ```
## Running Options ## Running Options
@@ -199,7 +200,7 @@ If you want to use the compressor in another project, it is recomended that you
use jemalloc `https://github.com/gnzlbg/jemallocator`. use jemalloc `https://github.com/gnzlbg/jemallocator`.
To prevent the progress bars from being shown, use the `no-progress-bars` feature. To prevent the progress bars from being shown, use the `no-progress-bars` feature.
(See `auto_compressor/Cargo.toml` for an example) (See `synapse_auto_compressor/Cargo.toml` for an example)
# Troubleshooting # Troubleshooting
@@ -225,17 +226,17 @@ setting in [`postgresql.conf`](https://www.postgresql.org/docs/current/runtime-c
The amount of output the tools produce can be altered by setting the RUST_LOG The amount of output the tools produce can be altered by setting the RUST_LOG
environment variable to something. environment variable to something.
To get more logs when running the auto_compressor tool try the following: To get more logs when running the synapse_auto_compressor tool try the following:
``` ```
$ RUST_LOG=debug auto_compressor -p postgresql://user:pass@localhost/synapse -c 50 -n 100 $ RUST_LOG=debug synapse_auto_compressor -p postgresql://user:pass@localhost/synapse -c 50 -n 100
``` ```
If you want to suppress all the debugging info you are getting from the If you want to suppress all the debugging info you are getting from the
Postgres client then try: Postgres client then try:
``` ```
RUST_LOG=auto_compressor=debug,synapse_compress_state=debug auto_compressor [etc.] RUST_LOG=synapse_auto_compressor=debug,synapse_compress_state=debug synapse_auto_compressor [etc.]
``` ```
This will only print the debugging information from those two packages. For more info see This will only print the debugging information from those two packages. For more info see
@@ -265,7 +266,7 @@ be a large problem.
## Compressor is trying to increase the number of rows ## Compressor is trying to increase the number of rows
Backfilling can lead to issues with compression. The auto_compressor will Backfilling can lead to issues with compression. The synapse_auto_compressor will
skip chunks it can't reduce the size of and so this should help jump over the backfilled skip chunks it can't reduce the size of and so this should help jump over the backfilled
state_groups. Lots of state resolution might also impact the ability to use the compressor. state_groups. Lots of state resolution might also impact the ability to use the compressor.

View File

@@ -13,9 +13,9 @@ postgres = "0.19.0"
postgres-openssl = "0.5.0" postgres-openssl = "0.5.0"
rand = "0.8.0" rand = "0.8.0"
synapse_compress_state = { path = "../", features = ["no-progress-bars"] } synapse_compress_state = { path = "../", features = ["no-progress-bars"] }
auto_compressor = { path = "../auto_compressor/" } synapse_auto_compressor = { path = "../synapse_auto_compressor/" }
env_logger = "0.9.0" env_logger = "0.9.0"
log = "0.4.14" log = "0.4.14"
[dependencies.state-map] [dependencies.state-map]
git = "https://github.com/matrix-org/rust-matrix-state-map" git = "https://github.com/matrix-org/rust-matrix-state-map"

View File

@@ -1,9 +1,9 @@
use auto_compressor::state_saving::{ use compressor_integration_tests::{clear_compressor_state, setup_logger, DB_URL};
use serial_test::serial;
use synapse_auto_compressor::state_saving::{
connect_to_database, create_tables_if_needed, read_room_compressor_state, connect_to_database, create_tables_if_needed, read_room_compressor_state,
write_room_compressor_state, write_room_compressor_state,
}; };
use compressor_integration_tests::{clear_compressor_state, setup_logger, DB_URL};
use serial_test::serial;
use synapse_compress_state::Level; use synapse_compress_state::Level;
#[test] #[test]

View File

@@ -18,7 +18,7 @@ the compressor is run.
3. Navigate to the correct location 3. Navigate to the correct location
For the automatic tool: For the automatic tool:
`$ cd /home/synapse/rust-synapse-compress-state/auto_compressor` `$ cd /home/synapse/rust-synapse-compress-state/synpase_auto_compressor`
For the manual tool: For the manual tool:
`$ cd /home/synapse/rust-synapse-compress-state` `$ cd /home/synapse/rust-synapse-compress-state`
@@ -30,9 +30,9 @@ This will install the relevant compressor tool into the activated virtual enviro
## Automatic tool example: ## Automatic tool example:
```python ```python
import auto_compressor import synapse_auto_compressor
auto_compressor.compress_state_events_table( synapse_auto_compressor.compress_state_events_table(
db_url="postgresql://localhost/synapse", db_url="postgresql://localhost/synapse",
chunk_size=500, chunk_size=500,
default_levels="100,50,25", default_levels="100,50,25",
@@ -51,4 +51,4 @@ synapse_compress_state.run_compression(
output_file="out.sql", output_file="out.sql",
transactions=True transactions=True
) )
``` ```

View File

@@ -788,7 +788,7 @@ fn synapse_compress_state(_py: Python, m: &PyModule) -> PyResult<()> {
let _ = pyo3_log::Logger::default() let _ = pyo3_log::Logger::default()
// don't send out anything lower than a warning from other crates // don't send out anything lower than a warning from other crates
.filter(LevelFilter::Warn) .filter(LevelFilter::Warn)
// don't log warnings from synapse_compress_state, the auto_compressor handles these // don't log warnings from synapse_compress_state, the synapse_auto_compressor handles these
// situations and provides better log messages // situations and provides better log messages
.filter_target("synapse_compress_state".to_owned(), LevelFilter::Debug) .filter_target("synapse_compress_state".to_owned(), LevelFilter::Debug)
.install(); .install();

View File

@@ -1,5 +1,5 @@
[package] [package]
name = "auto_compressor" name = "synapse_auto_compressor"
authors = ["William Ashton"] authors = ["William Ashton"]
version = "0.1.0" version = "0.1.0"
edition = "2018" edition = "2018"

View File

@@ -19,10 +19,10 @@
#[global_allocator] #[global_allocator]
static GLOBAL: jemallocator::Jemalloc = jemallocator::Jemalloc; static GLOBAL: jemallocator::Jemalloc = jemallocator::Jemalloc;
use auto_compressor::{manager, state_saving, LevelInfo};
use clap::{crate_authors, crate_description, crate_name, crate_version, value_t, App, Arg}; use clap::{crate_authors, crate_description, crate_name, crate_version, value_t, App, Arg};
use log::LevelFilter; use log::LevelFilter;
use std::{env, fs::OpenOptions}; use std::{env, fs::OpenOptions};
use synapse_auto_compressor::{manager, state_saving, LevelInfo};
/// Execution starts here /// Execution starts here
fn main() { fn main() {