com.yahoo.vespa.model.container.http.xml.FilterBuilder Maven / Gradle / Ivy

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.model.container.http.xml;

import com.yahoo.config.model.deploy.DeployState;
import com.yahoo.config.model.producer.AnyConfigProducer;
import com.yahoo.config.model.producer.TreeConfigProducer;
import com.yahoo.text.XML;
import com.yahoo.vespa.model.builder.xml.dom.DomComponentBuilder;
import com.yahoo.vespa.model.builder.xml.dom.VespaDomBuilder;
import com.yahoo.vespa.model.builder.xml.dom.chains.ChainedComponentModelBuilder;
import com.yahoo.vespa.model.container.http.Filter;
import com.yahoo.vespa.model.container.http.FilterConfigProvider;
import org.w3c.dom.Element;

/**
 * @author Tony Vaagenes
 * @author gjoranv
 */
public class FilterBuilder extends VespaDomBuilder.DomConfigProducerBuilderBase {

    @Override
    protected Filter doBuild(DeployState deployState, TreeConfigProducer ancestor, Element filterElement) {
        ChainedComponentModelBuilder modelBuilder = new ChainedComponentModelBuilder(filterElement);
        Filter filter =  new Filter(modelBuilder.build());
        DomComponentBuilder.addChildren(deployState, ancestor, filterElement, filter);
        addFilterConfig(filterElement, filter);

        return filter;
    }

    private static void addFilterConfig(Element filterElement, Filter filter) {
        Element filterConfigElement = XML.getChild(filterElement, "filter-config");
        if (filterConfigElement == null)
            return;

        FilterConfigProvider filterConfigProvider = filter.addAndInjectConfigProvider();
        putFilterConfig(filterConfigElement, filterConfigProvider);
    }

    private static void putFilterConfig(Element filterConfigElement, FilterConfigProvider filterConfigProvider) {
        for (Element e : XML.getChildren(filterConfigElement)) {
            filterConfigProvider.putConfig(e.getTagName(), XML.getValue(e));
        }
    }

}