That way there is a unified system for how to set values, which may be overridable by configuration variables or not. With this changes, the explicit application of environment variables for setting the cache isn't required anymore as everything happens using git-configuration, and automatically, while providing full control like before.
Showing the changes to git-repository/src (7 of 12 files). Show all
Unified · Split
| @@ -32,6 +32,7 @@ impl Cache { · 31 unmodified lines | ||
| 32 | 32 | ssh_prefix: _, |
| 33 | 33 | http_transport, |
| 34 | 34 | identity, |
| 35 | + gitoxide_prefix, | |
| 35 | 36 | }: repository::permissions::Environment, |
| 36 | 37 | repository::permissions::Config { |
| 37 | 38 | git_binary: use_installation, |
| @@ -133,7 +134,7 @@ impl Cache { · 95 unmodified lines | ||
| 133 | 134 | source: git_config::Source::Api, |
| 134 | 135 | })?; |
| 135 | 136 | } |
| 136 | − apply_environment_overrides(&mut globals, *git_prefix, http_transport, identity)?; | |
| 137 | + apply_environment_overrides(&mut globals, *git_prefix, http_transport, identity, gitoxide_prefix)?; | |
| 137 | 138 | globals |
| 138 | 139 | }; |
| 139 | 140 | |
| @@ -144,12 +145,16 @@ impl Cache { · 4 unmodified lines | ||
| 144 | 145 | let ignore_case = config_bool(&config, "core.ignoreCase", false, lenient_config)?; |
| 145 | 146 | let use_multi_pack_index = config_bool(&config, "core.multiPackIndex", true, lenient_config)?; |
| 146 | 147 | let object_kind_hint = util::disambiguate_hint(&config); |
| 148 | + let (pack_cache_bytes, object_cache_bytes) = | |
| 149 | + util::parse_object_caches(&config, lenient_config, filter_config_section)?; | |
| 147 | 150 | // NOTE: When adding a new initial cache, consider adjusting `reread_values_and_clear_caches()` as well. |
| 148 | 151 | Ok(Cache { |
| 149 | 152 | resolved: config.into(), |
| 150 | 153 | use_multi_pack_index, |
| 151 | 154 | object_hash, |
| 152 | 155 | object_kind_hint, |
| 156 | + pack_cache_bytes, | |
| 157 | + object_cache_bytes, | |
| 153 | 158 | reflog, |
| 154 | 159 | is_bare, |
| 155 | 160 | ignore_case, |
| @@ -203,6 +208,8 @@ impl Cache { · 47 unmodified lines | ||
| 203 | 208 | self.personas = Default::default(); |
| 204 | 209 | self.url_rewrite = Default::default(); |
| 205 | 210 | self.diff_algorithm = Default::default(); |
| 211 | + (self.pack_cache_bytes, self.object_cache_bytes) = | |
| 212 | + util::parse_object_caches(config, self.lenient_config, self.filter_config_section)?; | |
| 206 | 213 | #[cfg(any(feature = "blocking-network-client", feature = "async-network-client"))] |
| 207 | 214 | { |
| 208 | 215 | self.url_scheme = Default::default(); |
| @@ -244,6 +251,7 @@ fn apply_environment_overrides( · 35 unmodified lines | ||
| 244 | 251 | git_prefix: Permission, |
| 245 | 252 | http_transport: Permission, |
| 246 | 253 | identity: Permission, |
| 254 | + gitoxide_prefix: Permission, | |
| 247 | 255 | ) -> Result<(), Error> { |
| 248 | 256 | fn var_as_bstring(var: &str, perm: Permission) -> Option<BString> { |
| 249 | 257 | perm.check_opt(var) |
| @@ -418,11 +426,33 @@ fn apply_environment_overrides( · 168 unmodified lines | ||
| 418 | 426 | .new_section("gitoxide", Some(Cow::Borrowed("objects".into()))) |
| 419 | 427 | .expect("statically known valid section name"); |
| 420 | 428 | |
| 421 | − for (var, key) in [ | |
| 422 | − ("GIT_NO_REPLACE_OBJECTS", "noReplace"), | |
| 423 | − ("GIT_REPLACE_REF_BASE", "replaceRefBase"), | |
| 429 | + for (var, key, permission) in [ | |
| 430 | + ("GIT_NO_REPLACE_OBJECTS", "noReplace", git_prefix), | |
| 431 | + ("GIT_REPLACE_REF_BASE", "replaceRefBase", git_prefix), | |
| 432 | + ("GITOXIDE_OBJECT_CACHE_MEMORY", "cacheLimit", gitoxide_prefix), | |
| 424 | 433 | ] { |
| 425 | − if let Some(value) = var_as_bstring(var, git_prefix) { | |
| 434 | + if let Some(value) = var_as_bstring(var, permission) { | |
| 435 | + section.push_with_comment( | |
| 436 | + key.try_into().expect("statically known to be valid"), | |
| 437 | + Some(value.as_ref()), | |
| 438 | + format!("from {var}").as_str(), | |
| 439 | + ); | |
| 440 | + } | |
| 441 | + } | |
| 442 | + | |
| 443 | + if section.num_values() == 0 { | |
| 444 | + let id = section.id(); | |
| 445 | + env_override.remove_section_by_id(id); | |
| 446 | + } | |
| 447 | + } | |
| 448 | + | |
| 449 | + { | |
| 450 | + let mut section = env_override | |
| 451 | + .new_section("core", None) | |
| 452 | + .expect("statically known valid section name"); | |
| 453 | + | |
| 454 | + for (var, key) in [("GITOXIDE_PACK_CACHE_MEMORY", "deltaBaseCacheLimit")] { | |
| 455 | + if let Some(value) = var_as_bstring(var, gitoxide_prefix) { | |
| 426 | 456 | section.push_with_comment( |
| 427 | 457 | key.try_into().expect("statically known to be valid"), |
| 428 | 458 | Some(value.as_ref()), |
| @@ -104,6 +104,31 @@ pub(crate) fn reflog_or_default( · 103 unmodified lines | ||
| 104 | 104 | }) |
| 105 | 105 | } |
| 106 | 106 | |
| 107 | +/// Return `(pack_cache_bytes, object_cache_bytes)` as parsed from git-config | |
| 108 | +pub(crate) fn parse_object_caches( | |
| 109 | + config: &git_config::File<'static>, | |
| 110 | + lenient: bool, | |
| 111 | + mut filter_config_section: fn(&git_config::file::Metadata) -> bool, | |
| 112 | +) -> Result<(Option<usize>, usize), Error> { | |
| 113 | + let key = "core.deltaBaseCacheLimit"; | |
| 114 | + let pack_cache_bytes = config | |
| 115 | + .integer_filter_by_key(key, &mut filter_config_section) | |
| 116 | + .transpose() | |
| 117 | + .with_leniency(lenient) | |
| 118 | + .map_err(|err| Error::Value { source: err, key })?; | |
| 119 | + let key = "gitoxide.objects.cacheLimit"; | |
| 120 | + let object_cache_bytes = config | |
| 121 | + .integer_filter_by_key(key, &mut filter_config_section) | |
| 122 | + .transpose() | |
| 123 | + .with_leniency(lenient) | |
| 124 | + .map_err(|err| Error::Value { source: err, key })? | |
| 125 | + .unwrap_or_default(); | |
| 126 | + Ok(( | |
| 127 | + pack_cache_bytes.and_then(|v| v.try_into().ok()), | |
| 128 | + object_cache_bytes.try_into().unwrap_or_default(), | |
| 129 | + )) | |
| 130 | +} | |
| 131 | + | |
| 107 | 132 | pub(crate) fn parse_core_abbrev( |
| 108 | 133 | config: &git_config::File<'static>, |
| 109 | 134 | object_hash: git_hash::Kind, |
| @@ -195,6 +195,11 @@ pub(crate) struct Cache { · 194 unmodified lines | ||
| 195 | 195 | pub(crate) url_scheme: OnceCell<remote::url::SchemePermission>, |
| 196 | 196 | /// The algorithm to use when diffing blobs |
| 197 | 197 | pub(crate) diff_algorithm: OnceCell<git_diff::blob::Algorithm>, |
| 198 | + /// The amount of bytes to use for a memory backed delta pack cache. If `Some(0)`, no cache is used, if `None` | |
| 199 | + /// a standard cache is used which costs near to nothing and always pays for itself. | |
| 200 | + pub(crate) pack_cache_bytes: Option<usize>, | |
| 201 | + /// The amount of bytes to use for caching whole objects, or 0 to turn it off entirely. | |
| 202 | + pub(crate) object_cache_bytes: usize, | |
| 198 | 203 | /// The config section filter from the options used to initialize this instance. Keep these in sync! |
| 199 | 204 | filter_config_section: fn(&git_config::file::Metadata) -> bool, |
| 200 | 205 | /// The object kind to pick if a prefix is ambiguous. |
| @@ -34,9 +34,6 @@ · 33 unmodified lines | ||
| 34 | 34 | //! Use the `cache-efficiency-debug` cargo feature to learn how efficient the cache actually is - it's easy to end up with lowered |
| 35 | 35 | //! performance if the cache is not hit in 50% of the time. |
| 36 | 36 | //! |
| 37 | −//! Environment variables can also be used for configuration if the application is calling | |
| 38 | −//! [`apply_environment()`][crate::Repository::apply_environment()]. | |
| 39 | −//! | |
| 40 | 37 | //! ### Shortcomings & Limitations |
| 41 | 38 | //! |
| 42 | 39 | //! - Only a single `crate::object` or derivatives can be held in memory at a time, _per `Easy*`_. |
| @@ -26,67 +26,4 @@ impl crate::Repository { · 25 unmodified lines | ||
| 26 | 26 | self.object_cache_size(bytes) |
| 27 | 27 | } |
| 28 | 28 | } |
| 29 | − | |
| 30 | − /// Read well-known environment variables related to caches and apply them to this instance, but not to clones of it - each | |
| 31 | − /// needs their own configuration. | |
| 32 | − /// | |
| 33 | − /// Note that environment configuration never fails due to invalid environment values, but it should be used with caution as it | |
| 34 | − /// could be used to cause high memory consumption. | |
| 35 | − /// | |
| 36 | − /// Use the `GITOXIDE_DISABLE_PACK_CACHE` environment variable to turn off any pack cache, which can be beneficial when it's known that | |
| 37 | − /// the cache efficiency is low. Use `GITOXIDE_PACK_CACHE_MEMORY=512MB` to use up to 512MB of RAM for the pack delta base | |
| 38 | − /// cache. If none of these are set, the default cache is fast enough to nearly never cause a (marginal) slow-down while providing | |
| 39 | − /// some gains most of the time. Note that the value given is _per-thread_. | |
| 40 | − /// | |
| 41 | − /// Use the `GITOXIDE_OBJECT_CACHE_MEMORY=16mb` to set the given amount of memory to store full objects, on a per-thread basis. | |
| 42 | − pub fn apply_environment(self) -> Self { | |
| 43 | − // We have no cache types available without this flag currently. Maybe this should change at some point. | |
| 44 | − #[cfg(not(feature = "max-performance-safe"))] | |
| 45 | − return self; | |
| 46 | − #[cfg(feature = "max-performance-safe")] | |
| 47 | − { | |
| 48 | − let pack_cache_disabled = std::env::var_os("GITOXIDE_DISABLE_PACK_CACHE").is_some(); | |
| 49 | − let mut this = self; | |
| 50 | − if !pack_cache_disabled { | |
| 51 | − let bytes = parse_bytes_from_var("GITOXIDE_PACK_CACHE_MEMORY"); | |
| 52 | − let new_pack_cache = move || -> Box<git_odb::cache::PackCache> { | |
| 53 | − match bytes { | |
| 54 | − Some(bytes) => Box::new(git_pack::cache::lru::MemoryCappedHashmap::new(bytes)), | |
| 55 | − None => Box::new(git_pack::cache::lru::StaticLinkedList::<64>::default()), | |
| 56 | − } | |
| 57 | − }; | |
| 58 | − this.objects.set_pack_cache(new_pack_cache); | |
| 59 | − } else { | |
| 60 | − this.objects.unset_pack_cache(); | |
| 61 | − } | |
| 62 | − | |
| 63 | − if let Some(bytes) = parse_bytes_from_var("GITOXIDE_OBJECT_CACHE_MEMORY") { | |
| 64 | − this.objects | |
| 65 | − .set_object_cache(move || Box::new(git_pack::cache::object::MemoryCappedHashmap::new(bytes))); | |
| 66 | − } | |
| 67 | − this | |
| 68 | − } | |
| 69 | − } | |
| 70 | −} | |
| 71 | − | |
| 72 | −#[cfg(feature = "max-performance-safe")] | |
| 73 | −fn parse_bytes_from_var(name: &str) -> Option<usize> { | |
| 74 | − std::env::var(name) | |
| 75 | − .ok() | |
| 76 | − .and_then(|v| { | |
| 77 | − byte_unit::Byte::from_str(&v) | |
| 78 | − .map_err(|err| log::warn!("Failed to parse {:?} into byte unit for pack cache: {}", v, err)) | |
| 79 | − .ok() | |
| 80 | − }) | |
| 81 | − .and_then(|unit| { | |
| 82 | − unit.get_bytes() | |
| 83 | − .try_into() | |
| 84 | − .map_err(|err| { | |
| 85 | − log::warn!( | |
| 86 | − "Parsed bytes value is not representable as usize. Defaulting to standard pack cache: {}", | |
| 87 | − err | |
| 88 | − ) | |
| 89 | − }) | |
| 90 | − .ok() | |
| 91 | − }) | |
| 92 | 29 | } |
| @@ -10,20 +10,12 @@ impl crate::Repository { · 9 unmodified lines | ||
| 10 | 10 | linked_worktree_options: crate::open::Options, |
| 11 | 11 | index: crate::worktree::IndexStorage, |
| 12 | 12 | ) -> Self { |
| 13 | + let objects = setup_objects(objects, &config); | |
| 13 | 14 | crate::Repository { |
| 14 | 15 | bufs: RefCell::new(Vec::with_capacity(4)), |
| 15 | 16 | work_tree, |
| 16 | 17 | common_dir, |
| 17 | − objects: { | |
| 18 | − #[cfg(feature = "max-performance-safe")] | |
| 19 | − { | |
| 20 | − objects.with_pack_cache(|| Box::new(git_pack::cache::lru::StaticLinkedList::<64>::default())) | |
| 21 | − } | |
| 22 | − #[cfg(not(feature = "max-performance-safe"))] | |
| 23 | − { | |
| 24 | − objects | |
| 25 | − } | |
| 26 | − }, | |
| 18 | + objects, | |
| 27 | 19 | refs, |
| 28 | 20 | config, |
| 29 | 21 | options: linked_worktree_options, |
| @@ -36,3 +28,28 @@ impl crate::Repository { · 6 unmodified lines | ||
| 36 | 28 | self.into() |
| 37 | 29 | } |
| 38 | 30 | } |
| 31 | + | |
| 32 | +#[cfg_attr(not(feature = "max-performance-safe"), allow(unused_variables, unused_mut))] | |
| 33 | +fn setup_objects(mut objects: crate::OdbHandle, config: &crate::config::Cache) -> crate::OdbHandle { | |
| 34 | + #[cfg(feature = "max-performance-safe")] | |
| 35 | + { | |
| 36 | + match config.pack_cache_bytes { | |
| 37 | + None => objects.set_pack_cache(|| Box::new(git_pack::cache::lru::StaticLinkedList::<64>::default())), | |
| 38 | + Some(0) => objects.unset_pack_cache(), | |
| 39 | + Some(bytes) => objects.set_pack_cache(move || -> Box<git_odb::cache::PackCache> { | |
| 40 | + Box::new(git_pack::cache::lru::MemoryCappedHashmap::new(bytes)) | |
| 41 | + }), | |
| 42 | + }; | |
| 43 | + if config.object_cache_bytes == 0 { | |
| 44 | + objects.unset_object_cache(); | |
| 45 | + } else { | |
| 46 | + let bytes = config.object_cache_bytes; | |
| 47 | + objects.set_object_cache(move || Box::new(git_pack::cache::object::MemoryCappedHashmap::new(bytes))); | |
| 48 | + } | |
| 49 | + objects | |
| 50 | + } | |
| 51 | + #[cfg(not(feature = "max-performance-safe"))] | |
| 52 | + { | |
| 53 | + objects | |
| 54 | + } | |
| 55 | +} | |
| @@ -82,6 +82,8 @@ pub struct Environment { · 81 unmodified lines | ||
| 82 | 82 | /// Note that identity related environment variables prefixed with `GIT_` are falling under the |
| 83 | 83 | /// `git_prefix` permission, like `GIT_AUTHOR_NAME`. |
| 84 | 84 | pub identity: git_sec::Permission, |
| 85 | + /// Decide if `gitoxide` specific variables may be read, prefixed with `GITOXIDE_`. | |
| 86 | + pub gitoxide_prefix: git_sec::Permission, | |
| 85 | 87 | } |
| 86 | 88 | |
| 87 | 89 | impl Environment { |
| @@ -94,6 +96,7 @@ impl Environment { · 6 unmodified lines | ||
| 94 | 96 | ssh_prefix: git_sec::Permission::Allow, |
| 95 | 97 | http_transport: git_sec::Permission::Allow, |
| 96 | 98 | identity: git_sec::Permission::Allow, |
| 99 | + gitoxide_prefix: git_sec::Permission::Allow, | |
| 97 | 100 | } |
| 98 | 101 | } |
| 99 | 102 | } |
| @@ -140,6 +143,7 @@ impl Permissions { · 40 unmodified lines | ||
| 140 | 143 | git_prefix: deny, |
| 141 | 144 | http_transport: deny, |
| 142 | 145 | identity: deny, |
| 146 | + gitoxide_prefix: deny, | |
| 143 | 147 | } |
| 144 | 148 | }, |
| 145 | 149 | } |