+++ /dev/null
-/*
- * Copyright 2017-present Open Networking Foundation
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-package io.atomix.storage.journal;
-
-import java.util.NoSuchElementException;
-
-/**
- * Raft log reader.
- */
-public class SegmentedJournalReader<E> implements JournalReader<E> {
-
- private final SegmentedJournal<E> journal;
- private JournalSegment<E> currentSegment;
- private Indexed<E> previousEntry;
- private MappableJournalSegmentReader<E> currentReader;
- private final Mode mode;
-
- public SegmentedJournalReader(SegmentedJournal<E> journal, long index, Mode mode) {
- this.journal = journal;
- this.mode = mode;
- initialize(index);
- }
-
- /**
- * Initializes the reader to the given index.
- */
- private void initialize(long index) {
- currentSegment = journal.getSegment(index);
- currentSegment.acquire();
- currentReader = currentSegment.createReader();
- long nextIndex = getNextIndex();
- while (index > nextIndex && hasNext()) {
- next();
- nextIndex = getNextIndex();
- }
- }
-
- @Override
- public long getFirstIndex() {
- return journal.getFirstSegment().index();
- }
-
- @Override
- public long getCurrentIndex() {
- long currentIndex = currentReader.getCurrentIndex();
- if (currentIndex != 0) {
- return currentIndex;
- }
- if (previousEntry != null) {
- return previousEntry.index();
- }
- return 0;
- }
-
- @Override
- public Indexed<E> getCurrentEntry() {
- Indexed<E> currentEntry = currentReader.getCurrentEntry();
- if (currentEntry != null) {
- return currentEntry;
- }
- return previousEntry;
- }
-
- @Override
- public long getNextIndex() {
- return currentReader.getNextIndex();
- }
-
- @Override
- public void reset() {
- currentReader.close();
- currentSegment.release();
- currentSegment = journal.getFirstSegment();
- currentSegment.acquire();
- currentReader = currentSegment.createReader();
- previousEntry = null;
- }
-
- @Override
- public void reset(long index) {
- // If the current segment is not open, it has been replaced. Reset the segments.
- if (!currentSegment.isOpen()) {
- reset();
- }
-
- if (index < currentReader.getNextIndex()) {
- rewind(index);
- } else if (index > currentReader.getNextIndex()) {
- forward(index);
- } else {
- currentReader.reset(index);
- }
- }
-
- /**
- * Rewinds the journal to the given index.
- */
- private void rewind(long index) {
- if (currentSegment.index() >= index) {
- JournalSegment<E> segment = journal.getSegment(index - 1);
- if (segment != null) {
- currentReader.close();
- currentSegment.release();
- currentSegment = segment;
- currentSegment.acquire();
- currentReader = currentSegment.createReader();
- }
- }
-
- currentReader.reset(index);
- previousEntry = currentReader.getCurrentEntry();
- }
-
- /**
- * Fast forwards the journal to the given index.
- */
- private void forward(long index) {
- while (getNextIndex() < index && hasNext()) {
- next();
- }
- }
-
- @Override
- public boolean hasNext() {
- if (mode == Mode.ALL) {
- return hasNextEntry();
- }
-
- long nextIndex = getNextIndex();
- long commitIndex = journal.getCommitIndex();
- return nextIndex <= commitIndex && hasNextEntry();
- }
-
- private boolean hasNextEntry() {
- if (!currentReader.hasNext()) {
- JournalSegment<E> nextSegment = journal.getNextSegment(currentSegment.index());
- if (nextSegment != null && nextSegment.index() == getNextIndex()) {
- previousEntry = currentReader.getCurrentEntry();
- currentSegment.release();
- currentSegment = nextSegment;
- currentSegment.acquire();
- currentReader = currentSegment.createReader();
- return currentReader.hasNext();
- }
- return false;
- }
- return true;
- }
-
- @Override
- public Indexed<E> next() {
- if (!currentReader.hasNext()) {
- JournalSegment<E> nextSegment = journal.getNextSegment(currentSegment.index());
- if (nextSegment != null && nextSegment.index() == getNextIndex()) {
- previousEntry = currentReader.getCurrentEntry();
- currentSegment.release();
- currentSegment = nextSegment;
- currentSegment.acquire();
- currentReader = currentSegment.createReader();
- return currentReader.next();
- } else {
- throw new NoSuchElementException();
- }
- } else {
- previousEntry = currentReader.getCurrentEntry();
- return currentReader.next();
- }
- }
-
- @Override
- public void close() {
- currentReader.close();
- journal.closeReader(this);
- }
-}