| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You 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 org.apache.ignite.marshaller; |
| |
| import org.apache.ignite.IgniteCheckedException; |
| import org.apache.ignite.Ignition; |
| import org.apache.ignite.internal.util.IgniteUtils; |
| import org.apache.ignite.internal.util.typedef.internal.U; |
| import org.jetbrains.annotations.Nullable; |
| |
| import java.io.InputStream; |
| import java.io.OutputStream; |
| |
| /** |
| * Marshaller allowing for {@link Ignition#localIgnite()} calls. |
| */ |
| public abstract class AbstractNodeNameAwareMarshaller extends AbstractMarshaller { |
| /** Whether node name is set. */ |
| private volatile boolean nodeNameSet; |
| |
| /** Node name. */ |
| private volatile String nodeName = U.LOC_IGNITE_NAME_EMPTY; |
| |
| /** |
| * Set node name. |
| * |
| * @param nodeName Node name. |
| */ |
| public void nodeName(@Nullable String nodeName) { |
| if (!nodeNameSet) { |
| this.nodeName = nodeName; |
| |
| nodeNameSet = true; |
| } |
| } |
| |
| /** {@inheritDoc} */ |
| @Override public byte[] marshal(@Nullable Object obj) throws IgniteCheckedException { |
| String oldNodeName = IgniteUtils.setCurrentIgniteName(nodeName); |
| |
| try { |
| return marshal0(obj); |
| } |
| finally { |
| IgniteUtils.restoreOldIgniteName(oldNodeName, nodeName); |
| } |
| } |
| |
| /** {@inheritDoc} */ |
| @Override public void marshal(@Nullable Object obj, OutputStream out) throws IgniteCheckedException { |
| String oldNodeName = IgniteUtils.setCurrentIgniteName(nodeName); |
| |
| try { |
| marshal0(obj, out); |
| } |
| finally { |
| IgniteUtils.restoreOldIgniteName(oldNodeName, nodeName); |
| } |
| } |
| |
| /** {@inheritDoc} */ |
| @Override public <T> T unmarshal(byte[] arr, @Nullable ClassLoader clsLdr) throws IgniteCheckedException { |
| String oldNodeName = IgniteUtils.setCurrentIgniteName(nodeName); |
| |
| try { |
| return unmarshal0(arr, clsLdr); |
| } |
| finally { |
| IgniteUtils.restoreOldIgniteName(oldNodeName, nodeName); |
| } |
| } |
| |
| /** {@inheritDoc} */ |
| @Override public <T> T unmarshal(InputStream in, @Nullable ClassLoader clsLdr) throws IgniteCheckedException { |
| String oldNodeName = IgniteUtils.setCurrentIgniteName(nodeName); |
| |
| try { |
| return unmarshal0(in, clsLdr); |
| } |
| finally { |
| IgniteUtils.restoreOldIgniteName(oldNodeName, nodeName); |
| } |
| } |
| |
| /** |
| * Marshals object to the output stream. This method should not close |
| * given output stream. |
| * |
| * @param obj Object to marshal. {@code null} object will be marshaled to binary {@code null} representation. |
| * @param out Output stream to marshal into. |
| * @throws IgniteCheckedException If marshalling failed. |
| */ |
| protected abstract void marshal0(@Nullable Object obj, OutputStream out) throws IgniteCheckedException; |
| |
| /** |
| * Marshals object to byte array. |
| * |
| * @param obj Object to marshal. {@code null} object will be marshaled to binary {@code null} representation. |
| * @return Byte array. |
| * @throws IgniteCheckedException If marshalling failed. |
| */ |
| protected abstract byte[] marshal0(@Nullable Object obj) throws IgniteCheckedException; |
| |
| /** |
| * Unmarshals object from the input stream using given class loader. |
| * This method should not close given input stream. |
| * |
| * @param <T> Type of unmarshalled object. |
| * @param in Input stream. |
| * @param clsLdr If not {@code null} then given class loader will be used for unmarshal object. |
| * @return Unmarshalled object. |
| * @throws IgniteCheckedException If unmarshalling failed. |
| */ |
| protected abstract <T> T unmarshal0(InputStream in, @Nullable ClassLoader clsLdr) throws IgniteCheckedException; |
| |
| /** |
| * Unmarshals object from byte array using given class loader. |
| * |
| * @param <T> Type of unmarshalled object. |
| * @param arr Byte array. |
| * @param clsLdr If not {@code null} then given class loader will be used for unmarshal object. |
| * @return Unmarshalled object. |
| * @throws IgniteCheckedException If unmarshalling failed. |
| */ |
| protected abstract <T> T unmarshal0(byte[] arr, @Nullable ClassLoader clsLdr) throws IgniteCheckedException; |
| } |