2 * Copyright 2023 PANTHEON.tech, s.r.o.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
16 package io.atomix.utils.serializer;
18 import static java.util.Objects.requireNonNull;
20 import com.esotericsoftware.kryo.Kryo;
21 import com.esotericsoftware.kryo.KryoException;
22 import com.esotericsoftware.kryo.io.Input;
23 import com.esotericsoftware.kryo.serializers.JavaSerializer;
24 import io.atomix.storage.journal.JournalSerdes.EntryInput;
25 import java.io.IOException;
27 final class KryoEntryInput implements EntryInput {
28 private final Kryo kryo;
29 private final Input input;
30 private final JavaSerializer javaSerializer;
32 KryoEntryInput(final Kryo kryo, final Input input, final JavaSerializer javaSerializer) {
33 this.kryo = requireNonNull(kryo);
34 this.input = requireNonNull(input);
35 this.javaSerializer = requireNonNull(javaSerializer);
39 public byte[] readBytes(final int length) throws IOException {
41 return input.readBytes(length);
42 } catch (KryoException e) {
43 throw new IOException(e);
48 public long readLong() throws IOException {
50 return input.readLong(false);
51 } catch (KryoException e) {
52 throw new IOException(e);
57 public Object readObject() throws IOException {
59 return javaSerializer.read(kryo, input, null);
60 } catch (KryoException e) {
61 throw new IOException(e);
66 public String readString() throws IOException {
68 return input.readString();
69 } catch (KryoException e) {
70 throw new IOException(e);
75 public int readVarInt() throws IOException {
77 return input.readVarInt(true);
78 } catch (KryoException e) {
79 throw new IOException(e);