org.wildfly.clustering.marshalling.protostream.reflect.UnaryFieldMarshaller Maven / Gradle / Ivy

Go to download
/*
 * Copyright The WildFly Authors
 * SPDX-License-Identifier: Apache-2.0
 */

package org.wildfly.clustering.marshalling.protostream.reflect;

import java.lang.reflect.Constructor;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.security.PrivilegedAction;
import java.util.function.Function;

import org.wildfly.security.manager.WildFlySecurityManager;

/**
 * Generic marshaller based on a single non-public field.
 * @author Paul Ferraro
 */
public class UnaryFieldMarshaller extends UnaryMemberMarshaller {

    public UnaryFieldMarshaller(Class targetClass, Class fieldClass, Function factory) {
        super(targetClass, Reflect::getValue, Reflect::findField, fieldClass, factory);
    }

    public UnaryFieldMarshaller(Class targetClass, Class fieldClass) {
        this(targetClass, fieldClass, Reflect.getConstructor(targetClass, fieldClass));
    }

    private UnaryFieldMarshaller(Class targetClass, Class fieldClass, Constructor constructor) {
        this(targetClass, fieldClass, new Function<>() {
            @Override
            public T apply(F value) {
                return WildFlySecurityManager.doUnchecked(new PrivilegedAction<>() {
                    @Override
                    public T run() {
                        try {
                            return constructor.newInstance(value);
                        } catch (IllegalAccessException | InstantiationException | InvocationTargetException e) {
                            throw new IllegalStateException(e);
                        }
                    }
                });
            }
        });
    }
}