third-party/atomix/storage/src/main/java/io/atomix/utils/serializer/KryoJournalSerdesBuilder.java

   1 /*
   2  * Copyright 2014-2021 Open Networking Foundation
   3  * Copyright 2023 PANTHEON.tech, s.r.o.
   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.utils.serializer;
  18
  19 import static com.google.common.base.Preconditions.checkState;
  20 import static java.util.Objects.requireNonNull;
  21
  22 import io.atomix.storage.journal.JournalSerdes;
  23 import io.atomix.storage.journal.JournalSerdes.Builder;
  24 import io.atomix.storage.journal.JournalSerdes.EntrySerdes;
  25 import java.util.ArrayList;
  26 import java.util.List;
  27
  28 public final class KryoJournalSerdesBuilder implements Builder {
  29     private final List<RegisteredType> types = new ArrayList<>();
  30     private ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
  31
  32     @Override
  33     public KryoJournalSerdesBuilder register(final EntrySerdes<?> serdes, final Class<?>... classes) {
  34         types.add(new RegisteredType(new EntrySerializer<>(serdes), classes));
  35         return this;
  36     }
  37
  38     @Override
  39     public KryoJournalSerdesBuilder setClassLoader(final ClassLoader classLoader) {
  40         this.classLoader = requireNonNull(classLoader);
  41         return this;
  42     }
  43
  44     @Override
  45     public JournalSerdes build() {
  46         return build(KryoJournalSerdes.NO_NAME);
  47     }
  48
  49     @Override
  50     public JournalSerdes build(final String friendlyName) {
  51         checkState(!types.isEmpty(), "No serializers registered");
  52         return new KryoJournalSerdes(types, classLoader, friendlyName);
  53     }
  54 }