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.

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599
  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 java.io.File;
  48. import java.io.FileOutputStream;
  49. import java.io.IOException;
  50. import java.io.OutputStream;
  51. import java.io.PrintWriter;
  52. import java.io.StringWriter;
  53. import java.nio.channels.Channels;
  54. import java.nio.channels.FileChannel;
  55. import java.nio.file.DirectoryStream;
  56. import java.nio.file.Files;
  57. import java.nio.file.Path;
  58. import java.nio.file.StandardCopyOption;
  59. import java.text.MessageFormat;
  60. import java.text.ParseException;
  61. import java.time.Instant;
  62. import java.time.temporal.ChronoUnit;
  63. import java.util.ArrayList;
  64. import java.util.Collection;
  65. import java.util.Collections;
  66. import java.util.Comparator;
  67. import java.util.Date;
  68. import java.util.HashMap;
  69. import java.util.HashSet;
  70. import java.util.Iterator;
  71. import java.util.LinkedList;
  72. import java.util.List;
  73. import java.util.Map;
  74. import java.util.Objects;
  75. import java.util.Set;
  76. import java.util.TreeMap;
  77. import java.util.concurrent.Callable;
  78. import java.util.concurrent.ExecutorService;
  79. import java.util.regex.Pattern;
  80. import java.util.stream.Collectors;
  81. import java.util.stream.Stream;
  82. import org.eclipse.jgit.annotations.NonNull;
  83. import org.eclipse.jgit.dircache.DirCacheIterator;
  84. import org.eclipse.jgit.errors.CancelledException;
  85. import org.eclipse.jgit.errors.CorruptObjectException;
  86. import org.eclipse.jgit.errors.IncorrectObjectTypeException;
  87. import org.eclipse.jgit.errors.MissingObjectException;
  88. import org.eclipse.jgit.errors.NoWorkTreeException;
  89. import org.eclipse.jgit.internal.JGitText;
  90. import org.eclipse.jgit.internal.storage.pack.PackExt;
  91. import org.eclipse.jgit.internal.storage.pack.PackWriter;
  92. import org.eclipse.jgit.internal.storage.reftree.RefTreeNames;
  93. import org.eclipse.jgit.lib.ConfigConstants;
  94. import org.eclipse.jgit.lib.Constants;
  95. import org.eclipse.jgit.lib.FileMode;
  96. import org.eclipse.jgit.lib.NullProgressMonitor;
  97. import org.eclipse.jgit.lib.ObjectId;
  98. import org.eclipse.jgit.lib.ObjectIdSet;
  99. import org.eclipse.jgit.lib.ObjectLoader;
  100. import org.eclipse.jgit.lib.ObjectReader;
  101. import org.eclipse.jgit.lib.ProgressMonitor;
  102. import org.eclipse.jgit.lib.Ref;
  103. import org.eclipse.jgit.lib.Ref.Storage;
  104. import org.eclipse.jgit.lib.RefDatabase;
  105. import org.eclipse.jgit.lib.ReflogEntry;
  106. import org.eclipse.jgit.lib.ReflogReader;
  107. import org.eclipse.jgit.lib.internal.WorkQueue;
  108. import org.eclipse.jgit.revwalk.ObjectWalk;
  109. import org.eclipse.jgit.revwalk.RevObject;
  110. import org.eclipse.jgit.revwalk.RevWalk;
  111. import org.eclipse.jgit.storage.pack.PackConfig;
  112. import org.eclipse.jgit.treewalk.TreeWalk;
  113. import org.eclipse.jgit.treewalk.filter.TreeFilter;
  114. import org.eclipse.jgit.util.FileUtils;
  115. import org.eclipse.jgit.util.GitDateParser;
  116. import org.eclipse.jgit.util.SystemReader;
  117. import org.slf4j.Logger;
  118. import org.slf4j.LoggerFactory;
  119. /**
  120. * A garbage collector for git
  121. * {@link org.eclipse.jgit.internal.storage.file.FileRepository}. Instances of
  122. * this class are not thread-safe. Don't use the same instance from multiple
  123. * threads.
  124. *
  125. * This class started as a copy of DfsGarbageCollector from Shawn O. Pearce
  126. * adapted to FileRepositories.
  127. */
  128. public class GC {
  129. private final static Logger LOG = LoggerFactory
  130. .getLogger(GC.class);
  131. private static final String PRUNE_EXPIRE_DEFAULT = "2.weeks.ago"; //$NON-NLS-1$
  132. private static final String PRUNE_PACK_EXPIRE_DEFAULT = "1.hour.ago"; //$NON-NLS-1$
  133. private static final Pattern PATTERN_LOOSE_OBJECT = Pattern
  134. .compile("[0-9a-fA-F]{38}"); //$NON-NLS-1$
  135. private static final String PACK_EXT = "." + PackExt.PACK.getExtension();//$NON-NLS-1$
  136. private static final String BITMAP_EXT = "." //$NON-NLS-1$
  137. + PackExt.BITMAP_INDEX.getExtension();
  138. private static final String INDEX_EXT = "." + PackExt.INDEX.getExtension(); //$NON-NLS-1$
  139. private static final int DEFAULT_AUTOPACKLIMIT = 50;
  140. private static final int DEFAULT_AUTOLIMIT = 6700;
  141. private static volatile ExecutorService executor;
  142. /**
  143. * Set the executor for running auto-gc in the background. If no executor is
  144. * set JGit's own WorkQueue will be used.
  145. *
  146. * @param e
  147. * the executor to be used for running auto-gc
  148. * @since 4.8
  149. */
  150. public static void setExecutor(ExecutorService e) {
  151. executor = e;
  152. }
  153. private final FileRepository repo;
  154. private ProgressMonitor pm;
  155. private long expireAgeMillis = -1;
  156. private Date expire;
  157. private long packExpireAgeMillis = -1;
  158. private Date packExpire;
  159. private PackConfig pconfig = null;
  160. /**
  161. * the refs which existed during the last call to {@link #repack()}. This is
  162. * needed during {@link #prune(Set)} where we can optimize by looking at the
  163. * difference between the current refs and the refs which existed during
  164. * last {@link #repack()}.
  165. */
  166. private Collection<Ref> lastPackedRefs;
  167. /**
  168. * Holds the starting time of the last repack() execution. This is needed in
  169. * prune() to inspect only those reflog entries which have been added since
  170. * last repack().
  171. */
  172. private long lastRepackTime;
  173. /**
  174. * Whether gc should do automatic housekeeping
  175. */
  176. private boolean automatic;
  177. /**
  178. * Whether to run gc in a background thread
  179. */
  180. private boolean background;
  181. /**
  182. * Creates a new garbage collector with default values. An expirationTime of
  183. * two weeks and <code>null</code> as progress monitor will be used.
  184. *
  185. * @param repo
  186. * the repo to work on
  187. */
  188. public GC(FileRepository repo) {
  189. this.repo = repo;
  190. this.pm = NullProgressMonitor.INSTANCE;
  191. }
  192. /**
  193. * Runs a garbage collector on a
  194. * {@link org.eclipse.jgit.internal.storage.file.FileRepository}. It will
  195. * <ul>
  196. * <li>pack loose references into packed-refs</li>
  197. * <li>repack all reachable objects into new pack files and delete the old
  198. * pack files</li>
  199. * <li>prune all loose objects which are now reachable by packs</li>
  200. * </ul>
  201. *
  202. * If {@link #setAuto(boolean)} was set to {@code true} {@code gc} will
  203. * first check whether any housekeeping is required; if not, it exits
  204. * without performing any work.
  205. *
  206. * If {@link #setBackground(boolean)} was set to {@code true}
  207. * {@code collectGarbage} will start the gc in the background, and then
  208. * return immediately. In this case, errors will not be reported except in
  209. * gc.log.
  210. *
  211. * @return the collection of
  212. * {@link org.eclipse.jgit.internal.storage.file.PackFile}'s which
  213. * are newly created
  214. * @throws java.io.IOException
  215. * @throws java.text.ParseException
  216. * If the configuration parameter "gc.pruneexpire" couldn't be
  217. * parsed
  218. */
  219. // TODO(ms): in 5.0 change signature and return Future<Collection<PackFile>>
  220. public Collection<PackFile> gc() throws IOException, ParseException {
  221. if (!background) {
  222. return doGc();
  223. }
  224. final GcLog gcLog = new GcLog(repo);
  225. if (!gcLog.lock()) {
  226. // there is already a background gc running
  227. return Collections.emptyList();
  228. }
  229. Callable<Collection<PackFile>> gcTask = () -> {
  230. try {
  231. Collection<PackFile> newPacks = doGc();
  232. if (automatic && tooManyLooseObjects()) {
  233. String message = JGitText.get().gcTooManyUnpruned;
  234. gcLog.write(message);
  235. gcLog.commit();
  236. }
  237. return newPacks;
  238. } catch (IOException | ParseException e) {
  239. try {
  240. gcLog.write(e.getMessage());
  241. StringWriter sw = new StringWriter();
  242. e.printStackTrace(new PrintWriter(sw));
  243. gcLog.write(sw.toString());
  244. gcLog.commit();
  245. } catch (IOException e2) {
  246. e2.addSuppressed(e);
  247. LOG.error(e2.getMessage(), e2);
  248. }
  249. } finally {
  250. gcLog.unlock();
  251. }
  252. return Collections.emptyList();
  253. };
  254. // TODO(ms): in 5.0 change signature and return the Future
  255. executor().submit(gcTask);
  256. return Collections.emptyList();
  257. }
  258. private ExecutorService executor() {
  259. return (executor != null) ? executor : WorkQueue.getExecutor();
  260. }
  261. private Collection<PackFile> doGc() throws IOException, ParseException {
  262. if (automatic && !needGc()) {
  263. return Collections.emptyList();
  264. }
  265. pm.start(6 /* tasks */);
  266. packRefs();
  267. // TODO: implement reflog_expire(pm, repo);
  268. Collection<PackFile> newPacks = repack();
  269. prune(Collections.<ObjectId> emptySet());
  270. // TODO: implement rerere_gc(pm);
  271. return newPacks;
  272. }
  273. /**
  274. * Loosen objects in a pack file which are not also in the newly-created
  275. * pack files.
  276. *
  277. * @param inserter
  278. * @param reader
  279. * @param pack
  280. * @param existing
  281. * @throws IOException
  282. */
  283. private void loosen(ObjectDirectoryInserter inserter, ObjectReader reader, PackFile pack, HashSet<ObjectId> existing)
  284. throws IOException {
  285. for (PackIndex.MutableEntry entry : pack) {
  286. ObjectId oid = entry.toObjectId();
  287. if (existing.contains(oid)) {
  288. continue;
  289. }
  290. existing.add(oid);
  291. ObjectLoader loader = reader.open(oid);
  292. inserter.insert(loader.getType(),
  293. loader.getSize(),
  294. loader.openStream(),
  295. true /* create this object even though it's a duplicate */);
  296. }
  297. }
  298. /**
  299. * Delete old pack files. What is 'old' is defined by specifying a set of
  300. * old pack files and a set of new pack files. Each pack file contained in
  301. * old pack files but not contained in new pack files will be deleted. If
  302. * preserveOldPacks is set, keep a copy of the pack file in the preserve
  303. * directory. If an expirationDate is set then pack files which are younger
  304. * than the expirationDate will not be deleted nor preserved.
  305. * <p>
  306. * If we're not immediately expiring loose objects, loosen any objects
  307. * in the old pack files which aren't in the new pack files.
  308. *
  309. * @param oldPacks
  310. * @param newPacks
  311. * @throws ParseException
  312. * @throws IOException
  313. */
  314. private void deleteOldPacks(Collection<PackFile> oldPacks,
  315. Collection<PackFile> newPacks) throws ParseException, IOException {
  316. HashSet<ObjectId> ids = new HashSet<>();
  317. for (PackFile pack : newPacks) {
  318. for (PackIndex.MutableEntry entry : pack) {
  319. ids.add(entry.toObjectId());
  320. }
  321. }
  322. ObjectReader reader = repo.newObjectReader();
  323. ObjectDirectory dir = repo.getObjectDatabase();
  324. ObjectDirectoryInserter inserter = dir.newInserter();
  325. boolean shouldLoosen = !"now".equals(getPruneExpireStr()) && //$NON-NLS-1$
  326. getExpireDate() < Long.MAX_VALUE;
  327. prunePreserved();
  328. long packExpireDate = getPackExpireDate();
  329. oldPackLoop: for (PackFile oldPack : oldPacks) {
  330. checkCancelled();
  331. String oldName = oldPack.getPackName();
  332. // check whether an old pack file is also among the list of new
  333. // pack files. Then we must not delete it.
  334. for (PackFile newPack : newPacks)
  335. if (oldName.equals(newPack.getPackName()))
  336. continue oldPackLoop;
  337. if (!oldPack.shouldBeKept()
  338. && repo.getFS().lastModified(
  339. oldPack.getPackFile()) < packExpireDate) {
  340. oldPack.close();
  341. if (shouldLoosen) {
  342. loosen(inserter, reader, oldPack, ids);
  343. }
  344. prunePack(oldName);
  345. }
  346. }
  347. // close the complete object database. That's my only chance to force
  348. // rescanning and to detect that certain pack files are now deleted.
  349. repo.getObjectDatabase().close();
  350. }
  351. /**
  352. * Deletes old pack file, unless 'preserve-oldpacks' is set, in which case it
  353. * moves the pack file to the preserved directory
  354. *
  355. * @param packFile
  356. * @param packName
  357. * @param ext
  358. * @param deleteOptions
  359. * @throws IOException
  360. */
  361. private void removeOldPack(File packFile, String packName, PackExt ext,
  362. int deleteOptions) throws IOException {
  363. if (pconfig != null && pconfig.isPreserveOldPacks()) {
  364. File oldPackDir = repo.getObjectDatabase().getPreservedDirectory();
  365. FileUtils.mkdir(oldPackDir, true);
  366. String oldPackName = "pack-" + packName + ".old-" + ext.getExtension(); //$NON-NLS-1$ //$NON-NLS-2$
  367. File oldPackFile = new File(oldPackDir, oldPackName);
  368. FileUtils.rename(packFile, oldPackFile);
  369. } else {
  370. FileUtils.delete(packFile, deleteOptions);
  371. }
  372. }
  373. /**
  374. * Delete the preserved directory including all pack files within
  375. */
  376. private void prunePreserved() {
  377. if (pconfig != null && pconfig.isPrunePreserved()) {
  378. try {
  379. FileUtils.delete(repo.getObjectDatabase().getPreservedDirectory(),
  380. FileUtils.RECURSIVE | FileUtils.RETRY | FileUtils.SKIP_MISSING);
  381. } catch (IOException e) {
  382. // Deletion of the preserved pack files failed. Silently return.
  383. }
  384. }
  385. }
  386. /**
  387. * Delete files associated with a single pack file. First try to delete the
  388. * ".pack" file because on some platforms the ".pack" file may be locked and
  389. * can't be deleted. In such a case it is better to detect this early and
  390. * give up on deleting files for this packfile. Otherwise we may delete the
  391. * ".index" file and when failing to delete the ".pack" file we are left
  392. * with a ".pack" file without a ".index" file.
  393. *
  394. * @param packName
  395. */
  396. private void prunePack(String packName) {
  397. PackExt[] extensions = PackExt.values();
  398. try {
  399. // Delete the .pack file first and if this fails give up on deleting
  400. // the other files
  401. int deleteOptions = FileUtils.RETRY | FileUtils.SKIP_MISSING;
  402. for (PackExt ext : extensions)
  403. if (PackExt.PACK.equals(ext)) {
  404. File f = nameFor(packName, "." + ext.getExtension()); //$NON-NLS-1$
  405. removeOldPack(f, packName, ext, deleteOptions);
  406. break;
  407. }
  408. // The .pack file has been deleted. Delete as many as the other
  409. // files as you can.
  410. deleteOptions |= FileUtils.IGNORE_ERRORS;
  411. for (PackExt ext : extensions) {
  412. if (!PackExt.PACK.equals(ext)) {
  413. File f = nameFor(packName, "." + ext.getExtension()); //$NON-NLS-1$
  414. removeOldPack(f, packName, ext, deleteOptions);
  415. }
  416. }
  417. } catch (IOException e) {
  418. // Deletion of the .pack file failed. Silently return.
  419. }
  420. }
  421. /**
  422. * Like "git prune-packed" this method tries to prune all loose objects
  423. * which can be found in packs. If certain objects can't be pruned (e.g.
  424. * because the filesystem delete operation fails) this is silently ignored.
  425. *
  426. * @throws java.io.IOException
  427. */
  428. public void prunePacked() throws IOException {
  429. ObjectDirectory objdb = repo.getObjectDatabase();
  430. Collection<PackFile> packs = objdb.getPacks();
  431. File objects = repo.getObjectsDirectory();
  432. String[] fanout = objects.list();
  433. if (fanout != null && fanout.length > 0) {
  434. pm.beginTask(JGitText.get().pruneLoosePackedObjects, fanout.length);
  435. try {
  436. for (String d : fanout) {
  437. checkCancelled();
  438. pm.update(1);
  439. if (d.length() != 2)
  440. continue;
  441. String[] entries = new File(objects, d).list();
  442. if (entries == null)
  443. continue;
  444. for (String e : entries) {
  445. checkCancelled();
  446. if (e.length() != Constants.OBJECT_ID_STRING_LENGTH - 2)
  447. continue;
  448. ObjectId id;
  449. try {
  450. id = ObjectId.fromString(d + e);
  451. } catch (IllegalArgumentException notAnObject) {
  452. // ignoring the file that does not represent loose
  453. // object
  454. continue;
  455. }
  456. boolean found = false;
  457. for (PackFile p : packs) {
  458. checkCancelled();
  459. if (p.hasObject(id)) {
  460. found = true;
  461. break;
  462. }
  463. }
  464. if (found)
  465. FileUtils.delete(objdb.fileFor(id), FileUtils.RETRY
  466. | FileUtils.SKIP_MISSING
  467. | FileUtils.IGNORE_ERRORS);
  468. }
  469. }
  470. } finally {
  471. pm.endTask();
  472. }
  473. }
  474. }
  475. /**
  476. * Like "git prune" this method tries to prune all loose objects which are
  477. * unreferenced. If certain objects can't be pruned (e.g. because the
  478. * filesystem delete operation fails) this is silently ignored.
  479. *
  480. * @param objectsToKeep
  481. * a set of objects which should explicitly not be pruned
  482. * @throws java.io.IOException
  483. * @throws java.text.ParseException
  484. * If the configuration parameter "gc.pruneexpire" couldn't be
  485. * parsed
  486. */
  487. public void prune(Set<ObjectId> objectsToKeep) throws IOException,
  488. ParseException {
  489. long expireDate = getExpireDate();
  490. // Collect all loose objects which are old enough, not referenced from
  491. // the index and not in objectsToKeep
  492. Map<ObjectId, File> deletionCandidates = new HashMap<>();
  493. Set<ObjectId> indexObjects = null;
  494. File objects = repo.getObjectsDirectory();
  495. String[] fanout = objects.list();
  496. if (fanout == null || fanout.length == 0) {
  497. return;
  498. }
  499. pm.beginTask(JGitText.get().pruneLooseUnreferencedObjects,
  500. fanout.length);
  501. try {
  502. for (String d : fanout) {
  503. checkCancelled();
  504. pm.update(1);
  505. if (d.length() != 2)
  506. continue;
  507. File[] entries = new File(objects, d).listFiles();
  508. if (entries == null)
  509. continue;
  510. for (File f : entries) {
  511. checkCancelled();
  512. String fName = f.getName();
  513. if (fName.length() != Constants.OBJECT_ID_STRING_LENGTH - 2)
  514. continue;
  515. if (repo.getFS().lastModified(f) >= expireDate)
  516. continue;
  517. try {
  518. ObjectId id = ObjectId.fromString(d + fName);
  519. if (objectsToKeep.contains(id))
  520. continue;
  521. if (indexObjects == null)
  522. indexObjects = listNonHEADIndexObjects();
  523. if (indexObjects.contains(id))
  524. continue;
  525. deletionCandidates.put(id, f);
  526. } catch (IllegalArgumentException notAnObject) {
  527. // ignoring the file that does not represent loose
  528. // object
  529. continue;
  530. }
  531. }
  532. }
  533. } finally {
  534. pm.endTask();
  535. }
  536. if (deletionCandidates.isEmpty()) {
  537. return;
  538. }
  539. checkCancelled();
  540. // From the set of current refs remove all those which have been handled
  541. // during last repack(). Only those refs will survive which have been
  542. // added or modified since the last repack. Only these can save existing
  543. // loose refs from being pruned.
  544. Collection<Ref> newRefs;
  545. if (lastPackedRefs == null || lastPackedRefs.isEmpty())
  546. newRefs = getAllRefs();
  547. else {
  548. Map<String, Ref> last = new HashMap<>();
  549. for (Ref r : lastPackedRefs) {
  550. last.put(r.getName(), r);
  551. }
  552. newRefs = new ArrayList<>();
  553. for (Ref r : getAllRefs()) {
  554. Ref old = last.get(r.getName());
  555. if (!equals(r, old)) {
  556. newRefs.add(r);
  557. }
  558. }
  559. }
  560. if (!newRefs.isEmpty()) {
  561. // There are new/modified refs! Check which loose objects are now
  562. // referenced by these modified refs (or their reflogentries).
  563. // Remove these loose objects
  564. // from the deletionCandidates. When the last candidate is removed
  565. // leave this method.
  566. ObjectWalk w = new ObjectWalk(repo);
  567. try {
  568. for (Ref cr : newRefs) {
  569. checkCancelled();
  570. w.markStart(w.parseAny(cr.getObjectId()));
  571. }
  572. if (lastPackedRefs != null)
  573. for (Ref lpr : lastPackedRefs) {
  574. w.markUninteresting(w.parseAny(lpr.getObjectId()));
  575. }
  576. removeReferenced(deletionCandidates, w);
  577. } finally {
  578. w.dispose();
  579. }
  580. }
  581. if (deletionCandidates.isEmpty())
  582. return;
  583. // Since we have not left the method yet there are still
  584. // deletionCandidates. Last chance for these objects not to be pruned is
  585. // that they are referenced by reflog entries. Even refs which currently
  586. // point to the same object as during last repack() may have
  587. // additional reflog entries not handled during last repack()
  588. ObjectWalk w = new ObjectWalk(repo);
  589. try {
  590. for (Ref ar : getAllRefs())
  591. for (ObjectId id : listRefLogObjects(ar, lastRepackTime)) {
  592. checkCancelled();
  593. w.markStart(w.parseAny(id));
  594. }
  595. if (lastPackedRefs != null)
  596. for (Ref lpr : lastPackedRefs) {
  597. checkCancelled();
  598. w.markUninteresting(w.parseAny(lpr.getObjectId()));
  599. }
  600. removeReferenced(deletionCandidates, w);
  601. } finally {
  602. w.dispose();
  603. }
  604. if (deletionCandidates.isEmpty())
  605. return;
  606. checkCancelled();
  607. // delete all candidates which have survived: these are unreferenced
  608. // loose objects. Make a last check, though, to avoid deleting objects
  609. // that could have been referenced while the candidates list was being
  610. // built (by an incoming push, for example).
  611. Set<File> touchedFanout = new HashSet<>();
  612. for (File f : deletionCandidates.values()) {
  613. if (f.lastModified() < expireDate) {
  614. f.delete();
  615. touchedFanout.add(f.getParentFile());
  616. }
  617. }
  618. for (File f : touchedFanout) {
  619. FileUtils.delete(f,
  620. FileUtils.EMPTY_DIRECTORIES_ONLY | FileUtils.IGNORE_ERRORS);
  621. }
  622. repo.getObjectDatabase().close();
  623. }
  624. private long getExpireDate() throws ParseException {
  625. long expireDate = Long.MAX_VALUE;
  626. if (expire == null && expireAgeMillis == -1) {
  627. String pruneExpireStr = getPruneExpireStr();
  628. if (pruneExpireStr == null)
  629. pruneExpireStr = PRUNE_EXPIRE_DEFAULT;
  630. expire = GitDateParser.parse(pruneExpireStr, null, SystemReader
  631. .getInstance().getLocale());
  632. expireAgeMillis = -1;
  633. }
  634. if (expire != null)
  635. expireDate = expire.getTime();
  636. if (expireAgeMillis != -1)
  637. expireDate = System.currentTimeMillis() - expireAgeMillis;
  638. return expireDate;
  639. }
  640. private String getPruneExpireStr() {
  641. return repo.getConfig().getString(
  642. ConfigConstants.CONFIG_GC_SECTION, null,
  643. ConfigConstants.CONFIG_KEY_PRUNEEXPIRE);
  644. }
  645. private long getPackExpireDate() throws ParseException {
  646. long packExpireDate = Long.MAX_VALUE;
  647. if (packExpire == null && packExpireAgeMillis == -1) {
  648. String prunePackExpireStr = repo.getConfig().getString(
  649. ConfigConstants.CONFIG_GC_SECTION, null,
  650. ConfigConstants.CONFIG_KEY_PRUNEPACKEXPIRE);
  651. if (prunePackExpireStr == null)
  652. prunePackExpireStr = PRUNE_PACK_EXPIRE_DEFAULT;
  653. packExpire = GitDateParser.parse(prunePackExpireStr, null,
  654. SystemReader.getInstance().getLocale());
  655. packExpireAgeMillis = -1;
  656. }
  657. if (packExpire != null)
  658. packExpireDate = packExpire.getTime();
  659. if (packExpireAgeMillis != -1)
  660. packExpireDate = System.currentTimeMillis() - packExpireAgeMillis;
  661. return packExpireDate;
  662. }
  663. /**
  664. * Remove all entries from a map which key is the id of an object referenced
  665. * by the given ObjectWalk
  666. *
  667. * @param id2File
  668. * @param w
  669. * @throws MissingObjectException
  670. * @throws IncorrectObjectTypeException
  671. * @throws IOException
  672. */
  673. private void removeReferenced(Map<ObjectId, File> id2File,
  674. ObjectWalk w) throws MissingObjectException,
  675. IncorrectObjectTypeException, IOException {
  676. RevObject ro = w.next();
  677. while (ro != null) {
  678. checkCancelled();
  679. if (id2File.remove(ro.getId()) != null)
  680. if (id2File.isEmpty())
  681. return;
  682. ro = w.next();
  683. }
  684. ro = w.nextObject();
  685. while (ro != null) {
  686. checkCancelled();
  687. if (id2File.remove(ro.getId()) != null)
  688. if (id2File.isEmpty())
  689. return;
  690. ro = w.nextObject();
  691. }
  692. }
  693. private static boolean equals(Ref r1, Ref r2) {
  694. if (r1 == null || r2 == null)
  695. return false;
  696. if (r1.isSymbolic()) {
  697. if (!r2.isSymbolic())
  698. return false;
  699. return r1.getTarget().getName().equals(r2.getTarget().getName());
  700. } else {
  701. if (r2.isSymbolic()) {
  702. return false;
  703. }
  704. return Objects.equals(r1.getObjectId(), r2.getObjectId());
  705. }
  706. }
  707. /**
  708. * Packs all non-symbolic, loose refs into packed-refs.
  709. *
  710. * @throws java.io.IOException
  711. */
  712. public void packRefs() throws IOException {
  713. Collection<Ref> refs = repo.getRefDatabase().getRefs(Constants.R_REFS).values();
  714. List<String> refsToBePacked = new ArrayList<>(refs.size());
  715. pm.beginTask(JGitText.get().packRefs, refs.size());
  716. try {
  717. for (Ref ref : refs) {
  718. checkCancelled();
  719. if (!ref.isSymbolic() && ref.getStorage().isLoose())
  720. refsToBePacked.add(ref.getName());
  721. pm.update(1);
  722. }
  723. ((RefDirectory) repo.getRefDatabase()).pack(refsToBePacked);
  724. } finally {
  725. pm.endTask();
  726. }
  727. }
  728. /**
  729. * Packs all objects which reachable from any of the heads into one pack
  730. * file. Additionally all objects which are not reachable from any head but
  731. * which are reachable from any of the other refs (e.g. tags), special refs
  732. * (e.g. FETCH_HEAD) or index are packed into a separate pack file. Objects
  733. * included in pack files which have a .keep file associated are never
  734. * repacked. All old pack files which existed before are deleted.
  735. *
  736. * @return a collection of the newly created pack files
  737. * @throws java.io.IOException
  738. * when during reading of refs, index, packfiles, objects,
  739. * reflog-entries or during writing to the packfiles
  740. * {@link java.io.IOException} occurs
  741. */
  742. public Collection<PackFile> repack() throws IOException {
  743. Collection<PackFile> toBeDeleted = repo.getObjectDatabase().getPacks();
  744. long time = System.currentTimeMillis();
  745. Collection<Ref> refsBefore = getAllRefs();
  746. Set<ObjectId> allHeadsAndTags = new HashSet<>();
  747. Set<ObjectId> allHeads = new HashSet<>();
  748. Set<ObjectId> allTags = new HashSet<>();
  749. Set<ObjectId> nonHeads = new HashSet<>();
  750. Set<ObjectId> txnHeads = new HashSet<>();
  751. Set<ObjectId> tagTargets = new HashSet<>();
  752. Set<ObjectId> indexObjects = listNonHEADIndexObjects();
  753. RefDatabase refdb = repo.getRefDatabase();
  754. for (Ref ref : refsBefore) {
  755. checkCancelled();
  756. nonHeads.addAll(listRefLogObjects(ref, 0));
  757. if (ref.isSymbolic() || ref.getObjectId() == null) {
  758. continue;
  759. }
  760. if (isHead(ref)) {
  761. allHeads.add(ref.getObjectId());
  762. } else if (isTag(ref)) {
  763. allTags.add(ref.getObjectId());
  764. } else if (RefTreeNames.isRefTree(refdb, ref.getName())) {
  765. txnHeads.add(ref.getObjectId());
  766. } else {
  767. nonHeads.add(ref.getObjectId());
  768. }
  769. if (ref.getPeeledObjectId() != null) {
  770. tagTargets.add(ref.getPeeledObjectId());
  771. }
  772. }
  773. List<ObjectIdSet> excluded = new LinkedList<>();
  774. for (final PackFile f : repo.getObjectDatabase().getPacks()) {
  775. checkCancelled();
  776. if (f.shouldBeKept())
  777. excluded.add(f.getIndex());
  778. }
  779. // Don't exclude tags that are also branch tips
  780. allTags.removeAll(allHeads);
  781. allHeadsAndTags.addAll(allHeads);
  782. allHeadsAndTags.addAll(allTags);
  783. // Hoist all branch tips and tags earlier in the pack file
  784. tagTargets.addAll(allHeadsAndTags);
  785. nonHeads.addAll(indexObjects);
  786. // Combine the GC_REST objects into the GC pack if requested
  787. if (pconfig != null && pconfig.getSinglePack()) {
  788. allHeadsAndTags.addAll(nonHeads);
  789. nonHeads.clear();
  790. }
  791. List<PackFile> ret = new ArrayList<>(2);
  792. PackFile heads = null;
  793. if (!allHeadsAndTags.isEmpty()) {
  794. heads = writePack(allHeadsAndTags, PackWriter.NONE, allTags,
  795. tagTargets, excluded);
  796. if (heads != null) {
  797. ret.add(heads);
  798. excluded.add(0, heads.getIndex());
  799. }
  800. }
  801. if (!nonHeads.isEmpty()) {
  802. PackFile rest = writePack(nonHeads, allHeadsAndTags, PackWriter.NONE,
  803. tagTargets, excluded);
  804. if (rest != null)
  805. ret.add(rest);
  806. }
  807. if (!txnHeads.isEmpty()) {
  808. PackFile txn = writePack(txnHeads, PackWriter.NONE, PackWriter.NONE,
  809. null, excluded);
  810. if (txn != null)
  811. ret.add(txn);
  812. }
  813. try {
  814. deleteOldPacks(toBeDeleted, ret);
  815. } catch (ParseException e) {
  816. // TODO: the exception has to be wrapped into an IOException because
  817. // throwing the ParseException directly would break the API, instead
  818. // we should throw a ConfigInvalidException
  819. throw new IOException(e);
  820. }
  821. prunePacked();
  822. deleteEmptyRefsFolders();
  823. deleteOrphans();
  824. deleteTempPacksIdx();
  825. lastPackedRefs = refsBefore;
  826. lastRepackTime = time;
  827. return ret;
  828. }
  829. private static boolean isHead(Ref ref) {
  830. return ref.getName().startsWith(Constants.R_HEADS);
  831. }
  832. private static boolean isTag(Ref ref) {
  833. return ref.getName().startsWith(Constants.R_TAGS);
  834. }
  835. private void deleteEmptyRefsFolders() throws IOException {
  836. Path refs = repo.getDirectory().toPath().resolve(Constants.R_REFS);
  837. // Avoid deleting a folder that was created after the threshold so that concurrent
  838. // operations trying to create a reference are not impacted
  839. Instant threshold = Instant.now().minus(30, ChronoUnit.SECONDS);
  840. try (Stream<Path> entries = Files.list(refs)) {
  841. Iterator<Path> iterator = entries.iterator();
  842. while (iterator.hasNext()) {
  843. try (Stream<Path> s = Files.list(iterator.next())) {
  844. s.filter(path -> canBeSafelyDeleted(path, threshold)).forEach(this::deleteDir);
  845. }
  846. }
  847. }
  848. }
  849. private boolean canBeSafelyDeleted(Path path, Instant threshold) {
  850. try {
  851. return Files.getLastModifiedTime(path).toInstant().isBefore(threshold);
  852. }
  853. catch (IOException e) {
  854. LOG.warn(MessageFormat.format(
  855. JGitText.get().cannotAccessLastModifiedForSafeDeletion,
  856. path), e);
  857. return false;
  858. }
  859. }
  860. private void deleteDir(Path dir) {
  861. try (Stream<Path> dirs = Files.walk(dir)) {
  862. dirs.filter(this::isDirectory).sorted(Comparator.reverseOrder())
  863. .forEach(this::delete);
  864. } catch (IOException e) {
  865. LOG.error(e.getMessage(), e);
  866. }
  867. }
  868. private boolean isDirectory(Path p) {
  869. return p.toFile().isDirectory();
  870. }
  871. private void delete(Path d) {
  872. try {
  873. Files.delete(d);
  874. } catch (IOException e) {
  875. LOG.error(MessageFormat.format(JGitText.get().cannotDeleteFile, d),
  876. e);
  877. }
  878. }
  879. /**
  880. * Deletes orphans
  881. * <p>
  882. * A file is considered an orphan if it is either a "bitmap" or an index
  883. * file, and its corresponding pack file is missing in the list.
  884. * </p>
  885. */
  886. private void deleteOrphans() {
  887. Path packDir = repo.getObjectDatabase().getPackDirectory().toPath();
  888. List<String> fileNames = null;
  889. try (Stream<Path> files = Files.list(packDir)) {
  890. fileNames = files.map(path -> path.getFileName().toString())
  891. .filter(name -> {
  892. return (name.endsWith(PACK_EXT)
  893. || name.endsWith(BITMAP_EXT)
  894. || name.endsWith(INDEX_EXT));
  895. }).sorted(Collections.reverseOrder())
  896. .collect(Collectors.toList());
  897. } catch (IOException e1) {
  898. // ignore
  899. }
  900. if (fileNames == null) {
  901. return;
  902. }
  903. String base = null;
  904. for (String n : fileNames) {
  905. if (n.endsWith(PACK_EXT)) {
  906. base = n.substring(0, n.lastIndexOf('.'));
  907. } else {
  908. if (base == null || !n.startsWith(base)) {
  909. try {
  910. Files.delete(FileUtils.toPath(new File(packDir.toFile(), n)));
  911. } catch (IOException e) {
  912. LOG.error(e.getMessage(), e);
  913. }
  914. }
  915. }
  916. }
  917. }
  918. private void deleteTempPacksIdx() {
  919. Path packDir = repo.getObjectDatabase().getPackDirectory().toPath();
  920. Instant threshold = Instant.now().minus(1, ChronoUnit.DAYS);
  921. try (DirectoryStream<Path> stream =
  922. Files.newDirectoryStream(packDir, "gc_*_tmp")) { //$NON-NLS-1$
  923. stream.forEach(t -> {
  924. try {
  925. Instant lastModified = Files.getLastModifiedTime(t)
  926. .toInstant();
  927. if (lastModified.isBefore(threshold)) {
  928. Files.deleteIfExists(t);
  929. }
  930. } catch (IOException e) {
  931. LOG.error(e.getMessage(), e);
  932. }
  933. });
  934. } catch (IOException e) {
  935. LOG.error(e.getMessage(), e);
  936. }
  937. }
  938. /**
  939. * @param ref
  940. * the ref which log should be inspected
  941. * @param minTime only reflog entries not older then this time are processed
  942. * @return the {@link ObjectId}s contained in the reflog
  943. * @throws IOException
  944. */
  945. private Set<ObjectId> listRefLogObjects(Ref ref, long minTime) throws IOException {
  946. ReflogReader reflogReader = repo.getReflogReader(ref.getName());
  947. if (reflogReader == null) {
  948. return Collections.emptySet();
  949. }
  950. List<ReflogEntry> rlEntries = reflogReader
  951. .getReverseEntries();
  952. if (rlEntries == null || rlEntries.isEmpty())
  953. return Collections.<ObjectId> emptySet();
  954. Set<ObjectId> ret = new HashSet<>();
  955. for (ReflogEntry e : rlEntries) {
  956. if (e.getWho().getWhen().getTime() < minTime)
  957. break;
  958. ObjectId newId = e.getNewId();
  959. if (newId != null && !ObjectId.zeroId().equals(newId))
  960. ret.add(newId);
  961. ObjectId oldId = e.getOldId();
  962. if (oldId != null && !ObjectId.zeroId().equals(oldId))
  963. ret.add(oldId);
  964. }
  965. return ret;
  966. }
  967. /**
  968. * Returns a collection of all refs and additional refs.
  969. *
  970. * Additional refs which don't start with "refs/" are not returned because
  971. * they should not save objects from being garbage collected. Examples for
  972. * such references are ORIG_HEAD, MERGE_HEAD, FETCH_HEAD and
  973. * CHERRY_PICK_HEAD.
  974. *
  975. * @return a collection of refs pointing to live objects.
  976. * @throws IOException
  977. */
  978. private Collection<Ref> getAllRefs() throws IOException {
  979. RefDatabase refdb = repo.getRefDatabase();
  980. Collection<Ref> refs = refdb.getRefs(RefDatabase.ALL).values();
  981. List<Ref> addl = refdb.getAdditionalRefs();
  982. if (!addl.isEmpty()) {
  983. List<Ref> all = new ArrayList<>(refs.size() + addl.size());
  984. all.addAll(refs);
  985. // add additional refs which start with refs/
  986. for (Ref r : addl) {
  987. checkCancelled();
  988. if (r.getName().startsWith(Constants.R_REFS)) {
  989. all.add(r);
  990. }
  991. }
  992. return all;
  993. }
  994. return refs;
  995. }
  996. /**
  997. * Return a list of those objects in the index which differ from whats in
  998. * HEAD
  999. *
  1000. * @return a set of ObjectIds of changed objects in the index
  1001. * @throws IOException
  1002. * @throws CorruptObjectException
  1003. * @throws NoWorkTreeException
  1004. */
  1005. private Set<ObjectId> listNonHEADIndexObjects()
  1006. throws CorruptObjectException, IOException {
  1007. if (repo.isBare()) {
  1008. return Collections.emptySet();
  1009. }
  1010. try (TreeWalk treeWalk = new TreeWalk(repo)) {
  1011. treeWalk.addTree(new DirCacheIterator(repo.readDirCache()));
  1012. ObjectId headID = repo.resolve(Constants.HEAD);
  1013. if (headID != null) {
  1014. try (RevWalk revWalk = new RevWalk(repo)) {
  1015. treeWalk.addTree(revWalk.parseTree(headID));
  1016. }
  1017. }
  1018. treeWalk.setFilter(TreeFilter.ANY_DIFF);
  1019. treeWalk.setRecursive(true);
  1020. Set<ObjectId> ret = new HashSet<>();
  1021. while (treeWalk.next()) {
  1022. checkCancelled();
  1023. ObjectId objectId = treeWalk.getObjectId(0);
  1024. switch (treeWalk.getRawMode(0) & FileMode.TYPE_MASK) {
  1025. case FileMode.TYPE_MISSING:
  1026. case FileMode.TYPE_GITLINK:
  1027. continue;
  1028. case FileMode.TYPE_TREE:
  1029. case FileMode.TYPE_FILE:
  1030. case FileMode.TYPE_SYMLINK:
  1031. ret.add(objectId);
  1032. continue;
  1033. default:
  1034. throw new IOException(MessageFormat.format(
  1035. JGitText.get().corruptObjectInvalidMode3,
  1036. String.format("%o", //$NON-NLS-1$
  1037. Integer.valueOf(treeWalk.getRawMode(0))),
  1038. (objectId == null) ? "null" : objectId.name(), //$NON-NLS-1$
  1039. treeWalk.getPathString(), //
  1040. repo.getIndexFile()));
  1041. }
  1042. }
  1043. return ret;
  1044. }
  1045. }
  1046. private PackFile writePack(@NonNull Set<? extends ObjectId> want,
  1047. @NonNull Set<? extends ObjectId> have, @NonNull Set<ObjectId> tags,
  1048. Set<ObjectId> tagTargets, List<ObjectIdSet> excludeObjects)
  1049. throws IOException {
  1050. checkCancelled();
  1051. File tmpPack = null;
  1052. Map<PackExt, File> tmpExts = new TreeMap<>(
  1053. new Comparator<PackExt>() {
  1054. @Override
  1055. public int compare(PackExt o1, PackExt o2) {
  1056. // INDEX entries must be returned last, so the pack
  1057. // scanner does pick up the new pack until all the
  1058. // PackExt entries have been written.
  1059. if (o1 == o2)
  1060. return 0;
  1061. if (o1 == PackExt.INDEX)
  1062. return 1;
  1063. if (o2 == PackExt.INDEX)
  1064. return -1;
  1065. return Integer.signum(o1.hashCode() - o2.hashCode());
  1066. }
  1067. });
  1068. try (PackWriter pw = new PackWriter(
  1069. (pconfig == null) ? new PackConfig(repo) : pconfig,
  1070. repo.newObjectReader())) {
  1071. // prepare the PackWriter
  1072. pw.setDeltaBaseAsOffset(true);
  1073. pw.setReuseDeltaCommits(false);
  1074. if (tagTargets != null) {
  1075. pw.setTagTargets(tagTargets);
  1076. }
  1077. if (excludeObjects != null)
  1078. for (ObjectIdSet idx : excludeObjects)
  1079. pw.excludeObjects(idx);
  1080. pw.preparePack(pm, want, have, PackWriter.NONE, tags);
  1081. if (pw.getObjectCount() == 0)
  1082. return null;
  1083. checkCancelled();
  1084. // create temporary files
  1085. String id = pw.computeName().getName();
  1086. File packdir = repo.getObjectDatabase().getPackDirectory();
  1087. tmpPack = File.createTempFile("gc_", ".pack_tmp", packdir); //$NON-NLS-1$ //$NON-NLS-2$
  1088. final String tmpBase = tmpPack.getName()
  1089. .substring(0, tmpPack.getName().lastIndexOf('.'));
  1090. File tmpIdx = new File(packdir, tmpBase + ".idx_tmp"); //$NON-NLS-1$
  1091. tmpExts.put(INDEX, tmpIdx);
  1092. if (!tmpIdx.createNewFile())
  1093. throw new IOException(MessageFormat.format(
  1094. JGitText.get().cannotCreateIndexfile, tmpIdx.getPath()));
  1095. // write the packfile
  1096. FileOutputStream fos = new FileOutputStream(tmpPack);
  1097. FileChannel channel = fos.getChannel();
  1098. OutputStream channelStream = Channels.newOutputStream(channel);
  1099. try {
  1100. pw.writePack(pm, pm, channelStream);
  1101. } finally {
  1102. channel.force(true);
  1103. channelStream.close();
  1104. fos.close();
  1105. }
  1106. // write the packindex
  1107. fos = new FileOutputStream(tmpIdx);
  1108. FileChannel idxChannel = fos.getChannel();
  1109. OutputStream idxStream = Channels.newOutputStream(idxChannel);
  1110. try {
  1111. pw.writeIndex(idxStream);
  1112. } finally {
  1113. idxChannel.force(true);
  1114. idxStream.close();
  1115. fos.close();
  1116. }
  1117. if (pw.prepareBitmapIndex(pm)) {
  1118. File tmpBitmapIdx = new File(packdir, tmpBase + ".bitmap_tmp"); //$NON-NLS-1$
  1119. tmpExts.put(BITMAP_INDEX, tmpBitmapIdx);
  1120. if (!tmpBitmapIdx.createNewFile())
  1121. throw new IOException(MessageFormat.format(
  1122. JGitText.get().cannotCreateIndexfile,
  1123. tmpBitmapIdx.getPath()));
  1124. fos = new FileOutputStream(tmpBitmapIdx);
  1125. idxChannel = fos.getChannel();
  1126. idxStream = Channels.newOutputStream(idxChannel);
  1127. try {
  1128. pw.writeBitmapIndex(idxStream);
  1129. } finally {
  1130. idxChannel.force(true);
  1131. idxStream.close();
  1132. fos.close();
  1133. }
  1134. }
  1135. // rename the temporary files to real files
  1136. File realPack = nameFor(id, ".pack"); //$NON-NLS-1$
  1137. repo.getObjectDatabase().closeAllPackHandles(realPack);
  1138. tmpPack.setReadOnly();
  1139. FileUtils.rename(tmpPack, realPack, StandardCopyOption.ATOMIC_MOVE);
  1140. for (Map.Entry<PackExt, File> tmpEntry : tmpExts.entrySet()) {
  1141. File tmpExt = tmpEntry.getValue();
  1142. tmpExt.setReadOnly();
  1143. File realExt = nameFor(id,
  1144. "." + tmpEntry.getKey().getExtension()); //$NON-NLS-1$
  1145. try {
  1146. FileUtils.rename(tmpExt, realExt,
  1147. StandardCopyOption.ATOMIC_MOVE);
  1148. } catch (IOException e) {
  1149. File newExt = new File(realExt.getParentFile(),
  1150. realExt.getName() + ".new"); //$NON-NLS-1$
  1151. try {
  1152. FileUtils.rename(tmpExt, newExt,
  1153. StandardCopyOption.ATOMIC_MOVE);
  1154. } catch (IOException e2) {
  1155. newExt = tmpExt;
  1156. e = e2;
  1157. }
  1158. throw new IOException(MessageFormat.format(
  1159. JGitText.get().panicCantRenameIndexFile, newExt,
  1160. realExt), e);
  1161. }
  1162. }
  1163. return repo.getObjectDatabase().openPack(realPack);
  1164. } finally {
  1165. if (tmpPack != null && tmpPack.exists())
  1166. tmpPack.delete();
  1167. for (File tmpExt : tmpExts.values()) {
  1168. if (tmpExt.exists())
  1169. tmpExt.delete();
  1170. }
  1171. }
  1172. }
  1173. private File nameFor(String name, String ext) {
  1174. File packdir = repo.getObjectDatabase().getPackDirectory();
  1175. return new File(packdir, "pack-" + name + ext); //$NON-NLS-1$
  1176. }
  1177. private void checkCancelled() throws CancelledException {
  1178. if (pm.isCancelled()) {
  1179. throw new CancelledException(JGitText.get().operationCanceled);
  1180. }
  1181. }
  1182. /**
  1183. * A class holding statistical data for a FileRepository regarding how many
  1184. * objects are stored as loose or packed objects
  1185. */
  1186. public static class RepoStatistics {
  1187. /**
  1188. * The number of objects stored in pack files. If the same object is
  1189. * stored in multiple pack files then it is counted as often as it
  1190. * occurs in pack files.
  1191. */
  1192. public long numberOfPackedObjects;
  1193. /**
  1194. * The number of pack files
  1195. */
  1196. public long numberOfPackFiles;
  1197. /**
  1198. * The number of objects stored as loose objects.
  1199. */
  1200. public long numberOfLooseObjects;
  1201. /**
  1202. * The sum of the sizes of all files used to persist loose objects.
  1203. */
  1204. public long sizeOfLooseObjects;
  1205. /**
  1206. * The sum of the sizes of all pack files.
  1207. */
  1208. public long sizeOfPackedObjects;
  1209. /**
  1210. * The number of loose refs.
  1211. */
  1212. public long numberOfLooseRefs;
  1213. /**
  1214. * The number of refs stored in pack files.
  1215. */
  1216. public long numberOfPackedRefs;
  1217. /**
  1218. * The number of bitmaps in the bitmap indices.
  1219. */
  1220. public long numberOfBitmaps;
  1221. @Override
  1222. public String toString() {
  1223. final StringBuilder b = new StringBuilder();
  1224. b.append("numberOfPackedObjects=").append(numberOfPackedObjects); //$NON-NLS-1$
  1225. b.append(", numberOfPackFiles=").append(numberOfPackFiles); //$NON-NLS-1$
  1226. b.append(", numberOfLooseObjects=").append(numberOfLooseObjects); //$NON-NLS-1$
  1227. b.append(", numberOfLooseRefs=").append(numberOfLooseRefs); //$NON-NLS-1$
  1228. b.append(", numberOfPackedRefs=").append(numberOfPackedRefs); //$NON-NLS-1$
  1229. b.append(", sizeOfLooseObjects=").append(sizeOfLooseObjects); //$NON-NLS-1$
  1230. b.append(", sizeOfPackedObjects=").append(sizeOfPackedObjects); //$NON-NLS-1$
  1231. b.append(", numberOfBitmaps=").append(numberOfBitmaps); //$NON-NLS-1$
  1232. return b.toString();
  1233. }
  1234. }
  1235. /**
  1236. * Returns information about objects and pack files for a FileRepository.
  1237. *
  1238. * @return information about objects and pack files for a FileRepository
  1239. * @throws java.io.IOException
  1240. */
  1241. public RepoStatistics getStatistics() throws IOException {
  1242. RepoStatistics ret = new RepoStatistics();
  1243. Collection<PackFile> packs = repo.getObjectDatabase().getPacks();
  1244. for (PackFile f : packs) {
  1245. ret.numberOfPackedObjects += f.getIndex().getObjectCount();
  1246. ret.numberOfPackFiles++;
  1247. ret.sizeOfPackedObjects += f.getPackFile().length();
  1248. if (f.getBitmapIndex() != null)
  1249. ret.numberOfBitmaps += f.getBitmapIndex().getBitmapCount();
  1250. }
  1251. File objDir = repo.getObjectsDirectory();
  1252. String[] fanout = objDir.list();
  1253. if (fanout != null && fanout.length > 0) {
  1254. for (String d : fanout) {
  1255. if (d.length() != 2)
  1256. continue;
  1257. File[] entries = new File(objDir, d).listFiles();
  1258. if (entries == null)
  1259. continue;
  1260. for (File f : entries) {
  1261. if (f.getName().length() != Constants.OBJECT_ID_STRING_LENGTH - 2)
  1262. continue;
  1263. ret.numberOfLooseObjects++;
  1264. ret.sizeOfLooseObjects += f.length();
  1265. }
  1266. }
  1267. }
  1268. RefDatabase refDb = repo.getRefDatabase();
  1269. for (Ref r : refDb.getRefs(RefDatabase.ALL).values()) {
  1270. Storage storage = r.getStorage();
  1271. if (storage == Storage.LOOSE || storage == Storage.LOOSE_PACKED)
  1272. ret.numberOfLooseRefs++;
  1273. if (storage == Storage.PACKED || storage == Storage.LOOSE_PACKED)
  1274. ret.numberOfPackedRefs++;
  1275. }
  1276. return ret;
  1277. }
  1278. /**
  1279. * Set the progress monitor used for garbage collection methods.
  1280. *
  1281. * @param pm a {@link org.eclipse.jgit.lib.ProgressMonitor} object.
  1282. * @return this
  1283. */
  1284. public GC setProgressMonitor(ProgressMonitor pm) {
  1285. this.pm = (pm == null) ? NullProgressMonitor.INSTANCE : pm;
  1286. return this;
  1287. }
  1288. /**
  1289. * During gc() or prune() each unreferenced, loose object which has been
  1290. * created or modified in the last <code>expireAgeMillis</code> milliseconds
  1291. * will not be pruned. Only older objects may be pruned. If set to 0 then
  1292. * every object is a candidate for pruning.
  1293. *
  1294. * @param expireAgeMillis
  1295. * minimal age of objects to be pruned in milliseconds.
  1296. */
  1297. public void setExpireAgeMillis(long expireAgeMillis) {
  1298. this.expireAgeMillis = expireAgeMillis;
  1299. expire = null;
  1300. }
  1301. /**
  1302. * During gc() or prune() packfiles which are created or modified in the
  1303. * last <code>packExpireAgeMillis</code> milliseconds will not be deleted.
  1304. * Only older packfiles may be deleted. If set to 0 then every packfile is a
  1305. * candidate for deletion.
  1306. *
  1307. * @param packExpireAgeMillis
  1308. * minimal age of packfiles to be deleted in milliseconds.
  1309. */
  1310. public void setPackExpireAgeMillis(long packExpireAgeMillis) {
  1311. this.packExpireAgeMillis = packExpireAgeMillis;
  1312. expire = null;
  1313. }
  1314. /**
  1315. * Set the PackConfig used when (re-)writing packfiles. This allows to
  1316. * influence how packs are written and to implement something similar to
  1317. * "git gc --aggressive"
  1318. *
  1319. * @param pconfig
  1320. * the {@link org.eclipse.jgit.storage.pack.PackConfig} used when
  1321. * writing packs
  1322. */
  1323. public void setPackConfig(PackConfig pconfig) {
  1324. this.pconfig = pconfig;
  1325. }
  1326. /**
  1327. * During gc() or prune() each unreferenced, loose object which has been
  1328. * created or modified after or at <code>expire</code> will not be pruned.
  1329. * Only older objects may be pruned. If set to null then every object is a
  1330. * candidate for pruning.
  1331. *
  1332. * @param expire
  1333. * instant in time which defines object expiration
  1334. * objects with modification time before this instant are expired
  1335. * objects with modification time newer or equal to this instant
  1336. * are not expired
  1337. */
  1338. public void setExpire(Date expire) {
  1339. this.expire = expire;
  1340. expireAgeMillis = -1;
  1341. }
  1342. /**
  1343. * During gc() or prune() packfiles which are created or modified after or
  1344. * at <code>packExpire</code> will not be deleted. Only older packfiles may
  1345. * be deleted. If set to null then every packfile is a candidate for
  1346. * deletion.
  1347. *
  1348. * @param packExpire
  1349. * instant in time which defines packfile expiration
  1350. */
  1351. public void setPackExpire(Date packExpire) {
  1352. this.packExpire = packExpire;
  1353. packExpireAgeMillis = -1;
  1354. }
  1355. /**
  1356. * Set the {@code gc --auto} option.
  1357. *
  1358. * With this option, gc checks whether any housekeeping is required; if not,
  1359. * it exits without performing any work. Some JGit commands run
  1360. * {@code gc --auto} after performing operations that could create many
  1361. * loose objects.
  1362. * <p>
  1363. * Housekeeping is required if there are too many loose objects or too many
  1364. * packs in the repository. If the number of loose objects exceeds the value
  1365. * of the gc.auto option JGit GC consolidates all existing packs into a
  1366. * single pack (equivalent to {@code -A} option), whereas git-core would
  1367. * combine all loose objects into a single pack using {@code repack -d -l}.
  1368. * Setting the value of {@code gc.auto} to 0 disables automatic packing of
  1369. * loose objects.
  1370. * <p>
  1371. * If the number of packs exceeds the value of {@code gc.autoPackLimit},
  1372. * then existing packs (except those marked with a .keep file) are
  1373. * consolidated into a single pack by using the {@code -A} option of repack.
  1374. * Setting {@code gc.autoPackLimit} to 0 disables automatic consolidation of
  1375. * packs.
  1376. * <p>
  1377. * Like git the following jgit commands run auto gc:
  1378. * <ul>
  1379. * <li>fetch</li>
  1380. * <li>merge</li>
  1381. * <li>rebase</li>
  1382. * <li>receive-pack</li>
  1383. * </ul>
  1384. * The auto gc for receive-pack can be suppressed by setting the config
  1385. * option {@code receive.autogc = false}
  1386. *
  1387. * @param auto
  1388. * defines whether gc should do automatic housekeeping
  1389. */
  1390. public void setAuto(boolean auto) {
  1391. this.automatic = auto;
  1392. }
  1393. /**
  1394. * @param background
  1395. * whether to run the gc in a background thread.
  1396. */
  1397. void setBackground(boolean background) {
  1398. this.background = background;
  1399. }
  1400. private boolean needGc() {
  1401. if (tooManyPacks()) {
  1402. addRepackAllOption();
  1403. } else if (!tooManyLooseObjects()) {
  1404. return false;
  1405. }
  1406. // TODO run pre-auto-gc hook, if it fails return false
  1407. return true;
  1408. }
  1409. private void addRepackAllOption() {
  1410. // TODO: if JGit GC is enhanced to support repack's option -l this
  1411. // method needs to be implemented
  1412. }
  1413. /**
  1414. * @return {@code true} if number of packs > gc.autopacklimit (default 50)
  1415. */
  1416. boolean tooManyPacks() {
  1417. int autopacklimit = repo.getConfig().getInt(
  1418. ConfigConstants.CONFIG_GC_SECTION,
  1419. ConfigConstants.CONFIG_KEY_AUTOPACKLIMIT,
  1420. DEFAULT_AUTOPACKLIMIT);
  1421. if (autopacklimit <= 0) {
  1422. return false;
  1423. }
  1424. // JGit always creates two packfiles, one for the objects reachable from
  1425. // branches, and another one for the rest
  1426. return repo.getObjectDatabase().getPacks().size() > (autopacklimit + 1);
  1427. }
  1428. /**
  1429. * Quickly estimate number of loose objects, SHA1 is distributed evenly so
  1430. * counting objects in one directory (bucket 17) is sufficient
  1431. *
  1432. * @return {@code true} if number of loose objects > gc.auto (default 6700)
  1433. */
  1434. boolean tooManyLooseObjects() {
  1435. int auto = getLooseObjectLimit();
  1436. if (auto <= 0) {
  1437. return false;
  1438. }
  1439. int n = 0;
  1440. int threshold = (auto + 255) / 256;
  1441. Path dir = repo.getObjectsDirectory().toPath().resolve("17"); //$NON-NLS-1$
  1442. if (!Files.exists(dir)) {
  1443. return false;
  1444. }
  1445. try (DirectoryStream<Path> stream = Files.newDirectoryStream(dir,
  1446. new DirectoryStream.Filter<Path>() {
  1447. @Override
  1448. public boolean accept(Path file) throws IOException {
  1449. Path fileName = file.getFileName();
  1450. return Files.isRegularFile(file) && fileName != null
  1451. && PATTERN_LOOSE_OBJECT
  1452. .matcher(fileName.toString()).matches();
  1453. }
  1454. })) {
  1455. for (Iterator<Path> iter = stream.iterator(); iter.hasNext();
  1456. iter.next()) {
  1457. if (++n > threshold) {
  1458. return true;
  1459. }
  1460. }
  1461. } catch (IOException e) {
  1462. LOG.error(e.getMessage(), e);
  1463. }
  1464. return false;
  1465. }
  1466. private int getLooseObjectLimit() {
  1467. return repo.getConfig().getInt(ConfigConstants.CONFIG_GC_SECTION,
  1468. ConfigConstants.CONFIG_KEY_AUTO, DEFAULT_AUTOLIMIT);
  1469. }
  1470. }