org.dflib.jjava.jupyter.messages.Message Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jupyter-jvm-basekernel Show documentation
The newest version!
package org.dflib.jjava.jupyter.messages;

import java.util.Arrays;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

public class Message implements MessageContext {
    private List identities;

    private Header header;

    /**
     * Optional, in a chain of messages this is copied from
     * the parent so the client can better track where the messages
     * come from.
     */
    private Header parentHeader;

    private Map metadata;

    private T content;

    private List blobs;

    public Message(MessageContext ctx, MessageType type, T content) {
        this(ctx, type, content, null, null);
    }

    public Message(MessageContext ctx, MessageType type, T content, List blobs, Map metadata) {
        this(
                ctx != null ? ctx.getIdentities() : Collections.emptyList(),
                new Header<>(ctx, type),
                ctx != null ? ctx.getHeader() : null,
                metadata,
                content,
                blobs
        );
    }

    public Message(Header header, T content) {
        this(Collections.emptyList(), header, null, null, content, null);
    }

    public Message(Header header, T content, Map metadata, List blobs) {
        this(Collections.emptyList(), header, null, metadata, content, blobs);
    }

    public Message(List identities, Header header, T content) {
        this(identities, header, null, null, content, null);
    }

    public Message(List identities, Header header, Header parentHeader, Map metadata, T content, List blobs) {
        this.identities = identities;
        this.header = header;
        this.parentHeader = parentHeader;
        this.metadata = metadata;
        this.content = content;
        this.blobs = blobs;
    }

    @Override
    public List getIdentities() {
        return identities;
    }

    @Override
    public Header getHeader() {
        return header;
    }

    public boolean hasParentHeader() {
        return parentHeader != null;
    }

    public Header getParentHeader() {
        return parentHeader;
    }

    public boolean hasMetadata() {
        return metadata != null;
    }

    public Map getMetadata() {
        return metadata;
    }

    public Map getNonNullMetadata() {
        if (this.hasMetadata())
            return this.getMetadata();
        this.metadata = new LinkedHashMap<>();
        return this.metadata;
    }

    public T getContent() {
        return content;
    }

    public List getBlobs() {
        return blobs;
    }

    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder("Message {\n");
        sb.append("\tidentities = [\n");
        for (byte[] id : identities)
            sb.append("\t\t").append(Arrays.toString(id)).append("\n");
        sb.append("\t]\n");
        sb.append("\theader = ").append(header).append("\n");
        sb.append("\tparentHeader = ").append(parentHeader).append("\n");
        sb.append("\tmetadata = ").append(metadata).append("\n");
        sb.append("\tcontent = ").append(content).append("\n");
        sb.append("\tblobs = [\n");
        if (blobs != null)
            for (byte[] blob : blobs)
                sb.append("\t\t").append(Arrays.toString(blob)).append("\n");
        sb.append("\t]\n");
        sb.append("}\n");
        return sb.toString();
    }
}