com.github.frtu.dot.FieldStream Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of library-dot Show documentation
Library for dot language
The newest version!
package com.github.frtu.dot;

import com.github.frtu.dot.attributes.Attributes;
import com.github.frtu.dot.model.GraphEdge;
import com.github.frtu.dot.model.GraphNode;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.lang.reflect.Field;
import java.util.Arrays;
import java.util.function.BiConsumer;

/**
 * Util method to auto parse all class fields for rendering.
 *
 * @author frtu
 * @since 0.3.6
 */
public class FieldStream {
    private static final Logger LOGGER = LoggerFactory.getLogger(FieldStream.class);

    private static Field[] nodeFields = GraphNode.class.getDeclaredFields();
    private static Field[] edgeFields = GraphEdge.class.getDeclaredFields();

    static {
        if (LOGGER.isDebugEnabled()) {
            LOGGER.debug("== nodeFields ==");
            Arrays.stream(nodeFields).forEach(field -> LOGGER.debug(field.getName()));
            LOGGER.debug("== edgeFields ==");
            Arrays.stream(edgeFields).forEach(field -> LOGGER.debug(field.getName()));
        }
    }

    private T value;
    private Field[] fields;
    private int firstVisibleField = 0;

    FieldStream(GraphNode graphNode) {
        this.value = (T) graphNode;
        this.fields = nodeFields;
        this.firstVisibleField = GraphNode.FIRST_VISIBLE_FIELD_INDEX;
    }

    FieldStream(GraphEdge graphEdge) {
        this.value = (T) graphEdge;
        this.fields = edgeFields;
        this.firstVisibleField = GraphEdge.FIRST_VISIBLE_FIELD_INDEX;
    }

    FieldStream(Attributes attributes) {
        this.value = (T) attributes;
        this.fields = attributes.getClass().getDeclaredFields();
    }

    public static FieldStream attributes(Attributes attributes) {
        return new FieldStream(attributes);
    }

    public static FieldStream node(GraphNode graphNode) {
        return new FieldStream(graphNode);
    }

    public static FieldStream edge(GraphEdge graphEdge) {
        return new FieldStream(graphEdge);
    }

    public void apply(BiConsumer consumer) {
        for (int i = this.firstVisibleField; i < fields.length; i++) {
            final Object value;
            final Field nodeField = fields[i];
            try {
                nodeField.setAccessible(true);
                value = nodeField.get(this.value);
            } catch (IllegalAccessException e) {
                throw new IllegalStateException("SHOULD NEVER HAPPEN");
            }
            if (value != null) {
                consumer.accept(nodeField.getName(), value);
            }
        }
    }
}