org.neo4j.internal.DataCollectorManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of neo4j
Neo4j Community Edition
There is a newer version: 3.5.4
/*
 * Copyright (c) 2002-2019 "Neo4j,"
 * Neo4j Sweden AB [http://neo4j.com]
 *
 * This file is part of Neo4j.
 *
 * Neo4j is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see .
 */
package org.neo4j.internal;

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

import org.neo4j.graphdb.DependencyResolver;
import org.neo4j.internal.collector.DataCollectorModule;
import org.neo4j.io.IOUtils;
import org.neo4j.kernel.NeoStoreDataSource;
import org.neo4j.kernel.impl.core.EmbeddedProxySPI;
import org.neo4j.kernel.impl.proc.Procedures;
import org.neo4j.kernel.impl.transaction.state.DataSourceManager;
import org.neo4j.kernel.impl.util.DefaultValueMapper;
import org.neo4j.kernel.lifecycle.LifecycleAdapter;
import org.neo4j.kernel.monitoring.Monitors;
import org.neo4j.scheduler.JobScheduler;

public class DataCollectorManager extends LifecycleAdapter
{
    private final DataSourceManager dataSourceManager;
    private final JobScheduler jobScheduler;
    private final Procedures procedures;
    private final Monitors monitors;
    private final List dataCollectors;

    public DataCollectorManager( DataSourceManager dataSourceManager,
                                 JobScheduler jobScheduler,
                                 Procedures procedures,
                                 Monitors monitors )
    {
        this.dataSourceManager = dataSourceManager;
        this.jobScheduler = jobScheduler;
        this.procedures = procedures;
        this.monitors = monitors;
        this.dataCollectors = new ArrayList<>();
    }

    @Override
    public void start() throws Throwable
    {
        // When we have multiple dbs, this has to be suitably modified to get the right kernel and procedures
        NeoStoreDataSource dataSource = dataSourceManager.getDataSource();
        EmbeddedProxySPI embeddedProxySPI = dataSource.getDependencyResolver()
                .resolveDependency( EmbeddedProxySPI.class, DependencyResolver.SelectionStrategy.ONLY );
        dataCollectors.add( DataCollectorModule.setupDataCollector( procedures,
                                                                    jobScheduler,
                                                                    dataSource.getKernel(),
                                                                    monitors,
                                                                    new DefaultValueMapper( embeddedProxySPI ) ) );
    }

    @Override
    public void stop() throws Throwable
    {
        try
        {
            IOUtils.closeAll( dataCollectors );
        }
        finally
        {
            dataCollectors.clear();
        }
    }
}