Start9/start-os
2
0
Fork 0
mirror of https://github.com/Start9Labs/start-os.git synced 2026-03-26 02:11:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

prefer btreemap over hashmap

Browse Source
This commit is contained in:
Aiden McClelland
2021-09-28 14:00:45 -06:00
committed by Aiden McClelland
parent f8472135f5
commit 7622616856
16 changed files with 103 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
appmgr/src/bin/embassyd.rs
View File

@@ -1,4 +1,4 @@
use std::collections::HashMap;
use std::collections::BTreeMap;
use std::time::Duration;
use anyhow::anyhow;
@@ -37,7 +37,7 @@ fn err_to_500(e: Error) -> Response<Body> {
async fn inner_main(
cfg_path: Option<&str>,
log_level: LevelFilter,
module_logging: HashMap<String, LevelFilter>,
module_logging: BTreeMap<String, LevelFilter>,
) -> Result<Option<Shutdown>, Error> {
let rpc_ctx = RpcContext::init(cfg_path, log_level, module_logging).await?;
let mut shutdown_recv = rpc_ctx.shutdown.subscribe();
@@ -282,7 +282,7 @@ fn main() {
.expect(&format!("Invalid 'log-module' argument: {}", l)),
)
})
.collect::<HashMap<_, LevelFilter>>();
.collect::<BTreeMap<_, LevelFilter>>();
let cfg_path = matches.value_of("config");
let res = {

80
appmgr/src/config/spec.rs
View File

@@ -1,5 +1,5 @@
use std::borrow::{Borrow, Cow};
use std::collections::{BTreeMap, HashSet};
use std::collections::{BTreeMap, BTreeSet};
use std::fmt;
use std::fmt::Debug;
use std::hash::{Hash, Hasher};
@@ -46,7 +46,7 @@ pub trait ValueSpec {
value: &mut Value,
) -> Result<(), ConfigurationError>;
// returns all pointers that are live in the provided config
fn pointers(&self, value: &Value) -> Result<HashSet<ValueSpecPointer>, NoMatchWithPath>;
fn pointers(&self, value: &Value) -> Result<BTreeSet<ValueSpecPointer>, NoMatchWithPath>;
// requires returns whether the app id is the target of a pointer within it
fn requires(&self, id: &PackageId, value: &Value) -> bool;
// defines if 2 values of this type are equal for the purpose of uniqueness
@@ -165,7 +165,7 @@ where
.update(ctx, db, manifest, config_overrides, value)
.await
}
fn pointers(&self, value: &Value) -> Result<HashSet<ValueSpecPointer>, NoMatchWithPath> {
fn pointers(&self, value: &Value) -> Result<BTreeSet<ValueSpecPointer>, NoMatchWithPath> {
self.inner.pointers(value)
}
fn requires(&self, id: &PackageId, value: &Value) -> bool {
@@ -209,7 +209,7 @@ where
.update(ctx, db, manifest, config_overrides, value)
.await
}
fn pointers(&self, value: &Value) -> Result<HashSet<ValueSpecPointer>, NoMatchWithPath> {
fn pointers(&self, value: &Value) -> Result<BTreeSet<ValueSpecPointer>, NoMatchWithPath> {
self.inner.pointers(value)
}
fn requires(&self, id: &PackageId, value: &Value) -> bool {
@@ -286,7 +286,7 @@ where
.update(ctx, db, manifest, config_overrides, value)
.await
}
fn pointers(&self, value: &Value) -> Result<HashSet<ValueSpecPointer>, NoMatchWithPath> {
fn pointers(&self, value: &Value) -> Result<BTreeSet<ValueSpecPointer>, NoMatchWithPath> {
self.inner.pointers(value)
}
fn requires(&self, id: &PackageId, value: &Value) -> bool {
@@ -407,7 +407,7 @@ impl ValueSpec for ValueSpecAny {
ValueSpecAny::Pointer(a) => a.update(ctx, db, manifest, config_overrides, value).await,
}
}
fn pointers(&self, value: &Value) -> Result<HashSet<ValueSpecPointer>, NoMatchWithPath> {
fn pointers(&self, value: &Value) -> Result<BTreeSet<ValueSpecPointer>, NoMatchWithPath> {
match self {
ValueSpecAny::Boolean(a) => a.pointers(value),
ValueSpecAny::Enum(a) => a.pointers(value),
@@ -492,8 +492,8 @@ impl ValueSpec for ValueSpecBoolean {
) -> Result<(), ConfigurationError> {
Ok(())
}
fn pointers(&self, _value: &Value) -> Result<HashSet<ValueSpecPointer>, NoMatchWithPath> {
Ok(HashSet::new())
fn pointers(&self, _value: &Value) -> Result<BTreeSet<ValueSpecPointer>, NoMatchWithPath> {
Ok(BTreeSet::new())
}
fn requires(&self, _id: &PackageId, _value: &Value) -> bool {
false
@@ -581,8 +581,8 @@ impl ValueSpec for ValueSpecEnum {
) -> Result<(), ConfigurationError> {
Ok(())
}
fn pointers(&self, _value: &Value) -> Result<HashSet<ValueSpecPointer>, NoMatchWithPath> {
Ok(HashSet::new())
fn pointers(&self, _value: &Value) -> Result<BTreeSet<ValueSpecPointer>, NoMatchWithPath> {
Ok(BTreeSet::new())
}
fn requires(&self, _id: &PackageId, _value: &Value) -> bool {
false
@@ -685,8 +685,8 @@ where
)))
}
}
fn pointers(&self, _value: &Value) -> Result<HashSet<ValueSpecPointer>, NoMatchWithPath> {
Ok(HashSet::new())
fn pointers(&self, _value: &Value) -> Result<BTreeSet<ValueSpecPointer>, NoMatchWithPath> {
Ok(BTreeSet::new())
}
fn requires(&self, id: &PackageId, value: &Value) -> bool {
if let Value::Array(ref ls) = value {
@@ -780,7 +780,7 @@ impl ValueSpec for ValueSpecList {
ValueSpecList::Union(a) => a.update(ctx, db, manifest, config_overrides, value).await,
}
}
fn pointers(&self, value: &Value) -> Result<HashSet<ValueSpecPointer>, NoMatchWithPath> {
fn pointers(&self, value: &Value) -> Result<BTreeSet<ValueSpecPointer>, NoMatchWithPath> {
match self {
ValueSpecList::Enum(a) => a.pointers(value),
ValueSpecList::Number(a) => a.pointers(value),
@@ -898,8 +898,8 @@ impl ValueSpec for ValueSpecNumber {
) -> Result<(), ConfigurationError> {
Ok(())
}
fn pointers(&self, _value: &Value) -> Result<HashSet<ValueSpecPointer>, NoMatchWithPath> {
Ok(HashSet::new())
fn pointers(&self, _value: &Value) -> Result<BTreeSet<ValueSpecPointer>, NoMatchWithPath> {
Ok(BTreeSet::new())
}
fn requires(&self, _id: &PackageId, _value: &Value) -> bool {
false
@@ -969,7 +969,7 @@ impl ValueSpec for ValueSpecObject {
)))
}
}
fn pointers(&self, value: &Value) -> Result<HashSet<ValueSpecPointer>, NoMatchWithPath> {
fn pointers(&self, value: &Value) -> Result<BTreeSet<ValueSpecPointer>, NoMatchWithPath> {
if let Value::Object(o) = value {
self.spec.pointers(o)
} else {
@@ -1080,17 +1080,17 @@ impl ConfigSpec {
Ok(())
}
pub fn pointers(&self, cfg: &Config) -> Result<HashSet<ValueSpecPointer>, NoMatchWithPath> {
pub fn pointers(&self, cfg: &Config) -> Result<BTreeSet<ValueSpecPointer>, NoMatchWithPath> {
cfg.iter()
.filter_map(|(k, v)| self.0.get(k).map(|vs| (k, vs.pointers(v))))
.fold(Ok(HashSet::<ValueSpecPointer>::new()), |acc, v| {
.fold(Ok(BTreeSet::<ValueSpecPointer>::new()), |acc, v| {
match (acc, v) {
// propagate existing errors
(Err(e), _) => Err(e),
// create new error case
(Ok(_), (k, Err(e))) => Err(e.prepend(k.clone())),
// combine sets
(Ok(s0), (_, Ok(s1))) => Ok(HashSet::from_iter(s0.union(&s1).cloned())),
(Ok(s0), (_, Ok(s1))) => Ok(BTreeSet::from_iter(s0.union(&s1).cloned())),
}
})
}
@@ -1157,8 +1157,8 @@ impl ValueSpec for ValueSpecString {
) -> Result<(), ConfigurationError> {
Ok(())
}
fn pointers(&self, _value: &Value) -> Result<HashSet<ValueSpecPointer>, NoMatchWithPath> {
Ok(HashSet::new())
fn pointers(&self, _value: &Value) -> Result<BTreeSet<ValueSpecPointer>, NoMatchWithPath> {
Ok(BTreeSet::new())
}
fn requires(&self, _id: &PackageId, _value: &Value) -> bool {
false
@@ -1388,7 +1388,7 @@ impl ValueSpec for ValueSpecUnion {
)))
}
}
fn pointers(&self, value: &Value) -> Result<HashSet<ValueSpecPointer>, NoMatchWithPath> {
fn pointers(&self, value: &Value) -> Result<BTreeSet<ValueSpecPointer>, NoMatchWithPath> {
if let Value::Object(o) = value {
match o.get(&self.tag.id) {
None => Err(NoMatchWithPath::new(MatchError::MissingTag(
@@ -1461,7 +1461,7 @@ impl DefaultableWith for ValueSpecUnion {
}
}
#[derive(Clone, Debug, PartialEq, Eq, Hash, Serialize, Deserialize)]
#[derive(Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
#[serde(tag = "subtype")]
#[serde(rename_all = "kebab-case")]
pub enum ValueSpecPointer {
@@ -1517,8 +1517,8 @@ impl ValueSpec for ValueSpecPointer {
}
}
}
fn pointers(&self, _value: &Value) -> Result<HashSet<ValueSpecPointer>, NoMatchWithPath> {
let mut pointers = HashSet::new();
fn pointers(&self, _value: &Value) -> Result<BTreeSet<ValueSpecPointer>, NoMatchWithPath> {
let mut pointers = BTreeSet::new();
pointers.insert(self.clone());
Ok(pointers)
}
@@ -1533,7 +1533,7 @@ impl ValueSpec for ValueSpecPointer {
}
}
#[derive(Clone, Debug, PartialEq, Eq, Hash, Serialize, Deserialize)]
#[derive(Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
#[serde(tag = "target")]
#[serde(rename_all = "kebab-case")]
pub enum PackagePointerSpec {
@@ -1614,8 +1614,8 @@ impl ValueSpec for PackagePointerSpec {
*value = self.deref(ctx, db, manifest, config_overrides).await?;
Ok(())
}
fn pointers(&self, _value: &Value) -> Result<HashSet<ValueSpecPointer>, NoMatchWithPath> {
let mut pointers = HashSet::new();
fn pointers(&self, _value: &Value) -> Result<BTreeSet<ValueSpecPointer>, NoMatchWithPath> {
let mut pointers = BTreeSet::new();
pointers.insert(ValueSpecPointer::Package(self.clone()));
Ok(pointers)
}
@@ -1627,7 +1627,7 @@ impl ValueSpec for PackagePointerSpec {
}
}
#[derive(Clone, Debug, PartialEq, Eq, Hash, Serialize, Deserialize)]
#[derive(Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
#[serde(rename_all = "kebab-case")]
pub struct TorAddressPointer {
package_id: PackageId,
@@ -1658,7 +1658,7 @@ impl fmt::Display for TorAddressPointer {
}
}
#[derive(Clone, Debug, PartialEq, Eq, Hash, Serialize, Deserialize)]
#[derive(Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
#[serde(rename_all = "kebab-case")]
pub struct LanAddressPointer {
package_id: PackageId,
@@ -1688,7 +1688,7 @@ impl LanAddressPointer {
Ok(addr.to_owned().map(Value::String).unwrap_or(Value::Null))
}
}
#[derive(Clone, Debug, PartialEq, Eq, Hash, Serialize, Deserialize)]
#[derive(Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
#[serde(rename_all = "kebab-case")]
pub struct ConfigPointer {
package_id: PackageId,
@@ -1805,13 +1805,23 @@ impl PartialEq for ConfigSelector {
}
}
impl Eq for ConfigSelector {}
impl PartialOrd for ConfigSelector {
fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
self.src.partial_cmp(&other.src)
}
}
impl Ord for ConfigSelector {
fn cmp(&self, other: &Self) -> std::cmp::Ordering {
self.src.cmp(&other.src)
}
}
impl Hash for ConfigSelector {
fn hash<H: Hasher>(&self, state: &mut H) {
self.src.hash(state)
}
}
#[derive(Clone, Debug, PartialEq, Eq, Hash, Serialize, Deserialize)]
#[derive(Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
#[serde(rename_all = "kebab-case")]
#[serde(tag = "target")]
pub struct TorKeyPointer {
@@ -1849,7 +1859,7 @@ impl fmt::Display for TorKeyPointer {
}
}
#[derive(Clone, Debug, PartialEq, Eq, Hash, Serialize, Deserialize)]
#[derive(Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
#[serde(rename_all = "kebab-case")]
#[serde(tag = "target")]
pub enum SystemPointerSpec {}
@@ -1892,8 +1902,8 @@ impl ValueSpec for SystemPointerSpec {
*value = self.deref(db).await?;
Ok(())
}
fn pointers(&self, value: &Value) -> Result<HashSet<ValueSpecPointer>, NoMatchWithPath> {
let mut pointers = HashSet::new();
fn pointers(&self, value: &Value) -> Result<BTreeSet<ValueSpecPointer>, NoMatchWithPath> {
let mut pointers = BTreeSet::new();
pointers.insert(ValueSpecPointer::System(self.clone()));
Ok(pointers)
}

4
appmgr/src/context/rpc.rs
View File

@@ -1,5 +1,5 @@
use std::borrow::Cow;
use std::collections::{HashMap, VecDeque};
use std::collections::{BTreeMap, VecDeque};
use std::net::{IpAddr, SocketAddr};
use std::ops::Deref;
use std::path::{Path, PathBuf};
@@ -130,7 +130,7 @@ impl RpcContext {
pub async fn init<P: AsRef<Path>>(
cfg_path: Option<P>,
log_level: LevelFilter,
module_logging: HashMap<String, LevelFilter>,
module_logging: BTreeMap<String, LevelFilter>,
) -> Result<Self, Error> {
let base = RpcContextConfig::load(cfg_path).await?;
let log_epoch = Arc::new(AtomicU64::new(rand::random()));

2
appmgr/src/install/cleanup.rs
View File

@@ -1,4 +1,4 @@
use std::collections::HashMap;
use std::collections::{BTreeMap, HashMap};
use std::path::Path;
use anyhow::anyhow;

4
appmgr/src/install/mod.rs
View File

@@ -1,4 +1,4 @@
use std::collections::{BTreeMap, HashSet};
use std::collections::{BTreeMap, BTreeSet};
use std::io::SeekFrom;
use std::path::Path;
use std::process::Stdio;
@@ -606,7 +606,7 @@ pub async fn install_s9pk<R: AsyncRead + AsyncSeek + Unpin>(
current_dependents
.keys()
.chain(prev.current_dependents.keys())
.collect::<HashSet<_>>(),
.collect::<BTreeSet<_>>(),
)
.await?;
let mut configured = prev.status.configured;

12
appmgr/src/manager/mod.rs
View File

@@ -1,4 +1,4 @@
use std::collections::HashMap;
use std::collections::BTreeMap;
use std::future::Future;
use std::sync::atomic::AtomicUsize;
use std::sync::Arc;
@@ -21,7 +21,7 @@ use crate::util::{Container, NonDetachingJoinHandle, Version};
use crate::Error;
#[derive(Default)]
pub struct ManagerMap(RwLock<HashMap<(PackageId, Version), Arc<Manager>>>);
pub struct ManagerMap(RwLock<BTreeMap<(PackageId, Version), Arc<Manager>>>);
impl ManagerMap {
pub async fn init<Db: DbHandle, Ex>(
&self,
@@ -32,7 +32,7 @@ impl ManagerMap {
where
for<'a> &'a mut Ex: Executor<'a, Database = Sqlite>,
{
let mut res = HashMap::new();
let mut res = BTreeMap::new();
for package in crate::db::DatabaseModel::new()
.package_data()
.keys(db, true)
@@ -65,7 +65,7 @@ impl ManagerMap {
&self,
ctx: RpcContext,
manifest: Manifest,
tor_keys: HashMap<InterfaceId, TorSecretKeyV3>,
tor_keys: BTreeMap<InterfaceId, TorSecretKeyV3>,
) -> Result<(), Error> {
let mut lock = self.0.write().await;
let id = (manifest.id.clone(), manifest.version.clone());
@@ -125,7 +125,7 @@ struct ManagerSharedState {
on_stop: Sender<OnStop>,
manifest: Manifest,
container_name: String,
tor_keys: HashMap<InterfaceId, TorSecretKeyV3>,
tor_keys: BTreeMap<InterfaceId, TorSecretKeyV3>,
}
#[derive(Clone, Copy)]
@@ -246,7 +246,7 @@ impl Manager {
async fn create(
ctx: RpcContext,
manifest: Manifest,
tor_keys: HashMap<InterfaceId, TorSecretKeyV3>,
tor_keys: BTreeMap<InterfaceId, TorSecretKeyV3>,
) -> Result<Self, Error> {
let (on_stop, mut recv) = channel(OnStop::Sleep);
let shared = Arc::new(ManagerSharedState {

4
appmgr/src/net/interface.rs
View File

@@ -1,4 +1,4 @@
use std::collections::{BTreeMap, HashMap};
use std::collections::BTreeMap;
use std::path::Path;
use anyhow::anyhow;
@@ -72,7 +72,7 @@ impl Interfaces {
&self,
secrets: &mut Ex,
package_id: &PackageId,
) -> Result<HashMap<InterfaceId, TorSecretKeyV3>, Error>
) -> Result<BTreeMap<InterfaceId, TorSecretKeyV3>, Error>
where
for<'a> &'a mut Ex: Executor<'a, Database = Sqlite>,
{

6
appmgr/src/net/mdns.rs
View File

@@ -1,4 +1,4 @@
use std::collections::HashMap;
use std::collections::BTreeMap;
use avahi_sys::{
self, avahi_entry_group_add_service, avahi_entry_group_commit, avahi_entry_group_free,
@@ -35,7 +35,7 @@ impl MdnsController {
pub struct MdnsControllerInner {
hostname: Vec<u8>,
entry_group: *mut AvahiEntryGroup,
services: HashMap<(PackageId, InterfaceId), TorSecretKeyV3>,
services: BTreeMap<(PackageId, InterfaceId), TorSecretKeyV3>,
}
unsafe impl Send for MdnsControllerInner {}
unsafe impl Sync for MdnsControllerInner {}
@@ -115,7 +115,7 @@ impl MdnsControllerInner {
MdnsControllerInner {
hostname: hostname_buf,
entry_group: group,
services: HashMap::new(),
services: BTreeMap::new(),
}
}
}

10
appmgr/src/net/nginx.rs
View File

@@ -1,4 +1,4 @@
use std::collections::{BTreeMap, HashMap};
use std::collections::BTreeMap;
use std::net::Ipv4Addr;
use std::path::PathBuf;
@@ -35,14 +35,14 @@ impl NginxController {
pub struct NginxControllerInner {
nginx_root: PathBuf,
interfaces: HashMap<PackageId, PackageNetInfo>,
interfaces: BTreeMap<PackageId, PackageNetInfo>,
ssl_manager: SslManager,
}
impl NginxControllerInner {
async fn init(nginx_root: PathBuf, db: SqlitePool) -> Result<Self, Error> {
Ok(NginxControllerInner {
nginx_root,
interfaces: HashMap::new(),
interfaces: BTreeMap::new(),
ssl_manager: SslManager::init(db).await?,
})
}
@@ -60,7 +60,7 @@ impl NginxControllerInner {
// also don't add nginx unless it has at least one exposed port
&& meta.lan_config.len() > 0
})
.collect::<HashMap<InterfaceId, InterfaceMetadata>>();
.collect::<BTreeMap<InterfaceId, InterfaceMetadata>>();
for (id, meta) in interface_map.iter() {
for (port, lan_port_config) in meta.lan_config.iter() {
@@ -197,7 +197,7 @@ impl NginxControllerInner {
}
struct PackageNetInfo {
ip: Ipv4Addr,
interfaces: HashMap<InterfaceId, InterfaceMetadata>,
interfaces: BTreeMap<InterfaceId, InterfaceMetadata>,
}
pub struct InterfaceMetadata {
pub dns_base: String,

8
appmgr/src/net/tor.rs
View File

@@ -1,4 +1,4 @@
use std::collections::HashMap;
use std::collections::BTreeMap;
use std::net::{Ipv4Addr, SocketAddr};
use std::time::Duration;
@@ -137,7 +137,7 @@ pub struct TorControllerInner {
embassyd_tor_key: TorSecretKeyV3,
control_addr: SocketAddr,
connection: Option<AuthenticatedConnection>,
services: HashMap<(PackageId, InterfaceId), (TorSecretKeyV3, TorConfig, Ipv4Addr)>,
services: BTreeMap<(PackageId, InterfaceId), (TorSecretKeyV3, TorConfig, Ipv4Addr)>,
}
impl TorControllerInner {
async fn add<'a, I: IntoIterator<Item = (InterfaceId, TorConfig, TorSecretKeyV3)>>(
@@ -230,7 +230,7 @@ impl TorControllerInner {
embassyd_tor_key,
control_addr: tor_control,
connection: Some(connection),
services: HashMap::new(),
services: BTreeMap::new(),
};
controller.add_embassyd_onion().await?;
Ok(controller)
@@ -314,7 +314,7 @@ impl TorControllerInner {
self.connection.replace(new_connection);
// swap empty map for owned old service map
let old_services = std::mem::replace(&mut self.services, HashMap::new());
let old_services = std::mem::replace(&mut self.services, BTreeMap::new());
// re add all of the services on the new control socket
for ((package_id, interface_id), (tor_key, tor_cfg, ipv4)) in old_services {

6
appmgr/src/net/wifi.rs
View File

@@ -1,4 +1,4 @@
use std::collections::HashMap;
use std::collections::BTreeMap;
use std::path::Path;
use std::time::Duration;
@@ -353,7 +353,7 @@ impl<'a> WpaCli<'a> {
.await?;
Ok(())
}
pub async fn list_networks_low(&self) -> Result<HashMap<String, NetworkId>, Error> {
pub async fn list_networks_low(&self) -> Result<BTreeMap<String, NetworkId>, Error> {
let r = Command::new("wpa_cli")
.arg("-i")
.arg(self.interface)
@@ -369,7 +369,7 @@ impl<'a> WpaCli<'a> {
let ssid = cs.next()?.to_owned();
Some((ssid, nid))
})
.collect::<HashMap<String, NetworkId>>())
.collect::<BTreeMap<String, NetworkId>>())
}
pub async fn select_network_low(&self, id: &NetworkId) -> Result<(), Error> {
let _ = Command::new("wpa_cli")

4
appmgr/src/notifications.rs
View File

@@ -1,4 +1,4 @@
use std::collections::HashMap;
use std::collections::BTreeMap;
use std::fmt;
use std::str::FromStr;
@@ -187,7 +187,7 @@ pub enum NotificationSubtype {
BackupReport {
server_attempted: bool,
server_error: Option<String>,
packages: HashMap<String, Option<String>>,
packages: BTreeMap<String, Option<String>>,
},
}
impl NotificationSubtype {

6
appmgr/src/s9pk/header.rs
View File

@@ -1,4 +1,4 @@
use std::collections::HashMap;
use std::collections::BTreeMap;
use std::io::Write;
use anyhow::anyhow;
@@ -100,12 +100,12 @@ impl TableOfContents {
reader.read_exact(&mut toc_len).await?;
let toc_len = u32::from_be_bytes(toc_len);
let mut reader = reader.take(toc_len as u64);
let mut table = HashMap::new();
let mut table = BTreeMap::new();
while let Some((label, section)) = FileSection::deserialize_entry(&mut reader).await? {
table.insert(label, section);
}
fn from_table(
table: &HashMap<Vec<u8>, FileSection>,
table: &BTreeMap<Vec<u8>, FileSection>,
label: &str,
) -> std::io::Result<FileSection> {
table.get(label.as_bytes()).copied().ok_or_else(|| {

8
appmgr/src/status/mod.rs
View File

@@ -1,4 +1,4 @@
use std::collections::{BTreeMap, HashMap, HashSet};
use std::collections::{BTreeMap, BTreeSet};
use std::sync::Arc;
use anyhow::anyhow;
@@ -113,7 +113,7 @@ pub async fn check_all(ctx: &RpcContext) -> Result<(), Error> {
.0
.into_iter()
.map(|x| x.0)
.collect::<HashSet<PackageId>>();
.collect::<BTreeSet<PackageId>>();
status_manifest.push((
installed.clone().status(),
Arc::new(installed.clone().manifest().get(&mut db, true).await?),
@@ -151,7 +151,7 @@ pub async fn check_all(ctx: &RpcContext) -> Result<(), Error> {
Ok(res)
}
let (status_sender, mut statuses_recv) = tokio::sync::mpsc::channel(status_manifest.len() + 1);
let mut statuses = HashMap::with_capacity(status_manifest.len());
let mut statuses = BTreeMap::new();
futures::stream::iter(
status_manifest
.into_iter()
@@ -179,7 +179,7 @@ pub async fn check_all(ctx: &RpcContext) -> Result<(), Error> {
let statuses = Arc::new(statuses);
async fn dependency_status<Db: DbHandle>(
id: &PackageId,
statuses: Arc<HashMap<PackageId, MainStatus>>,
statuses: Arc<BTreeMap<PackageId, MainStatus>>,
model: InstalledPackageDataEntryModel,
current_deps: Arc<BTreeMap<PackageId, CurrentDependencyInfo>>,
mut db: Db,

16
appmgr/src/util/logger.rs
View File

@@ -1,4 +1,4 @@
use std::collections::HashMap;
use std::collections::BTreeMap;
use std::sync::atomic::{AtomicBool, AtomicU64, Ordering};
use std::sync::Arc;
@@ -12,7 +12,7 @@ pub struct ModuleMap {
trie: SequenceTrie<String, (LevelFilter, bool)>,
}
impl ModuleMap {
fn new(vals: HashMap<String, LevelFilter>) -> Self {
fn new(vals: BTreeMap<String, LevelFilter>) -> Self {
let mut module_trie = SequenceTrie::new();
for (k, v) in vals {
let mut include_submodules = false;
@@ -97,7 +97,7 @@ impl EmbassyLogger {
log_epoch: Arc<AtomicU64>,
share_dest: Option<Url>,
share_errors: bool,
module_map: HashMap<String, LevelFilter>,
module_map: BTreeMap<String, LevelFilter>,
) -> Self {
let share_dest = match share_dest {
None => Url::parse("https://beta-registry-0-3.start9labs.com/error-logs").unwrap(), // TODO
@@ -145,7 +145,7 @@ impl log::Log for EmbassyLogger {
}
if self.sharing.load(Ordering::SeqCst) {
if record.level() <= log::Level::Warn {
let mut body = HashMap::new();
let mut body = BTreeMap::new();
body.insert(
"log-epoch",
format!("{}", self.log_epoch.load(Ordering::SeqCst)),
@@ -180,7 +180,7 @@ proptest::proptest! {
#[test]
fn submodules_handled_by_parent(s0 in "[a-z][a-z0-9_]+", s1 in "[a-z][a-z0-9_]+", level in filter_strategy()) {
proptest::prop_assume!(level > LevelFilter::Off);
let mut hm = HashMap::new();
let mut hm = BTreeMap::new();
hm.insert(format!("{}::*", s0.clone()), level);
let mod_map = ModuleMap::new(hm);
proptest::prop_assert_eq!(mod_map.level_for(&format!("{}::{}", s0, s1)), &level)
@@ -188,7 +188,7 @@ proptest::proptest! {
#[test]
fn submodules_ignored_by_parent(s0 in "[a-z][a-z0-9_]+", s1 in "[a-z][a-z0-9_]+", level in filter_strategy()) {
proptest::prop_assume!(level > LevelFilter::Off);
let mut hm = HashMap::new();
let mut hm = BTreeMap::new();
hm.insert(s0.clone(), level);
let mod_map = ModuleMap::new(hm);
proptest::prop_assert_eq!(mod_map.level_for(&format!("{}::{}", s0, s1)), &LevelFilter::Off)
@@ -196,7 +196,7 @@ proptest::proptest! {
#[test]
fn duplicate_insertion_ignored(s0 in "[a-z][a-z0-9_]+", s1 in "[a-z][a-z0-9_]+", level in filter_strategy()) {
proptest::prop_assume!(level > LevelFilter::Off);
let mut hm = HashMap::new();
let mut hm = BTreeMap::new();
hm.insert(format!("{}::*", s0.clone()), level);
let sub = format!("{}::{}", s0, s1);
hm.insert(sub.clone(), level);
@@ -205,7 +205,7 @@ proptest::proptest! {
}
#[test]
fn parent_child_simul(s0 in "[a-z][a-z0-9_]+", s1 in "[a-z][a-z0-9_]+", level0 in filter_strategy(), level1 in filter_strategy()) {
let mut hm = HashMap::new();
let mut hm = BTreeMap::new();
hm.insert(s0.clone(), level0);
let sub = format!("{}::{}", s0, s1);
hm.insert(sub.clone(), level1);

10
appmgr/src/util/mod.rs
View File

@@ -387,6 +387,16 @@ impl PartialEq for Version {
}
}
impl Eq for Version {}
impl PartialOrd for Version {
fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
self.version.partial_cmp(&other.version)
}
}
impl Ord for Version {
fn cmp(&self, other: &Self) -> std::cmp::Ordering {
self.version.cmp(&other.version)
}
}
impl Hash for Version {
fn hash<H: Hasher>(&self, state: &mut H) {
self.version.hash(state)