io.deephaven.kafka.ingest.JsonNodeBooleanFieldCopier Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of deephaven-extensions-kafka Show documentation

Kafka: Integrating Engine tables with Kafka

The newest version!

//
// Copyright (c) 2016-2024 Deephaven Data Labs and Patent Pending
//
package io.deephaven.kafka.ingest;

import com.fasterxml.jackson.core.JsonPointer;
import com.fasterxml.jackson.databind.JsonNode;
import io.deephaven.UncheckedDeephavenException;
import io.deephaven.chunk.ObjectChunk;
import io.deephaven.chunk.WritableChunk;
import io.deephaven.chunk.WritableByteChunk;
import io.deephaven.chunk.attributes.Values;
import io.deephaven.util.BooleanUtils;

public class JsonNodeBooleanFieldCopier implements FieldCopier {
    private final JsonPointer fieldPointer;

    public JsonNodeBooleanFieldCopier(final String fieldPointerStr) {
        this.fieldPointer = JsonPointer.compile(fieldPointerStr);
    }

    @Override
    public void copyField(
            final ObjectChunk inputChunk,
            final WritableChunk publisherChunk,
            final int sourceOffset,
            final int destOffset,
            final int length) {
        final WritableByteChunk output = publisherChunk.asWritableByteChunk();
        for (int ii = 0; ii < length; ++ii) {
            final JsonNode node = (JsonNode) inputChunk.get(ii + sourceOffset);
            final String valueAsString = JsonNodeUtil.getString(node, fieldPointer, true, true);
            final Boolean valueAsBoolean;
            if (valueAsString == null) {
                valueAsBoolean = null;
            } else {
                switch (valueAsString.trim()) {
                    case "TRUE":
                    case "True":
                    case "true":
                    case "T":
                    case "t":
                    case "1":
                        valueAsBoolean = Boolean.TRUE;
                        break;
                    case "FALSE":
                    case "False":
                    case "false":
                    case "F":
                    case "f":
                        valueAsBoolean = Boolean.FALSE;
                        break;
                    case "":
                        valueAsBoolean = null;
                        break;
                    default:
                        throw new UncheckedDeephavenException(
                                "value " + valueAsString + " not recognized as Boolean for field " + fieldPointer);
                }
            }
            output.set(ii + destOffset, BooleanUtils.booleanAsByte(valueAsBoolean));
        }
    }
}