mirror of
https://gitlab.computer.surgery/matrix/grapevine.git
synced 2025-12-17 15:51:23 +01:00
parent
a854ce5cf6
commit
17a0b34309
5 changed files with 2 additions and 96 deletions
|
|
@ -27,8 +27,6 @@ pub struct Config {
|
||||||
pub db_cache_capacity_mb: f64,
|
pub db_cache_capacity_mb: f64,
|
||||||
#[serde(default = "true_fn")]
|
#[serde(default = "true_fn")]
|
||||||
pub enable_lightning_bolt: bool,
|
pub enable_lightning_bolt: bool,
|
||||||
#[serde(default = "true_fn")]
|
|
||||||
pub allow_check_for_updates: bool,
|
|
||||||
#[serde(default = "default_conduit_cache_capacity_modifier")]
|
#[serde(default = "default_conduit_cache_capacity_modifier")]
|
||||||
pub conduit_cache_capacity_modifier: f64,
|
pub conduit_cache_capacity_modifier: f64,
|
||||||
#[serde(default = "default_rocksdb_max_open_files")]
|
#[serde(default = "default_rocksdb_max_open_files")]
|
||||||
|
|
|
||||||
|
|
@ -12,7 +12,6 @@ use ruma::{
|
||||||
use crate::{database::KeyValueDatabase, service, services, utils, Error, Result};
|
use crate::{database::KeyValueDatabase, service, services, utils, Error, Result};
|
||||||
|
|
||||||
pub const COUNTER: &[u8] = b"c";
|
pub const COUNTER: &[u8] = b"c";
|
||||||
pub const LAST_CHECK_FOR_UPDATES_COUNT: &[u8] = b"u";
|
|
||||||
|
|
||||||
#[async_trait]
|
#[async_trait]
|
||||||
impl service::globals::Data for KeyValueDatabase {
|
impl service::globals::Data for KeyValueDatabase {
|
||||||
|
|
@ -28,23 +27,6 @@ impl service::globals::Data for KeyValueDatabase {
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
fn last_check_for_updates_id(&self) -> Result<u64> {
|
|
||||||
self.global
|
|
||||||
.get(LAST_CHECK_FOR_UPDATES_COUNT)?
|
|
||||||
.map_or(Ok(0_u64), |bytes| {
|
|
||||||
utils::u64_from_bytes(&bytes).map_err(|_| {
|
|
||||||
Error::bad_database("last check for updates count has invalid bytes.")
|
|
||||||
})
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
fn update_check_for_updates_id(&self, id: u64) -> Result<()> {
|
|
||||||
self.global
|
|
||||||
.insert(LAST_CHECK_FOR_UPDATES_COUNT, &id.to_be_bytes())?;
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn watch(&self, user_id: &UserId, device_id: &DeviceId) -> Result<()> {
|
async fn watch(&self, user_id: &UserId, device_id: &DeviceId) -> Result<()> {
|
||||||
let userid_bytes = user_id.as_bytes().to_vec();
|
let userid_bytes = user_id.as_bytes().to_vec();
|
||||||
let mut userid_prefix = userid_bytes.clone();
|
let mut userid_prefix = userid_bytes.clone();
|
||||||
|
|
|
||||||
|
|
@ -19,7 +19,6 @@ use ruma::{
|
||||||
CanonicalJsonValue, EventId, OwnedDeviceId, OwnedEventId, OwnedRoomId, OwnedUserId, RoomId,
|
CanonicalJsonValue, EventId, OwnedDeviceId, OwnedEventId, OwnedRoomId, OwnedUserId, RoomId,
|
||||||
UserId,
|
UserId,
|
||||||
};
|
};
|
||||||
use serde::Deserialize;
|
|
||||||
use std::{
|
use std::{
|
||||||
collections::{BTreeMap, HashMap, HashSet},
|
collections::{BTreeMap, HashMap, HashSet},
|
||||||
fs::{self, remove_dir_all},
|
fs::{self, remove_dir_all},
|
||||||
|
|
@ -27,9 +26,7 @@ use std::{
|
||||||
mem::size_of,
|
mem::size_of,
|
||||||
path::Path,
|
path::Path,
|
||||||
sync::{Arc, Mutex, RwLock},
|
sync::{Arc, Mutex, RwLock},
|
||||||
time::Duration,
|
|
||||||
};
|
};
|
||||||
use tokio::time::interval;
|
|
||||||
|
|
||||||
use tracing::{debug, error, info, warn};
|
use tracing::{debug, error, info, warn};
|
||||||
|
|
||||||
|
|
@ -986,9 +983,6 @@ impl KeyValueDatabase {
|
||||||
services().sending.start_handler();
|
services().sending.start_handler();
|
||||||
|
|
||||||
Self::start_cleanup_task().await;
|
Self::start_cleanup_task().await;
|
||||||
if services().globals.allow_check_for_updates() {
|
|
||||||
Self::start_check_for_updates_task();
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
@ -1004,62 +998,10 @@ impl KeyValueDatabase {
|
||||||
res
|
res
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tracing::instrument]
|
|
||||||
pub fn start_check_for_updates_task() {
|
|
||||||
tokio::spawn(async move {
|
|
||||||
let timer_interval = Duration::from_secs(60 * 60);
|
|
||||||
let mut i = interval(timer_interval);
|
|
||||||
loop {
|
|
||||||
i.tick().await;
|
|
||||||
let _ = Self::try_handle_updates().await;
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn try_handle_updates() -> Result<()> {
|
|
||||||
let response = services()
|
|
||||||
.globals
|
|
||||||
.default_client()
|
|
||||||
.get("https://conduit.rs/check-for-updates/stable")
|
|
||||||
.send()
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
|
||||||
struct CheckForUpdatesResponseEntry {
|
|
||||||
id: u64,
|
|
||||||
date: String,
|
|
||||||
message: String,
|
|
||||||
}
|
|
||||||
#[derive(Deserialize)]
|
|
||||||
struct CheckForUpdatesResponse {
|
|
||||||
updates: Vec<CheckForUpdatesResponseEntry>,
|
|
||||||
}
|
|
||||||
|
|
||||||
let response = serde_json::from_str::<CheckForUpdatesResponse>(&response.text().await?)
|
|
||||||
.map_err(|_| Error::BadServerResponse("Bad version check response"))?;
|
|
||||||
|
|
||||||
let mut last_update_id = services().globals.last_check_for_updates_id()?;
|
|
||||||
for update in response.updates {
|
|
||||||
last_update_id = last_update_id.max(update.id);
|
|
||||||
if update.id > services().globals.last_check_for_updates_id()? {
|
|
||||||
println!("{}", update.message);
|
|
||||||
services()
|
|
||||||
.admin
|
|
||||||
.send_message(RoomMessageEventContent::text_plain(format!(
|
|
||||||
"@room: The following is a message from the Conduit developers. It was sent on '{}':\n\n{}",
|
|
||||||
update.date, update.message
|
|
||||||
)))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
services()
|
|
||||||
.globals
|
|
||||||
.update_check_for_updates_id(last_update_id)?;
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tracing::instrument]
|
#[tracing::instrument]
|
||||||
pub async fn start_cleanup_task() {
|
pub async fn start_cleanup_task() {
|
||||||
|
use tokio::time::interval;
|
||||||
|
|
||||||
#[cfg(unix)]
|
#[cfg(unix)]
|
||||||
use tokio::signal::unix::{signal, SignalKind};
|
use tokio::signal::unix::{signal, SignalKind};
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -13,8 +13,6 @@ use crate::Result;
|
||||||
pub trait Data: Send + Sync {
|
pub trait Data: Send + Sync {
|
||||||
fn next_count(&self) -> Result<u64>;
|
fn next_count(&self) -> Result<u64>;
|
||||||
fn current_count(&self) -> Result<u64>;
|
fn current_count(&self) -> Result<u64>;
|
||||||
fn last_check_for_updates_id(&self) -> Result<u64>;
|
|
||||||
fn update_check_for_updates_id(&self, id: u64) -> Result<()>;
|
|
||||||
async fn watch(&self, user_id: &UserId, device_id: &DeviceId) -> Result<()>;
|
async fn watch(&self, user_id: &UserId, device_id: &DeviceId) -> Result<()>;
|
||||||
fn cleanup(&self) -> Result<()>;
|
fn cleanup(&self) -> Result<()>;
|
||||||
fn memory_usage(&self) -> String;
|
fn memory_usage(&self) -> String;
|
||||||
|
|
|
||||||
|
|
@ -255,16 +255,6 @@ impl Service {
|
||||||
self.db.current_count()
|
self.db.current_count()
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tracing::instrument(skip(self))]
|
|
||||||
pub fn last_check_for_updates_id(&self) -> Result<u64> {
|
|
||||||
self.db.last_check_for_updates_id()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tracing::instrument(skip(self))]
|
|
||||||
pub fn update_check_for_updates_id(&self, id: u64) -> Result<()> {
|
|
||||||
self.db.update_check_for_updates_id(id)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn watch(&self, user_id: &UserId, device_id: &DeviceId) -> Result<()> {
|
pub async fn watch(&self, user_id: &UserId, device_id: &DeviceId) -> Result<()> {
|
||||||
self.db.watch(user_id, device_id).await
|
self.db.watch(user_id, device_id).await
|
||||||
}
|
}
|
||||||
|
|
@ -313,10 +303,6 @@ impl Service {
|
||||||
self.config.enable_lightning_bolt
|
self.config.enable_lightning_bolt
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn allow_check_for_updates(&self) -> bool {
|
|
||||||
self.config.allow_check_for_updates
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn trusted_servers(&self) -> &[OwnedServerName] {
|
pub fn trusted_servers(&self) -> &[OwnedServerName] {
|
||||||
&self.config.trusted_servers
|
&self.config.trusted_servers
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue