You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

GC.java 32KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990
  1. /*
  2. * Copyright (C) 2012, Christian Halstrick <christian.halstrick@sap.com>
  3. * Copyright (C) 2011, Shawn O. Pearce <spearce@spearce.org>
  4. * and other copyright owners as documented in the project's IP log.
  5. *
  6. * This program and the accompanying materials are made available
  7. * under the terms of the Eclipse Distribution License v1.0 which
  8. * accompanies this distribution, is reproduced below, and is
  9. * available at http://www.eclipse.org/org/documents/edl-v10.php
  10. *
  11. * All rights reserved.
  12. *
  13. * Redistribution and use in source and binary forms, with or
  14. * without modification, are permitted provided that the following
  15. * conditions are met:
  16. *
  17. * - Redistributions of source code must retain the above copyright
  18. * notice, this list of conditions and the following disclaimer.
  19. *
  20. * - Redistributions in binary form must reproduce the above
  21. * copyright notice, this list of conditions and the following
  22. * disclaimer in the documentation and/or other materials provided
  23. * with the distribution.
  24. *
  25. * - Neither the name of the Eclipse Foundation, Inc. nor the
  26. * names of its contributors may be used to endorse or promote
  27. * products derived from this software without specific prior
  28. * written permission.
  29. *
  30. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
  31. * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
  32. * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  33. * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  34. * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
  35. * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  36. * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  37. * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  38. * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  39. * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  40. * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  41. * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  42. * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  43. */
  44. package org.eclipse.jgit.internal.storage.file;
  45. import static org.eclipse.jgit.internal.storage.pack.PackExt.BITMAP_INDEX;
  46. import static org.eclipse.jgit.internal.storage.pack.PackExt.INDEX;
  47. import static org.eclipse.jgit.lib.RefDatabase.ALL;
  48. import java.io.File;
  49. import java.io.FileOutputStream;
  50. import java.io.IOException;
  51. import java.io.OutputStream;
  52. import java.nio.channels.Channels;
  53. import java.nio.channels.FileChannel;
  54. import java.text.MessageFormat;
  55. import java.text.ParseException;
  56. import java.util.ArrayList;
  57. import java.util.Collection;
  58. import java.util.Collections;
  59. import java.util.Comparator;
  60. import java.util.Date;
  61. import java.util.HashMap;
  62. import java.util.HashSet;
  63. import java.util.Iterator;
  64. import java.util.LinkedList;
  65. import java.util.List;
  66. import java.util.Map;
  67. import java.util.Map.Entry;
  68. import java.util.Set;
  69. import java.util.TreeMap;
  70. import org.eclipse.jgit.dircache.DirCacheIterator;
  71. import org.eclipse.jgit.errors.CorruptObjectException;
  72. import org.eclipse.jgit.errors.IncorrectObjectTypeException;
  73. import org.eclipse.jgit.errors.MissingObjectException;
  74. import org.eclipse.jgit.errors.NoWorkTreeException;
  75. import org.eclipse.jgit.internal.JGitText;
  76. import org.eclipse.jgit.internal.storage.pack.PackExt;
  77. import org.eclipse.jgit.internal.storage.pack.PackWriter;
  78. import org.eclipse.jgit.internal.storage.pack.PackWriter.ObjectIdSet;
  79. import org.eclipse.jgit.lib.AnyObjectId;
  80. import org.eclipse.jgit.lib.ConfigConstants;
  81. import org.eclipse.jgit.lib.Constants;
  82. import org.eclipse.jgit.lib.FileMode;
  83. import org.eclipse.jgit.lib.NullProgressMonitor;
  84. import org.eclipse.jgit.lib.ObjectId;
  85. import org.eclipse.jgit.lib.ProgressMonitor;
  86. import org.eclipse.jgit.lib.Ref;
  87. import org.eclipse.jgit.lib.Ref.Storage;
  88. import org.eclipse.jgit.lib.RefDatabase;
  89. import org.eclipse.jgit.lib.ReflogEntry;
  90. import org.eclipse.jgit.revwalk.ObjectWalk;
  91. import org.eclipse.jgit.revwalk.RevObject;
  92. import org.eclipse.jgit.revwalk.RevWalk;
  93. import org.eclipse.jgit.storage.pack.PackConfig;
  94. import org.eclipse.jgit.treewalk.TreeWalk;
  95. import org.eclipse.jgit.treewalk.filter.TreeFilter;
  96. import org.eclipse.jgit.util.FileUtils;
  97. import org.eclipse.jgit.util.GitDateParser;
  98. import org.eclipse.jgit.util.SystemReader;
  99. /**
  100. * A garbage collector for git {@link FileRepository}. Instances of this class
  101. * are not thread-safe. Don't use the same instance from multiple threads.
  102. *
  103. * This class started as a copy of DfsGarbageCollector from Shawn O. Pearce
  104. * adapted to FileRepositories.
  105. */
  106. public class GC {
  107. private static final String PRUNE_EXPIRE_DEFAULT = "2.weeks.ago"; //$NON-NLS-1$
  108. private final FileRepository repo;
  109. private ProgressMonitor pm;
  110. private long expireAgeMillis = -1;
  111. private Date expire;
  112. private PackConfig pconfig = null;
  113. /**
  114. * the refs which existed during the last call to {@link #repack()}. This is
  115. * needed during {@link #prune(Set)} where we can optimize by looking at the
  116. * difference between the current refs and the refs which existed during
  117. * last {@link #repack()}.
  118. */
  119. private Map<String, Ref> lastPackedRefs;
  120. /**
  121. * Holds the starting time of the last repack() execution. This is needed in
  122. * prune() to inspect only those reflog entries which have been added since
  123. * last repack().
  124. */
  125. private long lastRepackTime;
  126. /**
  127. * Creates a new garbage collector with default values. An expirationTime of
  128. * two weeks and <code>null</code> as progress monitor will be used.
  129. *
  130. * @param repo
  131. * the repo to work on
  132. */
  133. public GC(FileRepository repo) {
  134. this.repo = repo;
  135. this.pm = NullProgressMonitor.INSTANCE;
  136. }
  137. /**
  138. * Runs a garbage collector on a {@link FileRepository}. It will
  139. * <ul>
  140. * <li>pack loose references into packed-refs</li>
  141. * <li>repack all reachable objects into new pack files and delete the old
  142. * pack files</li>
  143. * <li>prune all loose objects which are now reachable by packs</li>
  144. * </ul>
  145. *
  146. * @return the collection of {@link PackFile}'s which are newly created
  147. * @throws IOException
  148. * @throws ParseException
  149. * If the configuration parameter "gc.pruneexpire" couldn't be
  150. * parsed
  151. */
  152. public Collection<PackFile> gc() throws IOException, ParseException {
  153. pm.start(6 /* tasks */);
  154. packRefs();
  155. // TODO: implement reflog_expire(pm, repo);
  156. Collection<PackFile> newPacks = repack();
  157. prune(Collections.<ObjectId> emptySet());
  158. // TODO: implement rerere_gc(pm);
  159. return newPacks;
  160. }
  161. /**
  162. * Delete old pack files. What is 'old' is defined by specifying a set of
  163. * old pack files and a set of new pack files. Each pack file contained in
  164. * old pack files but not contained in new pack files will be deleted.
  165. *
  166. * @param oldPacks
  167. * @param newPacks
  168. */
  169. private void deleteOldPacks(Collection<PackFile> oldPacks,
  170. Collection<PackFile> newPacks) {
  171. oldPackLoop: for (PackFile oldPack : oldPacks) {
  172. String oldName = oldPack.getPackName();
  173. // check whether an old pack file is also among the list of new
  174. // pack files. Then we must not delete it.
  175. for (PackFile newPack : newPacks)
  176. if (oldName.equals(newPack.getPackName()))
  177. continue oldPackLoop;
  178. if (!oldPack.shouldBeKept()) {
  179. oldPack.close();
  180. prunePack(oldName);
  181. }
  182. }
  183. // close the complete object database. Thats my only chance to force
  184. // rescanning and to detect that certain pack files are now deleted.
  185. repo.getObjectDatabase().close();
  186. }
  187. /**
  188. * Delete files associated with a single pack file. First try to delete the
  189. * ".pack" file because on some platforms the ".pack" file may be locked and
  190. * can't be deleted. In such a case it is better to detect this early and
  191. * give up on deleting files for this packfile. Otherwise we may delete the
  192. * ".index" file and when failing to delete the ".pack" file we are left
  193. * with a ".pack" file without a ".index" file.
  194. *
  195. * @param packName
  196. */
  197. private void prunePack(String packName) {
  198. PackExt[] extensions = PackExt.values();
  199. try {
  200. // Delete the .pack file first and if this fails give up on deleting
  201. // the other files
  202. int deleteOptions = FileUtils.RETRY | FileUtils.SKIP_MISSING;
  203. for (PackExt ext : extensions)
  204. if (PackExt.PACK.equals(ext)) {
  205. File f = nameFor(packName, "." + ext.getExtension()); //$NON-NLS-1$
  206. FileUtils.delete(f, deleteOptions);
  207. break;
  208. }
  209. // The .pack file has been deleted. Delete as many as the other
  210. // files as you can.
  211. deleteOptions |= FileUtils.IGNORE_ERRORS;
  212. for (PackExt ext : extensions) {
  213. if (!PackExt.PACK.equals(ext)) {
  214. File f = nameFor(packName, "." + ext.getExtension()); //$NON-NLS-1$
  215. FileUtils.delete(f, deleteOptions);
  216. }
  217. }
  218. } catch (IOException e) {
  219. // Deletion of the .pack file failed. Silently return.
  220. }
  221. }
  222. /**
  223. * Like "git prune-packed" this method tries to prune all loose objects
  224. * which can be found in packs. If certain objects can't be pruned (e.g.
  225. * because the filesystem delete operation fails) this is silently ignored.
  226. *
  227. * @throws IOException
  228. */
  229. public void prunePacked() throws IOException {
  230. ObjectDirectory objdb = repo.getObjectDatabase();
  231. Collection<PackFile> packs = objdb.getPacks();
  232. File objects = repo.getObjectsDirectory();
  233. String[] fanout = objects.list();
  234. if (fanout != null && fanout.length > 0) {
  235. pm.beginTask(JGitText.get().pruneLoosePackedObjects, fanout.length);
  236. try {
  237. for (String d : fanout) {
  238. pm.update(1);
  239. if (d.length() != 2)
  240. continue;
  241. String[] entries = new File(objects, d).list();
  242. if (entries == null)
  243. continue;
  244. for (String e : entries) {
  245. if (e.length() != Constants.OBJECT_ID_STRING_LENGTH - 2)
  246. continue;
  247. ObjectId id;
  248. try {
  249. id = ObjectId.fromString(d + e);
  250. } catch (IllegalArgumentException notAnObject) {
  251. // ignoring the file that does not represent loose
  252. // object
  253. continue;
  254. }
  255. boolean found = false;
  256. for (PackFile p : packs)
  257. if (p.hasObject(id)) {
  258. found = true;
  259. break;
  260. }
  261. if (found)
  262. FileUtils.delete(objdb.fileFor(id), FileUtils.RETRY
  263. | FileUtils.SKIP_MISSING
  264. | FileUtils.IGNORE_ERRORS);
  265. }
  266. }
  267. } finally {
  268. pm.endTask();
  269. }
  270. }
  271. }
  272. /**
  273. * Like "git prune" this method tries to prune all loose objects which are
  274. * unreferenced. If certain objects can't be pruned (e.g. because the
  275. * filesystem delete operation fails) this is silently ignored.
  276. *
  277. * @param objectsToKeep
  278. * a set of objects which should explicitly not be pruned
  279. *
  280. * @throws IOException
  281. * @throws ParseException
  282. * If the configuration parameter "gc.pruneexpire" couldn't be
  283. * parsed
  284. */
  285. public void prune(Set<ObjectId> objectsToKeep) throws IOException,
  286. ParseException {
  287. long expireDate = Long.MAX_VALUE;
  288. if (expire == null && expireAgeMillis == -1) {
  289. String pruneExpireStr = repo.getConfig().getString(
  290. ConfigConstants.CONFIG_GC_SECTION, null,
  291. ConfigConstants.CONFIG_KEY_PRUNEEXPIRE);
  292. if (pruneExpireStr == null)
  293. pruneExpireStr = PRUNE_EXPIRE_DEFAULT;
  294. expire = GitDateParser.parse(pruneExpireStr, null, SystemReader
  295. .getInstance().getLocale());
  296. expireAgeMillis = -1;
  297. }
  298. if (expire != null)
  299. expireDate = expire.getTime();
  300. if (expireAgeMillis != -1)
  301. expireDate = System.currentTimeMillis() - expireAgeMillis;
  302. // Collect all loose objects which are old enough, not referenced from
  303. // the index and not in objectsToKeep
  304. Map<ObjectId, File> deletionCandidates = new HashMap<ObjectId, File>();
  305. Set<ObjectId> indexObjects = null;
  306. File objects = repo.getObjectsDirectory();
  307. String[] fanout = objects.list();
  308. if (fanout != null && fanout.length > 0) {
  309. pm.beginTask(JGitText.get().pruneLooseUnreferencedObjects,
  310. fanout.length);
  311. try {
  312. for (String d : fanout) {
  313. pm.update(1);
  314. if (d.length() != 2)
  315. continue;
  316. File[] entries = new File(objects, d).listFiles();
  317. if (entries == null)
  318. continue;
  319. for (File f : entries) {
  320. String fName = f.getName();
  321. if (fName.length() != Constants.OBJECT_ID_STRING_LENGTH - 2)
  322. continue;
  323. if (f.lastModified() >= expireDate)
  324. continue;
  325. try {
  326. ObjectId id = ObjectId.fromString(d + fName);
  327. if (objectsToKeep.contains(id))
  328. continue;
  329. if (indexObjects == null)
  330. indexObjects = listNonHEADIndexObjects();
  331. if (indexObjects.contains(id))
  332. continue;
  333. deletionCandidates.put(id, f);
  334. } catch (IllegalArgumentException notAnObject) {
  335. // ignoring the file that does not represent loose
  336. // object
  337. continue;
  338. }
  339. }
  340. }
  341. } finally {
  342. pm.endTask();
  343. }
  344. }
  345. if (deletionCandidates.isEmpty())
  346. return;
  347. // From the set of current refs remove all those which have been handled
  348. // during last repack(). Only those refs will survive which have been
  349. // added or modified since the last repack. Only these can save existing
  350. // loose refs from being pruned.
  351. Map<String, Ref> newRefs;
  352. if (lastPackedRefs == null || lastPackedRefs.isEmpty())
  353. newRefs = getAllRefs();
  354. else {
  355. newRefs = new HashMap<String, Ref>();
  356. for (Iterator<Map.Entry<String, Ref>> i = getAllRefs().entrySet()
  357. .iterator(); i.hasNext();) {
  358. Entry<String, Ref> newEntry = i.next();
  359. Ref old = lastPackedRefs.get(newEntry.getKey());
  360. if (!equals(newEntry.getValue(), old))
  361. newRefs.put(newEntry.getKey(), newEntry.getValue());
  362. }
  363. }
  364. if (!newRefs.isEmpty()) {
  365. // There are new/modified refs! Check which loose objects are now
  366. // referenced by these modified refs (or their reflogentries).
  367. // Remove these loose objects
  368. // from the deletionCandidates. When the last candidate is removed
  369. // leave this method.
  370. ObjectWalk w = new ObjectWalk(repo);
  371. try {
  372. for (Ref cr : newRefs.values())
  373. w.markStart(w.parseAny(cr.getObjectId()));
  374. if (lastPackedRefs != null)
  375. for (Ref lpr : lastPackedRefs.values())
  376. w.markUninteresting(w.parseAny(lpr.getObjectId()));
  377. removeReferenced(deletionCandidates, w);
  378. } finally {
  379. w.dispose();
  380. }
  381. }
  382. if (deletionCandidates.isEmpty())
  383. return;
  384. // Since we have not left the method yet there are still
  385. // deletionCandidates. Last chance for these objects not to be pruned is
  386. // that they are referenced by reflog entries. Even refs which currently
  387. // point to the same object as during last repack() may have
  388. // additional reflog entries not handled during last repack()
  389. ObjectWalk w = new ObjectWalk(repo);
  390. try {
  391. for (Ref ar : getAllRefs().values())
  392. for (ObjectId id : listRefLogObjects(ar, lastRepackTime))
  393. w.markStart(w.parseAny(id));
  394. if (lastPackedRefs != null)
  395. for (Ref lpr : lastPackedRefs.values())
  396. w.markUninteresting(w.parseAny(lpr.getObjectId()));
  397. removeReferenced(deletionCandidates, w);
  398. } finally {
  399. w.dispose();
  400. }
  401. if (deletionCandidates.isEmpty())
  402. return;
  403. // delete all candidates which have survived: these are unreferenced
  404. // loose objects
  405. for (File f : deletionCandidates.values())
  406. f.delete();
  407. repo.getObjectDatabase().close();
  408. }
  409. /**
  410. * Remove all entries from a map which key is the id of an object referenced
  411. * by the given ObjectWalk
  412. *
  413. * @param id2File
  414. * @param w
  415. * @throws MissingObjectException
  416. * @throws IncorrectObjectTypeException
  417. * @throws IOException
  418. */
  419. private void removeReferenced(Map<ObjectId, File> id2File,
  420. ObjectWalk w) throws MissingObjectException,
  421. IncorrectObjectTypeException, IOException {
  422. RevObject ro = w.next();
  423. while (ro != null) {
  424. if (id2File.remove(ro.getId()) != null)
  425. if (id2File.isEmpty())
  426. return;
  427. ro = w.next();
  428. }
  429. ro = w.nextObject();
  430. while (ro != null) {
  431. if (id2File.remove(ro.getId()) != null)
  432. if (id2File.isEmpty())
  433. return;
  434. ro = w.nextObject();
  435. }
  436. }
  437. private static boolean equals(Ref r1, Ref r2) {
  438. if (r1 == null || r2 == null)
  439. return false;
  440. if (r1.isSymbolic()) {
  441. if (!r2.isSymbolic())
  442. return false;
  443. return r1.getTarget().getName().equals(r2.getTarget().getName());
  444. } else {
  445. if (r2.isSymbolic())
  446. return false;
  447. return r1.getObjectId().equals(r2.getObjectId());
  448. }
  449. }
  450. /**
  451. * Packs all non-symbolic, loose refs into packed-refs.
  452. *
  453. * @throws IOException
  454. */
  455. public void packRefs() throws IOException {
  456. Collection<Ref> refs = repo.getRefDatabase().getRefs(Constants.R_REFS).values();
  457. List<String> refsToBePacked = new ArrayList<String>(refs.size());
  458. pm.beginTask(JGitText.get().packRefs, refs.size());
  459. try {
  460. for (Ref ref : refs) {
  461. if (!ref.isSymbolic() && ref.getStorage().isLoose())
  462. refsToBePacked.add(ref.getName());
  463. pm.update(1);
  464. }
  465. ((RefDirectory) repo.getRefDatabase()).pack(refsToBePacked);
  466. } finally {
  467. pm.endTask();
  468. }
  469. }
  470. /**
  471. * Packs all objects which reachable from any of the heads into one pack
  472. * file. Additionally all objects which are not reachable from any head but
  473. * which are reachable from any of the other refs (e.g. tags), special refs
  474. * (e.g. FETCH_HEAD) or index are packed into a separate pack file. Objects
  475. * included in pack files which have a .keep file associated are never
  476. * repacked. All old pack files which existed before are deleted.
  477. *
  478. * @return a collection of the newly created pack files
  479. * @throws IOException
  480. * when during reading of refs, index, packfiles, objects,
  481. * reflog-entries or during writing to the packfiles
  482. * {@link IOException} occurs
  483. */
  484. public Collection<PackFile> repack() throws IOException {
  485. Collection<PackFile> toBeDeleted = repo.getObjectDatabase().getPacks();
  486. long time = System.currentTimeMillis();
  487. Map<String, Ref> refsBefore = getAllRefs();
  488. Set<ObjectId> allHeads = new HashSet<ObjectId>();
  489. Set<ObjectId> nonHeads = new HashSet<ObjectId>();
  490. Set<ObjectId> tagTargets = new HashSet<ObjectId>();
  491. Set<ObjectId> indexObjects = listNonHEADIndexObjects();
  492. for (Ref ref : refsBefore.values()) {
  493. nonHeads.addAll(listRefLogObjects(ref, 0));
  494. if (ref.isSymbolic() || ref.getObjectId() == null)
  495. continue;
  496. if (ref.getName().startsWith(Constants.R_HEADS))
  497. allHeads.add(ref.getObjectId());
  498. else
  499. nonHeads.add(ref.getObjectId());
  500. if (ref.getPeeledObjectId() != null)
  501. tagTargets.add(ref.getPeeledObjectId());
  502. }
  503. List<ObjectIdSet> excluded = new LinkedList<ObjectIdSet>();
  504. for (final PackFile f : repo.getObjectDatabase().getPacks())
  505. if (f.shouldBeKept())
  506. excluded.add(objectIdSet(f.getIndex()));
  507. tagTargets.addAll(allHeads);
  508. nonHeads.addAll(indexObjects);
  509. List<PackFile> ret = new ArrayList<PackFile>(2);
  510. PackFile heads = null;
  511. if (!allHeads.isEmpty()) {
  512. heads = writePack(allHeads, Collections.<ObjectId> emptySet(),
  513. tagTargets, excluded);
  514. if (heads != null) {
  515. ret.add(heads);
  516. excluded.add(0, objectIdSet(heads.getIndex()));
  517. }
  518. }
  519. if (!nonHeads.isEmpty()) {
  520. PackFile rest = writePack(nonHeads, allHeads, tagTargets, excluded);
  521. if (rest != null)
  522. ret.add(rest);
  523. }
  524. deleteOldPacks(toBeDeleted, ret);
  525. prunePacked();
  526. lastPackedRefs = refsBefore;
  527. lastRepackTime = time;
  528. return ret;
  529. }
  530. /**
  531. * @param ref
  532. * the ref which log should be inspected
  533. * @param minTime only reflog entries not older then this time are processed
  534. * @return the {@link ObjectId}s contained in the reflog
  535. * @throws IOException
  536. */
  537. private Set<ObjectId> listRefLogObjects(Ref ref, long minTime) throws IOException {
  538. List<ReflogEntry> rlEntries = repo.getReflogReader(ref.getName())
  539. .getReverseEntries();
  540. if (rlEntries == null || rlEntries.isEmpty())
  541. return Collections.<ObjectId> emptySet();
  542. Set<ObjectId> ret = new HashSet<ObjectId>();
  543. for (ReflogEntry e : rlEntries) {
  544. if (e.getWho().getWhen().getTime() < minTime)
  545. break;
  546. ObjectId newId = e.getNewId();
  547. if (newId != null && !ObjectId.zeroId().equals(newId))
  548. ret.add(newId);
  549. ObjectId oldId = e.getOldId();
  550. if (oldId != null && !ObjectId.zeroId().equals(oldId))
  551. ret.add(oldId);
  552. }
  553. return ret;
  554. }
  555. /**
  556. * Returns a map of all refs and additional refs (e.g. FETCH_HEAD,
  557. * MERGE_HEAD, ...)
  558. *
  559. * @return a map where names of refs point to ref objects
  560. * @throws IOException
  561. */
  562. private Map<String, Ref> getAllRefs() throws IOException {
  563. Map<String, Ref> ret = repo.getRefDatabase().getRefs(ALL);
  564. for (Ref ref : repo.getRefDatabase().getAdditionalRefs())
  565. ret.put(ref.getName(), ref);
  566. return ret;
  567. }
  568. /**
  569. * Return a list of those objects in the index which differ from whats in
  570. * HEAD
  571. *
  572. * @return a set of ObjectIds of changed objects in the index
  573. * @throws IOException
  574. * @throws CorruptObjectException
  575. * @throws NoWorkTreeException
  576. */
  577. private Set<ObjectId> listNonHEADIndexObjects()
  578. throws CorruptObjectException, IOException {
  579. RevWalk revWalk = null;
  580. try {
  581. if (repo.getIndexFile() == null)
  582. return Collections.emptySet();
  583. } catch (NoWorkTreeException e) {
  584. return Collections.emptySet();
  585. }
  586. TreeWalk treeWalk = new TreeWalk(repo);
  587. try {
  588. treeWalk.addTree(new DirCacheIterator(repo.readDirCache()));
  589. ObjectId headID = repo.resolve(Constants.HEAD);
  590. if (headID != null) {
  591. revWalk = new RevWalk(repo);
  592. treeWalk.addTree(revWalk.parseTree(headID));
  593. revWalk.dispose();
  594. revWalk = null;
  595. }
  596. treeWalk.setFilter(TreeFilter.ANY_DIFF);
  597. treeWalk.setRecursive(true);
  598. Set<ObjectId> ret = new HashSet<ObjectId>();
  599. while (treeWalk.next()) {
  600. ObjectId objectId = treeWalk.getObjectId(0);
  601. switch (treeWalk.getRawMode(0) & FileMode.TYPE_MASK) {
  602. case FileMode.TYPE_MISSING:
  603. case FileMode.TYPE_GITLINK:
  604. continue;
  605. case FileMode.TYPE_TREE:
  606. case FileMode.TYPE_FILE:
  607. case FileMode.TYPE_SYMLINK:
  608. ret.add(objectId);
  609. continue;
  610. default:
  611. throw new IOException(MessageFormat.format(
  612. JGitText.get().corruptObjectInvalidMode3,
  613. String.format("%o", //$NON-NLS-1$
  614. Integer.valueOf(treeWalk.getRawMode(0))),
  615. (objectId == null) ? "null" : objectId.name(), //$NON-NLS-1$
  616. treeWalk.getPathString(), //
  617. repo.getIndexFile()));
  618. }
  619. }
  620. return ret;
  621. } finally {
  622. if (revWalk != null)
  623. revWalk.dispose();
  624. treeWalk.release();
  625. }
  626. }
  627. private PackFile writePack(Set<? extends ObjectId> want,
  628. Set<? extends ObjectId> have, Set<ObjectId> tagTargets,
  629. List<ObjectIdSet> excludeObjects) throws IOException {
  630. File tmpPack = null;
  631. Map<PackExt, File> tmpExts = new TreeMap<PackExt, File>(
  632. new Comparator<PackExt>() {
  633. public int compare(PackExt o1, PackExt o2) {
  634. // INDEX entries must be returned last, so the pack
  635. // scanner does pick up the new pack until all the
  636. // PackExt entries have been written.
  637. if (o1 == o2)
  638. return 0;
  639. if (o1 == PackExt.INDEX)
  640. return 1;
  641. if (o2 == PackExt.INDEX)
  642. return -1;
  643. return Integer.signum(o1.hashCode() - o2.hashCode());
  644. }
  645. });
  646. PackWriter pw = new PackWriter((pconfig == null) ? new PackConfig(repo) : pconfig, repo.newObjectReader());
  647. try {
  648. // prepare the PackWriter
  649. pw.setDeltaBaseAsOffset(true);
  650. pw.setReuseDeltaCommits(false);
  651. if (tagTargets != null)
  652. pw.setTagTargets(tagTargets);
  653. if (excludeObjects != null)
  654. for (ObjectIdSet idx : excludeObjects)
  655. pw.excludeObjects(idx);
  656. pw.preparePack(pm, want, have);
  657. if (pw.getObjectCount() == 0)
  658. return null;
  659. // create temporary files
  660. String id = pw.computeName().getName();
  661. File packdir = new File(repo.getObjectsDirectory(), "pack"); //$NON-NLS-1$
  662. tmpPack = File.createTempFile("gc_", ".pack_tmp", packdir); //$NON-NLS-1$ //$NON-NLS-2$
  663. final String tmpBase = tmpPack.getName()
  664. .substring(0, tmpPack.getName().lastIndexOf('.'));
  665. File tmpIdx = new File(packdir, tmpBase + ".idx_tmp"); //$NON-NLS-1$
  666. tmpExts.put(INDEX, tmpIdx);
  667. if (!tmpIdx.createNewFile())
  668. throw new IOException(MessageFormat.format(
  669. JGitText.get().cannotCreateIndexfile, tmpIdx.getPath()));
  670. // write the packfile
  671. FileOutputStream fos = new FileOutputStream(tmpPack);
  672. FileChannel channel = fos.getChannel();
  673. OutputStream channelStream = Channels.newOutputStream(channel);
  674. try {
  675. pw.writePack(pm, pm, channelStream);
  676. } finally {
  677. channel.force(true);
  678. channelStream.close();
  679. fos.close();
  680. }
  681. // write the packindex
  682. fos = new FileOutputStream(tmpIdx);
  683. FileChannel idxChannel = fos.getChannel();
  684. OutputStream idxStream = Channels.newOutputStream(idxChannel);
  685. try {
  686. pw.writeIndex(idxStream);
  687. } finally {
  688. idxChannel.force(true);
  689. idxStream.close();
  690. fos.close();
  691. }
  692. if (pw.prepareBitmapIndex(pm)) {
  693. File tmpBitmapIdx = new File(packdir, tmpBase + ".bitmap_tmp"); //$NON-NLS-1$
  694. tmpExts.put(BITMAP_INDEX, tmpBitmapIdx);
  695. if (!tmpBitmapIdx.createNewFile())
  696. throw new IOException(MessageFormat.format(
  697. JGitText.get().cannotCreateIndexfile,
  698. tmpBitmapIdx.getPath()));
  699. fos = new FileOutputStream(tmpBitmapIdx);
  700. idxChannel = fos.getChannel();
  701. idxStream = Channels.newOutputStream(idxChannel);
  702. try {
  703. pw.writeBitmapIndex(idxStream);
  704. } finally {
  705. idxChannel.force(true);
  706. idxStream.close();
  707. fos.close();
  708. }
  709. }
  710. // rename the temporary files to real files
  711. File realPack = nameFor(id, ".pack"); //$NON-NLS-1$
  712. // if the packfile already exists (because we are rewriting a
  713. // packfile for the same set of objects maybe with different
  714. // PackConfig) then make sure we get rid of all handles on the file.
  715. // Windows will not allow for rename otherwise.
  716. if (realPack.exists())
  717. for (PackFile p : repo.getObjectDatabase().getPacks())
  718. if (realPack.getPath().equals(p.getPackFile().getPath())) {
  719. p.close();
  720. break;
  721. }
  722. tmpPack.setReadOnly();
  723. boolean delete = true;
  724. try {
  725. FileUtils.rename(tmpPack, realPack);
  726. delete = false;
  727. for (Map.Entry<PackExt, File> tmpEntry : tmpExts.entrySet()) {
  728. File tmpExt = tmpEntry.getValue();
  729. tmpExt.setReadOnly();
  730. File realExt = nameFor(
  731. id, "." + tmpEntry.getKey().getExtension()); //$NON-NLS-1$
  732. try {
  733. FileUtils.rename(tmpExt, realExt);
  734. } catch (IOException e) {
  735. File newExt = new File(realExt.getParentFile(),
  736. realExt.getName() + ".new"); //$NON-NLS-1$
  737. if (!tmpExt.renameTo(newExt))
  738. newExt = tmpExt;
  739. throw new IOException(MessageFormat.format(
  740. JGitText.get().panicCantRenameIndexFile, newExt,
  741. realExt));
  742. }
  743. }
  744. } finally {
  745. if (delete) {
  746. if (tmpPack.exists())
  747. tmpPack.delete();
  748. for (File tmpExt : tmpExts.values()) {
  749. if (tmpExt.exists())
  750. tmpExt.delete();
  751. }
  752. }
  753. }
  754. return repo.getObjectDatabase().openPack(realPack);
  755. } finally {
  756. pw.release();
  757. if (tmpPack != null && tmpPack.exists())
  758. tmpPack.delete();
  759. for (File tmpExt : tmpExts.values()) {
  760. if (tmpExt.exists())
  761. tmpExt.delete();
  762. }
  763. }
  764. }
  765. private File nameFor(String name, String ext) {
  766. File packdir = new File(repo.getObjectsDirectory(), "pack"); //$NON-NLS-1$
  767. return new File(packdir, "pack-" + name + ext); //$NON-NLS-1$
  768. }
  769. /**
  770. * A class holding statistical data for a FileRepository regarding how many
  771. * objects are stored as loose or packed objects
  772. */
  773. public class RepoStatistics {
  774. /**
  775. * The number of objects stored in pack files. If the same object is
  776. * stored in multiple pack files then it is counted as often as it
  777. * occurs in pack files.
  778. */
  779. public long numberOfPackedObjects;
  780. /**
  781. * The number of pack files
  782. */
  783. public long numberOfPackFiles;
  784. /**
  785. * The number of objects stored as loose objects.
  786. */
  787. public long numberOfLooseObjects;
  788. /**
  789. * The sum of the sizes of all files used to persist loose objects.
  790. */
  791. public long sizeOfLooseObjects;
  792. /**
  793. * The sum of the sizes of all pack files.
  794. */
  795. public long sizeOfPackedObjects;
  796. /**
  797. * The number of loose refs.
  798. */
  799. public long numberOfLooseRefs;
  800. /**
  801. * The number of refs stored in pack files.
  802. */
  803. public long numberOfPackedRefs;
  804. public String toString() {
  805. final StringBuilder b = new StringBuilder();
  806. b.append("numberOfPackedObjects=").append(numberOfPackedObjects); //$NON-NLS-1$
  807. b.append(", numberOfPackFiles=").append(numberOfPackFiles); //$NON-NLS-1$
  808. b.append(", numberOfLooseObjects=").append(numberOfLooseObjects); //$NON-NLS-1$
  809. b.append(", numberOfLooseRefs=").append(numberOfLooseRefs); //$NON-NLS-1$
  810. b.append(", numberOfPackedRefs=").append(numberOfPackedRefs); //$NON-NLS-1$
  811. b.append(", sizeOfLooseObjects=").append(sizeOfLooseObjects); //$NON-NLS-1$
  812. b.append(", sizeOfPackedObjects=").append(sizeOfPackedObjects); //$NON-NLS-1$
  813. return b.toString();
  814. }
  815. }
  816. /**
  817. * Returns the number of objects stored in pack files. If an object is
  818. * contained in multiple pack files it is counted as often as it occurs.
  819. *
  820. * @return the number of objects stored in pack files
  821. * @throws IOException
  822. */
  823. public RepoStatistics getStatistics() throws IOException {
  824. RepoStatistics ret = new RepoStatistics();
  825. Collection<PackFile> packs = repo.getObjectDatabase().getPacks();
  826. for (PackFile f : packs) {
  827. ret.numberOfPackedObjects += f.getIndex().getObjectCount();
  828. ret.numberOfPackFiles++;
  829. ret.sizeOfPackedObjects += f.getPackFile().length();
  830. }
  831. File objDir = repo.getObjectsDirectory();
  832. String[] fanout = objDir.list();
  833. if (fanout != null && fanout.length > 0) {
  834. for (String d : fanout) {
  835. if (d.length() != 2)
  836. continue;
  837. File[] entries = new File(objDir, d).listFiles();
  838. if (entries == null)
  839. continue;
  840. for (File f : entries) {
  841. if (f.getName().length() != Constants.OBJECT_ID_STRING_LENGTH - 2)
  842. continue;
  843. ret.numberOfLooseObjects++;
  844. ret.sizeOfLooseObjects += f.length();
  845. }
  846. }
  847. }
  848. RefDatabase refDb = repo.getRefDatabase();
  849. for (Ref r : refDb.getRefs(RefDatabase.ALL).values()) {
  850. Storage storage = r.getStorage();
  851. if (storage == Storage.LOOSE || storage == Storage.LOOSE_PACKED)
  852. ret.numberOfLooseRefs++;
  853. if (storage == Storage.PACKED || storage == Storage.LOOSE_PACKED)
  854. ret.numberOfPackedRefs++;
  855. }
  856. return ret;
  857. }
  858. /**
  859. * Set the progress monitor used for garbage collection methods.
  860. *
  861. * @param pm
  862. * @return this
  863. */
  864. public GC setProgressMonitor(ProgressMonitor pm) {
  865. this.pm = (pm == null) ? NullProgressMonitor.INSTANCE : pm;
  866. return this;
  867. }
  868. /**
  869. * During gc() or prune() each unreferenced, loose object which has been
  870. * created or modified in the last <code>expireAgeMillis</code> milliseconds
  871. * will not be pruned. Only older objects may be pruned. If set to 0 then
  872. * every object is a candidate for pruning.
  873. *
  874. * @param expireAgeMillis
  875. * minimal age of objects to be pruned in milliseconds.
  876. */
  877. public void setExpireAgeMillis(long expireAgeMillis) {
  878. this.expireAgeMillis = expireAgeMillis;
  879. expire = null;
  880. }
  881. /**
  882. * Set the PackConfig used when (re-)writing packfiles. This allows to
  883. * influence how packs are written and to implement something similar to
  884. * "git gc --aggressive"
  885. *
  886. * @since 3.6
  887. * @param pconfig
  888. * the {@link PackConfig} used when writing packs
  889. */
  890. public void setPackConfig(PackConfig pconfig) {
  891. this.pconfig = pconfig;
  892. }
  893. /**
  894. * During gc() or prune() each unreferenced, loose object which has been
  895. * created or modified after or at <code>expire</code> will not be pruned.
  896. * Only older objects may be pruned. If set to null then every object is a
  897. * candidate for pruning.
  898. *
  899. * @param expire
  900. * instant in time which defines object expiration
  901. * objects with modification time before this instant are expired
  902. * objects with modification time newer or equal to this instant
  903. * are not expired
  904. */
  905. public void setExpire(Date expire) {
  906. this.expire = expire;
  907. expireAgeMillis = -1;
  908. }
  909. private static ObjectIdSet objectIdSet(final PackIndex idx) {
  910. return new ObjectIdSet() {
  911. public boolean contains(AnyObjectId objectId) {
  912. return idx.hasObject(objectId);
  913. }
  914. };
  915. }
  916. }