io.github.qsy7.serialization.modules.snakeyaml.MessageRepresenter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of snakeyaml Show documentation

There is a newer version: 0.3.3

package io.github.qsy7.serialization.modules.snakeyaml;

import io.github.qsy7.datastore.api.model.entity.AbstractEntity;
import io.github.qsy7.serialization.api.annotation.PrivateField;
import io.github.qsy7.serialization.modules.snakeyaml.types.DurationRepresenter;
import java.lang.reflect.Field;
import java.time.Duration;
import java.util.HashSet;
import java.util.Set;
import org.yaml.snakeyaml.introspector.Property;
import org.yaml.snakeyaml.nodes.NodeTuple;
import org.yaml.snakeyaml.nodes.Tag;
import org.yaml.snakeyaml.representer.Representer;

public class MessageRepresenter extends Representer {
  public MessageRepresenter() {
    //    /**
    //     * TODO: this doesn't do anything instead, iterate through all model objects inheriting
    // from
    //     * AbstractMessage / portage and then add the class/tag combination
    //     */
    // this doesn't do anything because this is basically instructing yaml how to serialize this
    // class which we could care less about
    //    addClassTag(getClass(), new Tag("!" + getClass().getSimpleName()));
    this.representers.put(Duration.class, new DurationRepresenter());
    //    this.(Duration.class, new DurationConstructor());
  }

  @Override
  protected Set getProperties(Class type) {
    final Set properties = super.getProperties(type);
    final Set safeProperties = new HashSet<>();

    for (final Property property : properties) {
      if (!isFieldPrivate(type, property)) {
        safeProperties.add(property);
      }
    }

    return (safeProperties);
  }

  private static boolean isFieldPrivate(final Class type, final Property property) {
    if (!isExtendsAbstractEntity(type)) return true;

    try {
      final Field field = type.getDeclaredField(property.getName());
      return (field.isAnnotationPresent(PrivateField.class));
    } catch (NoSuchFieldException e) {
      return (isFieldPrivate(type.getSuperclass(), property));
    }
  }

  private static boolean isExtendsAbstractEntity(final Class type) {
    return AbstractEntity.class.isAssignableFrom(type);
  }

  @Override
  protected NodeTuple representJavaBeanProperty(
      Object javaBean, Property property, Object propertyValue, Tag customTag) {
    if (propertyValue == null) {
      return null;
    }

    return super.representJavaBeanProperty(javaBean, property, propertyValue, customTag);
  }
}