/* * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.addthis.hydra.data.filter.bundle; import java.util.ArrayList; import java.util.Arrays; import java.util.Collections; import java.util.List; import com.addthis.bundle.core.Bundle; import com.addthis.bundle.core.BundleField; import com.addthis.bundle.util.AutoField; import com.addthis.codec.annotations.FieldConfig; import com.fasterxml.jackson.annotation.JsonCreator; import com.fasterxml.jackson.annotation.JsonProperty; /** * This {@link BundleFilter BundleFilter} <span class="hydra-summary">randomly selects an input field and copies it to an output field</span>. * <p/> * <p>The possible input fields are specified with {@link #inFields inFields}. Of the possible * input fields only those with non-null values are considered. One of the eligible values * is randomly selected and copied into the output field.</p> * <p/> * <p>Example:</p> * <pre> * {random-field {inFields: ["INPUT1", "INPUT2", "INPUT3"], out: "OUTPUT"}} * </pre> * * @user-reference */ public class BundleFilterRandomField implements BundleFilter { /** * The possible input bundle fields from which one will be selected. This field is required. */ private AutoField[] inFields; /** * The name of the output bundle field. This field is required. */ private AutoField out; @JsonCreator public BundleFilterRandomField(@JsonProperty(value = "inFields", required = true) AutoField[] inFields, @JsonProperty(value = "out", required = true) AutoField out) { this.inFields = inFields; this.out = out; } @Override public boolean filter(Bundle bundle) { List<AutoField> inFieldsShuffle = new ArrayList<>(Arrays.asList(inFields)); Collections.shuffle(inFieldsShuffle); for (AutoField bf : inFieldsShuffle) { if (bf != null && bf.getValue(bundle) != null) { out.setValue(bundle, bf.getValue(bundle)); break; } } return true; } }