atomix-storage/src/main/java/io/atomix/storage/journal/SegmentedJournalReader.java

   1 /*
   2  * Copyright 2017-2022 Open Networking Foundation and others.  All rights reserved.
   3  * Copyright (c) 2024 PANTHEON.tech, s.r.o. and others.
   4  *
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  * http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  */
  17 package io.atomix.storage.journal;
  18
  19 import static java.util.Objects.requireNonNull;
  20
  21 import org.eclipse.jdt.annotation.NonNullByDefault;
  22 import org.eclipse.jdt.annotation.Nullable;
  23 import org.opendaylight.controller.raft.journal.EntryReader;
  24 import org.opendaylight.controller.raft.journal.FromByteBufMapper;
  25
  26 /**
  27  * A {@link JournalReader} backed by a {@link EntryReader}.
  28  */
  29 @NonNullByDefault
  30 final class SegmentedJournalReader<E> implements JournalReader<E> {
  31     private final FromByteBufMapper<E> mapper;
  32     private final EntryReader reader;
  33
  34     SegmentedJournalReader(final EntryReader reader, final FromByteBufMapper<E> mapper) {
  35         this.reader = requireNonNull(reader);
  36         this.mapper = requireNonNull(mapper);
  37     }
  38
  39     @Override
  40     public long getNextIndex() {
  41         return reader.nextIndex();
  42     }
  43
  44     @Override
  45     public void reset() {
  46         reader.reset();
  47     }
  48
  49     @Override
  50     public void reset(final long index) {
  51         reader.reset(index);
  52     }
  53
  54     @Override
  55     public <T> @Nullable T tryNext(final EntryMapper<E, T> entryMapper) {
  56         return reader.tryNext((index, buf) -> {
  57             final var size = buf.readableBytes();
  58             return requireNonNull(entryMapper.mapEntry(index, mapper.bytesToObject(index, buf), size));
  59         });
  60     }
  61
  62     @Override
  63     public void close() {
  64         reader.close();
  65     }
  66 }