dev.langchain4j.store.embedding.infinispan.LangChainMetadataMarshaller Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of langchain4j-infinispan Show documentation

There is a newer version: 0.36.2

Show newest version

package dev.langchain4j.store.embedding.infinispan;

import org.infinispan.protostream.MessageMarshaller;

import java.io.IOException;

/**
 * Marshaller to read and write metadata to Infinispan
 */
public class LangChainMetadataMarshaller implements MessageMarshaller {

    private final String typeName;

    /**
     * Constructor for the LangChainMetadata Marshaller
     * @param typeName, the full type of the protobuf entity
     */
    public LangChainMetadataMarshaller(String typeName) {
        this.typeName = typeName;
    }

    @Override
    public LangChainMetadata readFrom(ProtoStreamReader reader) throws IOException {
        String name = reader.readString("name");
        String value = reader.readString("value");
        return new LangChainMetadata(name, value);
    }

    @Override
    public void writeTo(ProtoStreamWriter writer, LangChainMetadata item)
            throws IOException {
        writer.writeString("name", item.name());
        writer.writeString("value", item.value());
    }

    @Override
    public Class getJavaClass() {
        return LangChainMetadata.class;
    }

    @Override
    public String getTypeName() {
        return typeName;
    }
}