signer-daemon 0.3.1

Signer daemon package.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

use crate::{
  SignerDaemonCore, SignerDaemonState, entity::server,
  signer_remote::SignerRemote,
};
use sea_orm::prelude::*;
use std::collections::HashSet;

pub struct RemoteController {
  core: SignerDaemonCore,
}

impl RemoteController {
  pub fn new(core: SignerDaemonCore) -> Self {
    Self { core }
  }

  pub async fn reconcile(
    &self,
    state: &mut SignerDaemonState,
  ) -> anyhow::Result<()> {
    let mut remove_set = state
      .remotes
      .keys()
      .map(|i| i.to_string())
      .collect::<HashSet<String>>();

    let servers = server::Entity::find()
      .filter(server::Column::Enable.eq(true))
      .all(&self.core.db)
      .await?;
    for server in servers {
      if remove_set.remove(&server.addr.to_string()) {
        continue;
      }

      let mut remote = SignerRemote::new(&server.addr);
      remote
        .open_eventsource(state.daemon.as_ref().unwrap().clone())
        .await?;
      state.remotes.insert(server.addr.to_string(), remote);
    }

    for remove in remove_set {
      if let Some(mut rm) = state.remotes.remove(&remove) {
        rm.close().await;
      }
    }

    Ok(())
  }
}