container-core/src/main/java/com/yahoo/container/core/documentapi/MessageBusDocumentAccessProvider.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

package com.yahoo.container.core.documentapi;

import com.google.inject.Inject;
import com.yahoo.component.AbstractComponent;
import com.yahoo.container.di.componentgraph.Provider;
import com.yahoo.document.config.DocumentmanagerConfig;
import com.yahoo.documentapi.DocumentAccess;
import com.yahoo.documentapi.messagebus.MessageBusDocumentAccess;
import com.yahoo.documentapi.messagebus.MessageBusParams;
import com.yahoo.documentapi.messagebus.loadtypes.LoadTypeSet;
import com.yahoo.vespa.config.content.LoadTypeConfig;

/**
 * Has a lazily populated reference to a {@link MessageBusDocumentAccess}.
 *
 * @author jonmv
 */
public class MessageBusDocumentAccessProvider extends AbstractComponent implements Provider<DocumentAccess> {

    private final DocumentmanagerConfig documentmanagerConfig;
    private final LoadTypeConfig loadTypeConfig;
    private final Object monitor = new Object();
    private boolean shutDown = false;
    private DocumentAccess access = null;

    @Inject
    public MessageBusDocumentAccessProvider(DocumentmanagerConfig documentmanagerConfig, LoadTypeConfig loadTypeConfig) {
        this.documentmanagerConfig = documentmanagerConfig;
        this.loadTypeConfig = loadTypeConfig;
    }

    @Override
    public DocumentAccess get() {
        synchronized (monitor) {
            if (access == null) {
                if (shutDown)
                    throw new IllegalStateException("This document access has been shut down");

                access = new MessageBusDocumentAccess((MessageBusParams) new MessageBusParams(new LoadTypeSet(loadTypeConfig)).setDocumentmanagerConfig(documentmanagerConfig));
            }
            return access;
        }
    }

    @Override
    public void deconstruct() {
        synchronized (monitor) {
            shutDown = true;
            if (access != null)
                access.shutdown();
        }
    }

}