commit dc308fede5abde7865d4d8d1bb928b0c9be9740d · nekomimi.pet/microcosm-rs

+6 -9

Cargo.lock

···

       225
        
       

     

       226
        
       [[package]]

     

       227
        
       name = "atrium-api"

     

       228
       -
       version = "0.25.2"

     

       229
       -
       source = "registry+https://github.com/rust-lang/crates.io-index"

     

       230
       -
       checksum = "0d4eb9b4787aba546015c8ccda1d3924c157cee13d67848997fba74ac8144a07"

     

       231
        
       dependencies = [

     

       232
        
        "atrium-common",

     

       233
        
        "atrium-xrpc",

     
···

       245
        
       

     

       246
        
       [[package]]

     

       247
        
       name = "atrium-common"

     

       248
       -
       version = "0.1.1"

     

       249
       -
       source = "registry+https://github.com/rust-lang/crates.io-index"

     

       250
       -
       checksum = "ba30d2f9e1a8b3db8fc97d0a5f91ee5a28f8acdddb771ad74c1b08eda357ca3d"

     

       251
        
       dependencies = [

     

       252
        
        "dashmap",

     

       253
        
        "lru",

     
···

       260
        
       

     

       261
        
       [[package]]

     

       262
        
       name = "atrium-xrpc"

     

       263
       -
       version = "0.12.2"

     

       264
       -
       source = "registry+https://github.com/rust-lang/crates.io-index"

     

       265
       -
       checksum = "18a9e526cb2ed3e0a2ca78c3ce2a943d9041a68e067dadf42923b523771e07df"

     

       266
        
       dependencies = [

     

       267
        
        "http",

     

       268
        
        "serde",

···

       225
        
       

     

       226
        
       [[package]]

     

       227
        
       name = "atrium-api"

     

       228
       +
       version = "0.25.3"

     

       229
       +
       source = "git+https://github.com/uniphil/atrium?branch=fix%2Fnsid-allow-nonleading-name-digits#c4364f318d337bbc3e3e3aaf97c9f971e95f5f7e"

     

       0
        
       
     

       230
        
       dependencies = [

     

       231
        
        "atrium-common",

     

       232
        
        "atrium-xrpc",

     
···

       244
        
       

     

       245
        
       [[package]]

     

       246
        
       name = "atrium-common"

     

       247
       +
       version = "0.1.2"

     

       248
       +
       source = "git+https://github.com/uniphil/atrium?branch=fix%2Fnsid-allow-nonleading-name-digits#c4364f318d337bbc3e3e3aaf97c9f971e95f5f7e"

     

       0
        
       
     

       249
        
       dependencies = [

     

       250
        
        "dashmap",

     

       251
        
        "lru",

     
···

       258
        
       

     

       259
        
       [[package]]

     

       260
        
       name = "atrium-xrpc"

     

       261
       +
       version = "0.12.3"

     

       262
       +
       source = "git+https://github.com/uniphil/atrium?branch=fix%2Fnsid-allow-nonleading-name-digits#c4364f318d337bbc3e3e3aaf97c9f971e95f5f7e"

     

       0
        
       
     

       263
        
       dependencies = [

     

       264
        
        "http",

     

       265
        
        "serde",

+1 -1

jetstream/Cargo.toml

···

       10
        
       

     

       11
        
       [dependencies]

     

       12
        
       async-trait = "0.1.83"

     

       13
       -
       atrium-api = { version = "0.25.2", default-features = false, features = [

     

       14
        
           "namespace-appbsky",

     

       15
        
       ] }

     

       16
        
       tokio = { version = "1.44.2", features = ["full", "sync", "time"] }

···

       10
        
       

     

       11
        
       [dependencies]

     

       12
        
       async-trait = "0.1.83"

     

       13
       +
       atrium-api = { git = "https://github.com/uniphil/atrium", branch = "fix/nsid-allow-nonleading-name-digits", default-features = false, features = [

     

       14
        
           "namespace-appbsky",

     

       15
        
       ] }

     

       16
        
       tokio = { version = "1.44.2", features = ["full", "sync", "time"] }

+2 -2

jetstream/src/lib.rs

···

       439
        
       

     

       440
        
                               if let Some(last) = last_cursor {

     

       441
        
                                   if event_cursor <= *last {

     

       442
       -
                                       log::warn!("event cursor {event_cursor:?} was older than the last one: {last:?}. dropping event.");

     

       443
        
                                       continue;

     

       444
        
                                   }

     

       445
        
                               }

     
···

       475
        
       

     

       476
        
                               if let Some(last) = last_cursor {

     

       477
        
                                   if event_cursor <= *last {

     

       478
       -
                                       log::warn!("event cursor {event_cursor:?} was older than the last one: {last:?}. dropping event.");

     

       479
        
                                       continue;

     

       480
        
                                   }

     

       481
        
                               }

···

       439
        
       

     

       440
        
                               if let Some(last) = last_cursor {

     

       441
        
                                   if event_cursor <= *last {

     

       442
       +
                                       log::warn!("event cursor {event_cursor:?} was not newer than the last one: {last:?}. dropping event.");

     

       443
        
                                       continue;

     

       444
        
                                   }

     

       445
        
                               }

     
···

       475
        
       

     

       476
        
                               if let Some(last) = last_cursor {

     

       477
        
                                   if event_cursor <= *last {

     

       478
       +
                                       log::warn!("event cursor {event_cursor:?} was not newer than the last one: {last:?}. dropping event.");

     

       479
        
                                       continue;

     

       480
        
                                   }

     

       481
        
                               }

+32 -12

ufos/src/consumer.rs

···

       8
        
       use std::mem;

     

       9
        
       use std::time::Duration;

     

       10
        
       use tokio::sync::mpsc::{channel, Receiver, Sender};

     

       0
        
       
     

       11
        
       

     

       12
        
       use crate::error::{BatchInsertError, FirehoseEventError};

     

       13
        
       use crate::{DeleteAccount, EventBatch, UFOsCommit};

     
···

       34
        
           batch_sender: Sender<LimitedBatch>,

     

       35
        
           current_batch: CurrentBatch,

     

       36
        
           sketch_secret: SketchSecretPrefix,

     

       0
        
       
     

       37
        
       }

     

       38
        
       

     

       39
        
       pub async fn consume(

     
···

       64
        
               .await?;

     

       65
        
           let (batch_sender, batch_reciever) = channel::<LimitedBatch>(BATCH_QUEUE_SIZE);

     

       66
        
           let mut batcher = Batcher::new(jetstream_receiver, batch_sender, sketch_secret);

     

       67
       -
           tokio::task::spawn(async move { batcher.run().await });

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       68
        
           Ok(batch_reciever)

     

       69
        
       }

     

       70
        
       

     
···

       74
        
               batch_sender: Sender<LimitedBatch>,

     

       75
        
               sketch_secret: SketchSecretPrefix,

     

       76
        
           ) -> Self {

     

       0
        
       
     

       0
        
       
     

       77
        
               Self {

     

       78
        
                   jetstream_receiver,

     

       79
        
                   batch_sender,

     

       80
        
                   current_batch: Default::default(),

     

       81
        
                   sketch_secret,

     

       0
        
       
     

       82
        
               }

     

       83
        
           }

     

       84
        
       

     

       85
        
           pub async fn run(&mut self) -> anyhow::Result<()> {

     

       0
        
       
     

       86
        
               loop {

     

       87
       -
                   if let Some(event) = self.jetstream_receiver.recv().await {

     

       88
       -
                       self.handle_event(event).await?

     

       89
       -
                   } else {

     

       90
       -
                       anyhow::bail!("channel closed");

     

       91
        
                   }

     

       92
        
               }

     

       93
        
           }

     

       94
        
       

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       95
        
           async fn handle_event(&mut self, event: JetstreamEvent) -> anyhow::Result<()> {

     

       96
        
               if let Some(earliest) = &self.current_batch.initial_cursor {

     

       97
        
                   if event.cursor.duration_since(earliest)? > Duration::from_secs_f64(MAX_BATCH_SPAN_SECS)

     

       98
        
                   {

     

       99
       -
                       self.send_current_batch_now(false).await?;

     

       0
        
       
     

       100
        
                   }

     

       101
        
               } else {

     

       102
        
                   self.current_batch.initial_cursor = Some(event.cursor);

     
···

       126
        
                   if event.cursor.duration_since(earliest)?.as_secs_f64() > MIN_BATCH_SPAN_SECS

     

       127
        
                       && self.batch_sender.capacity() == BATCH_QUEUE_SIZE

     

       128
        
                   {

     

       129
       -
                       self.send_current_batch_now(true).await?;

     

       130
        
                   }

     

       131
        
               }

     

       132
        
               Ok(())

     
···

       141
        
               );

     

       142
        
       

     

       143
        
               if let Err(BatchInsertError::BatchFull(commit)) = optimistic_res {

     

       144
       -
                   self.send_current_batch_now(false).await?;

     

       145
        
                   self.current_batch.batch.insert_commit_by_nsid(

     

       146
        
                       &collection,

     

       147
        
                       commit,

     
···

       157
        
       

     

       158
        
           async fn handle_delete_account(&mut self, did: Did, cursor: Cursor) -> anyhow::Result<()> {

     

       159
        
               if self.current_batch.batch.account_removes.len() >= MAX_ACCOUNT_REMOVES {

     

       160
       -
                   self.send_current_batch_now(false).await?;

     

       161
        
               }

     

       162
        
               self.current_batch

     

       163
        
                   .batch

     
···

       168
        
       

     

       169
        
           // holds up all consumer progress until it can send to the channel

     

       170
        
           // use this when the current batch is too full to add more to it

     

       171
       -
           async fn send_current_batch_now(&mut self, small: bool) -> anyhow::Result<()> {

     

       172
        
               let beginning = match self.current_batch.initial_cursor.map(|c| c.elapsed()) {

     

       173
        
                   None => "unknown".to_string(),

     

       174
        
                   Some(Ok(t)) => format!("{:?}", t),

     

       175
        
                   Some(Err(e)) => format!("+{:?}", e.duration()),

     

       176
        
               };

     

       177
       -
               log::info!(

     

       178
       -
                   "sending batch now from {beginning}, {}, queue capacity: {}",

     

       179
        
                   if small { "small" } else { "full" },

     

       180
        
                   self.batch_sender.capacity(),

     

       181
        
               );

     

       182
        
               let current = mem::take(&mut self.current_batch);

     

       0
        
       
     

       183
        
               self.batch_sender

     

       184
        
                   .send_timeout(current.batch, Duration::from_secs_f64(SEND_TIMEOUT_S))

     

       185
        
                   .await?;

···

       8
        
       use std::mem;

     

       9
        
       use std::time::Duration;

     

       10
        
       use tokio::sync::mpsc::{channel, Receiver, Sender};

     

       11
       +
       use tokio::time::{timeout, Interval};

     

       12
        
       

     

       13
        
       use crate::error::{BatchInsertError, FirehoseEventError};

     

       14
        
       use crate::{DeleteAccount, EventBatch, UFOsCommit};

     
···

       35
        
           batch_sender: Sender<LimitedBatch>,

     

       36
        
           current_batch: CurrentBatch,

     

       37
        
           sketch_secret: SketchSecretPrefix,

     

       38
       +
           rate_limit: Interval,

     

       39
        
       }

     

       40
        
       

     

       41
        
       pub async fn consume(

     
···

       66
        
               .await?;

     

       67
        
           let (batch_sender, batch_reciever) = channel::<LimitedBatch>(BATCH_QUEUE_SIZE);

     

       68
        
           let mut batcher = Batcher::new(jetstream_receiver, batch_sender, sketch_secret);

     

       69
       +
           tokio::task::spawn(async move {

     

       70
       +
               let r = batcher.run().await;

     

       71
       +
               log::info!("batcher ended: {r:?}");

     

       72
       +
           });

     

       73
        
           Ok(batch_reciever)

     

       74
        
       }

     

       75
        
       

     
···

       79
        
               batch_sender: Sender<LimitedBatch>,

     

       80
        
               sketch_secret: SketchSecretPrefix,

     

       81
        
           ) -> Self {

     

       82
       +
               let mut rate_limit = tokio::time::interval(std::time::Duration::from_millis(5));

     

       83
       +
               rate_limit.set_missed_tick_behavior(tokio::time::MissedTickBehavior::Delay);

     

       84
        
               Self {

     

       85
        
                   jetstream_receiver,

     

       86
        
                   batch_sender,

     

       87
        
                   current_batch: Default::default(),

     

       88
        
                   sketch_secret,

     

       89
       +
                   rate_limit,

     

       90
        
               }

     

       91
        
           }

     

       92
        
       

     

       93
        
           pub async fn run(&mut self) -> anyhow::Result<()> {

     

       94
       +
               // TODO: report errors *from here* probably, since this gets shipped off into a spawned task that might just vanish

     

       95
        
               loop {

     

       96
       +
                   match timeout(Duration::from_millis(9_000), self.jetstream_receiver.recv()).await {

     

       97
       +
                       Err(_elapsed) => self.no_events_step().await?,

     

       98
       +
                       Ok(Some(event)) => self.handle_event(event).await?,

     

       99
       +
                       Ok(None) => anyhow::bail!("channel closed"),

     

       100
        
                   }

     

       101
        
               }

     

       102
        
           }

     

       103
        
       

     

       104
       +
           async fn no_events_step(&mut self) -> anyhow::Result<()> {

     

       105
       +
               let empty = self.current_batch.batch.is_empty();

     

       106
       +
               log::info!("no events received, stepping batcher (empty? {empty})");

     

       107
       +
               if !empty {

     

       108
       +
                   self.send_current_batch_now(true, "no events step").await?;

     

       109
       +
               }

     

       110
       +
               Ok(())

     

       111
       +
           }

     

       112
       +
       

     

       113
        
           async fn handle_event(&mut self, event: JetstreamEvent) -> anyhow::Result<()> {

     

       114
        
               if let Some(earliest) = &self.current_batch.initial_cursor {

     

       115
        
                   if event.cursor.duration_since(earliest)? > Duration::from_secs_f64(MAX_BATCH_SPAN_SECS)

     

       116
        
                   {

     

       117
       +
                       self.send_current_batch_now(false, "time since event")

     

       118
       +
                           .await?;

     

       119
        
                   }

     

       120
        
               } else {

     

       121
        
                   self.current_batch.initial_cursor = Some(event.cursor);

     
···

       145
        
                   if event.cursor.duration_since(earliest)?.as_secs_f64() > MIN_BATCH_SPAN_SECS

     

       146
        
                       && self.batch_sender.capacity() == BATCH_QUEUE_SIZE

     

       147
        
                   {

     

       148
       +
                       self.send_current_batch_now(true, "available queue").await?;

     

       149
        
                   }

     

       150
        
               }

     

       151
        
               Ok(())

     
···

       160
        
               );

     

       161
        
       

     

       162
        
               if let Err(BatchInsertError::BatchFull(commit)) = optimistic_res {

     

       163
       +
                   self.send_current_batch_now(false, "handle commit").await?;

     

       164
        
                   self.current_batch.batch.insert_commit_by_nsid(

     

       165
        
                       &collection,

     

       166
        
                       commit,

     
···

       176
        
       

     

       177
        
           async fn handle_delete_account(&mut self, did: Did, cursor: Cursor) -> anyhow::Result<()> {

     

       178
        
               if self.current_batch.batch.account_removes.len() >= MAX_ACCOUNT_REMOVES {

     

       179
       +
                   self.send_current_batch_now(false, "delete account").await?;

     

       180
        
               }

     

       181
        
               self.current_batch

     

       182
        
                   .batch

     
···

       187
        
       

     

       188
        
           // holds up all consumer progress until it can send to the channel

     

       189
        
           // use this when the current batch is too full to add more to it

     

       190
       +
           async fn send_current_batch_now(&mut self, small: bool, referrer: &str) -> anyhow::Result<()> {

     

       191
        
               let beginning = match self.current_batch.initial_cursor.map(|c| c.elapsed()) {

     

       192
        
                   None => "unknown".to_string(),

     

       193
        
                   Some(Ok(t)) => format!("{:?}", t),

     

       194
        
                   Some(Err(e)) => format!("+{:?}", e.duration()),

     

       195
        
               };

     

       196
       +
               log::trace!(

     

       197
       +
                   "sending batch now from {beginning}, {}, queue capacity: {}, referrer: {referrer}",

     

       198
        
                   if small { "small" } else { "full" },

     

       199
        
                   self.batch_sender.capacity(),

     

       200
        
               );

     

       201
        
               let current = mem::take(&mut self.current_batch);

     

       202
       +
               self.rate_limit.tick().await;

     

       203
        
               self.batch_sender

     

       204
        
                   .send_timeout(current.batch, Duration::from_secs_f64(SEND_TIMEOUT_S))

     

       205
        
                   .await?;

+24 -8

ufos/src/file_consumer.rs

···

       12
        
       async fn read_jsonl(f: File, sender: Sender<JetstreamEvent>) -> Result<()> {

     

       13
        
           let mut lines = BufReader::new(f).lines();

     

       14
        
           while let Some(line) = lines.next_line().await? {

     

       15
       -
               let event: JetstreamEvent =

     

       16
       -
                   serde_json::from_str(&line).map_err(JetstreamEventError::ReceivedMalformedJSON)?;

     

       17
       -
               if sender.send(event).await.is_err() {

     

       18
       -
                   log::warn!("All receivers for the jsonl fixture have been dropped, bye.");

     

       19
       -
                   return Err(JetstreamEventError::ReceiverClosedError.into());

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       20
        
               }

     

       21
        
           }

     

       22
       -
           Ok(())

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       23
        
       }

     

       24
        
       

     

       25
        
       pub async fn consume(

     
···

       30
        
           let (jsonl_sender, jsonl_receiver) = channel::<JetstreamEvent>(16);

     

       31
        
           let (batch_sender, batch_reciever) = channel::<LimitedBatch>(BATCH_QUEUE_SIZE);

     

       32
        
           let mut batcher = Batcher::new(jsonl_receiver, batch_sender, sketch_secret);

     

       33
       -
           tokio::task::spawn(async move { read_jsonl(f, jsonl_sender).await });

     

       34
       -
           tokio::task::spawn(async move { batcher.run().await });

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       35
        
           Ok(batch_reciever)

     

       36
        
       }

···

       12
        
       async fn read_jsonl(f: File, sender: Sender<JetstreamEvent>) -> Result<()> {

     

       13
        
           let mut lines = BufReader::new(f).lines();

     

       14
        
           while let Some(line) = lines.next_line().await? {

     

       15
       +
               match serde_json::from_str::<JetstreamEvent>(&line) {

     

       16
       +
                   Ok(event) => match sender.send(event).await {

     

       17
       +
                       Ok(_) => {}

     

       18
       +
                       Err(e) => {

     

       19
       +
                           log::warn!("All receivers for the jsonl fixture have been dropped, bye: {e:?}");

     

       20
       +
                           return Err(JetstreamEventError::ReceiverClosedError.into());

     

       21
       +
                       }

     

       22
       +
                   },

     

       23
       +
                   Err(parse_err) => {

     

       24
       +
                       log::warn!("failed to parse event: {parse_err:?} from event:\n{line}");

     

       25
       +
                       continue;

     

       26
       +
                   }

     

       27
        
               }

     

       28
        
           }

     

       29
       +
           log::info!("reached end of jsonl file, looping on noop to keep server alive.");

     

       30
       +
           loop {

     

       31
       +
               tokio::time::sleep(std::time::Duration::from_secs_f64(10.)).await;

     

       32
       +
           }

     

       33
        
       }

     

       34
        
       

     

       35
        
       pub async fn consume(

     
···

       40
        
           let (jsonl_sender, jsonl_receiver) = channel::<JetstreamEvent>(16);

     

       41
        
           let (batch_sender, batch_reciever) = channel::<LimitedBatch>(BATCH_QUEUE_SIZE);

     

       42
        
           let mut batcher = Batcher::new(jsonl_receiver, batch_sender, sketch_secret);

     

       43
       +
           tokio::task::spawn(async move {

     

       44
       +
               let r = read_jsonl(f, jsonl_sender).await;

     

       45
       +
               log::info!("read_jsonl finished: {r:?}");

     

       46
       +
           });

     

       47
       +
           tokio::task::spawn(async move {

     

       48
       +
               let r = batcher.run().await;

     

       49
       +
               log::info!("batcher finished: {r:?}");

     

       50
       +
           });

     

       51
        
           Ok(batch_reciever)

     

       52
        
       }

ufos/src/lib.rs

···

       230
        
               endpoint: String,

     

       231
        
               started_at: u64,

     

       232
        
               latest_cursor: Option<u64>,

     

       0
        
       
     

       233
        
           },

     

       234
        
       }

     

       235

···

       230
        
               endpoint: String,

     

       231
        
               started_at: u64,

     

       232
        
               latest_cursor: Option<u64>,

     

       233
       +
               rollup_cursor: Option<u64>,

     

       234
        
           },

     

       235
        
       }

     

       236

+143 -4

ufos/src/main.rs

···

       1
        
       use clap::Parser;

     

       2
        
       use jetstream::events::Cursor;

     

       3
        
       use std::path::PathBuf;

     

       0
        
       
     

       4
        
       use ufos::consumer;

     

       5
        
       use ufos::file_consumer;

     

       6
        
       use ufos::server;

     
···

       8
        
       use ufos::storage_fjall::FjallStorage;

     

       9
        
       use ufos::storage_mem::MemStorage;

     

       10
        
       use ufos::store_types::SketchSecretPrefix;

     

       0
        
       
     

       11
        
       

     

       12
        
       #[cfg(not(target_env = "msvc"))]

     

       13
        
       use tikv_jemallocator::Jemalloc;

     
···

       36
        
           #[arg(long)]

     

       37
        
           data: PathBuf,

     

       38
        
           /// DEBUG: don't start the jetstream consumer or its write loop

     

       39
       -
           /// todo: restore this

     

       40
        
           #[arg(long, action)]

     

       41
        
           pause_writer: bool,

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       42
        
           /// DEBUG: force the rw loop to fall behind  by pausing it

     

       43
        
           /// todo: restore this

     

       44
        
           #[arg(long, action)]

     
···

       46
        
           /// DEBUG: use an in-memory store instead of fjall

     

       47
        
           #[arg(long, action)]

     

       48
        
           in_mem: bool,

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       49
        
           /// DEBUG: interpret jetstream as a file fixture

     

       50
        
           #[arg(long, action)]

     

       51
        
           jetstream_fixture: bool,

     
···

       68
        
                   args.jetstream,

     

       69
        
                   args.jetstream_fixture,

     

       70
        
                   args.pause_writer,

     

       0
        
       
     

       0
        
       
     

       71
        
                   read_store,

     

       72
        
                   write_store,

     

       73
        
                   cursor,

     
···

       85
        
                   args.jetstream,

     

       86
        
                   args.jetstream_fixture,

     

       87
        
                   args.pause_writer,

     

       0
        
       
     

       0
        
       
     

       88
        
                   read_store,

     

       89
        
                   write_store,

     

       90
        
                   cursor,

     
···

       96
        
           Ok(())

     

       97
        
       }

     

       98
        
       

     

       0
        
       
     

       99
        
       async fn go<B: StoreBackground>(

     

       100
        
           jetstream: String,

     

       101
        
           jetstream_fixture: bool,

     

       102
        
           pause_writer: bool,

     

       103
       -
           read_store: impl StoreReader + 'static,

     

       0
        
       
     

       0
        
       
     

       104
        
           mut write_store: impl StoreWriter<B> + 'static,

     

       105
        
           cursor: Option<Cursor>,

     

       106
        
           sketch_secret: SketchSecretPrefix,

     

       107
        
       ) -> anyhow::Result<()> {

     

       108
        
           println!("starting server with storage...");

     

       109
       -
           let serving = server::serve(read_store);

     

       110
        
       

     

       111
        
           if pause_writer {

     

       112
        
               log::info!("not starting jetstream or the write loop.");

     
···

       125
        
               consumer::consume(&jetstream, cursor, false, sketch_secret).await?

     

       126
        
           };

     

       127
        
       

     

       128
       -
           let rolling = write_store.background_tasks()?.run();

     

       129
        
           let storing = write_store.receive_batches(batches);

     

       0
        
       
     

       0
        
       
     

       130
        
       

     

       131
        
           tokio::select! {

     

       132
        
               z = serving => log::warn!("serve task ended: {z:?}"),

     

       133
        
               z = rolling => log::warn!("rollup task ended: {z:?}"),

     

       134
        
               z = storing => log::warn!("storage task ended: {z:?}"),

     

       0
        
       
     

       135
        
           };

     

       136
        
       

     

       137
        
           println!("bye!");

     

       138
        
       

     

       139
        
           Ok(())

     

       140
        
       }

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0

···

       1
        
       use clap::Parser;

     

       2
        
       use jetstream::events::Cursor;

     

       3
        
       use std::path::PathBuf;

     

       4
       +
       use std::time::{Duration, SystemTime};

     

       5
        
       use ufos::consumer;

     

       6
        
       use ufos::file_consumer;

     

       7
        
       use ufos::server;

     
···

       9
        
       use ufos::storage_fjall::FjallStorage;

     

       10
        
       use ufos::storage_mem::MemStorage;

     

       11
        
       use ufos::store_types::SketchSecretPrefix;

     

       12
       +
       use ufos::ConsumerInfo;

     

       13
        
       

     

       14
        
       #[cfg(not(target_env = "msvc"))]

     

       15
        
       use tikv_jemallocator::Jemalloc;

     
···

       38
        
           #[arg(long)]

     

       39
        
           data: PathBuf,

     

       40
        
           /// DEBUG: don't start the jetstream consumer or its write loop

     

       0
        
       
     

       41
        
           #[arg(long, action)]

     

       42
        
           pause_writer: bool,

     

       43
       +
           /// Adjust runtime settings like background task intervals for efficient backfill

     

       44
       +
           #[arg(long, action)]

     

       45
       +
           backfill: bool,

     

       46
        
           /// DEBUG: force the rw loop to fall behind  by pausing it

     

       47
        
           /// todo: restore this

     

       48
        
           #[arg(long, action)]

     
···

       50
        
           /// DEBUG: use an in-memory store instead of fjall

     

       51
        
           #[arg(long, action)]

     

       52
        
           in_mem: bool,

     

       53
       +
           /// reset the rollup cursor, scrape through missed things in the past (backfill)

     

       54
       +
           #[arg(long, action)]

     

       55
       +
           reroll: bool,

     

       56
        
           /// DEBUG: interpret jetstream as a file fixture

     

       57
        
           #[arg(long, action)]

     

       58
        
           jetstream_fixture: bool,

     
···

       75
        
                   args.jetstream,

     

       76
        
                   args.jetstream_fixture,

     

       77
        
                   args.pause_writer,

     

       78
       +
                   args.backfill,

     

       79
       +
                   args.reroll,

     

       80
        
                   read_store,

     

       81
        
                   write_store,

     

       82
        
                   cursor,

     
···

       94
        
                   args.jetstream,

     

       95
        
                   args.jetstream_fixture,

     

       96
        
                   args.pause_writer,

     

       97
       +
                   args.backfill,

     

       98
       +
                   args.reroll,

     

       99
        
                   read_store,

     

       100
        
                   write_store,

     

       101
        
                   cursor,

     
···

       107
        
           Ok(())

     

       108
        
       }

     

       109
        
       

     

       110
       +
       #[allow(clippy::too_many_arguments)]

     

       111
        
       async fn go<B: StoreBackground>(

     

       112
        
           jetstream: String,

     

       113
        
           jetstream_fixture: bool,

     

       114
        
           pause_writer: bool,

     

       115
       +
           backfill: bool,

     

       116
       +
           reroll: bool,

     

       117
       +
           read_store: impl StoreReader + 'static + Clone,

     

       118
        
           mut write_store: impl StoreWriter<B> + 'static,

     

       119
        
           cursor: Option<Cursor>,

     

       120
        
           sketch_secret: SketchSecretPrefix,

     

       121
        
       ) -> anyhow::Result<()> {

     

       122
        
           println!("starting server with storage...");

     

       123
       +
           let serving = server::serve(read_store.clone());

     

       124
        
       

     

       125
        
           if pause_writer {

     

       126
        
               log::info!("not starting jetstream or the write loop.");

     
···

       139
        
               consumer::consume(&jetstream, cursor, false, sketch_secret).await?

     

       140
        
           };

     

       141
        
       

     

       142
       +
           let rolling = write_store.background_tasks(reroll)?.run(backfill);

     

       143
        
           let storing = write_store.receive_batches(batches);

     

       144
       +
       

     

       145
       +
           let stating = do_update_stuff(read_store);

     

       146
        
       

     

       147
        
           tokio::select! {

     

       148
        
               z = serving => log::warn!("serve task ended: {z:?}"),

     

       149
        
               z = rolling => log::warn!("rollup task ended: {z:?}"),

     

       150
        
               z = storing => log::warn!("storage task ended: {z:?}"),

     

       151
       +
               z = stating => log::warn!("status task ended: {z:?}"),

     

       152
        
           };

     

       153
        
       

     

       154
        
           println!("bye!");

     

       155
        
       

     

       156
        
           Ok(())

     

       157
        
       }

     

       158
       +
       

     

       159
       +
       async fn do_update_stuff(read_store: impl StoreReader) {

     

       160
       +
           let started_at = std::time::SystemTime::now();

     

       161
       +
           let mut first_cursor = None;

     

       162
       +
           let mut first_rollup = None;

     

       163
       +
           let mut last_at = std::time::SystemTime::now();

     

       164
       +
           let mut last_cursor = None;

     

       165
       +
           let mut last_rollup = None;

     

       166
       +
           loop {

     

       167
       +
               tokio::time::sleep(std::time::Duration::from_secs_f64(4.)).await;

     

       168
       +
               match read_store.get_consumer_info().await {

     

       169
       +
                   Err(e) => log::warn!("failed to get jetstream consumer info: {e:?}"),

     

       170
       +
                   Ok(ConsumerInfo::Jetstream {

     

       171
       +
                       latest_cursor,

     

       172
       +
                       rollup_cursor,

     

       173
       +
                       ..

     

       174
       +
                   }) => {

     

       175
       +
                       let now = std::time::SystemTime::now();

     

       176
       +
                       let latest_cursor = latest_cursor.map(Cursor::from_raw_u64);

     

       177
       +
                       let rollup_cursor = rollup_cursor.map(Cursor::from_raw_u64);

     

       178
       +
                       backfill_info(

     

       179
       +
                           latest_cursor,

     

       180
       +
                           rollup_cursor,

     

       181
       +
                           last_cursor,

     

       182
       +
                           last_rollup,

     

       183
       +
                           last_at,

     

       184
       +
                           first_cursor,

     

       185
       +
                           first_rollup,

     

       186
       +
                           started_at,

     

       187
       +
                           now,

     

       188
       +
                       );

     

       189
       +
                       first_cursor = first_cursor.or(latest_cursor);

     

       190
       +
                       first_rollup = first_rollup.or(rollup_cursor);

     

       191
       +
                       last_cursor = latest_cursor;

     

       192
       +
                       last_rollup = rollup_cursor;

     

       193
       +
                       last_at = now;

     

       194
       +
                   }

     

       195
       +
               }

     

       196
       +
           }

     

       197
       +
       }

     

       198
       +
       

     

       199
       +
       #[allow(clippy::too_many_arguments)]

     

       200
       +
       fn backfill_info(

     

       201
       +
           latest_cursor: Option<Cursor>,

     

       202
       +
           rollup_cursor: Option<Cursor>,

     

       203
       +
           last_cursor: Option<Cursor>,

     

       204
       +
           last_rollup: Option<Cursor>,

     

       205
       +
           last_at: SystemTime,

     

       206
       +
           first_cursor: Option<Cursor>,

     

       207
       +
           first_rollup: Option<Cursor>,

     

       208
       +
           started_at: SystemTime,

     

       209
       +
           now: SystemTime,

     

       210
       +
       ) {

     

       211
       +
           let nice_duration = |dt: Duration| {

     

       212
       +
               let secs = dt.as_secs_f64();

     

       213
       +
               if secs < 1. {

     

       214
       +
                   return format!("{:.0}ms", secs * 1000.);

     

       215
       +
               }

     

       216
       +
               if secs < 60. {

     

       217
       +
                   return format!("{secs:.02}s");

     

       218
       +
               }

     

       219
       +
               let mins = (secs / 60.).floor();

     

       220
       +
               let rsecs = secs - (mins * 60.);

     

       221
       +
               if mins < 60. {

     

       222
       +
                   return format!("{mins:.0}m{rsecs:.0}s");

     

       223
       +
               }

     

       224
       +
               let hrs = (mins / 60.).floor();

     

       225
       +
               let rmins = mins - (hrs * 60.);

     

       226
       +
               if hrs < 24. {

     

       227
       +
                   return format!("{hrs:.0}h{rmins:.0}m{rsecs:.0}s");

     

       228
       +
               }

     

       229
       +
               let days = (hrs / 24.).floor();

     

       230
       +
               let rhrs = hrs - (days * 24.);

     

       231
       +
               format!("{days:.0}d{rhrs:.0}h{rmins:.0}m{rsecs:.0}s")

     

       232
       +
           };

     

       233
       +
       

     

       234
       +
           let nice_dt_two_maybes = |earlier: Option<Cursor>, later: Option<Cursor>| match (earlier, later)

     

       235
       +
           {

     

       236
       +
               (Some(earlier), Some(later)) => match later.duration_since(&earlier) {

     

       237
       +
                   Ok(dt) => nice_duration(dt),

     

       238
       +
                   Err(e) => {

     

       239
       +
                       let rev_dt = e.duration();

     

       240
       +
                       format!("+{}", nice_duration(rev_dt))

     

       241
       +
                   }

     

       242
       +
               },

     

       243
       +
               _ => "unknown".to_string(),

     

       244
       +
           };

     

       245
       +
       

     

       246
       +
           let rate = |mlatest: Option<Cursor>, msince: Option<Cursor>, real: Duration| {

     

       247
       +
               mlatest

     

       248
       +
                   .zip(msince)

     

       249
       +
                   .map(|(latest, since)| {

     

       250
       +
                       latest

     

       251
       +
                           .duration_since(&since)

     

       252
       +
                           .unwrap_or(Duration::from_millis(1))

     

       253
       +
                   })

     

       254
       +
                   .map(|dtc| format!("{:.2}", dtc.as_secs_f64() / real.as_secs_f64()))

     

       255
       +
                   .unwrap_or("??".into())

     

       256
       +
           };

     

       257
       +
       

     

       258
       +
           let dt_real = now

     

       259
       +
               .duration_since(last_at)

     

       260
       +
               .unwrap_or(Duration::from_millis(1));

     

       261
       +
       

     

       262
       +
           let dt_real_total = now

     

       263
       +
               .duration_since(started_at)

     

       264
       +
               .unwrap_or(Duration::from_millis(1));

     

       265
       +
       

     

       266
       +
           let cursor_rate = rate(latest_cursor, last_cursor, dt_real);

     

       267
       +
           let cursor_avg = rate(latest_cursor, first_cursor, dt_real_total);

     

       268
       +
       

     

       269
       +
           let rollup_rate = rate(rollup_cursor, last_rollup, dt_real);

     

       270
       +
           let rollup_avg = rate(rollup_cursor, first_rollup, dt_real_total);

     

       271
       +
       

     

       272
       +
           log::info!(

     

       273
       +
               "cursor: {} behind (→{}, {cursor_rate}x, {cursor_avg}x avg). rollup: {} behind (→{}, {rollup_rate}x, {rollup_avg}x avg).",

     

       274
       +
               latest_cursor.map(|c| c.elapsed().map(nice_duration).unwrap_or("++".to_string())).unwrap_or("?".to_string()),

     

       275
       +
               nice_dt_two_maybes(last_cursor, latest_cursor),

     

       276
       +
               rollup_cursor.map(|c| c.elapsed().map(nice_duration).unwrap_or("++".to_string())).unwrap_or("?".to_string()),

     

       277
       +
               nice_dt_two_maybes(last_rollup, rollup_cursor),

     

       278
       +
           );

     

       279
       +
       }

+4 -3

ufos/src/storage.rs

···

       26
        
       where

     

       27
        
           Self: 'static,

     

       28
        
       {

     

       29
       -
           fn background_tasks(&mut self) -> StorageResult<B>;

     

       30
        
       

     

       31
        
           fn receive_batches<const LIMIT: usize>(

     

       32
        
               mut self,

     
···

       53
        
       

     

       54
        
           fn step_rollup(&mut self) -> StorageResult<(usize, HashSet<Nsid>)>;

     

       55
        
       

     

       56
       -
           fn trim_collection(&mut self, collection: &Nsid, limit: usize) -> StorageResult<()>;

     

       0
        
       
     

       57
        
       

     

       58
        
           fn delete_account(&mut self, did: &Did) -> StorageResult<usize>;

     

       59
        
       }

     

       60
        
       

     

       61
        
       #[async_trait]

     

       62
        
       pub trait StoreBackground: Send + Sync {

     

       63
       -
           async fn run(mut self) -> StorageResult<()>;

     

       64
        
       }

     

       65
        
       

     

       66
        
       #[async_trait]

···

       26
        
       where

     

       27
        
           Self: 'static,

     

       28
        
       {

     

       29
       +
           fn background_tasks(&mut self, reroll: bool) -> StorageResult<B>;

     

       30
        
       

     

       31
        
           fn receive_batches<const LIMIT: usize>(

     

       32
        
               mut self,

     
···

       53
        
       

     

       54
        
           fn step_rollup(&mut self) -> StorageResult<(usize, HashSet<Nsid>)>;

     

       55
        
       

     

       56
       +
           fn trim_collection(&mut self, collection: &Nsid, limit: usize)

     

       57
       +
               -> StorageResult<(usize, usize)>;

     

       58
        
       

     

       59
        
           fn delete_account(&mut self, did: &Did) -> StorageResult<usize>;

     

       60
        
       }

     

       61
        
       

     

       62
        
       #[async_trait]

     

       63
        
       pub trait StoreBackground: Send + Sync {

     

       64
       +
           async fn run(mut self, backfill: bool) -> StorageResult<()>;

     

       65
        
       }

     

       66
        
       

     

       67
        
       #[async_trait]

+41 -27

ufos/src/storage_fjall.rs

···

       149
        
               let keyspace = {

     

       150
        
                   let config = Config::new(path);

     

       151
        
       

     

       152
       -
                   #[cfg(not(test))]

     

       153
       -
                   let config = config.fsync_ms(Some(4_000));

     

       154
        
       

     

       155
        
                   config.open()?

     

       156
        
               };

     
···

       187
        
                           )?;

     

       188
        
                       } else {

     

       189
        
                           return Err(StorageError::InitError(format!(

     

       190
       -
                               "stored js_endpoint {stored:?} differs from provided {endpoint:?}, refusing to start.")));

     

       191
        
                       }

     

       192
        
                   }

     

       193
        
                   stored_secret

     
···

       360
        
                   get_snapshot_static_neu::<JetstreamCursorKey, JetstreamCursorValue>(&global)?

     

       361
        
                       .map(|c| c.to_raw_u64());

     

       362
        
       

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       363
        
               Ok(ConsumerInfo::Jetstream {

     

       364
        
                   endpoint,

     

       365
        
                   started_at,

     

       366
        
                   latest_cursor,

     

       0
        
       
     

       367
        
               })

     

       368
        
           }

     

       369
        
       

     
···

       801
        
       }

     

       802
        
       

     

       803
        
       impl StoreWriter<FjallBackground> for FjallWriter {

     

       804
       -
           fn background_tasks(&mut self) -> StorageResult<FjallBackground> {

     

       805
        
               if self.bg_taken.swap(true, Ordering::SeqCst) {

     

       806
        
                   Err(StorageError::BackgroundAlreadyStarted)

     

       807
        
               } else {

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       808
        
                   Ok(FjallBackground(self.clone()))

     

       809
        
               }

     

       810
        
           }

     
···

       948
        
               Ok((cursors_stepped, dirty_nsids))

     

       949
        
           }

     

       950
        
       

     

       951
       -
           fn trim_collection(&mut self, collection: &Nsid, limit: usize) -> StorageResult<()> {

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       952
        
               let mut dangling_feed_keys_cleaned = 0;

     

       953
        
               let mut records_deleted = 0;

     

       954
        
       

     
···

       967
        
               let mut live_records_found = 0;

     

       968
        
               let mut latest_expired_feed_cursor = None;

     

       969
        
               let mut batch = self.keyspace.batch();

     

       970
       -
               for kv in self.feeds.range(live_range).rev() {

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       971
        
                   let (key_bytes, val_bytes) = kv?;

     

       972
        
                   let feed_key = db_complete::<NsidRecordFeedKey>(&key_bytes)?;

     

       973
        
                   let feed_val = db_complete::<NsidRecordFeedVal>(&val_bytes)?;

     
···

       1022
        
       

     

       1023
        
               batch.commit()?;

     

       1024
        
       

     

       1025
       -
               log::info!("trim_collection ({collection:?}) removed {dangling_feed_keys_cleaned} dangling feed entries and {records_deleted} records");

     

       1026
       -
               Ok(())

     

       1027
        
           }

     

       1028
        
       

     

       1029
        
           fn delete_account(&mut self, did: &Did) -> Result<usize, StorageError> {

     
···

       1048
        
       

     

       1049
        
       #[async_trait]

     

       1050
        
       impl StoreBackground for FjallBackground {

     

       1051
       -
           async fn run(mut self) -> StorageResult<()> {

     

       1052
        
               let mut dirty_nsids = HashSet::new();

     

       1053
        
       

     

       1054
       -
               let mut rollup = tokio::time::interval(Duration::from_millis(81));

     

       0
        
       
     

       1055
        
               rollup.set_missed_tick_behavior(tokio::time::MissedTickBehavior::Delay);

     

       1056
        
       

     

       1057
       -
               let mut trim = tokio::time::interval(Duration::from_millis(6_000));

     

       0
        
       
     

       1058
        
               trim.set_missed_tick_behavior(tokio::time::MissedTickBehavior::Skip);

     

       1059
        
       

     

       1060
        
               loop {

     
···

       1065
        
                               rollup.reset_after(Duration::from_millis(1_200)); // we're caught up, take a break

     

       1066
        
                           }

     

       1067
        
                           dirty_nsids.extend(dirty);

     

       1068
       -
                           log::info!("rolled up {n} items ({} collections now dirty)", dirty_nsids.len());

     

       1069
        
                       },

     

       1070
        
                       _ = trim.tick() => {

     

       1071
       -
                           log::info!("trimming {} nsids: {dirty_nsids:?}", dirty_nsids.len());

     

       0
        
       
     

       1072
        
                           let t0 = Instant::now();

     

       0
        
       
     

       1073
        
                           for collection in &dirty_nsids {

     

       1074
       -
                               self.0.trim_collection(collection, 512).inspect_err(|e| log::error!("trim error: {e:?}"))?;

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       1075
        
                           }

     

       1076
       -
                           log::info!("finished trimming in {:?}", t0.elapsed());

     

       1077
        
                           dirty_nsids.clear();

     

       1078
        
                       },

     

       1079
        
                   };

     
···

       1154
        
       }

     

       1155
        
       

     

       1156
        
       ////////// temp stuff to remove:

     

       1157
       -
       

     

       1158
       -
       // fn summarize_batch<const LIMIT: usize>(batch: &EventBatch<LIMIT>) -> String {

     

       1159
       -
       //     format!(

     

       1160
       -
       //         "batch of {: >3} samples from {: >4} records in {: >2} collections from ~{: >4} DIDs, {} acct removes, cursor {: <12?}",

     

       1161
       -
       //         batch.total_records(),

     

       1162
       -
       //         batch.total_seen(),

     

       1163
       -
       //         batch.total_collections(),

     

       1164
       -
       //         batch.estimate_dids(),

     

       1165
       -
       //         batch.account_removes(),

     

       1166
       -
       //         batch.latest_cursor().map(|c| c.elapsed()),

     

       1167
       -
       //     )

     

       1168
       -
       // }

     

       1169
        
       

     

       1170
        
       #[cfg(test)]

     

       1171
        
       mod tests {

···

       149
        
               let keyspace = {

     

       150
        
                   let config = Config::new(path);

     

       151
        
       

     

       152
       +
                   // #[cfg(not(test))]

     

       153
       +
                   // let config = config.fsync_ms(Some(4_000));

     

       154
        
       

     

       155
        
                   config.open()?

     

       156
        
               };

     
···

       187
        
                           )?;

     

       188
        
                       } else {

     

       189
        
                           return Err(StorageError::InitError(format!(

     

       190
       +
                               "stored js_endpoint {stored:?} differs from provided {endpoint:?}, refusing to start without --jetstream-force.")));

     

       191
        
                       }

     

       192
        
                   }

     

       193
        
                   stored_secret

     
···

       360
        
                   get_snapshot_static_neu::<JetstreamCursorKey, JetstreamCursorValue>(&global)?

     

       361
        
                       .map(|c| c.to_raw_u64());

     

       362
        
       

     

       363
       +
               let rollup_cursor =

     

       364
       +
                   get_snapshot_static_neu::<NewRollupCursorKey, NewRollupCursorValue>(&global)?

     

       365
       +
                       .map(|c| c.to_raw_u64());

     

       366
       +
       

     

       367
        
               Ok(ConsumerInfo::Jetstream {

     

       368
        
                   endpoint,

     

       369
        
                   started_at,

     

       370
        
                   latest_cursor,

     

       371
       +
                   rollup_cursor,

     

       372
        
               })

     

       373
        
           }

     

       374
        
       

     
···

       806
        
       }

     

       807
        
       

     

       808
        
       impl StoreWriter<FjallBackground> for FjallWriter {

     

       809
       +
           fn background_tasks(&mut self, reroll: bool) -> StorageResult<FjallBackground> {

     

       810
        
               if self.bg_taken.swap(true, Ordering::SeqCst) {

     

       811
        
                   Err(StorageError::BackgroundAlreadyStarted)

     

       812
        
               } else {

     

       813
       +
                   if reroll {

     

       814
       +
                       insert_static_neu::<NewRollupCursorKey>(&self.global, Cursor::from_start())?;

     

       815
       +
                   }

     

       816
        
                   Ok(FjallBackground(self.clone()))

     

       817
        
               }

     

       818
        
           }

     
···

       956
        
               Ok((cursors_stepped, dirty_nsids))

     

       957
        
           }

     

       958
        
       

     

       959
       +
           fn trim_collection(

     

       960
       +
               &mut self,

     

       961
       +
               collection: &Nsid,

     

       962
       +
               limit: usize,

     

       963
       +
           ) -> StorageResult<(usize, usize)> {

     

       964
        
               let mut dangling_feed_keys_cleaned = 0;

     

       965
        
               let mut records_deleted = 0;

     

       966
        
       

     
···

       979
        
               let mut live_records_found = 0;

     

       980
        
               let mut latest_expired_feed_cursor = None;

     

       981
        
               let mut batch = self.keyspace.batch();

     

       982
       +
               for (i, kv) in self.feeds.range(live_range).rev().enumerate() {

     

       983
       +
                   if i > 1_000_000 {

     

       984
       +
                       log::info!("stopping collection trim early: already scanned 1M elements");

     

       985
       +
                       break;

     

       986
       +
                   }

     

       987
        
                   let (key_bytes, val_bytes) = kv?;

     

       988
        
                   let feed_key = db_complete::<NsidRecordFeedKey>(&key_bytes)?;

     

       989
        
                   let feed_val = db_complete::<NsidRecordFeedVal>(&val_bytes)?;

     
···

       1038
        
       

     

       1039
        
               batch.commit()?;

     

       1040
        
       

     

       1041
       +
               log::trace!("trim_collection ({collection:?}) removed {dangling_feed_keys_cleaned} dangling feed entries and {records_deleted} records");

     

       1042
       +
               Ok((dangling_feed_keys_cleaned, records_deleted))

     

       1043
        
           }

     

       1044
        
       

     

       1045
        
           fn delete_account(&mut self, did: &Did) -> Result<usize, StorageError> {

     
···

       1064
        
       

     

       1065
        
       #[async_trait]

     

       1066
        
       impl StoreBackground for FjallBackground {

     

       1067
       +
           async fn run(mut self, backfill: bool) -> StorageResult<()> {

     

       1068
        
               let mut dirty_nsids = HashSet::new();

     

       1069
        
       

     

       1070
       +
               let mut rollup =

     

       1071
       +
                   tokio::time::interval(Duration::from_millis(if backfill { 1 } else { 81 }));

     

       1072
        
               rollup.set_missed_tick_behavior(tokio::time::MissedTickBehavior::Delay);

     

       1073
        
       

     

       1074
       +
               let mut trim =

     

       1075
       +
                   tokio::time::interval(Duration::from_millis(if backfill { 3_000 } else { 6_000 }));

     

       1076
        
               trim.set_missed_tick_behavior(tokio::time::MissedTickBehavior::Skip);

     

       1077
        
       

     

       1078
        
               loop {

     
···

       1083
        
                               rollup.reset_after(Duration::from_millis(1_200)); // we're caught up, take a break

     

       1084
        
                           }

     

       1085
        
                           dirty_nsids.extend(dirty);

     

       1086
       +
                           log::trace!("rolled up {n} items ({} collections now dirty)", dirty_nsids.len());

     

       1087
        
                       },

     

       1088
        
                       _ = trim.tick() => {

     

       1089
       +
                           let n = dirty_nsids.len();

     

       1090
       +
                           log::trace!("trimming {n} nsids: {dirty_nsids:?}");

     

       1091
        
                           let t0 = Instant::now();

     

       1092
       +
                           let (mut total_danglers, mut total_deleted) = (0, 0);

     

       1093
        
                           for collection in &dirty_nsids {

     

       1094
       +
                               let (danglers, deleted) = self.0.trim_collection(collection, 512).inspect_err(|e| log::error!("trim error: {e:?}"))?;

     

       1095
       +
                               total_danglers += danglers;

     

       1096
       +
                               total_deleted += deleted;

     

       1097
       +
                               if total_deleted > 1_000_000 {

     

       1098
       +
                                   log::info!("trim stopped early, more than 1M records already deleted.");

     

       1099
       +
                                   break;

     

       1100
       +
                               }

     

       1101
        
                           }

     

       1102
       +
                           log::info!("finished trimming {n} nsids in {:?}: {total_danglers} dangling and {total_deleted} total removed.", t0.elapsed());

     

       1103
        
                           dirty_nsids.clear();

     

       1104
        
                       },

     

       1105
        
                   };

     
···

       1180
        
       }

     

       1181
        
       

     

       1182
        
       ////////// temp stuff to remove:

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       1183
        
       

     

       1184
        
       #[cfg(test)]

     

       1185
        
       mod tests {

+9 -4

ufos/src/storage_mem.rs

···

       441
        
                   get_snapshot_static_neu::<JetstreamCursorKey, JetstreamCursorValue>(&global)?

     

       442
        
                       .map(|c| c.to_raw_u64());

     

       443
        
       

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       444
        
               Ok(ConsumerInfo::Jetstream {

     

       445
        
                   endpoint,

     

       446
        
                   started_at,

     

       447
        
                   latest_cursor,

     

       0
        
       
     

       448
        
               })

     

       449
        
           }

     

       450
        
       

     
···

       797
        
       }

     

       798
        
       

     

       799
        
       impl StoreWriter<MemBackground> for MemWriter {

     

       800
       -
           fn background_tasks(&mut self) -> StorageResult<MemBackground> {

     

       801
        
               Ok(MemBackground {})

     

       802
        
           }

     

       803
        
       

     
···

       959
        
               collection: &Nsid,

     

       960
        
               limit: usize,

     

       961
        
               // TODO: could add a start cursor limit to avoid iterating deleted stuff at the start (/end)

     

       962
       -
           ) -> StorageResult<()> {

     

       963
        
               let mut dangling_feed_keys_cleaned = 0;

     

       964
        
               let mut records_deleted = 0;

     

       965
        
       

     
···

       1015
        
               batch.commit()?;

     

       1016
        
       

     

       1017
        
               log::info!("trim_collection ({collection:?}) removed {dangling_feed_keys_cleaned} dangling feed entries and {records_deleted} records");

     

       1018
       -
               Ok(())

     

       1019
        
           }

     

       1020
        
       

     

       1021
        
           fn delete_account(&mut self, did: &Did) -> Result<usize, StorageError> {

     
···

       1040
        
       

     

       1041
        
       #[async_trait]

     

       1042
        
       impl StoreBackground for MemBackground {

     

       1043
       -
           async fn run(mut self) -> StorageResult<()> {

     

       1044
        
               // noop for mem (is there a nicer way to do this?)

     

       1045
        
               loop {

     

       1046
        
                   tokio::time::sleep(std::time::Duration::from_secs_f64(10.)).await;

···

       441
        
                   get_snapshot_static_neu::<JetstreamCursorKey, JetstreamCursorValue>(&global)?

     

       442
        
                       .map(|c| c.to_raw_u64());

     

       443
        
       

     

       444
       +
               let rollup_cursor =

     

       445
       +
                   get_snapshot_static_neu::<NewRollupCursorKey, JetstreamCursorValue>(&global)?

     

       446
       +
                       .map(|c| c.to_raw_u64());

     

       447
       +
       

     

       448
        
               Ok(ConsumerInfo::Jetstream {

     

       449
        
                   endpoint,

     

       450
        
                   started_at,

     

       451
        
                   latest_cursor,

     

       452
       +
                   rollup_cursor,

     

       453
        
               })

     

       454
        
           }

     

       455
        
       

     
···

       802
        
       }

     

       803
        
       

     

       804
        
       impl StoreWriter<MemBackground> for MemWriter {

     

       805
       +
           fn background_tasks(&mut self, _reroll: bool) -> StorageResult<MemBackground> {

     

       806
        
               Ok(MemBackground {})

     

       807
        
           }

     

       808
        
       

     
···

       964
        
               collection: &Nsid,

     

       965
        
               limit: usize,

     

       966
        
               // TODO: could add a start cursor limit to avoid iterating deleted stuff at the start (/end)

     

       967
       +
           ) -> StorageResult<(usize, usize)> {

     

       968
        
               let mut dangling_feed_keys_cleaned = 0;

     

       969
        
               let mut records_deleted = 0;

     

       970
        
       

     
···

       1020
        
               batch.commit()?;

     

       1021
        
       

     

       1022
        
               log::info!("trim_collection ({collection:?}) removed {dangling_feed_keys_cleaned} dangling feed entries and {records_deleted} records");

     

       1023
       +
               Ok((dangling_feed_keys_cleaned, records_deleted))

     

       1024
        
           }

     

       1025
        
       

     

       1026
        
           fn delete_account(&mut self, did: &Did) -> Result<usize, StorageError> {

     
···

       1045
        
       

     

       1046
        
       #[async_trait]

     

       1047
        
       impl StoreBackground for MemBackground {

     

       1048
       +
           async fn run(mut self, _backfill: bool) -> StorageResult<()> {

     

       1049
        
               // noop for mem (is there a nicer way to do this?)

     

       1050
        
               loop {

     

       1051
        
                   tokio::time::sleep(std::time::Duration::from_secs_f64(10.)).await;