org.modelmapper.protobuf.ProtobufValueWriter Maven / Gradle / Ivy

Go to download
/*
 * Copyright 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 org.modelmapper.protobuf;

import com.google.protobuf.Message.Builder;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.Collection;
import org.modelmapper.internal.Errors;
import org.modelmapper.spi.ValueWriter;

/**
 * Protocol buffer ValueReader implementation.
 *
 * @author Chun Han Hsiao
 */
public class ProtobufValueWriter implements ValueWriter {
  @Override
  public void setValue(Builder destination, Object value, String memberName) {
    try {
      if (ProtobufHelper.hasBuilder(destination.getClass(), memberName)) {
        Method method = ProtobufHelper.setterForBuilder(destination.getClass(), memberName);
        method.invoke(destination, value);
      } else {
        Method method = ProtobufHelper.getter(destination.getClass(), memberName);
        method.invoke(destination, value);
      }
    } catch (NoSuchMethodException e) {
      throw new Errors().addMessage(e, "Cannot get the member").toMappingException();
    } catch (IllegalAccessException e) {
      throw new Errors().addMessage(e, "Cannot set the member").toMappingException();
    } catch (InvocationTargetException e) {
      throw new Errors().addMessage(e, "Cannot set the member").toMappingException();
    }
  }

  @Override
  public Member getMember(Class destinationType, String memberName) {
    try {
      final Class memberType = ProtobufHelper.getter(destinationType, memberName)
          .getReturnType();
      final Method setter = ProtobufHelper.setter(destinationType, memberName);
      return new Member(memberType) {
        @Override
        public void setValue(Builder destination, Object value) {
          try {
            if (value != null)
              setter.invoke(destination, value);
          } catch (IllegalAccessException e) {
            throw new Errors().addMessage(e, "Cannot set the member").toMappingException();
          } catch (InvocationTargetException e) {
            throw new Errors().addMessage(e, "Cannot set the member").toMappingException();
          }
        }
      };
    } catch (NoSuchMethodException e) {
      throw new Errors().addMessage(e, "Cannot get the member").toMappingException();
    }
  }

  @Override
  public Collection memberNames(Class destinationType) {
    return ProtobufHelper.fields(destinationType);
  }

  @Override
  public boolean isResolveMembersSupport() {
    return true;
  }
}