Mercurial > hg4j
diff src/org/tmatesoft/hg/internal/COWTransaction.java @ 635:4ec2d44e2bf3
Compound test scenario for checkout, add, remove, revert and commit
author | Artem Tikhomirov <tikhomirov.artem@gmail.com> |
---|---|
date | Thu, 06 Jun 2013 18:42:38 +0200 |
parents | 99ad1e3a4e4d |
children | 12a4f60ea972 |
line wrap: on
line diff
--- a/src/org/tmatesoft/hg/internal/COWTransaction.java Thu Jun 06 14:21:11 2013 +0200 +++ b/src/org/tmatesoft/hg/internal/COWTransaction.java Thu Jun 06 18:42:38 2013 +0200 @@ -46,19 +46,12 @@ @Override public File prepare(File f) throws HgIOException { - if (!f.exists()) { - record(f, null); - try { - f.getParentFile().mkdirs(); - f.createNewFile(); - return f; - } catch (IOException ex) { - throw new HgIOException("Failed to create new file", ex, f); - } - } if (known(f)) { return f; } + if (!f.exists()) { + return recordNonExistent(f); + } final File parentDir = f.getParentFile(); assert parentDir.canWrite(); File copy = new File(parentDir, f.getName() + ".hg4j.copy"); @@ -85,6 +78,9 @@ if (known(origin)) { return origin; } + if (!origin.exists()) { + return recordNonExistent(origin); + } fileHelper.copy(origin, backup); final RollbackEntry e = record(origin, backup); e.keepBackup = true; @@ -139,6 +135,17 @@ } } + private File recordNonExistent(File f) throws HgIOException { + record(f, null); + try { + f.getParentFile().mkdirs(); + f.createNewFile(); + return f; + } catch (IOException ex) { + throw new HgIOException("Failed to create new file", ex, f); + } + } + private RollbackEntry record(File origin, File backup) { final RollbackEntry e = new RollbackEntry(origin, backup); entries.add(e); @@ -146,23 +153,28 @@ } private boolean known(File f) { - for (RollbackEntry e : entries) { - if (e.origin.equals(f)) { - return true; - } + RollbackEntry e = lookup(f); + return e != null; + } + + private RollbackEntry find(File f) { + RollbackEntry e = lookup(f); + if (e != null) { + return e; } - return false; + assert false; + return new RollbackEntry(f,f); } - private RollbackEntry find(File f) { + + private RollbackEntry lookup(File f) { for (RollbackEntry e : entries) { if (e.origin.equals(f)) { return e; } } - assert false; - return new RollbackEntry(f,f); + return null; } - + private static class RollbackEntry { public final File origin; public final File backup; // may be null to indicate file didn't exist