container-disc/src/main/java/com/yahoo/container/jdisc/FilterBindingsProvider.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.container.jdisc;

import com.yahoo.component.ComponentId;
import com.yahoo.component.provider.ComponentRegistry;
import com.yahoo.container.di.componentgraph.Provider;
import com.yahoo.container.http.filter.FilterChainRepository;
import com.yahoo.jdisc.application.BindingRepository;
import com.yahoo.jdisc.http.ServerConfig;
import com.yahoo.jdisc.http.filter.RequestFilter;
import com.yahoo.jdisc.http.filter.ResponseFilter;
import com.yahoo.jdisc.http.filter.SecurityRequestFilter;
import com.yahoo.jdisc.http.server.FilterBindings;

import java.util.ArrayList;
import java.util.List;

/**
 * Provides filter bindings based on vespa config.
 *
 * @author bakksjo
 */
public class FilterBindingsProvider implements Provider<FilterBindings> {

    final BindingRepository<RequestFilter> requestFilters = new BindingRepository<>();
    final BindingRepository<ResponseFilter> responseFilters = new BindingRepository<>();

    public FilterBindingsProvider(ComponentId componentId,
                                  ServerConfig config,
                                  FilterChainRepository filterChainRepository,
                                  ComponentRegistry<SecurityRequestFilter> legacyRequestFilters) {
        ComponentId serverId = componentId.getNamespace();
        try {
            FilterUtil.setupFilters(
                    componentId,
                    legacyRequestFilters,
                    toFilterSpecs(config.filter()),
                    filterChainRepository,
                    requestFilters,
                    responseFilters);
        } catch (Exception e) {
            throw new RuntimeException("Invalid config for http server " + serverId, e);
        }
    }

    private List<FilterUtil.FilterSpec> toFilterSpecs(List<ServerConfig.Filter> inFilters) {
        List<FilterUtil.FilterSpec> outFilters = new ArrayList<>();
        for (ServerConfig.Filter inFilter : inFilters) {
            outFilters.add(new FilterUtil.FilterSpec(inFilter.id(), inFilter.binding()));
        }
        return outFilters;
    }

    @Override
    public FilterBindings get() {
        return new FilterBindings(requestFilters, responseFilters);
    }

    @Override
    public void deconstruct() {}

}