container-core/src/main/java/com/yahoo/jdisc/http/server/jetty/testutils/ConnectorFactoryRegistryModule.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

// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.jdisc.http.server.jetty.testutils;

import com.google.inject.Binder;
import com.google.inject.Module;
import com.google.inject.Provides;
import com.yahoo.component.ComponentId;
import com.yahoo.component.provider.ComponentRegistry;
import com.yahoo.jdisc.http.ConnectorConfig;

import com.yahoo.jdisc.http.server.jetty.ConnectorFactory;
import com.yahoo.jdisc.http.ssl.impl.ConfiguredSslContextFactoryProvider;

/**
 * Guice module for test ConnectorFactories
 *
 * @author Tony Vaagenes
 */
public class ConnectorFactoryRegistryModule implements Module {

    private final ConnectorConfig config;

    public ConnectorFactoryRegistryModule(ConnectorConfig config) {
        this.config = config;
    }

    public ConnectorFactoryRegistryModule() {
        this(new ConnectorConfig(new ConnectorConfig.Builder()));
    }

    @SuppressWarnings("unused")
    @Provides
    public ComponentRegistry<ConnectorFactory> connectorFactoryComponentRegistry() {
        ComponentRegistry<ConnectorFactory> registry = new ComponentRegistry<>();
        registry.register(ComponentId.createAnonymousComponentId("connector-factory"),
                new StaticKeyDbConnectorFactory(config));

        registry.freeze();
        return registry;
    }

    @Override public void configure(Binder binder) {}

    private static class StaticKeyDbConnectorFactory extends ConnectorFactory {

        public StaticKeyDbConnectorFactory(ConnectorConfig connectorConfig) {
            super(connectorConfig, new ConfiguredSslContextFactoryProvider(connectorConfig));
        }

    }

}