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.

UsageMap.java 29KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941
  1. /*
  2. Copyright (c) 2005 Health Market Science, Inc.
  3. This library is free software; you can redistribute it and/or
  4. modify it under the terms of the GNU Lesser General Public
  5. License as published by the Free Software Foundation; either
  6. version 2.1 of the License, or (at your option) any later version.
  7. This library is distributed in the hope that it will be useful,
  8. but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  10. Lesser General Public License for more details.
  11. You should have received a copy of the GNU Lesser General Public
  12. License along with this library; if not, write to the Free Software
  13. Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  14. USA
  15. You can contact Health Market Science at info@healthmarketscience.com
  16. or at the following address:
  17. Health Market Science
  18. 2700 Horizon Drive
  19. Suite 200
  20. King of Prussia, PA 19406
  21. */
  22. package com.healthmarketscience.jackcess;
  23. import java.io.IOException;
  24. import java.nio.ByteBuffer;
  25. import java.util.BitSet;
  26. import org.apache.commons.logging.Log;
  27. import org.apache.commons.logging.LogFactory;
  28. /**
  29. * Describes which database pages a particular table uses
  30. * @author Tim McCune
  31. */
  32. public class UsageMap
  33. {
  34. private static final Log LOG = LogFactory.getLog(UsageMap.class);
  35. /** Inline map type */
  36. public static final byte MAP_TYPE_INLINE = 0x0;
  37. /** Reference map type, for maps that are too large to fit inline */
  38. public static final byte MAP_TYPE_REFERENCE = 0x1;
  39. /** bit index value for an invalid page number */
  40. private static final int INVALID_BIT_INDEX = -1;
  41. /** owning database */
  42. private final Database _database;
  43. /** Page number of the map table declaration */
  44. private final int _tablePageNum;
  45. /** Offset of the data page at which the usage map data starts */
  46. private int _startOffset;
  47. /** Offset of the data page at which the usage map declaration starts */
  48. private final short _rowStart;
  49. /** First page that this usage map applies to */
  50. private int _startPage;
  51. /** Last page that this usage map applies to */
  52. private int _endPage;
  53. /** bits representing page numbers used, offset from _startPage */
  54. private BitSet _pageNumbers = new BitSet();
  55. /** Buffer that contains the usage map table declaration page */
  56. private final ByteBuffer _tableBuffer;
  57. /** modification count on the usage map, used to keep the cursors in
  58. sync */
  59. private int _modCount;
  60. /** the current handler implementation for reading/writing the specific
  61. usage map type. note, this may change over time. */
  62. private Handler _handler;
  63. /** Error message prefix used when map type is unrecognized. */
  64. static final String MSG_PREFIX_UNRECOGNIZED_MAP = "Unrecognized map type: ";
  65. /**
  66. * @param database database that contains this usage map
  67. * @param tableBuffer Buffer that contains this map's declaration
  68. * @param pageNum Page number that this usage map is contained in
  69. * @param rowStart Offset at which the declaration starts in the buffer
  70. */
  71. private UsageMap(Database database, ByteBuffer tableBuffer,
  72. int pageNum, short rowStart)
  73. throws IOException
  74. {
  75. _database = database;
  76. _tableBuffer = tableBuffer;
  77. _tablePageNum = pageNum;
  78. _rowStart = rowStart;
  79. _tableBuffer.position(_rowStart + getFormat().OFFSET_USAGE_MAP_START);
  80. _startOffset = _tableBuffer.position();
  81. if (LOG.isDebugEnabled()) {
  82. LOG.debug("Usage map block:\n" + ByteUtil.toHexString(_tableBuffer, _rowStart,
  83. tableBuffer.limit() - _rowStart));
  84. }
  85. }
  86. public Database getDatabase() {
  87. return _database;
  88. }
  89. public JetFormat getFormat() {
  90. return getDatabase().getFormat();
  91. }
  92. public PageChannel getPageChannel() {
  93. return getDatabase().getPageChannel();
  94. }
  95. /**
  96. * @param database database that contains this usage map
  97. * @param pageNum Page number that this usage map is contained in
  98. * @param rowNum Number of the row on the page that contains this usage map
  99. * @return Either an InlineUsageMap or a ReferenceUsageMap, depending on
  100. * which type of map is found
  101. */
  102. public static UsageMap read(Database database, int pageNum,
  103. int rowNum, boolean assumeOutOfRangeBitsOn)
  104. throws IOException
  105. {
  106. JetFormat format = database.getFormat();
  107. PageChannel pageChannel = database.getPageChannel();
  108. ByteBuffer tableBuffer = pageChannel.createPageBuffer();
  109. pageChannel.readPage(tableBuffer, pageNum);
  110. short rowStart = Table.findRowStart(tableBuffer, rowNum, format);
  111. int rowEnd = Table.findRowEnd(tableBuffer, rowNum, format);
  112. tableBuffer.limit(rowEnd);
  113. byte mapType = tableBuffer.get(rowStart);
  114. UsageMap rtn = new UsageMap(database, tableBuffer, pageNum, rowStart);
  115. rtn.initHandler(mapType, assumeOutOfRangeBitsOn);
  116. return rtn;
  117. }
  118. private void initHandler(byte mapType, boolean assumeOutOfRangeBitsOn)
  119. throws IOException
  120. {
  121. if (mapType == MAP_TYPE_INLINE) {
  122. _handler = new InlineHandler(assumeOutOfRangeBitsOn);
  123. } else if (mapType == MAP_TYPE_REFERENCE) {
  124. _handler = new ReferenceHandler();
  125. } else {
  126. throw new IOException(MSG_PREFIX_UNRECOGNIZED_MAP + mapType);
  127. }
  128. }
  129. public PageCursor cursor() {
  130. return new PageCursor();
  131. }
  132. protected short getRowStart() {
  133. return _rowStart;
  134. }
  135. protected int getRowEnd() {
  136. return getTableBuffer().limit();
  137. }
  138. protected void setStartOffset(int startOffset) {
  139. _startOffset = startOffset;
  140. }
  141. protected int getStartOffset() {
  142. return _startOffset;
  143. }
  144. protected ByteBuffer getTableBuffer() {
  145. return _tableBuffer;
  146. }
  147. protected int getTablePageNumber() {
  148. return _tablePageNum;
  149. }
  150. protected int getStartPage() {
  151. return _startPage;
  152. }
  153. protected int getEndPage() {
  154. return _endPage;
  155. }
  156. protected BitSet getPageNumbers() {
  157. return _pageNumbers;
  158. }
  159. protected void setPageRange(int newStartPage, int newEndPage) {
  160. _startPage = newStartPage;
  161. _endPage = newEndPage;
  162. }
  163. protected boolean isPageWithinRange(int pageNumber)
  164. {
  165. return((pageNumber >= _startPage) && (pageNumber < _endPage));
  166. }
  167. protected int getFirstPageNumber() {
  168. return bitIndexToPageNumber(getNextBitIndex(-1), RowId.LAST_PAGE_NUMBER);
  169. }
  170. protected int getNextPageNumber(int curPage) {
  171. return bitIndexToPageNumber(
  172. getNextBitIndex(pageNumberToBitIndex(curPage)),
  173. RowId.LAST_PAGE_NUMBER);
  174. }
  175. protected int getNextBitIndex(int curIndex) {
  176. return _pageNumbers.nextSetBit(curIndex + 1);
  177. }
  178. protected int getLastPageNumber() {
  179. return bitIndexToPageNumber(getPrevBitIndex(_pageNumbers.length()),
  180. RowId.FIRST_PAGE_NUMBER);
  181. }
  182. protected int getPrevPageNumber(int curPage) {
  183. return bitIndexToPageNumber(
  184. getPrevBitIndex(pageNumberToBitIndex(curPage)),
  185. RowId.FIRST_PAGE_NUMBER);
  186. }
  187. protected int getPrevBitIndex(int curIndex) {
  188. --curIndex;
  189. while((curIndex >= 0) && !_pageNumbers.get(curIndex)) {
  190. --curIndex;
  191. }
  192. return curIndex;
  193. }
  194. protected int bitIndexToPageNumber(int bitIndex,
  195. int invalidPageNumber) {
  196. return((bitIndex >= 0) ? (_startPage + bitIndex) : invalidPageNumber);
  197. }
  198. protected int pageNumberToBitIndex(int pageNumber) {
  199. return((pageNumber >= 0) ? (pageNumber - _startPage) :
  200. INVALID_BIT_INDEX);
  201. }
  202. protected void clearTableAndPages()
  203. {
  204. // reset some values
  205. _pageNumbers.clear();
  206. _startPage = 0;
  207. _endPage = 0;
  208. ++_modCount;
  209. // clear out the table data (everything except map type)
  210. int tableStart = getRowStart() + 1;
  211. int tableEnd = getRowEnd();
  212. ByteUtil.clearRange(_tableBuffer, tableStart, tableEnd);
  213. }
  214. protected void writeTable()
  215. throws IOException
  216. {
  217. // note, we only want to write the row data with which we are working
  218. getPageChannel().writePage(_tableBuffer, _tablePageNum, _rowStart);
  219. }
  220. /**
  221. * Read in the page numbers in this inline map
  222. */
  223. protected void processMap(ByteBuffer buffer, int bufferStartPage)
  224. {
  225. int byteCount = 0;
  226. while (buffer.hasRemaining()) {
  227. byte b = buffer.get();
  228. if(b != (byte)0) {
  229. for (int i = 0; i < 8; i++) {
  230. if ((b & (1 << i)) != 0) {
  231. int pageNumberOffset = (byteCount * 8 + i) + bufferStartPage;
  232. int pageNumber = bitIndexToPageNumber(
  233. pageNumberOffset,
  234. PageChannel.INVALID_PAGE_NUMBER);
  235. if(!isPageWithinRange(pageNumber)) {
  236. throw new IllegalStateException(
  237. "found page number " + pageNumber
  238. + " in usage map outside of expected range " +
  239. _startPage + " to " + _endPage);
  240. }
  241. _pageNumbers.set(pageNumberOffset);
  242. }
  243. }
  244. }
  245. byteCount++;
  246. }
  247. }
  248. /**
  249. * Determines if the given page number is contained in this map.
  250. */
  251. public boolean containsPageNumber(int pageNumber) {
  252. return _handler.containsPageNumber(pageNumber);
  253. }
  254. /**
  255. * Add a page number to this usage map
  256. */
  257. public void addPageNumber(int pageNumber) throws IOException {
  258. ++_modCount;
  259. _handler.addOrRemovePageNumber(pageNumber, true);
  260. }
  261. /**
  262. * Remove a page number from this usage map
  263. */
  264. public void removePageNumber(int pageNumber) throws IOException {
  265. ++_modCount;
  266. _handler.addOrRemovePageNumber(pageNumber, false);
  267. }
  268. protected void updateMap(int absolutePageNumber,
  269. int bufferRelativePageNumber,
  270. ByteBuffer buffer, boolean add)
  271. throws IOException
  272. {
  273. //Find the byte to which to apply the bitmask and create the bitmask
  274. int offset = bufferRelativePageNumber / 8;
  275. int bitmask = 1 << (bufferRelativePageNumber % 8);
  276. byte b = buffer.get(_startOffset + offset);
  277. // check current value for this page number
  278. int pageNumberOffset = pageNumberToBitIndex(absolutePageNumber);
  279. boolean isOn = _pageNumbers.get(pageNumberOffset);
  280. if(isOn == add) {
  281. throw new IOException("Page number " + absolutePageNumber + " already " +
  282. ((add) ? "added to" : "removed from") +
  283. " usage map, expected range " +
  284. _startPage + " to " + _endPage);
  285. }
  286. //Apply the bitmask
  287. if (add) {
  288. b |= bitmask;
  289. _pageNumbers.set(pageNumberOffset);
  290. } else {
  291. b &= ~bitmask;
  292. _pageNumbers.clear(pageNumberOffset);
  293. }
  294. buffer.put(_startOffset + offset, b);
  295. }
  296. /**
  297. * Promotes and inline usage map to a reference usage map.
  298. */
  299. private void promoteInlineHandlerToReferenceHandler(int newPageNumber)
  300. throws IOException
  301. {
  302. // copy current page number info to new references and then clear old
  303. int oldStartPage = _startPage;
  304. BitSet oldPageNumbers = (BitSet)_pageNumbers.clone();
  305. // clear out the main table (inline usage map data and start page)
  306. clearTableAndPages();
  307. // set the new map type
  308. _tableBuffer.put(getRowStart(), MAP_TYPE_REFERENCE);
  309. // write the new table data
  310. writeTable();
  311. // set new handler
  312. _handler = new ReferenceHandler();
  313. // update new handler with old data
  314. reAddPages(oldStartPage, oldPageNumbers, newPageNumber);
  315. }
  316. private void reAddPages(int oldStartPage, BitSet oldPageNumbers,
  317. int newPageNumber)
  318. throws IOException
  319. {
  320. // add all the old pages back in
  321. for(int i = oldPageNumbers.nextSetBit(0); i >= 0;
  322. i = oldPageNumbers.nextSetBit(i + 1)) {
  323. addPageNumber(oldStartPage + i);
  324. }
  325. if(newPageNumber > PageChannel.INVALID_PAGE_NUMBER) {
  326. // and then add the new page
  327. addPageNumber(newPageNumber);
  328. }
  329. }
  330. @Override
  331. public String toString() {
  332. StringBuilder builder = new StringBuilder(
  333. "(" + _handler.getClass().getSimpleName() +
  334. ") page numbers (range " + _startPage + " " + _endPage + "): [");
  335. PageCursor pCursor = cursor();
  336. while(true) {
  337. int nextPage = pCursor.getNextPage();
  338. if(nextPage < 0) {
  339. break;
  340. }
  341. builder.append(nextPage).append(", ");
  342. }
  343. builder.append("]");
  344. return builder.toString();
  345. }
  346. private abstract class Handler
  347. {
  348. protected Handler() {
  349. }
  350. public boolean containsPageNumber(int pageNumber) {
  351. return(isPageWithinRange(pageNumber) &&
  352. getPageNumbers().get(pageNumberToBitIndex(pageNumber)));
  353. }
  354. /**
  355. * @param pageNumber Page number to add or remove from this map
  356. * @param add True to add it, false to remove it
  357. */
  358. public abstract void addOrRemovePageNumber(int pageNumber, boolean add)
  359. throws IOException;
  360. }
  361. /**
  362. * Usage map whose map is written inline in the same page. For Jet4, this
  363. * type of map can usually contains a maximum of 512 pages. Free space maps
  364. * are always inline, used space maps may be inline or reference. It has a
  365. * start page, which all page numbers in its map are calculated as starting
  366. * from.
  367. * @author Tim McCune
  368. */
  369. private class InlineHandler extends Handler
  370. {
  371. private final boolean _assumeOutOfRangeBitsOn;
  372. private final int _maxInlinePages;
  373. private InlineHandler(boolean assumeOutOfRangeBitsOn)
  374. throws IOException
  375. {
  376. _assumeOutOfRangeBitsOn = assumeOutOfRangeBitsOn;
  377. _maxInlinePages = (getInlineDataEnd() - getInlineDataStart()) * 8;
  378. int startPage = getTableBuffer().getInt(getRowStart() + 1);
  379. setInlinePageRange(startPage);
  380. processMap(getTableBuffer(), 0);
  381. }
  382. private int getMaxInlinePages() {
  383. return _maxInlinePages;
  384. }
  385. private int getInlineDataStart() {
  386. return getRowStart() + getFormat().OFFSET_USAGE_MAP_START;
  387. }
  388. private int getInlineDataEnd() {
  389. return getRowEnd();
  390. }
  391. /**
  392. * Sets the page range for an inline usage map starting from the given
  393. * page.
  394. */
  395. private void setInlinePageRange(int startPage) {
  396. setPageRange(startPage, startPage + getMaxInlinePages());
  397. }
  398. @Override
  399. public boolean containsPageNumber(int pageNumber) {
  400. return(super.containsPageNumber(pageNumber) ||
  401. (_assumeOutOfRangeBitsOn && (pageNumber >= 0) &&
  402. !isPageWithinRange(pageNumber)));
  403. }
  404. @Override
  405. public void addOrRemovePageNumber(int pageNumber, boolean add)
  406. throws IOException
  407. {
  408. if(isPageWithinRange(pageNumber)) {
  409. // easy enough, just update the inline data
  410. int bufferRelativePageNumber = pageNumberToBitIndex(pageNumber);
  411. updateMap(pageNumber, bufferRelativePageNumber, getTableBuffer(), add);
  412. // Write the updated map back to disk
  413. writeTable();
  414. } else {
  415. // uh-oh, we've split our britches. what now? determine what our
  416. // status is
  417. int firstPage = getFirstPageNumber();
  418. int lastPage = getLastPageNumber();
  419. if(add) {
  420. // we can ignore out-of-range page addition if we are already
  421. // assuming out-of-range bits are "on". Note, we are leaving small
  422. // holes in the database here (leaving behind some free pages), but
  423. // it's not the end of the world.
  424. if(!_assumeOutOfRangeBitsOn) {
  425. // we are adding, can we shift the bits and stay inline?
  426. if(firstPage <= PageChannel.INVALID_PAGE_NUMBER) {
  427. // no pages currently
  428. firstPage = pageNumber;
  429. lastPage = pageNumber;
  430. } else if(pageNumber > lastPage) {
  431. lastPage = pageNumber;
  432. } else {
  433. firstPage = pageNumber;
  434. }
  435. if((lastPage - firstPage + 1) < getMaxInlinePages()) {
  436. // we can still fit within an inline map
  437. moveToNewStartPage(firstPage, pageNumber);
  438. } else {
  439. // not going to happen, need to promote the usage map to a
  440. // reference map
  441. promoteInlineHandlerToReferenceHandler(pageNumber);
  442. }
  443. }
  444. } else {
  445. // we are removing, what does that mean?
  446. if(_assumeOutOfRangeBitsOn) {
  447. // we are using an inline map and assuming that anything not
  448. // within the current range is "on". so, if we attempt to set a
  449. // bit which is before the current page, ignore it, we are not
  450. // going back for it.
  451. if((firstPage <= PageChannel.INVALID_PAGE_NUMBER) ||
  452. (pageNumber > lastPage)) {
  453. // move to new start page, filling in as we move
  454. moveToNewStartPageForRemove(firstPage, pageNumber);
  455. }
  456. } else {
  457. // this should not happen, we are removing a page which is not in
  458. // the map
  459. throw new IOException("Page number " + pageNumber +
  460. " already removed from usage map" +
  461. ", expected range " +
  462. _startPage + " to " + _endPage);
  463. }
  464. }
  465. }
  466. }
  467. /**
  468. * Shifts the inline usage map so that it now starts with the given page.
  469. * @param newStartPage new page at which to start
  470. * @param newPageNumber optional page number to add once the map has been
  471. * shifted to the new start page
  472. */
  473. private void moveToNewStartPage(int newStartPage, int newPageNumber)
  474. throws IOException
  475. {
  476. int oldStartPage = getStartPage();
  477. BitSet oldPageNumbers = (BitSet)getPageNumbers().clone();
  478. // clear out the main table (inline usage map data and start page)
  479. clearTableAndPages();
  480. // write new start page
  481. ByteBuffer tableBuffer = getTableBuffer();
  482. tableBuffer.position(getRowStart() + 1);
  483. tableBuffer.putInt(newStartPage);
  484. // write the new table data
  485. writeTable();
  486. // set new page range
  487. setInlinePageRange(newStartPage);
  488. // put the pages back in
  489. reAddPages(oldStartPage, oldPageNumbers, newPageNumber);
  490. }
  491. /**
  492. * Shifts the inline usage map so that it now starts with the given
  493. * firstPage (if valid), otherwise the newPageNumber. Any page numbers
  494. * added to the end of the usage map are set to "on".
  495. * @param firstPage current first used page
  496. * @param newPageNumber page number to remove once the map has been
  497. * shifted to the new start page
  498. */
  499. private void moveToNewStartPageForRemove(int firstPage, int newPageNumber)
  500. throws IOException
  501. {
  502. int oldEndPage = getEndPage();
  503. int newStartPage =
  504. ((firstPage <= PageChannel.INVALID_PAGE_NUMBER) ? newPageNumber :
  505. // just shift a little and discard any initial unused pages.
  506. (newPageNumber - (getMaxInlinePages() / 2)));
  507. // move the current data
  508. moveToNewStartPage(newStartPage, PageChannel.INVALID_PAGE_NUMBER);
  509. if(firstPage <= PageChannel.INVALID_PAGE_NUMBER) {
  510. // this is the common case where we left everything behind
  511. ByteUtil.fillRange(_tableBuffer, getInlineDataStart(),
  512. getInlineDataEnd());
  513. // write out the updated table
  514. writeTable();
  515. // "add" all the page numbers
  516. getPageNumbers().set(0, getMaxInlinePages());
  517. } else {
  518. // add every new page manually
  519. for(int i = oldEndPage; i < getEndPage(); ++i) {
  520. addPageNumber(i);
  521. }
  522. }
  523. // lastly, remove the new page
  524. removePageNumber(newPageNumber);
  525. }
  526. }
  527. /**
  528. * Usage map whose map is written across one or more entire separate pages
  529. * of page type USAGE_MAP. For Jet4, this type of map can contain 32736
  530. * pages per reference page, and a maximum of 17 reference map pages for a
  531. * total maximum of 556512 pages (2 GB).
  532. * @author Tim McCune
  533. */
  534. private class ReferenceHandler extends Handler
  535. {
  536. /** Buffer that contains the current reference map page */
  537. private final TempPageHolder _mapPageHolder =
  538. TempPageHolder.newHolder(TempBufferHolder.Type.SOFT);
  539. private ReferenceHandler()
  540. throws IOException
  541. {
  542. int numUsagePages = (getRowEnd() - getRowStart() - 1) / 4;
  543. setStartOffset(getFormat().OFFSET_USAGE_MAP_PAGE_DATA);
  544. setPageRange(0, (numUsagePages * getMaxPagesPerUsagePage()));
  545. // there is no "start page" for a reference usage map, so we get an
  546. // extra page reference on top of the number of page references that fit
  547. // in the table
  548. for (int i = 0; i < numUsagePages; i++) {
  549. int mapPageNum = getTableBuffer().getInt(
  550. calculateMapPagePointerOffset(i));
  551. if (mapPageNum > 0) {
  552. ByteBuffer mapPageBuffer =
  553. _mapPageHolder.setPage(getPageChannel(), mapPageNum);
  554. byte pageType = mapPageBuffer.get();
  555. if (pageType != PageTypes.USAGE_MAP) {
  556. throw new IOException("Looking for usage map at page " +
  557. mapPageNum + ", but page type is " +
  558. pageType);
  559. }
  560. mapPageBuffer.position(getFormat().OFFSET_USAGE_MAP_PAGE_DATA);
  561. processMap(mapPageBuffer, (getMaxPagesPerUsagePage() * i));
  562. }
  563. }
  564. }
  565. private int getMaxPagesPerUsagePage() {
  566. return((getFormat().PAGE_SIZE - getFormat().OFFSET_USAGE_MAP_PAGE_DATA)
  567. * 8);
  568. }
  569. @Override
  570. public void addOrRemovePageNumber(int pageNumber, boolean add)
  571. throws IOException
  572. {
  573. if(!isPageWithinRange(pageNumber)) {
  574. throw new IOException("Page number " + pageNumber +
  575. " is out of supported range");
  576. }
  577. int pageIndex = (pageNumber / getMaxPagesPerUsagePage());
  578. int mapPageNum = getTableBuffer().getInt(
  579. calculateMapPagePointerOffset(pageIndex));
  580. ByteBuffer mapPageBuffer = null;
  581. if(mapPageNum > 0) {
  582. mapPageBuffer = _mapPageHolder.setPage(getPageChannel(), mapPageNum);
  583. } else {
  584. // Need to create a new usage map page
  585. mapPageBuffer = createNewUsageMapPage(pageIndex);
  586. mapPageNum = _mapPageHolder.getPageNumber();
  587. }
  588. updateMap(pageNumber,
  589. (pageNumber - (getMaxPagesPerUsagePage() * pageIndex)),
  590. mapPageBuffer, add);
  591. getPageChannel().writePage(mapPageBuffer, mapPageNum);
  592. }
  593. /**
  594. * Create a new usage map page and update the map declaration with a
  595. * pointer to it.
  596. * @param pageIndex Index of the page reference within the map declaration
  597. */
  598. private ByteBuffer createNewUsageMapPage(int pageIndex) throws IOException
  599. {
  600. ByteBuffer mapPageBuffer = _mapPageHolder.setNewPage(getPageChannel());
  601. mapPageBuffer.put(PageTypes.USAGE_MAP);
  602. mapPageBuffer.put((byte) 0x01); //Unknown
  603. mapPageBuffer.putShort((short) 0); //Unknown
  604. int mapPageNum = _mapPageHolder.getPageNumber();
  605. getTableBuffer().putInt(calculateMapPagePointerOffset(pageIndex),
  606. mapPageNum);
  607. writeTable();
  608. return mapPageBuffer;
  609. }
  610. private int calculateMapPagePointerOffset(int pageIndex) {
  611. return getRowStart() + getFormat().OFFSET_REFERENCE_MAP_PAGE_NUMBERS +
  612. (pageIndex * 4);
  613. }
  614. }
  615. /**
  616. * Utility class to traverse over the pages in the UsageMap. Remains valid
  617. * in the face of usage map modifications.
  618. */
  619. public final class PageCursor
  620. {
  621. /** handler for moving the page cursor forward */
  622. private final DirHandler _forwardDirHandler = new ForwardDirHandler();
  623. /** handler for moving the page cursor backward */
  624. private final DirHandler _reverseDirHandler = new ReverseDirHandler();
  625. /** the current used page number */
  626. private int _curPageNumber;
  627. /** the previous used page number */
  628. private int _prevPageNumber;
  629. /** the last read modification count on the UsageMap. we track this so
  630. that the cursor can detect updates to the usage map while traversing
  631. and act accordingly */
  632. private int _lastModCount;
  633. private PageCursor() {
  634. reset();
  635. }
  636. public UsageMap getUsageMap() {
  637. return UsageMap.this;
  638. }
  639. /**
  640. * Returns the DirHandler for the given direction
  641. */
  642. private DirHandler getDirHandler(boolean moveForward) {
  643. return (moveForward ? _forwardDirHandler : _reverseDirHandler);
  644. }
  645. /**
  646. * Returns {@code true} if this cursor is up-to-date with respect to its
  647. * usage map.
  648. */
  649. public boolean isUpToDate() {
  650. return(UsageMap.this._modCount == _lastModCount);
  651. }
  652. /**
  653. * @return valid page number if there was another page to read,
  654. * {@link RowId#LAST_PAGE_NUMBER} otherwise
  655. */
  656. public int getNextPage() {
  657. return getAnotherPage(Cursor.MOVE_FORWARD);
  658. }
  659. /**
  660. * @return valid page number if there was another page to read,
  661. * {@link RowId#FIRST_PAGE_NUMBER} otherwise
  662. */
  663. public int getPreviousPage() {
  664. return getAnotherPage(Cursor.MOVE_REVERSE);
  665. }
  666. /**
  667. * Gets another page in the given direction, returning the new page.
  668. */
  669. private int getAnotherPage(boolean moveForward) {
  670. DirHandler handler = getDirHandler(moveForward);
  671. if(_curPageNumber == handler.getEndPageNumber()) {
  672. if(!isUpToDate()) {
  673. restorePosition(_prevPageNumber);
  674. // drop through and retry moving to another page
  675. } else {
  676. // at end, no more
  677. return _curPageNumber;
  678. }
  679. }
  680. checkForModification();
  681. _prevPageNumber = _curPageNumber;
  682. _curPageNumber = handler.getAnotherPageNumber(_curPageNumber);
  683. return _curPageNumber;
  684. }
  685. /**
  686. * After calling this method, getNextPage will return the first page in
  687. * the map
  688. */
  689. public void reset() {
  690. beforeFirst();
  691. }
  692. /**
  693. * After calling this method, {@link #getNextPage} will return the first
  694. * page in the map
  695. */
  696. public void beforeFirst() {
  697. reset(Cursor.MOVE_FORWARD);
  698. }
  699. /**
  700. * After calling this method, {@link #getPreviousPage} will return the
  701. * last page in the map
  702. */
  703. public void afterLast() {
  704. reset(Cursor.MOVE_REVERSE);
  705. }
  706. /**
  707. * Resets this page cursor for traversing the given direction.
  708. */
  709. protected void reset(boolean moveForward) {
  710. _curPageNumber = getDirHandler(moveForward).getBeginningPageNumber();
  711. _prevPageNumber = _curPageNumber;
  712. _lastModCount = UsageMap.this._modCount;
  713. }
  714. /**
  715. * Restores a current position for the cursor (current position becomes
  716. * previous position).
  717. */
  718. private void restorePosition(int curPageNumber)
  719. {
  720. restorePosition(curPageNumber, _curPageNumber);
  721. }
  722. /**
  723. * Restores a current and previous position for the cursor.
  724. */
  725. protected void restorePosition(int curPageNumber, int prevPageNumber)
  726. {
  727. if((curPageNumber != _curPageNumber) ||
  728. (prevPageNumber != _prevPageNumber))
  729. {
  730. _prevPageNumber = updatePosition(prevPageNumber);
  731. _curPageNumber = updatePosition(curPageNumber);
  732. _lastModCount = UsageMap.this._modCount;
  733. } else {
  734. checkForModification();
  735. }
  736. }
  737. /**
  738. * Checks the usage map for modifications an updates state accordingly.
  739. */
  740. private void checkForModification() {
  741. if(!isUpToDate()) {
  742. _prevPageNumber = updatePosition(_prevPageNumber);
  743. _curPageNumber = updatePosition(_curPageNumber);
  744. _lastModCount = UsageMap.this._modCount;
  745. }
  746. }
  747. private int updatePosition(int pageNumber) {
  748. if(pageNumber < UsageMap.this.getFirstPageNumber()) {
  749. pageNumber = RowId.FIRST_PAGE_NUMBER;
  750. } else if(pageNumber > UsageMap.this.getLastPageNumber()) {
  751. pageNumber = RowId.LAST_PAGE_NUMBER;
  752. }
  753. return pageNumber;
  754. }
  755. @Override
  756. public String toString() {
  757. return getClass().getSimpleName() + " CurPosition " + _curPageNumber +
  758. ", PrevPosition " + _prevPageNumber;
  759. }
  760. /**
  761. * Handles moving the cursor in a given direction. Separates cursor
  762. * logic from value storage.
  763. */
  764. private abstract class DirHandler {
  765. public abstract int getAnotherPageNumber(int curPageNumber);
  766. public abstract int getBeginningPageNumber();
  767. public abstract int getEndPageNumber();
  768. }
  769. /**
  770. * Handles moving the cursor forward.
  771. */
  772. private final class ForwardDirHandler extends DirHandler {
  773. @Override
  774. public int getAnotherPageNumber(int curPageNumber) {
  775. if(curPageNumber == getBeginningPageNumber()) {
  776. return UsageMap.this.getFirstPageNumber();
  777. }
  778. return UsageMap.this.getNextPageNumber(curPageNumber);
  779. }
  780. @Override
  781. public int getBeginningPageNumber() {
  782. return RowId.FIRST_PAGE_NUMBER;
  783. }
  784. @Override
  785. public int getEndPageNumber() {
  786. return RowId.LAST_PAGE_NUMBER;
  787. }
  788. }
  789. /**
  790. * Handles moving the cursor backward.
  791. */
  792. private final class ReverseDirHandler extends DirHandler {
  793. @Override
  794. public int getAnotherPageNumber(int curPageNumber) {
  795. if(curPageNumber == getBeginningPageNumber()) {
  796. return UsageMap.this.getLastPageNumber();
  797. }
  798. return UsageMap.this.getPrevPageNumber(curPageNumber);
  799. }
  800. @Override
  801. public int getBeginningPageNumber() {
  802. return RowId.LAST_PAGE_NUMBER;
  803. }
  804. @Override
  805. public int getEndPageNumber() {
  806. return RowId.FIRST_PAGE_NUMBER;
  807. }
  808. }
  809. }
  810. }