-
Notifications
You must be signed in to change notification settings - Fork 2.3k
/
DatabaseMigrator.java
102 lines (95 loc) · 4.55 KB
/
DatabaseMigrator.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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
/*
* Copyright (c) 2002-2017 "Neo Technology,"
* Network Engine for Objects in Lund AB [http://neotechnology.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 <http://www.gnu.org/licenses/>.
*/
package org.neo4j.kernel.impl.storemigration;
import java.io.File;
import java.util.Map;
import org.neo4j.io.fs.FileSystemAbstraction;
import org.neo4j.io.pagecache.PageCache;
import org.neo4j.kernel.configuration.Config;
import org.neo4j.kernel.impl.api.index.SchemaIndexProviderMap;
import org.neo4j.kernel.impl.logging.LogService;
import org.neo4j.kernel.impl.store.format.RecordFormats;
import org.neo4j.kernel.impl.storemigration.monitoring.MigrationProgressMonitor;
import org.neo4j.kernel.impl.storemigration.participant.CountsMigrator;
import org.neo4j.kernel.impl.storemigration.participant.LegacyIndexMigrator;
import org.neo4j.kernel.impl.storemigration.participant.NativeLabelScanStoreMigrator;
import org.neo4j.kernel.impl.storemigration.participant.StoreMigrator;
import org.neo4j.kernel.impl.transaction.log.LogTailScanner;
import org.neo4j.kernel.spi.legacyindex.IndexImplementation;
import org.neo4j.logging.LogProvider;
/**
* DatabaseMigrator collects all dependencies required for store migration,
* prepare and construct all store upgrade participants in correct order and allow clients just migrate store
* specified by provided location.
*
* @see StoreUpgrader
*/
public class DatabaseMigrator
{
private final MigrationProgressMonitor progressMonitor;
private final FileSystemAbstraction fs;
private final Config config;
private final LogService logService;
private final SchemaIndexProviderMap schemaIndexProviderMap;
private final Map<String,IndexImplementation> indexProviders;
private final PageCache pageCache;
private final RecordFormats format;
private final LogTailScanner tailScanner;
public DatabaseMigrator(
MigrationProgressMonitor progressMonitor, FileSystemAbstraction fs,
Config config, LogService logService, SchemaIndexProviderMap schemaIndexProviderMap,
Map<String,IndexImplementation> indexProviders, PageCache pageCache,
RecordFormats format, LogTailScanner tailScanner )
{
this.progressMonitor = progressMonitor;
this.fs = fs;
this.config = config;
this.logService = logService;
this.schemaIndexProviderMap = schemaIndexProviderMap;
this.indexProviders = indexProviders;
this.pageCache = pageCache;
this.format = format;
this.tailScanner = tailScanner;
}
/**
* Performs construction of {@link StoreUpgrader} and all of the necessary participants and performs store
* migration if that is required.
* @param storeDir store to migrate
*/
public void migrate( File storeDir )
{
LogProvider logProvider = logService.getInternalLogProvider();
UpgradableDatabase upgradableDatabase =
new UpgradableDatabase( new StoreVersionCheck( pageCache ), format, tailScanner );
StoreUpgrader storeUpgrader = new StoreUpgrader( upgradableDatabase, progressMonitor, config, fs, pageCache,
logProvider );
LegacyIndexMigrator legacyIndexMigrator = new LegacyIndexMigrator( fs, indexProviders, logProvider );
StoreMigrator storeMigrator = new StoreMigrator( fs, pageCache, config, logService );
NativeLabelScanStoreMigrator nativeLabelScanStoreMigrator = new NativeLabelScanStoreMigrator( fs, pageCache );
CountsMigrator countsMigrator = new CountsMigrator( fs, pageCache, config );
schemaIndexProviderMap.accept(
provider -> storeUpgrader.addParticipant( provider.storeMigrationParticipant( fs, pageCache ) ) );
storeUpgrader.addParticipant( legacyIndexMigrator );
storeUpgrader.addParticipant( storeMigrator );
storeUpgrader.addParticipant( nativeLabelScanStoreMigrator );
storeUpgrader.addParticipant( countsMigrator );
storeUpgrader.migrateIfNeeded( storeDir );
}
}