ch.rasc.xodusqueue.serializer.DefaultXodusQueueSerializer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of xodus-queue Show documentation

Persistent java.util.Queue implementation with Xodus

There is a newer version: 1.0.2

/**
 * Copyright 2018-2018 the original author or authors.
 *
 * 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 ch.rasc.xodusqueue.serializer;

import com.esotericsoftware.kryo.Kryo;
import com.esotericsoftware.kryo.io.Input;
import com.esotericsoftware.kryo.io.Output;
import com.esotericsoftware.kryo.pool.KryoFactory;
import com.esotericsoftware.kryo.pool.KryoPool;

import jetbrains.exodus.ArrayByteIterable;
import jetbrains.exodus.ByteIterable;

public class DefaultXodusQueueSerializer implements XodusQueueSerializer {

	private final KryoPool kryoPool;

	final Class entryClass;

	public DefaultXodusQueueSerializer(final Class entryClass) {
		KryoFactory factory = () -> {
			Kryo kryo = new Kryo();
			kryo.register(entryClass);
			return kryo;
		};
		this.kryoPool = new KryoPool.Builder(factory).build();
		this.entryClass = entryClass;
	}

	@Override
	public T fromEntry(ByteIterable value) {
		ArrayByteIterable abi = new ArrayByteIterable(value);
		try (Input input = new Input(abi.getBytesUnsafe(), 0, abi.getLength())) {
			return this.kryoPool.run(kryo -> kryo.readObject(input, this.entryClass));
		}
	}

	@Override
	public ByteIterable toEntry(T element) {
		Kryo kryo = this.kryoPool.borrow();
		try (Output output = new Output(64, -1)) {
			kryo.writeObject(output, element);
			return new ArrayByteIterable(output.toBytes());
		}
		finally {
			this.kryoPool.release(kryo);
		}
	}

}