Вы не можете выбрать более 25 тем Темы должны начинаться с буквы или цифры, могут содержать дефисы(-) и должны содержать не более 35 символов.

ObjectDirectory.java 28KB

Added read/write support for pack bitmap index. A pack bitmap index is an additional index of compressed bitmaps of the object graph. Furthermore, a logical API of the index functionality is included, as it is expected to be used by the PackWriter. Compressed bitmaps are created using the javaewah library, which is a word-aligned compressed variant of the Java bitset class based on run-length encoding. The library only works with positive integer values. Thus, the maximum number of ObjectIds in a pack file that this index can currently support is limited to Integer.MAX_VALUE. Every ObjectId is given an integer mapping. The integer is the position of the ObjectId in the complete ObjectId list, sorted by offset, for the pack file. That integer is what the bitmaps use to reference the ObjectId. Currently, the new index format can only be used with pack files that contain a complete closure of the object graph e.g. the result of a garbage collection. The index file includes four bitmaps for the Git object types i.e. commits, trees, blobs, and tags. In addition, a collection of bitmaps keyed by an ObjectId is also included. The bitmap for each entry in the collection represents the full closure of ObjectIds reachable from the keyed ObjectId (including the keyed ObjectId itself). The bitmaps are further compressed by XORing the current bitmaps against prior bitmaps in the index, and selecting the smallest representation. The XOR'd bitmap and offset from the current entry to the position of the bitmap to XOR against is the actual representation of the entry in the index file. Each entry contains one byte, which is currently used to note whether the bitmap should be blindly reused. Change-Id: Id328724bf6b4c8366a088233098c18643edcf40f
11 лет назад
Fix concurrent creation of fan-out object directories If multiple threads attempted to insert loose objects into the same new fan-out directory, the creation of that directory was subject to a race condition that could lead to an unnecessary IOException being thrown - because an inserter could not 'create' a directory that had just been generated by a different thread. All we require is that the directory does indeed *exist*, so not being able to _create_ it is not actually a fatal problem. Setting 'skipExisting' to 'true' on the call to mkdir() fixes the issue. I found this issue as a real world occurrence while working on The BFG Repo Cleaner (https://github.com/rtyley/bfg-repo-cleaner), a tool which concurrently performs a lot of object creation. In order to demonstrate the problem here I've added a small test case which reliably reproduces the issue on the few different hardware systems I've tried. The error thrown when the race-condition arises is this: java.io.IOException: Creating directory /home/roberto/repo.git/objects/e6 failed at org.eclipse.jgit.util.FileUtils.mkdir(FileUtils.java:182) at org.eclipse.jgit.storage.file.ObjectDirectory.insertUnpackedObject(ObjectDirectory.java:590) at org.eclipse.jgit.storage.file.ObjectDirectoryInserter.insertOneObject(ObjectDirectoryInserter.java:113) at org.eclipse.jgit.storage.file.ObjectDirectoryInserter.insert(ObjectDirectoryInserter.java:91) at org.eclipse.jgit.lib.ObjectInserter.insert(ObjectInserter.java:329) Change-Id: I88eac49bc600c56ba9ad290e6133d8a7113125ab
11 лет назад
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964
  1. /*
  2. * Copyright (C) 2009, Google Inc.
  3. * and other copyright owners as documented in the project's IP log.
  4. *
  5. * This program and the accompanying materials are made available
  6. * under the terms of the Eclipse Distribution License v1.0 which
  7. * accompanies this distribution, is reproduced below, and is
  8. * available at http://www.eclipse.org/org/documents/edl-v10.php
  9. *
  10. * All rights reserved.
  11. *
  12. * Redistribution and use in source and binary forms, with or
  13. * without modification, are permitted provided that the following
  14. * conditions are met:
  15. *
  16. * - Redistributions of source code must retain the above copyright
  17. * notice, this list of conditions and the following disclaimer.
  18. *
  19. * - Redistributions in binary form must reproduce the above
  20. * copyright notice, this list of conditions and the following
  21. * disclaimer in the documentation and/or other materials provided
  22. * with the distribution.
  23. *
  24. * - Neither the name of the Eclipse Foundation, Inc. nor the
  25. * names of its contributors may be used to endorse or promote
  26. * products derived from this software without specific prior
  27. * written permission.
  28. *
  29. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
  30. * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
  31. * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  32. * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  33. * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
  34. * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  35. * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  36. * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  37. * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  38. * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  39. * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  40. * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  41. * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  42. */
  43. package org.eclipse.jgit.internal.storage.file;
  44. import static org.eclipse.jgit.internal.storage.pack.PackExt.INDEX;
  45. import static org.eclipse.jgit.internal.storage.pack.PackExt.PACK;
  46. import java.io.BufferedReader;
  47. import java.io.File;
  48. import java.io.FileInputStream;
  49. import java.io.FileNotFoundException;
  50. import java.io.FileReader;
  51. import java.io.IOException;
  52. import java.io.PrintWriter;
  53. import java.io.StringWriter;
  54. import java.text.MessageFormat;
  55. import java.util.ArrayList;
  56. import java.util.Arrays;
  57. import java.util.Collection;
  58. import java.util.Collections;
  59. import java.util.HashMap;
  60. import java.util.HashSet;
  61. import java.util.List;
  62. import java.util.Map;
  63. import java.util.Set;
  64. import java.util.concurrent.atomic.AtomicReference;
  65. import org.eclipse.jgit.errors.CorruptObjectException;
  66. import org.eclipse.jgit.errors.PackInvalidException;
  67. import org.eclipse.jgit.errors.PackMismatchException;
  68. import org.eclipse.jgit.internal.JGitText;
  69. import org.eclipse.jgit.internal.storage.pack.ObjectToPack;
  70. import org.eclipse.jgit.internal.storage.pack.PackExt;
  71. import org.eclipse.jgit.internal.storage.pack.PackWriter;
  72. import org.eclipse.jgit.lib.AbbreviatedObjectId;
  73. import org.eclipse.jgit.lib.AnyObjectId;
  74. import org.eclipse.jgit.lib.Config;
  75. import org.eclipse.jgit.lib.ConfigConstants;
  76. import org.eclipse.jgit.lib.Constants;
  77. import org.eclipse.jgit.lib.ObjectDatabase;
  78. import org.eclipse.jgit.lib.ObjectId;
  79. import org.eclipse.jgit.lib.ObjectLoader;
  80. import org.eclipse.jgit.lib.RepositoryCache;
  81. import org.eclipse.jgit.lib.RepositoryCache.FileKey;
  82. import org.eclipse.jgit.util.FS;
  83. import org.eclipse.jgit.util.FileUtils;
  84. /**
  85. * Traditional file system based {@link ObjectDatabase}.
  86. * <p>
  87. * This is the classical object database representation for a Git repository,
  88. * where objects are stored loose by hashing them into directories by their
  89. * {@link ObjectId}, or are stored in compressed containers known as
  90. * {@link PackFile}s.
  91. * <p>
  92. * Optionally an object database can reference one or more alternates; other
  93. * ObjectDatabase instances that are searched in addition to the current
  94. * database.
  95. * <p>
  96. * Databases are divided into two halves: a half that is considered to be fast
  97. * to search (the {@code PackFile}s), and a half that is considered to be slow
  98. * to search (loose objects). When alternates are present the fast half is fully
  99. * searched (recursively through all alternates) before the slow half is
  100. * considered.
  101. */
  102. public class ObjectDirectory extends FileObjectDatabase {
  103. private static final PackList NO_PACKS = new PackList(
  104. FileSnapshot.DIRTY, new PackFile[0]);
  105. /** Maximum number of candidates offered as resolutions of abbreviation. */
  106. private static final int RESOLVE_ABBREV_LIMIT = 256;
  107. private final Config config;
  108. private final File objects;
  109. private final File infoDirectory;
  110. private final File packDirectory;
  111. private final File alternatesFile;
  112. private final AtomicReference<PackList> packList;
  113. private final FS fs;
  114. private final AtomicReference<AlternateHandle[]> alternates;
  115. private final UnpackedObjectCache unpackedObjectCache;
  116. private final File shallowFile;
  117. private FileSnapshot shallowFileSnapshot = FileSnapshot.DIRTY;
  118. private Set<ObjectId> shallowCommitsIds;
  119. /**
  120. * Initialize a reference to an on-disk object directory.
  121. *
  122. * @param cfg
  123. * configuration this directory consults for write settings.
  124. * @param dir
  125. * the location of the <code>objects</code> directory.
  126. * @param alternatePaths
  127. * a list of alternate object directories
  128. * @param fs
  129. * the file system abstraction which will be necessary to perform
  130. * certain file system operations.
  131. * @param shallowFile
  132. * file which contains IDs of shallow commits, null if shallow
  133. * commits handling should be turned off
  134. * @throws IOException
  135. * an alternate object cannot be opened.
  136. */
  137. public ObjectDirectory(final Config cfg, final File dir,
  138. File[] alternatePaths, FS fs, File shallowFile) throws IOException {
  139. config = cfg;
  140. objects = dir;
  141. infoDirectory = new File(objects, "info"); //$NON-NLS-1$
  142. packDirectory = new File(objects, "pack"); //$NON-NLS-1$
  143. alternatesFile = new File(infoDirectory, "alternates"); //$NON-NLS-1$
  144. packList = new AtomicReference<PackList>(NO_PACKS);
  145. unpackedObjectCache = new UnpackedObjectCache();
  146. this.fs = fs;
  147. this.shallowFile = shallowFile;
  148. alternates = new AtomicReference<AlternateHandle[]>();
  149. if (alternatePaths != null) {
  150. AlternateHandle[] alt;
  151. alt = new AlternateHandle[alternatePaths.length];
  152. for (int i = 0; i < alternatePaths.length; i++)
  153. alt[i] = openAlternate(alternatePaths[i]);
  154. alternates.set(alt);
  155. }
  156. }
  157. /**
  158. * @return the location of the <code>objects</code> directory.
  159. */
  160. public final File getDirectory() {
  161. return objects;
  162. }
  163. @Override
  164. public boolean exists() {
  165. return fs.exists(objects);
  166. }
  167. @Override
  168. public void create() throws IOException {
  169. FileUtils.mkdirs(objects);
  170. FileUtils.mkdir(infoDirectory);
  171. FileUtils.mkdir(packDirectory);
  172. }
  173. @Override
  174. public ObjectDirectoryInserter newInserter() {
  175. return new ObjectDirectoryInserter(this, config);
  176. }
  177. @Override
  178. public void close() {
  179. unpackedObjectCache.clear();
  180. final PackList packs = packList.get();
  181. if (packs != NO_PACKS && packList.compareAndSet(packs, NO_PACKS)) {
  182. for (PackFile p : packs.packs)
  183. p.close();
  184. }
  185. // Fully close all loaded alternates and clear the alternate list.
  186. AlternateHandle[] alt = alternates.get();
  187. if (alt != null && alternates.compareAndSet(alt, null)) {
  188. for(final AlternateHandle od : alt)
  189. od.close();
  190. }
  191. }
  192. /**
  193. * @return unmodifiable collection of all known pack files local to this
  194. * directory. Most recent packs are presented first. Packs most
  195. * likely to contain more recent objects appear before packs
  196. * containing objects referenced by commits further back in the
  197. * history of the repository.
  198. */
  199. @Override
  200. public Collection<PackFile> getPacks() {
  201. PackList list = packList.get();
  202. if (list == NO_PACKS)
  203. list = scanPacks(list);
  204. PackFile[] packs = list.packs;
  205. return Collections.unmodifiableCollection(Arrays.asList(packs));
  206. }
  207. /**
  208. * Add a single existing pack to the list of available pack files.
  209. *
  210. * @param pack
  211. * path of the pack file to open.
  212. * @return the pack that was opened and added to the database.
  213. * @throws IOException
  214. * index file could not be opened, read, or is not recognized as
  215. * a Git pack file index.
  216. */
  217. public PackFile openPack(final File pack)
  218. throws IOException {
  219. final String p = pack.getName();
  220. if (p.length() != 50 || !p.startsWith("pack-") || !p.endsWith(".pack")) //$NON-NLS-1$ //$NON-NLS-2$
  221. throw new IOException(MessageFormat.format(JGitText.get().notAValidPack, pack));
  222. // The pack and index are assumed to exist. The existence of other
  223. // extensions needs to be explicitly checked.
  224. //
  225. int extensions = PACK.getBit() | INDEX.getBit();
  226. final String base = p.substring(0, p.length() - 4);
  227. for (PackExt ext : PackExt.values()) {
  228. if ((extensions & ext.getBit()) == 0) {
  229. final String name = base + ext.getExtension();
  230. if (new File(pack.getParentFile(), name).exists())
  231. extensions |= ext.getBit();
  232. }
  233. }
  234. PackFile res = new PackFile(pack, extensions);
  235. insertPack(res);
  236. return res;
  237. }
  238. @Override
  239. public String toString() {
  240. return "ObjectDirectory[" + getDirectory() + "]"; //$NON-NLS-1$ //$NON-NLS-2$
  241. }
  242. @Override
  243. public boolean has(AnyObjectId objectId) {
  244. return unpackedObjectCache.isUnpacked(objectId)
  245. || hasPackedInSelfOrAlternate(objectId)
  246. || hasLooseInSelfOrAlternate(objectId);
  247. }
  248. private boolean hasPackedInSelfOrAlternate(AnyObjectId objectId) {
  249. if (hasPackedObject(objectId))
  250. return true;
  251. for (AlternateHandle alt : myAlternates()) {
  252. if (alt.db.hasPackedInSelfOrAlternate(objectId))
  253. return true;
  254. }
  255. return false;
  256. }
  257. private boolean hasLooseInSelfOrAlternate(AnyObjectId objectId) {
  258. if (fileFor(objectId).exists())
  259. return true;
  260. for (AlternateHandle alt : myAlternates()) {
  261. if (alt.db.hasLooseInSelfOrAlternate(objectId))
  262. return true;
  263. }
  264. return false;
  265. }
  266. boolean hasPackedObject(AnyObjectId objectId) {
  267. PackList pList;
  268. do {
  269. pList = packList.get();
  270. for (PackFile p : pList.packs) {
  271. try {
  272. if (p.hasObject(objectId))
  273. return true;
  274. } catch (IOException e) {
  275. // The hasObject call should have only touched the index,
  276. // so any failure here indicates the index is unreadable
  277. // by this process, and the pack is likewise not readable.
  278. removePack(p);
  279. }
  280. }
  281. } while (searchPacksAgain(pList));
  282. return false;
  283. }
  284. @Override
  285. void resolve(Set<ObjectId> matches, AbbreviatedObjectId id)
  286. throws IOException {
  287. // Go through the packs once. If we didn't find any resolutions
  288. // scan for new packs and check once more.
  289. int oldSize = matches.size();
  290. PackList pList;
  291. do {
  292. pList = packList.get();
  293. for (PackFile p : pList.packs) {
  294. try {
  295. p.resolve(matches, id, RESOLVE_ABBREV_LIMIT);
  296. } catch (IOException e) {
  297. handlePackError(e, p);
  298. }
  299. if (matches.size() > RESOLVE_ABBREV_LIMIT)
  300. return;
  301. }
  302. } while (matches.size() == oldSize && searchPacksAgain(pList));
  303. String fanOut = id.name().substring(0, 2);
  304. String[] entries = new File(getDirectory(), fanOut).list();
  305. if (entries != null) {
  306. for (String e : entries) {
  307. if (e.length() != Constants.OBJECT_ID_STRING_LENGTH - 2)
  308. continue;
  309. try {
  310. ObjectId entId = ObjectId.fromString(fanOut + e);
  311. if (id.prefixCompare(entId) == 0)
  312. matches.add(entId);
  313. } catch (IllegalArgumentException notId) {
  314. continue;
  315. }
  316. if (matches.size() > RESOLVE_ABBREV_LIMIT)
  317. return;
  318. }
  319. }
  320. for (AlternateHandle alt : myAlternates()) {
  321. alt.db.resolve(matches, id);
  322. if (matches.size() > RESOLVE_ABBREV_LIMIT)
  323. return;
  324. }
  325. }
  326. @Override
  327. ObjectLoader openObject(WindowCursor curs, AnyObjectId objectId)
  328. throws IOException {
  329. if (unpackedObjectCache.isUnpacked(objectId)) {
  330. ObjectLoader ldr = openLooseObject(curs, objectId);
  331. if (ldr != null)
  332. return ldr;
  333. }
  334. ObjectLoader ldr = openPackedFromSelfOrAlternate(curs, objectId);
  335. if (ldr != null)
  336. return ldr;
  337. return openLooseFromSelfOrAlternate(curs, objectId);
  338. }
  339. private ObjectLoader openPackedFromSelfOrAlternate(WindowCursor curs,
  340. AnyObjectId objectId) {
  341. ObjectLoader ldr = openPackedObject(curs, objectId);
  342. if (ldr != null)
  343. return ldr;
  344. for (AlternateHandle alt : myAlternates()) {
  345. ldr = alt.db.openPackedFromSelfOrAlternate(curs, objectId);
  346. if (ldr != null)
  347. return ldr;
  348. }
  349. return null;
  350. }
  351. private ObjectLoader openLooseFromSelfOrAlternate(WindowCursor curs,
  352. AnyObjectId objectId) throws IOException {
  353. ObjectLoader ldr = openLooseObject(curs, objectId);
  354. if (ldr != null)
  355. return ldr;
  356. for (AlternateHandle alt : myAlternates()) {
  357. ldr = alt.db.openLooseFromSelfOrAlternate(curs, objectId);
  358. if (ldr != null)
  359. return ldr;
  360. }
  361. return null;
  362. }
  363. ObjectLoader openPackedObject(WindowCursor curs, AnyObjectId objectId) {
  364. PackList pList;
  365. do {
  366. SEARCH: for (;;) {
  367. pList = packList.get();
  368. for (PackFile p : pList.packs) {
  369. try {
  370. ObjectLoader ldr = p.get(curs, objectId);
  371. if (ldr != null)
  372. return ldr;
  373. } catch (PackMismatchException e) {
  374. // Pack was modified; refresh the entire pack list.
  375. if (searchPacksAgain(pList))
  376. continue SEARCH;
  377. } catch (IOException e) {
  378. handlePackError(e, p);
  379. }
  380. }
  381. break SEARCH;
  382. }
  383. } while (searchPacksAgain(pList));
  384. return null;
  385. }
  386. ObjectLoader openLooseObject(WindowCursor curs, AnyObjectId id)
  387. throws IOException {
  388. try {
  389. File path = fileFor(id);
  390. FileInputStream in = new FileInputStream(path);
  391. try {
  392. unpackedObjectCache.add(id);
  393. return UnpackedObject.open(in, path, id, curs);
  394. } finally {
  395. in.close();
  396. }
  397. } catch (FileNotFoundException noFile) {
  398. unpackedObjectCache.remove(id);
  399. return null;
  400. }
  401. }
  402. long getObjectSize(WindowCursor curs, AnyObjectId id)
  403. throws IOException {
  404. if (unpackedObjectCache.isUnpacked(id)) {
  405. long len = getLooseObjectSize(curs, id);
  406. if (0 <= len)
  407. return len;
  408. }
  409. long len = getPackedSizeFromSelfOrAlternate(curs, id);
  410. if (0 <= len)
  411. return len;
  412. return getLooseSizeFromSelfOrAlternate(curs, id);
  413. }
  414. private long getPackedSizeFromSelfOrAlternate(WindowCursor curs,
  415. AnyObjectId id) {
  416. long len = getPackedObjectSize(curs, id);
  417. if (0 <= len)
  418. return len;
  419. for (AlternateHandle alt : myAlternates()) {
  420. len = alt.db.getPackedSizeFromSelfOrAlternate(curs, id);
  421. if (0 <= len)
  422. return len;
  423. }
  424. return -1;
  425. }
  426. private long getLooseSizeFromSelfOrAlternate(WindowCursor curs,
  427. AnyObjectId id) throws IOException {
  428. long len = getLooseObjectSize(curs, id);
  429. if (0 <= len)
  430. return len;
  431. for (AlternateHandle alt : myAlternates()) {
  432. len = alt.db.getLooseSizeFromSelfOrAlternate(curs, id);
  433. if (0 <= len)
  434. return len;
  435. }
  436. return -1;
  437. }
  438. private long getPackedObjectSize(WindowCursor curs, AnyObjectId id) {
  439. PackList pList;
  440. do {
  441. SEARCH: for (;;) {
  442. pList = packList.get();
  443. for (PackFile p : pList.packs) {
  444. try {
  445. long len = p.getObjectSize(curs, id);
  446. if (0 <= len)
  447. return len;
  448. } catch (PackMismatchException e) {
  449. // Pack was modified; refresh the entire pack list.
  450. if (searchPacksAgain(pList))
  451. continue SEARCH;
  452. } catch (IOException e) {
  453. handlePackError(e, p);
  454. }
  455. }
  456. break SEARCH;
  457. }
  458. } while (searchPacksAgain(pList));
  459. return -1;
  460. }
  461. private long getLooseObjectSize(WindowCursor curs, AnyObjectId id)
  462. throws IOException {
  463. try {
  464. FileInputStream in = new FileInputStream(fileFor(id));
  465. try {
  466. unpackedObjectCache.add(id);
  467. return UnpackedObject.getSize(in, id, curs);
  468. } finally {
  469. in.close();
  470. }
  471. } catch (FileNotFoundException noFile) {
  472. unpackedObjectCache.remove(id);
  473. return -1;
  474. }
  475. }
  476. @Override
  477. void selectObjectRepresentation(PackWriter packer, ObjectToPack otp,
  478. WindowCursor curs) throws IOException {
  479. PackList pList = packList.get();
  480. SEARCH: for (;;) {
  481. for (final PackFile p : pList.packs) {
  482. try {
  483. LocalObjectRepresentation rep = p.representation(curs, otp);
  484. if (rep != null)
  485. packer.select(otp, rep);
  486. } catch (PackMismatchException e) {
  487. // Pack was modified; refresh the entire pack list.
  488. //
  489. pList = scanPacks(pList);
  490. continue SEARCH;
  491. } catch (IOException e) {
  492. handlePackError(e, p);
  493. }
  494. }
  495. break SEARCH;
  496. }
  497. for (AlternateHandle h : myAlternates())
  498. h.db.selectObjectRepresentation(packer, otp, curs);
  499. }
  500. private void handlePackError(IOException e, PackFile p) {
  501. String tmpl;
  502. if ((e instanceof CorruptObjectException)
  503. || (e instanceof PackInvalidException)) {
  504. tmpl = JGitText.get().corruptPack;
  505. // Assume the pack is corrupted, and remove it from the list.
  506. removePack(p);
  507. } else if (e instanceof FileNotFoundException) {
  508. tmpl = JGitText.get().packWasDeleted;
  509. removePack(p);
  510. } else {
  511. tmpl = JGitText.get().exceptionWhileReadingPack;
  512. // Don't remove the pack from the list, as the error may be
  513. // transient.
  514. }
  515. StringBuilder buf = new StringBuilder(MessageFormat.format(tmpl,
  516. p.getPackFile().getAbsolutePath()));
  517. StringWriter sw = new StringWriter();
  518. e.printStackTrace(new PrintWriter(sw));
  519. buf.append('\n');
  520. buf.append(sw.toString());
  521. // TODO instead of syserr we should use a logging framework
  522. System.err.println(buf.toString());
  523. }
  524. @Override
  525. InsertLooseObjectResult insertUnpackedObject(File tmp, ObjectId id,
  526. boolean createDuplicate) throws IOException {
  527. // If the object is already in the repository, remove temporary file.
  528. //
  529. if (unpackedObjectCache.isUnpacked(id)) {
  530. FileUtils.delete(tmp, FileUtils.RETRY);
  531. return InsertLooseObjectResult.EXISTS_LOOSE;
  532. }
  533. if (!createDuplicate && has(id)) {
  534. FileUtils.delete(tmp, FileUtils.RETRY);
  535. return InsertLooseObjectResult.EXISTS_PACKED;
  536. }
  537. final File dst = fileFor(id);
  538. if (fs.exists(dst)) {
  539. // We want to be extra careful and avoid replacing an object
  540. // that already exists. We can't be sure renameTo() would
  541. // fail on all platforms if dst exists, so we check first.
  542. //
  543. FileUtils.delete(tmp, FileUtils.RETRY);
  544. return InsertLooseObjectResult.EXISTS_LOOSE;
  545. }
  546. if (tmp.renameTo(dst)) {
  547. dst.setReadOnly();
  548. unpackedObjectCache.add(id);
  549. return InsertLooseObjectResult.INSERTED;
  550. }
  551. // Maybe the directory doesn't exist yet as the object
  552. // directories are always lazily created. Note that we
  553. // try the rename first as the directory likely does exist.
  554. //
  555. FileUtils.mkdir(dst.getParentFile(), true);
  556. if (tmp.renameTo(dst)) {
  557. dst.setReadOnly();
  558. unpackedObjectCache.add(id);
  559. return InsertLooseObjectResult.INSERTED;
  560. }
  561. if (!createDuplicate && has(id)) {
  562. FileUtils.delete(tmp, FileUtils.RETRY);
  563. return InsertLooseObjectResult.EXISTS_PACKED;
  564. }
  565. // The object failed to be renamed into its proper
  566. // location and it doesn't exist in the repository
  567. // either. We really don't know what went wrong, so
  568. // fail.
  569. //
  570. FileUtils.delete(tmp, FileUtils.RETRY);
  571. return InsertLooseObjectResult.FAILURE;
  572. }
  573. private boolean searchPacksAgain(PackList old) {
  574. // Whether to trust the pack folder's modification time. If set
  575. // to false we will always scan the .git/objects/pack folder to
  576. // check for new pack files. If set to true (default) we use the
  577. // lastmodified attribute of the folder and assume that no new
  578. // pack files can be in this folder if his modification time has
  579. // not changed.
  580. boolean trustFolderStat = config.getBoolean(
  581. ConfigConstants.CONFIG_CORE_SECTION,
  582. ConfigConstants.CONFIG_KEY_TRUSTFOLDERSTAT, true);
  583. return ((!trustFolderStat) || old.snapshot.isModified(packDirectory))
  584. && old != scanPacks(old);
  585. }
  586. Config getConfig() {
  587. return config;
  588. }
  589. @Override
  590. FS getFS() {
  591. return fs;
  592. }
  593. @Override
  594. Set<ObjectId> getShallowCommits() throws IOException {
  595. if (shallowFile == null || !shallowFile.isFile())
  596. return Collections.emptySet();
  597. if (shallowFileSnapshot == null
  598. || shallowFileSnapshot.isModified(shallowFile)) {
  599. shallowCommitsIds = new HashSet<ObjectId>();
  600. final BufferedReader reader = open(shallowFile);
  601. try {
  602. String line;
  603. while ((line = reader.readLine()) != null)
  604. shallowCommitsIds.add(ObjectId.fromString(line));
  605. } finally {
  606. reader.close();
  607. }
  608. shallowFileSnapshot = FileSnapshot.save(shallowFile);
  609. }
  610. return shallowCommitsIds;
  611. }
  612. private void insertPack(final PackFile pf) {
  613. PackList o, n;
  614. do {
  615. o = packList.get();
  616. // If the pack in question is already present in the list
  617. // (picked up by a concurrent thread that did a scan?) we
  618. // do not want to insert it a second time.
  619. //
  620. final PackFile[] oldList = o.packs;
  621. final String name = pf.getPackFile().getName();
  622. for (PackFile p : oldList) {
  623. if (PackFile.SORT.compare(pf, p) < 0)
  624. break;
  625. if (name.equals(p.getPackFile().getName()))
  626. return;
  627. }
  628. final PackFile[] newList = new PackFile[1 + oldList.length];
  629. newList[0] = pf;
  630. System.arraycopy(oldList, 0, newList, 1, oldList.length);
  631. n = new PackList(o.snapshot, newList);
  632. } while (!packList.compareAndSet(o, n));
  633. }
  634. private void removePack(final PackFile deadPack) {
  635. PackList o, n;
  636. do {
  637. o = packList.get();
  638. final PackFile[] oldList = o.packs;
  639. final int j = indexOf(oldList, deadPack);
  640. if (j < 0)
  641. break;
  642. final PackFile[] newList = new PackFile[oldList.length - 1];
  643. System.arraycopy(oldList, 0, newList, 0, j);
  644. System.arraycopy(oldList, j + 1, newList, j, newList.length - j);
  645. n = new PackList(o.snapshot, newList);
  646. } while (!packList.compareAndSet(o, n));
  647. deadPack.close();
  648. }
  649. private static int indexOf(final PackFile[] list, final PackFile pack) {
  650. for (int i = 0; i < list.length; i++) {
  651. if (list[i] == pack)
  652. return i;
  653. }
  654. return -1;
  655. }
  656. private PackList scanPacks(final PackList original) {
  657. synchronized (packList) {
  658. PackList o, n;
  659. do {
  660. o = packList.get();
  661. if (o != original) {
  662. // Another thread did the scan for us, while we
  663. // were blocked on the monitor above.
  664. //
  665. return o;
  666. }
  667. n = scanPacksImpl(o);
  668. if (n == o)
  669. return n;
  670. } while (!packList.compareAndSet(o, n));
  671. return n;
  672. }
  673. }
  674. private PackList scanPacksImpl(final PackList old) {
  675. final Map<String, PackFile> forReuse = reuseMap(old);
  676. final FileSnapshot snapshot = FileSnapshot.save(packDirectory);
  677. final Set<String> names = listPackDirectory();
  678. final List<PackFile> list = new ArrayList<PackFile>(names.size() >> 2);
  679. boolean foundNew = false;
  680. for (final String indexName : names) {
  681. // Must match "pack-[0-9a-f]{40}.idx" to be an index.
  682. //
  683. if (indexName.length() != 49 || !indexName.endsWith(".idx")) //$NON-NLS-1$
  684. continue;
  685. final String base = indexName.substring(0, indexName.length() - 3);
  686. int extensions = 0;
  687. for (PackExt ext : PackExt.values()) {
  688. if (names.contains(base + ext.getExtension()))
  689. extensions |= ext.getBit();
  690. }
  691. if ((extensions & PACK.getBit()) == 0) {
  692. // Sometimes C Git's HTTP fetch transport leaves a
  693. // .idx file behind and does not download the .pack.
  694. // We have to skip over such useless indexes.
  695. //
  696. continue;
  697. }
  698. final String packName = base + PACK.getExtension();
  699. final PackFile oldPack = forReuse.remove(packName);
  700. if (oldPack != null) {
  701. list.add(oldPack);
  702. continue;
  703. }
  704. final File packFile = new File(packDirectory, packName);
  705. list.add(new PackFile(packFile, extensions));
  706. foundNew = true;
  707. }
  708. // If we did not discover any new files, the modification time was not
  709. // changed, and we did not remove any files, then the set of files is
  710. // the same as the set we were given. Instead of building a new object
  711. // return the same collection.
  712. //
  713. if (!foundNew && forReuse.isEmpty() && snapshot.equals(old.snapshot)) {
  714. old.snapshot.setClean(snapshot);
  715. return old;
  716. }
  717. for (final PackFile p : forReuse.values()) {
  718. p.close();
  719. }
  720. if (list.isEmpty())
  721. return new PackList(snapshot, NO_PACKS.packs);
  722. final PackFile[] r = list.toArray(new PackFile[list.size()]);
  723. Arrays.sort(r, PackFile.SORT);
  724. return new PackList(snapshot, r);
  725. }
  726. private static Map<String, PackFile> reuseMap(final PackList old) {
  727. final Map<String, PackFile> forReuse = new HashMap<String, PackFile>();
  728. for (final PackFile p : old.packs) {
  729. if (p.invalid()) {
  730. // The pack instance is corrupted, and cannot be safely used
  731. // again. Do not include it in our reuse map.
  732. //
  733. p.close();
  734. continue;
  735. }
  736. final PackFile prior = forReuse.put(p.getPackFile().getName(), p);
  737. if (prior != null) {
  738. // This should never occur. It should be impossible for us
  739. // to have two pack files with the same name, as all of them
  740. // came out of the same directory. If it does, we promised to
  741. // close any PackFiles we did not reuse, so close the second,
  742. // readers are likely to be actively using the first.
  743. //
  744. forReuse.put(prior.getPackFile().getName(), prior);
  745. p.close();
  746. }
  747. }
  748. return forReuse;
  749. }
  750. private Set<String> listPackDirectory() {
  751. final String[] nameList = packDirectory.list();
  752. if (nameList == null)
  753. return Collections.emptySet();
  754. final Set<String> nameSet = new HashSet<String>(nameList.length << 1);
  755. for (final String name : nameList) {
  756. if (name.startsWith("pack-")) //$NON-NLS-1$
  757. nameSet.add(name);
  758. }
  759. return nameSet;
  760. }
  761. AlternateHandle[] myAlternates() {
  762. AlternateHandle[] alt = alternates.get();
  763. if (alt == null) {
  764. synchronized (alternates) {
  765. alt = alternates.get();
  766. if (alt == null) {
  767. try {
  768. alt = loadAlternates();
  769. } catch (IOException e) {
  770. alt = new AlternateHandle[0];
  771. }
  772. alternates.set(alt);
  773. }
  774. }
  775. }
  776. return alt;
  777. }
  778. private AlternateHandle[] loadAlternates() throws IOException {
  779. final List<AlternateHandle> l = new ArrayList<AlternateHandle>(4);
  780. final BufferedReader br = open(alternatesFile);
  781. try {
  782. String line;
  783. while ((line = br.readLine()) != null) {
  784. l.add(openAlternate(line));
  785. }
  786. } finally {
  787. br.close();
  788. }
  789. return l.toArray(new AlternateHandle[l.size()]);
  790. }
  791. private static BufferedReader open(final File f)
  792. throws FileNotFoundException {
  793. return new BufferedReader(new FileReader(f));
  794. }
  795. private AlternateHandle openAlternate(final String location)
  796. throws IOException {
  797. final File objdir = fs.resolve(objects, location);
  798. return openAlternate(objdir);
  799. }
  800. private AlternateHandle openAlternate(File objdir) throws IOException {
  801. final File parent = objdir.getParentFile();
  802. if (FileKey.isGitRepository(parent, fs)) {
  803. FileKey key = FileKey.exact(parent, fs);
  804. FileRepository db = (FileRepository) RepositoryCache.open(key);
  805. return new AlternateRepository(db);
  806. }
  807. ObjectDirectory db = new ObjectDirectory(config, objdir, null, fs, null);
  808. return new AlternateHandle(db);
  809. }
  810. /**
  811. * Compute the location of a loose object file.
  812. *
  813. * @param objectId
  814. * identity of the loose object to map to the directory.
  815. * @return location of the object, if it were to exist as a loose object.
  816. */
  817. public File fileFor(AnyObjectId objectId) {
  818. String n = objectId.name();
  819. String d = n.substring(0, 2);
  820. String f = n.substring(2);
  821. return new File(new File(getDirectory(), d), f);
  822. }
  823. private static final class PackList {
  824. /** State just before reading the pack directory. */
  825. final FileSnapshot snapshot;
  826. /** All known packs, sorted by {@link PackFile#SORT}. */
  827. final PackFile[] packs;
  828. PackList(final FileSnapshot monitor, final PackFile[] packs) {
  829. this.snapshot = monitor;
  830. this.packs = packs;
  831. }
  832. }
  833. static class AlternateHandle {
  834. final ObjectDirectory db;
  835. AlternateHandle(ObjectDirectory db) {
  836. this.db = db;
  837. }
  838. void close() {
  839. db.close();
  840. }
  841. }
  842. static class AlternateRepository extends AlternateHandle {
  843. final FileRepository repository;
  844. AlternateRepository(FileRepository r) {
  845. super(r.getObjectDatabase());
  846. repository = r;
  847. }
  848. void close() {
  849. repository.close();
  850. }
  851. }
  852. @Override
  853. public ObjectDatabase newCachedDatabase() {
  854. return newCachedFileObjectDatabase();
  855. }
  856. CachedObjectDirectory newCachedFileObjectDatabase() {
  857. return new CachedObjectDirectory(this);
  858. }
  859. }