node-admin/src/main/java/com/yahoo/vespa/hosted/node/admin/provider/NodeAdminProvider.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

// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.node.admin.provider;

import com.google.inject.Inject;
import com.yahoo.concurrent.classlock.ClassLocking;
import com.yahoo.container.di.componentgraph.Provider;
import com.yahoo.vespa.athenz.identity.SiaIdentityProvider;
import com.yahoo.vespa.hosted.dockerapi.Docker;
import com.yahoo.vespa.hosted.dockerapi.metrics.MetricReceiverWrapper;
import com.yahoo.vespa.hosted.node.admin.component.ConfigServerInfo;
import com.yahoo.vespa.hosted.node.admin.component.DockerAdminComponent;
import com.yahoo.vespa.hosted.node.admin.config.ConfigServerConfig;
import com.yahoo.vespa.hosted.node.admin.configserver.ConfigServerClients;
import com.yahoo.vespa.hosted.node.admin.configserver.RealConfigServerClients;

public class NodeAdminProvider implements Provider<NodeAdminStateUpdater> {
    private final DockerAdminComponent dockerAdmin;

    @Inject
    public NodeAdminProvider(ConfigServerConfig configServerConfig,
                             SiaIdentityProvider identityProvider,
                             Docker docker,
                             MetricReceiverWrapper metricReceiver,
                             ClassLocking classLocking) {
        ConfigServerClients clients =
                new RealConfigServerClients(identityProvider, new ConfigServerInfo(configServerConfig));

        dockerAdmin = new DockerAdminComponent(configServerConfig,
                identityProvider,
                docker,
                metricReceiver,
                classLocking,
                clients);
        dockerAdmin.enable();
    }

    @Override
    public NodeAdminStateUpdater get() {
        return dockerAdmin.getNodeAdminStateUpdater();
    }

    @Override
    public void deconstruct() {
        dockerAdmin.disable();
    }
}