| /* |
| * Copyright (c) 1998, 2021 Oracle and/or its affiliates. All rights reserved. |
| * |
| * This program and the accompanying materials are made available under the |
| * terms of the Eclipse Public License v. 2.0 which is available at |
| * http://www.eclipse.org/legal/epl-2.0, |
| * or the Eclipse Distribution License v. 1.0 which is available at |
| * http://www.eclipse.org/org/documents/edl-v10.php. |
| * |
| * SPDX-License-Identifier: EPL-2.0 OR BSD-3-Clause |
| */ |
| |
| // Contributors: |
| // Oracle - initial impl |
| package org.eclipse.persistence.sessions.serializers.kryo; |
| |
| import java.io.ByteArrayInputStream; |
| import java.io.ByteArrayOutputStream; |
| import java.io.InputStream; |
| import java.io.OutputStream; |
| import java.lang.reflect.Constructor; |
| import java.lang.reflect.Method; |
| |
| import org.eclipse.persistence.exceptions.ValidationException; |
| import org.eclipse.persistence.sessions.Session; |
| import org.eclipse.persistence.sessions.serializers.AbstractSerializer; |
| |
| /** |
| * Uses Kryo to serialize the object. |
| * @author James Sutherland |
| */ |
| public class KryoSerializer extends AbstractSerializer { |
| /** Kryo is not thread safe, so need thread local. */ |
| transient ThreadLocal<Object> kryo; |
| transient Constructor<?> kryoConstructor; |
| transient Constructor<?> outputConstructor; |
| transient Constructor<?> inputConstructor; |
| transient Method writeMethod; |
| transient Method readMethod; |
| transient Method inputCloseMethod; |
| transient Method outputCloseMethod; |
| |
| /** |
| * Default constructor. |
| */ |
| public KryoSerializer() { |
| try { |
| Class<?> kryoClass = Class.forName("com.esotericsoftware.kryo.Kryo"); |
| this.kryoConstructor = kryoClass.getConstructor(); |
| Class<?> inputClass = Class.forName("com.esotericsoftware.kryo.io.Input"); |
| this.inputConstructor = inputClass.getConstructor(InputStream.class); |
| Class<?> outputClass = Class.forName("com.esotericsoftware.kryo.io.Output"); |
| this.outputConstructor = outputClass.getConstructor(OutputStream.class); |
| this.writeMethod = kryoClass.getMethod("writeClassAndObject", outputClass, Object.class); |
| this.readMethod = kryoClass.getMethod("readClassAndObject", inputClass); |
| this.inputCloseMethod = inputClass.getMethod("close"); |
| this.outputCloseMethod = outputClass.getMethod("close"); |
| this.kryo = new ThreadLocal<>(); |
| } catch (Exception exception) { |
| throw ValidationException.reflectiveExceptionWhileCreatingClassInstance("com.esotericsoftware.kryo.Kryo", exception); |
| } |
| } |
| |
| @Override |
| public Class<?> getType() { |
| return byte[].class; |
| } |
| |
| /** |
| * Return an instance of {@code com.esotericsoftware.kryo.Kryo}. |
| * |
| * @return the instance of {@code com.esotericsoftware.kryo.Kryo} |
| */ |
| public Object getKryo() { |
| Object value = this.kryo.get(); |
| if (value == null) { |
| try { |
| value = this.kryoConstructor.newInstance(); |
| this.kryo.set(value); |
| } catch (Exception exception) { |
| throw ValidationException.reflectiveExceptionWhileCreatingClassInstance("com.esotericsoftware.kryo.Kryo", exception); |
| } |
| } |
| return value; |
| } |
| |
| @Override |
| public Object serialize(Object object, Session session) { |
| try { |
| ByteArrayOutputStream stream = new ByteArrayOutputStream(); |
| Object output = this.outputConstructor.newInstance(stream); |
| this.writeMethod.invoke(getKryo(), output, object); |
| this.outputCloseMethod.invoke(output); |
| return stream.toByteArray(); |
| } catch (Exception exception) { |
| throw new RuntimeException(exception); |
| } |
| } |
| |
| @Override |
| public Object deserialize(Object bytes, Session session) { |
| try { |
| ByteArrayInputStream stream = new ByteArrayInputStream((byte[])bytes); |
| Object input = this.inputConstructor.newInstance(stream); |
| Object result = this.readMethod.invoke(getKryo(), input); |
| this.inputCloseMethod.invoke(input); |
| return result; |
| } catch (Exception exception) { |
| throw new RuntimeException(exception); |
| } |
| } |
| |
| @Override |
| public String toString() { |
| return getClass().getSimpleName(); |
| } |
| } |