mirrors/jj
1
0
Fork 1
mirror of https://github.com/martinvonz/jj.git synced 2024-10-23 23:10:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

working_copy: use OnceCell to manage {operation_id, workspace_id}

Browse source 
For consistency with the tree_state handling. This isn't that simple and
concise compared to the tree_state one, so I'm fine to drop the series.

I've extracted {operation_id, workspace_id} pair so these values can be
safely initialized by OnceCell. The extracted struct is named after the
"checkout" file.
This commit is contained in:
Yuya Nishihara 2022-10-02 18:42:20 +09:00 committed by Martin von Zweigbergk
parent a2f750171a
commit c0c1eade91
1 changed files with 35 additions and 33 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

68
lib/src/working_copy.rs
View file

@ -12,7 +12,6 @@
// See the License for the specific language governing permissions and
// limitations under the License.
use std::cell::RefCell;
use std::collections::{BTreeMap, HashSet};
use std::ffi::OsString;
use std::fs;
@ -958,12 +957,18 @@ impl TreeState {
}
}
/// Working copy state stored in "checkout" file.
#[derive(Clone, Debug)]
struct CheckoutState {
operation_id: OperationId,
workspace_id: WorkspaceId,
}
pub struct WorkingCopy {
store: Arc<Store>,
working_copy_path: PathBuf,
state_path: PathBuf,
operation_id: RefCell<Option<OperationId>>,
workspace_id: RefCell<Option<WorkspaceId>>,
checkout_state: OnceCell<CheckoutState>,
tree_state: OnceCell<TreeState>,
}
@ -991,8 +996,7 @@ impl WorkingCopy {
store,
working_copy_path,
state_path,
operation_id: RefCell::new(Some(operation_id)),
workspace_id: RefCell::new(Some(workspace_id)),
checkout_state: OnceCell::new(),
tree_state: OnceCell::new(),
}
}
@ -1002,8 +1006,7 @@ impl WorkingCopy {
store,
working_copy_path,
state_path,
operation_id: RefCell::new(None),
workspace_id: RefCell::new(None),
checkout_state: OnceCell::new(),
tree_state: OnceCell::new(),
}
}
@ -1024,36 +1027,35 @@ impl WorkingCopy {
temp_file.persist(self.state_path.join("checkout")).unwrap();
}
fn load_proto(&self) {
let mut file = File::open(self.state_path.join("checkout")).unwrap();
let proto: crate::protos::working_copy::Checkout =
Message::parse_from_reader(&mut file).unwrap();
self.operation_id
.replace(Some(OperationId::new(proto.operation_id)));
let workspace_id = if proto.workspace_id.is_empty() {
// For compatibility with old working copies.
// TODO: Delete in mid 2022 or so
WorkspaceId::default()
} else {
WorkspaceId::new(proto.workspace_id)
};
self.workspace_id.replace(Some(workspace_id));
fn checkout_state(&self) -> &CheckoutState {
self.checkout_state.get_or_init(|| {
let mut file = File::open(self.state_path.join("checkout")).unwrap();
let proto: crate::protos::working_copy::Checkout =
Message::parse_from_reader(&mut file).unwrap();
CheckoutState {
operation_id: OperationId::new(proto.operation_id),
workspace_id: if proto.workspace_id.is_empty() {
// For compatibility with old working copies.
// TODO: Delete in mid 2022 or so
WorkspaceId::default()
} else {
WorkspaceId::new(proto.workspace_id)
},
}
})
}
fn checkout_state_mut(&mut self) -> &mut CheckoutState {
self.checkout_state(); // ensure loaded
self.checkout_state.get_mut().unwrap()
}
pub fn operation_id(&self) -> OperationId {
if self.operation_id.borrow().is_none() {
self.load_proto();
}
self.operation_id.borrow().as_ref().unwrap().clone()
self.checkout_state().operation_id.clone()
}
pub fn workspace_id(&self) -> WorkspaceId {
if self.workspace_id.borrow().is_none() {
self.load_proto();
}
self.workspace_id.borrow().as_ref().unwrap().clone()
self.checkout_state().workspace_id.clone()
}
fn tree_state(&self) -> &TreeState {
@ -1095,7 +1097,7 @@ impl WorkingCopy {
let lock = FileLock::lock(lock_path);
// Re-read from disk after taking the lock
self.load_proto();
self.checkout_state.take();
// TODO: It's expensive to reload the whole tree. We should first check if it
// has changed.
self.tree_state.take();
@ -1204,7 +1206,7 @@ impl LockedWorkingCopy<'_> {
self.wc.tree_state_mut().save();
}
if self.old_operation_id != operation_id {
self.wc.operation_id.replace(Some(operation_id));
self.wc.checkout_state_mut().operation_id = operation_id;
self.wc.save();
}
// TODO: Clear the "pending_checkout" file here.