com.redhat.insights.kafka.connect.transforms.DropIf Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of connect-transforms Show documentation

A set of general-purpose SMTs developed as part of Red Hat Insights

There is a newer version: 0.1.2

package com.redhat.insights.kafka.connect.transforms;

import org.apache.kafka.common.config.ConfigDef;
import org.apache.kafka.connect.connector.ConnectRecord;

/**
 * Drops a message key or value based on the given ECMAScript predicate.
 */
public abstract class DropIf> extends BooleanPredicateTransform implements KeyOrValueTransformation  {

    public DropIf(){
        super(new ConfigDef()
            .define(CONFIG_FIELD, ConfigDef.Type.STRING, null, ConfigDef.Importance.HIGH,
                "ECMAScript predicate to be evaluated for each message. If the predicate evaluates to true the message key/value is dropped.")
            // https://cwiki.apache.org/confluence/display/KAFKA/KIP-585
            .define(CONFIG_FIELD_LEGACY, ConfigDef.Type.STRING, null, ConfigDef.Importance.HIGH,
                "Alias of the 'if' property. The alias exists for backward compatibility."));
    }

    @Override
    public T apply(T record) {
        if (evalPredicate(record)) {
            return newRecord(record, null, null);
        }

        return record;
    }

    public static class Key> extends DropIf implements KeyOrValueTransformation.Key {}
    public static class Value> extends DropIf implements KeyOrValueTransformation.Value {}
}