atomix-storage/src/main/java/io/atomix/storage/journal/SegmentedJournalWriter.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
  23 /**
  24  * A {@link JournalWriter} backed by a {@link ByteBufWriter}.
  25  */
  26 @NonNullByDefault
  27 final class SegmentedJournalWriter<E> implements JournalWriter<E> {
  28     private final ByteBufMapper<E> mapper;
  29     private final ByteBufWriter writer;
  30
  31     SegmentedJournalWriter(final ByteBufWriter writer, final ByteBufMapper<E> mapper) {
  32         this.writer = requireNonNull(writer);
  33         this.mapper = requireNonNull(mapper);
  34     }
  35
  36     @Override
  37     public long getNextIndex() {
  38         return writer.nextIndex();
  39     }
  40
  41     @Override
  42     public void commit(final long index) {
  43         writer.commit(index);
  44     }
  45
  46     @Override
  47     public <T extends E> Indexed<T> append(final T entry) {
  48         final var index = writer.nextIndex();
  49         return new Indexed<>(index, entry, writer.append(mapper, entry));
  50     }
  51
  52     @Override
  53     public void reset(final long index) {
  54         writer.reset(index);
  55     }
  56
  57     @Override
  58     public void flush() {
  59         writer.flush();
  60     }
  61 }